Compare commits
7 Commits
Author | SHA1 | Date | |
---|---|---|---|
0c9b20d47f | |||
1e33734ab1 | |||
0c6e90e644 | |||
3d5a4a1482 | |||
a0b561237a | |||
ad1e06229f | |||
e99a8dbcd6 |
@ -1,9 +1,7 @@
|
|||||||
package sse
|
package sse
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"gitea.bvbej.com/bvbej/base-golang/pkg/mux"
|
"gitea.bvbej.com/bvbej/base-golang/pkg/mux"
|
||||||
"gitea.bvbej.com/bvbej/base-golang/pkg/ticker"
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
@ -19,6 +17,7 @@ type Server interface {
|
|||||||
GinHandlerFunc(auth func(c *gin.Context) (string, error)) gin.HandlerFunc
|
GinHandlerFunc(auth func(c *gin.Context) (string, error)) gin.HandlerFunc
|
||||||
Push(user any, name, msg string) bool
|
Push(user any, name, msg string) bool
|
||||||
Broadcast(name, msg string)
|
Broadcast(name, msg string)
|
||||||
|
ClientCount() int32
|
||||||
}
|
}
|
||||||
|
|
||||||
type clientChan struct {
|
type clientChan struct {
|
||||||
@ -36,7 +35,6 @@ type event struct {
|
|||||||
Count atomic.Int32
|
Count atomic.Int32
|
||||||
Register chan clientChan
|
Register chan clientChan
|
||||||
Unregister chan any
|
Unregister chan any
|
||||||
Ticker ticker.Ticker
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewServer() Server {
|
func NewServer() Server {
|
||||||
@ -45,12 +43,8 @@ func NewServer() Server {
|
|||||||
Count: atomic.Int32{},
|
Count: atomic.Int32{},
|
||||||
Register: make(chan clientChan),
|
Register: make(chan clientChan),
|
||||||
Unregister: make(chan any),
|
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()
|
go e.listen()
|
||||||
|
|
||||||
return e
|
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("Connection", "keep-alive")
|
||||||
c.Context().Writer.Header().Set("Transfer-Encoding", "chunked")
|
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 {
|
c.Context().Stream(func(w io.Writer) bool {
|
||||||
if msg, ok := <-e; ok {
|
if msg, ok := <-e; ok {
|
||||||
c.Context().SSEvent(msg.Name, msg.Message)
|
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("Connection", "keep-alive")
|
||||||
c.Writer.Header().Set("Transfer-Encoding", "chunked")
|
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 {
|
c.Stream(func(w io.Writer) bool {
|
||||||
if msg, ok := <-e; ok {
|
if msg, ok := <-e; ok {
|
||||||
c.SSEvent(msg.Name, msg.Message)
|
c.SSEvent(msg.Name, msg.Message)
|
||||||
@ -156,3 +158,7 @@ func (stream *event) Broadcast(name, msg string) {
|
|||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (stream *event) ClientCount() int32 {
|
||||||
|
return stream.Count.Load()
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user