Merge pull request '[🚀] ClientCount' (#16) from dev into main

Reviewed-on: #16

pkg/sse/server.go

@@ -1,9 +1,7 @@
 package sse
 
 import (
-	"fmt"
 	"gitea.bvbej.com/bvbej/base-golang/pkg/mux"
-	"gitea.bvbej.com/bvbej/base-golang/pkg/ticker"
 	"github.com/gin-gonic/gin"
 	"io"
 	"net/http"
@@ -19,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 {
@@ -36,7 +35,6 @@ type event struct {
 	Count       atomic.Int32
 	Register    chan clientChan
 	Unregister  chan any
-	Ticker      ticker.Ticker
 }
 
 func NewServer() Server {
@@ -45,12 +43,8 @@ func NewServer() Server {
 		Count:       atomic.Int32{},
 		Register:    make(chan clientChan),
 		Unregister:  make(chan any),
-		Ticker:      ticker.New(time.Second * 3),
 	}
 
-	go e.Ticker.Process(func() {
-		e.Broadcast("ping", fmt.Sprintf("%d", time.Now().Unix()))
-	})
 	go e.listen()
 
 	return e
@@ -97,6 +91,10 @@ func (stream *event) HandlerFunc() mux.HandlerFunc {
 		c.Context().Writer.Header().Set("Connection", "keep-alive")
 		c.Context().Writer.Header().Set("Transfer-Encoding", "chunked")
 
+		time.AfterFunc(time.Second, func() {
+			e <- msgChan{Name: "message", Message: "success"}
+		})
+
 		c.Context().Stream(func(w io.Writer) bool {
 			if msg, ok := <-e; ok {
 				c.Context().SSEvent(msg.Name, msg.Message)
@@ -130,6 +128,10 @@ func (stream *event) GinHandlerFunc(auth func(c *gin.Context) (string, error)) g
 		c.Writer.Header().Set("Connection", "keep-alive")
 		c.Writer.Header().Set("Transfer-Encoding", "chunked")
 
+		time.AfterFunc(time.Second, func() {
+			e <- msgChan{Name: "message", Message: "success"}
+		})
+
 		c.Stream(func(w io.Writer) bool {
 			if msg, ok := <-e; ok {
 				c.SSEvent(msg.Name, msg.Message)
@@ -156,3 +158,7 @@ func (stream *event) Broadcast(name, msg string) {
 		return true
 	})
 }
+
+func (stream *event) ClientCount() int32 {
+	return stream.Count.Load()
+}