From a3f3830ad824fc46a5512e26ea7a3ffba4f163e2 Mon Sep 17 00:00:00 2001
From: Trevor Judice <tjudice@gfx.io>
Date: Fri, 30 Sep 2022 16:01:57 -0500
Subject: [PATCH] changes

---
 lib/gat/gatling/gatling.go |  1 -
 lib/metrics/gat.go         | 10 ++++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/lib/gat/gatling/gatling.go b/lib/gat/gatling/gatling.go
index bbe640d2..0a2f2107 100644
--- a/lib/gat/gatling/gatling.go
+++ b/lib/gat/gatling/gatling.go
@@ -164,7 +164,6 @@ func (g *Gatling) ListenAndServe(ctx context.Context) error {
 
 			err = <-errch
 			if err != nil {
-				// metrics.Counter("pggat", "accept connection errors")
 				log.Println("failed to accept connection:", err)
 			}
 		}
diff --git a/lib/metrics/gat.go b/lib/metrics/gat.go
index 223d288d..7d7ecda2 100644
--- a/lib/metrics/gat.go
+++ b/lib/metrics/gat.go
@@ -56,3 +56,13 @@ func RecordActiveConnections(change int) {
 	g := GatMetrics()
 	g.ActiveConnections.Add(float64(change))
 }
+
+func RecordConnectionError(err error) {
+	if !On() {
+		return
+	}
+	g := GatMetrics()
+	if err != nil {
+		g.ConnectionErrorCounter.WithLabelValues(err.Error()).Inc()
+	}
+}
-- 
GitLab