mirror of
https://github.com/hoernschen/dendrite.git
synced 2024-12-26 15:08:28 +00:00
Define InternalAPIError and friends
This commit is contained in:
parent
59cf8e936e
commit
c15a941284
1 changed files with 86 additions and 9 deletions
|
@ -27,19 +27,73 @@ import (
|
||||||
"github.com/opentracing/opentracing-go/ext"
|
"github.com/opentracing/opentracing-go/ext"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type InternalAPIError interface {
|
||||||
|
error
|
||||||
|
Temporary() bool
|
||||||
|
Remote() bool
|
||||||
|
isInternalAPIError()
|
||||||
|
}
|
||||||
|
|
||||||
|
// internalAPICallError represents an error reaching an internal API.
|
||||||
|
type internalAPICallError struct {
|
||||||
|
err error
|
||||||
|
temporary bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *internalAPICallError) isInternalAPIError() {}
|
||||||
|
|
||||||
|
func (e *internalAPICallError) Error() string {
|
||||||
|
return fmt.Sprintf("internal API call failed: %s", e.err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *internalAPICallError) Remote() bool {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *internalAPICallError) Temporary() bool {
|
||||||
|
return e.temporary
|
||||||
|
}
|
||||||
|
|
||||||
|
// internalAPIRemoteError represents an error returned from a internal API.
|
||||||
|
type internalAPIRemoteError struct {
|
||||||
|
code int
|
||||||
|
url string
|
||||||
|
err string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *internalAPIRemoteError) isInternalAPIError() {}
|
||||||
|
|
||||||
|
func (e *internalAPIRemoteError) Error() string {
|
||||||
|
return fmt.Sprintf("internal API %s returned HTTP %d: %s", e.url, e.code, e.err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *internalAPIRemoteError) Remote() bool {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *internalAPIRemoteError) Temporary() bool {
|
||||||
|
return e.code >= 500
|
||||||
|
}
|
||||||
|
|
||||||
// PostJSON performs a POST request with JSON on an internal HTTP API
|
// PostJSON performs a POST request with JSON on an internal HTTP API
|
||||||
func PostJSON(
|
func PostJSON(
|
||||||
ctx context.Context, span opentracing.Span, httpClient *http.Client,
|
ctx context.Context, span opentracing.Span, httpClient *http.Client,
|
||||||
apiURL string, request, response interface{},
|
apiURL string, request, response interface{},
|
||||||
) error {
|
) InternalAPIError {
|
||||||
jsonBytes, err := json.Marshal(request)
|
jsonBytes, err := json.Marshal(request)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return &internalAPICallError{
|
||||||
|
err: err,
|
||||||
|
temporary: false,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
parsedAPIURL, err := url.Parse(apiURL)
|
parsedAPIURL, err := url.Parse(apiURL)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return &internalAPICallError{
|
||||||
|
err: err,
|
||||||
|
temporary: false,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
parsedAPIURL.Path = InternalPathPrefix + strings.TrimLeft(parsedAPIURL.Path, "/")
|
parsedAPIURL.Path = InternalPathPrefix + strings.TrimLeft(parsedAPIURL.Path, "/")
|
||||||
|
@ -47,7 +101,10 @@ func PostJSON(
|
||||||
|
|
||||||
req, err := http.NewRequest(http.MethodPost, apiURL, bytes.NewReader(jsonBytes))
|
req, err := http.NewRequest(http.MethodPost, apiURL, bytes.NewReader(jsonBytes))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return &internalAPICallError{
|
||||||
|
err: err,
|
||||||
|
temporary: false,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Mark the span as being an RPC client.
|
// Mark the span as being an RPC client.
|
||||||
|
@ -56,7 +113,10 @@ func PostJSON(
|
||||||
tracer := opentracing.GlobalTracer()
|
tracer := opentracing.GlobalTracer()
|
||||||
|
|
||||||
if err = tracer.Inject(span.Context(), opentracing.HTTPHeaders, carrier); err != nil {
|
if err = tracer.Inject(span.Context(), opentracing.HTTPHeaders, carrier); err != nil {
|
||||||
return err
|
return &internalAPICallError{
|
||||||
|
err: err,
|
||||||
|
temporary: true,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
req.Header.Set("Content-Type", "application/json")
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
@ -66,16 +126,33 @@ func PostJSON(
|
||||||
defer (func() { err = res.Body.Close() })()
|
defer (func() { err = res.Body.Close() })()
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return &internalAPICallError{
|
||||||
|
err: err,
|
||||||
|
temporary: true,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
if res.StatusCode != http.StatusOK {
|
if res.StatusCode != http.StatusOK {
|
||||||
var errorBody struct {
|
var errorBody struct {
|
||||||
Message string `json:"message"`
|
Message string `json:"message"`
|
||||||
}
|
}
|
||||||
if msgerr := json.NewDecoder(res.Body).Decode(&errorBody); msgerr == nil {
|
if msgerr := json.NewDecoder(res.Body).Decode(&errorBody); msgerr == nil {
|
||||||
return fmt.Errorf("internal API: %d from %s: %s", res.StatusCode, apiURL, errorBody.Message)
|
return &internalAPIRemoteError{
|
||||||
|
err: errorBody.Message,
|
||||||
|
url: apiURL,
|
||||||
|
code: res.StatusCode,
|
||||||
}
|
}
|
||||||
return fmt.Errorf("internal API: %d from %s", res.StatusCode, apiURL)
|
|
||||||
}
|
}
|
||||||
return json.NewDecoder(res.Body).Decode(response)
|
return &internalAPIRemoteError{
|
||||||
|
err: "unknown error",
|
||||||
|
url: apiURL,
|
||||||
|
code: res.StatusCode,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err := json.NewDecoder(res.Body).Decode(response); err != nil {
|
||||||
|
return &internalAPICallError{
|
||||||
|
err: err,
|
||||||
|
temporary: false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue