diff --git a/pkg/sse/server.go b/pkg/sse/server.go index 4ac7ed7..824a1aa 100644 --- a/pkg/sse/server.go +++ b/pkg/sse/server.go @@ -17,6 +17,7 @@ type Server interface { GinHandlerFunc(auth func(c *gin.Context) (string, error)) gin.HandlerFunc Push(user any, name, msg string) bool Broadcast(name, msg string) + ClientCount() int32 } type clientChan struct { @@ -157,3 +158,7 @@ func (stream *event) Broadcast(name, msg string) { return true }) } + +func (stream *event) ClientCount() int32 { + return stream.Count.Load() +}