Added session duration metrics with device analytics along with device

analytics metrics in the DB to aggregate and DeviceID abuse prevention

1 files changed, 17 insertions, 1 deletions

diff --git a/internal/server/server.go b/internal/server/server.go
index ecb5234..d94dca6 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -276,8 +276,10 @@ func (server *server) handleConnection(conn net.Conn) {
 	sess := session.NewSession(server, addr, cancel, &writerWg)
 	go func() {
 		<-ctx.Done()
-		// Remove session after cancellation
 		if sess.IsAuthenticated() {
+			server.handleSessionMetrics(ctx, sess)
+
+			// Remove session after cancellation
 			sameAddress := addr.String() == server.Session(sess.ID()).Addr().String()
 			// false if the user signed in from a different connection
 			if sameAddress {
@@ -680,3 +682,17 @@ func formatIPv4(ip uint32) string {
 	n = strconv.AppendUint(n, uint64(ip&0xFF), 10)
 	return string(n)
 }
+
+func (s *server) handleSessionMetrics(ctx context.Context, sess *session.Session) {
+	duration := sess.Duration()
+	analytics := sess.Analytics()
+	if api.IsValidAnalytics(ctx, analytics) {
+		metrics.SessionDuration.WithLabelValues(
+			analytics.OS, analytics.Arch, analytics.Term, analytics.Colorterm,
+		).Observe(duration.Seconds())
+	} else {
+		metrics.SessionDuration.WithLabelValues(
+			"", "", "", "",
+		).Observe(duration.Seconds())
+	}
+}