From 8b573c94895dc0ac0e1d9d59cf3e8745e8b539ca Mon Sep 17 00:00:00 2001
From: GitLab Bot <gitlab-bot@gitlab.com>
Date: Thu, 17 Dec 2020 11:59:07 +0000
Subject: Add latest changes from gitlab-org/gitlab@13-7-stable-ee

---
 workhorse/internal/git/responsewriter.go | 75 ++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 workhorse/internal/git/responsewriter.go

(limited to 'workhorse/internal/git/responsewriter.go')

diff --git a/workhorse/internal/git/responsewriter.go b/workhorse/internal/git/responsewriter.go
new file mode 100644
index 00000000000..c4d4ac252d4
--- /dev/null
+++ b/workhorse/internal/git/responsewriter.go
@@ -0,0 +1,75 @@
+package git
+
+import (
+	"net/http"
+	"strconv"
+
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promauto"
+
+	"gitlab.com/gitlab-org/gitlab-workhorse/internal/helper"
+)
+
+const (
+	directionIn  = "in"
+	directionOut = "out"
+)
+
+var (
+	gitHTTPSessionsActive = promauto.NewGauge(prometheus.GaugeOpts{
+		Name: "gitlab_workhorse_git_http_sessions_active",
+		Help: "Number of Git HTTP request-response cycles currently being handled by gitlab-workhorse.",
+	})
+
+	gitHTTPRequests = promauto.NewCounterVec(
+		prometheus.CounterOpts{
+			Name: "gitlab_workhorse_git_http_requests",
+			Help: "How many Git HTTP requests have been processed by gitlab-workhorse, partitioned by request type and agent.",
+		},
+		[]string{"method", "code", "service", "agent"},
+	)
+
+	gitHTTPBytes = promauto.NewCounterVec(
+		prometheus.CounterOpts{
+			Name: "gitlab_workhorse_git_http_bytes",
+			Help: "How many Git HTTP bytes have been sent by gitlab-workhorse, partitioned by request type, agent and direction.",
+		},
+		[]string{"method", "code", "service", "agent", "direction"},
+	)
+)
+
+type HttpResponseWriter struct {
+	helper.CountingResponseWriter
+}
+
+func NewHttpResponseWriter(rw http.ResponseWriter) *HttpResponseWriter {
+	gitHTTPSessionsActive.Inc()
+	return &HttpResponseWriter{
+		CountingResponseWriter: helper.NewCountingResponseWriter(rw),
+	}
+}
+
+func (w *HttpResponseWriter) Log(r *http.Request, writtenIn int64) {
+	service := getService(r)
+	agent := getRequestAgent(r)
+
+	gitHTTPSessionsActive.Dec()
+	gitHTTPRequests.WithLabelValues(r.Method, strconv.Itoa(w.Status()), service, agent).Inc()
+	gitHTTPBytes.WithLabelValues(r.Method, strconv.Itoa(w.Status()), service, agent, directionIn).
+		Add(float64(writtenIn))
+	gitHTTPBytes.WithLabelValues(r.Method, strconv.Itoa(w.Status()), service, agent, directionOut).
+		Add(float64(w.Count()))
+}
+
+func getRequestAgent(r *http.Request) string {
+	u, _, ok := r.BasicAuth()
+	if !ok {
+		return "anonymous"
+	}
+
+	if u == "gitlab-ci-token" {
+		return "gitlab-ci"
+	}
+
+	return "logged"
+}
-- 
cgit v1.2.3