From ad1e06229f2adb78088475ad240b804e4fadfa74 Mon Sep 17 00:00:00 2001
From: bvbej <admin@bvbej.com>
Date: Sat, 7 Sep 2024 15:55:14 +0800
Subject: [PATCH] =?UTF-8?q?[=F0=9F=9A=80]=20=20message?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/sse/server.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/pkg/sse/server.go b/pkg/sse/server.go
index b724268..c502c25 100644
--- a/pkg/sse/server.go
+++ b/pkg/sse/server.go
@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"sync"
 	"sync/atomic"
+	"time"
 )
 
 var _ Server = (*event)(nil)
@@ -122,6 +123,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)
@@ -129,8 +134,6 @@ func (stream *event) GinHandlerFunc(auth func(c *gin.Context) (string, error)) g
 			}
 			return false
 		})
-
-		e <- msgChan{Name: "message", Message: "success"}
 	}
 }