Compare commits
9 Commits
Author | SHA1 | Date | |
---|---|---|---|
0c9b20d47f | |||
1e33734ab1 | |||
0c6e90e644 | |||
3d5a4a1482 | |||
a0b561237a | |||
ad1e06229f | |||
e99a8dbcd6 | |||
d3513ae4fa | |||
99264c6715 |
@ -7,6 +7,7 @@ import (
|
||||
"net/http"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
)
|
||||
|
||||
var _ Server = (*event)(nil)
|
||||
@ -16,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 {
|
||||
@ -31,9 +33,8 @@ type msgChan struct {
|
||||
type event struct {
|
||||
SessionList sync.Map
|
||||
Count atomic.Int32
|
||||
|
||||
Register chan clientChan
|
||||
Unregister chan any
|
||||
Register chan clientChan
|
||||
Unregister chan any
|
||||
}
|
||||
|
||||
func NewServer() Server {
|
||||
@ -90,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)
|
||||
@ -123,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)
|
||||
@ -149,3 +158,7 @@ func (stream *event) Broadcast(name, msg string) {
|
||||
return true
|
||||
})
|
||||
}
|
||||
|
||||
func (stream *event) ClientCount() int32 {
|
||||
return stream.Count.Load()
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user