make latency available on client

Kelwing · May 17, 2023 · 17f4a05 · 17f4a05
1 parent 4b682cf
commit 17f4a05
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/gateway/shard/heartbeat.go b/gateway/shard/heartbeat.go
@@ -1,6 +1,7 @@
 package shard
 
 import (
+	"math"
 	"sync"
 	"time"
 
@@ -17,6 +18,7 @@ type Heartbeat struct {
 	lastHeartbeat time.Time
 	logger        zerolog.Logger
 	wg            sync.WaitGroup
+	latency       *atomic.Duration
 }
 
 func NewHeartbeat(interval int64, shard *Shard, logger zerolog.Logger) *Heartbeat {
@@ -25,6 +27,7 @@ func NewHeartbeat(interval int64, shard *Shard, logger zerolog.Logger) *Heartbea
 		acked:    atomic.NewBool(true),
 		gw:       shard,
 		logger:   logger,
+    latency: atomic.NewDuration(time.Duration(math.Inf(1))),
 	}
 }
 
@@ -76,5 +79,6 @@ func (h *Heartbeat) Stop() {
 
 func (h *Heartbeat) ACK() {
 	h.acked.Store(true)
-	h.logger.Debug().Dur("latency", time.Since(h.lastHeartbeat)).Msg("Received heartbeat ACK")
+	h.latency.Store(time.Since(h.lastHeartbeat))
+	h.logger.Debug().Dur("latency", h.latency.Load()).Msg("Received heartbeat ACK")
 }
diff --git a/gateway/shard/shard.go b/gateway/shard/shard.go
@@ -278,6 +278,10 @@ func (s *Shard) process(p objects.Payload) error {
 	return nil
 }
 
+func (s *Shard) Latency() time.Duration {
+  return s.heartbeat.latency.Load()
+}
+
 func (s *Shard) Run() error {
 	var err error
 	for {