Files
weave-scope/app/server_helpers.go
Bryan Boreham 5264b61951 improvement: stop rendering if Context is cancelled
Typically this means the http caller has closed the connection,
so no point responding to them.

Also check at the point we send a response back, and log to OpenTracing.
2020-06-11 11:13:38 +00:00

36 lines
935 B
Go

package app
import (
"context"
"net/http"
opentracing "github.com/opentracing/opentracing-go"
"github.com/ugorji/go/codec"
log "github.com/sirupsen/logrus"
)
func respondWith(ctx context.Context, w http.ResponseWriter, code int, response interface{}) {
if err, ok := response.(error); ok {
log.Errorf("Error %d: %v", code, err)
response = err.Error()
} else if 500 <= code && code < 600 {
log.Errorf("Non-error %d: %v", code, response)
} else if ctx.Err() != nil {
log.Debugf("Context error %v", ctx.Err())
code = 499
response = nil
}
if span := opentracing.SpanFromContext(ctx); span != nil {
span.LogKV("response-code", code)
}
w.Header().Set("Content-Type", "application/json")
w.Header().Add("Cache-Control", "no-cache")
w.WriteHeader(code)
encoder := codec.NewEncoder(w, &codec.JsonHandle{})
if err := encoder.Encode(response); err != nil {
log.Errorf("Error encoding response: %v", err)
}
}