Function readBytesAndClose

backend/plugin/db/elasticsearch/sync.go:390–426

(anyResp any)

}

func readBytesAndClose(anyResp any) ([]byte, error) {
	var body io.ReadCloser
	var statusCode int
	var isError bool

	// get body closer and status code.
	switch resp := anyResp.(type) {
	case *http.Response:
		body = resp.Body
		statusCode = resp.StatusCode
		isError = resp.StatusCode >= 400

	case *esapi.Response:
		body = resp.Body
		statusCode = resp.StatusCode
		isError = resp.IsError()

	default:
		return nil, errors.New("not supported response type")
	}

	// read bytes.
	bytes, err := io.ReadAll(body)
	if err != nil {
		return nil, errors.Wrap(err, "failed to read response body")
	}
	if err := body.Close(); err != nil {
		return nil, errors.Wrap(err, "failed to close response body")
	}

	// Check for error status codes
	if isError {
		return bytes, errors.Errorf("request failed with status code %d: %s", statusCode, string(bytes))
	}

	return bytes, nil
}