mirror of
https://github.com/christianselig/apollo-backend
synced 2024-11-13 07:27:43 +00:00
add statsd to API responses
This commit is contained in:
parent
f4e5d7e979
commit
7234628428
1 changed files with 11 additions and 2 deletions
|
@ -165,10 +165,12 @@ func (a *api) loggingMiddleware(next http.Handler) http.Handler {
|
|||
|
||||
start := time.Now()
|
||||
lrw := &LoggingResponseWriter{w: w}
|
||||
// Do stuff here
|
||||
|
||||
// Call the next handler, which can be another middleware in the chain, or the final handler.
|
||||
next.ServeHTTP(lrw, r)
|
||||
|
||||
duration := time.Since(start).Milliseconds()
|
||||
|
||||
remoteAddr := r.Header.Get("X-Forwarded-For")
|
||||
if remoteAddr == "" {
|
||||
if ip, _, err := net.SplitHostPort(r.RemoteAddr); err != nil {
|
||||
|
@ -179,7 +181,7 @@ func (a *api) loggingMiddleware(next http.Handler) http.Handler {
|
|||
}
|
||||
|
||||
fields := []zap.Field{
|
||||
zap.Int64("duration", time.Since(start).Milliseconds()),
|
||||
zap.Int64("duration", duration),
|
||||
zap.String("method", r.Method),
|
||||
zap.String("remote#addr", remoteAddr),
|
||||
zap.Int("response#bytes", lrw.bytes),
|
||||
|
@ -193,5 +195,12 @@ func (a *api) loggingMiddleware(next http.Handler) http.Handler {
|
|||
err := lrw.Header().Get("X-Apollo-Error")
|
||||
a.logger.Error(err, fields...)
|
||||
}
|
||||
|
||||
tags := []string{fmt.Sprintf("status:%d", lrw.statusCode)}
|
||||
_ = a.statsd.Histogram("api.latency", float64(duration), tags, 0.1)
|
||||
_ = a.statsd.Incr("api.calls", nil, 0.1)
|
||||
if lrw.statusCode >= 500 {
|
||||
_ = a.statsd.Incr("api.errors", nil, 0.1)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue