From 3b46655dbb0dc077530a9dd601fc65a0d8d7b16e Mon Sep 17 00:00:00 2001
From: Josh Bleecher Snyder <josh@tailscale.com>
Date: Thu, 29 Oct 2020 15:44:46 -0700
Subject: [PATCH] tsweb: add StatusCodeCounters to HandlerOptions

This lets servers using tsweb register expvars
that will track the number of requests ending
in 200s/300s/400s/500s.

Signed-off-by: Josh Bleecher Snyder <josh@tailscale.com>
---
 tsweb/tsweb.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tsweb/tsweb.go b/tsweb/tsweb.go
index 8675fae0c..134f034cf 100644
--- a/tsweb/tsweb.go
+++ b/tsweb/tsweb.go
@@ -161,6 +161,11 @@ type HandlerOptions struct {
 	Quiet200s bool // if set, do not log successfully handled HTTP requests
 	Logf      logger.Logf
 	Now       func() time.Time // if nil, defaults to time.Now
+
+	// If non-nil, StatusCodeCounters maintains counters
+	// of status codes for handled responses.
+	// The keys are "1xx", "2xx", "3xx", "4xx", and "5xx".
+	StatusCodeCounters *expvar.Map
 }
 
 // StdHandler converts a ReturnHandler into a standard http.Handler.
@@ -278,6 +283,10 @@ func (h retHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	if msg.Code != 200 || !h.opts.Quiet200s {
 		h.opts.Logf("%s", msg)
 	}
+
+	if h.opts.StatusCodeCounters != nil {
+		key := fmt.Sprintf("%dxx", msg.Code/100)
+		h.opts.StatusCodeCounters.Add(key, 1)
 	}
 }