client_http_response.go

// Copyright (c) 2024 Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

package zanzibar

import (
	"fmt"
	"io"
	"net/http"
	"strings"
	"time"

	"github.com/pkg/errors"
	"github.com/uber/zanzibar/v2/runtime/jsonwrapper"

	"go.uber.org/multierr"
	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"
)

// ClientHTTPResponse is the struct managing the client response
// when making outbound http calls.
type ClientHTTPResponse struct {
	req              *ClientHTTPRequest
	finishTime       time.Time
	finished         bool
	rawResponse      *http.Response
	responseRead     bool
	rawResponseBytes []byte
	StatusCode       int
	Duration         time.Duration
	Header           http.Header
	jsonWrapper      jsonwrapper.JSONWrapper
}

// NewClientHTTPResponse allocates a client http response object
// to track http response.
func NewClientHTTPResponse(
	req *ClientHTTPRequest,
) *ClientHTTPResponse {
	return &ClientHTTPResponse{
		req:         req,
		jsonWrapper: req.jsonWrapper,
	}
}

func (res *ClientHTTPResponse) setRawHTTPResponse(httpRes *http.Response) {
	res.rawResponse = httpRes
	res.StatusCode = httpRes.StatusCode
	res.Header = httpRes.Header
}

// ReadAll reads bytes from response.
func (res *ClientHTTPResponse) ReadAll() ([]byte, error) {
	// if response is already read
	if res.responseRead {
		return res.GetRawBody(), nil
	}

	// mark response read to true
	defer func() {
		res.responseRead = true
	}()

	rawBody, err := io.ReadAll(res.rawResponse.Body)

	cerr := res.rawResponse.Body.Close()
	if cerr != nil {
		/* coverage ignore next line */
		res.req.ContextLogger.WarnZ(res.req.ctx, "Could not close response body", zap.Error(cerr))
	}

	if err != nil {
		res.finish()
		return nil, errors.Wrapf(
			err, "Could not read %s.%s response body",
			res.req.ClientID, res.req.MethodName,
		)
	}

	res.rawResponseBytes = rawBody
	res.finish()
	return rawBody, nil
}

// GetRawBody returns the body as byte array if it has been read.
func (res *ClientHTTPResponse) GetRawBody() []byte {
	return res.rawResponseBytes
}

// ReadAndUnmarshalBody will try to unmarshal non pointer value or fail
func (res *ClientHTTPResponse) ReadAndUnmarshalBody(v interface{}) error {
	rawBody, err := res.ReadAll()
	if err != nil {
		return err
	}
	return res.UnmarshalBody(v, rawBody)
}

// UnmarshalBody helper to unmarshal body into struct
func (res *ClientHTTPResponse) UnmarshalBody(v interface{}, rawBody []byte) error {
	err := res.jsonWrapper.Unmarshal(rawBody, v)
	if err != nil {
		res.req.Metrics.IncCounter(res.req.ctx, clientHTTPUnmarshalError, 1)
		return errors.Wrapf(
			err, "Could not parse %s.%s response json",
			res.req.ClientID, res.req.MethodName,
		)
	}
	return nil
}

// ReadAndUnmarshalBodyMultipleOptions will try to unmarshal non pointer value to one of the provided types or fail
// It will return the deserialized struct (if any) that succeeded
func (res *ClientHTTPResponse) ReadAndUnmarshalBodyMultipleOptions(vs []interface{}) (interface{}, error) {
	rawBody, err := res.ReadAll()
	if err != nil {
		/* coverage ignore next line */
		return nil, err
	}

	var merr error
	for _, v := range vs {
		err = res.jsonWrapper.Unmarshal(rawBody, v)
		if err == nil {
			// All done -- successfully deserialized
			return v, nil
		}
		merr = multierr.Append(merr, err)
	}

	return nil, errors.Wrapf(
		merr, "Could not parse %s.%s response json into any of provided interfaces",
		res.req.ClientID, res.req.MethodName,
	)
}

// CheckOKResponse checks if the status code is OK.
func (res *ClientHTTPResponse) CheckOKResponse(okResponses []int) {
	for _, okResponse := range okResponses {
		if res.rawResponse.StatusCode == okResponse {
			return
		}
	}
	res.req.ContextLogger.WarnZ(res.req.ctx, "Received unexpected client response status code")
}

// finish will handle final logic, like metrics
func (res *ClientHTTPResponse) finish() {
	if !res.req.started {
		/* coverage ignore next line */
		res.req.ContextLogger.Error(res.req.ctx, "Forgot to start client request")
		/* coverage ignore next line */
		return
	}
	if res.finished {
		/* coverage ignore next line */
		res.req.ContextLogger.Error(res.req.ctx, "Finished a client response multiple times")
		/* coverage ignore next line */
		return
	}
	res.finished = true
	res.finishTime = time.Now()

	// emit metrics
	delta := res.finishTime.Sub(res.req.startTime)
	res.req.Metrics.RecordTimer(res.req.ctx, clientLatency, delta)
	res.req.Metrics.RecordHistogramDuration(res.req.ctx, clientLatencyHist, delta)
	res.Duration = delta

	_, known := knownStatusCodes[res.StatusCode]
	if !known {
		res.req.ContextLogger.WarnZ(res.req.ctx, "Received unknown status code from client")
	} else {
		scopeTags := map[string]string{scopeTagStatus: fmt.Sprintf("%d", res.StatusCode)}
		res.req.ctx = WithScopeTagsDefault(res.req.ctx, scopeTags, res.req.Metrics.Scope())
		res.req.Metrics.IncCounter(res.req.ctx, clientStatus, 1)
	}
	if !known || res.StatusCode >= 400 && res.StatusCode < 600 {
		res.req.Metrics.IncCounter(res.req.ctx, clientErrors, 1)
	}
	AppendLogFieldsToContext(res.req.ctx, clientHTTPLogFields(res.req, res)...)
}

func clientHTTPLogFields(req *ClientHTTPRequest, res *ClientHTTPResponse) []zapcore.Field {
	fields := []zapcore.Field{
		zap.Int(logFieldClientStatusCode, res.StatusCode),
	}
	for k, v := range req.httpReq.Header {
		if len(v) > 0 {
			fields = append(fields, zap.String(
				fmt.Sprintf("%s-%s", logFieldClientRequestHeaderPrefix, k),
				strings.Join(v, ", "),
			))
		}
	}
	for k, v := range req.res.Header {
		if len(v) > 0 {
			fields = append(fields, zap.String(
				fmt.Sprintf("%s-%s", logFieldClientResponseHeaderPrefix, k),
				strings.Join(v, ", "),
			))
		}
	}

	// TODO: log jaeger trace span

	return fields
}