From 1e33734ab1aed0058f9bf28267edab4dbaa70c0b Mon Sep 17 00:00:00 2001 From: bvbej Date: Sat, 7 Sep 2024 17:13:43 +0800 Subject: [PATCH] =?UTF-8?q?[=F0=9F=9A=80]=20=20ClientCount?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/sse/server.go | 5 +++++ 1 file changed, 5 insertions(+) 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() +}