// ServeHTTP passes the request to the underlying http.Handler and then times
// the response by its HTTP status code via go-metrics.
func (c *TimeByStatus) ServeHTTP(w0 http.ResponseWriter, r *http.Request) {
w := middleware.NewResponseWriter(w0)
// Mark the start time.
start := time.Now()
// Run the request.
c.handler.ServeHTTP(w, r)
c.Lock()
_, ok := c.timers[w.StatusCode]
if !ok {
// Generate a new timer if required.
timer := metrics.NewTimer()
c.timers[w.StatusCode] = timer
// Gets an existing metric or creates and registers a new one.
m := c.registry.GetOrRegister(
fmt.Sprintf("timer-%s-%d", c.name, w.StatusCode),
timer,
)
if _, ok := m.(error); ok {
c.Unlock()
panic(m)
}
}
c.Unlock()
c.timers[w.StatusCode].UpdateSince(start)
}
// ServeHTTP passes the request to the underlying http.Handler and then counts
// the response by its HTTP status code via go-metrics.
func (c *MeterByStatus) ServeHTTP(w0 http.ResponseWriter, r *http.Request) {
w := middleware.NewResponseWriter(w0)
c.handler.ServeHTTP(w, r)
c.Lock()
_, ok := c.meters[w.StatusCode]
if !ok {
// Generate a new meter if required.
meter := metrics.NewMeter()
c.meters[w.StatusCode] = meter
// Gets an existing metric or creates and registers a new one.
m := c.registry.GetOrRegister(
fmt.Sprintf("meter-%s-%d", c.name, w.StatusCode),
meter,
)
if _, ok := m.(error); ok {
c.Unlock()
panic(m)
}
}
c.Unlock()
c.meters[w.StatusCode].Mark(1)
}
