From 1a79bdeb6bde4adc4e6f0bae727c37842ed5fdbf Mon Sep 17 00:00:00 2001 From: Justin Li Date: Thu, 24 Jul 2014 17:48:03 -0400 Subject: [PATCH] Add RequestsBlocked stat counter --- http/routes.go | 3 +++ stats/stats.go | 5 +++++ 2 files changed, 8 insertions(+) diff --git a/http/routes.go b/http/routes.go index 5ea0441..4bdd0f2 100644 --- a/http/routes.go +++ b/http/routes.go @@ -44,6 +44,7 @@ func (s *Server) serveAnnounce(w http.ResponseWriter, r *http.Request, p httprou if err == models.ErrMalformedRequest { writer.WriteError(err) + stats.RecordEvent(stats.BlockedRequest) return http.StatusOK, nil } else if err != nil { return http.StatusInternalServerError, err @@ -53,6 +54,7 @@ func (s *Server) serveAnnounce(w http.ResponseWriter, r *http.Request, p httprou if err == models.ErrBadRequest { writer.WriteError(err) + stats.RecordEvent(stats.BlockedRequest) return http.StatusOK, nil } else if err != nil { return http.StatusInternalServerError, err @@ -68,6 +70,7 @@ func (s *Server) serveScrape(w http.ResponseWriter, r *http.Request, p httproute if err == models.ErrMalformedRequest { writer.WriteError(err) + stats.RecordEvent(stats.BlockedRequest) return http.StatusOK, nil } else if err != nil { return http.StatusInternalServerError, err diff --git a/stats/stats.go b/stats/stats.go index cfe1614..4352455 100644 --- a/stats/stats.go +++ b/stats/stats.go @@ -32,6 +32,7 @@ const ( ClosedConnection HandledRequest + BlockedRequest ErroredRequest ResponseTime @@ -82,6 +83,7 @@ type Stats struct { RequestsHandled uint64 `json:"requests_handled"` RequestsErrored uint64 `json:"requests_errored"` + RequestsBlocked uint64 `json:"requests_blocked"` ResponseTime PercentileTimes `json:"response_time"` MemStats *MemStatsWrapper `json:"mem,omitempty"` @@ -198,6 +200,9 @@ func (s *Stats) handleEvent(event int) { case HandledRequest: s.RequestsHandled++ + case BlockedRequest: + s.RequestsBlocked++ + case ErroredRequest: s.RequestsErrored++