websocket 增加多分组 fork https://github.com/olahol/melody
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
775 B
37 lines
775 B
package main
|
|
|
|
import (
|
|
"log"
|
|
"melody"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func main() {
|
|
m := melody.New()
|
|
router := gin.Default()
|
|
router.GET("/chat", func(context *gin.Context) {
|
|
if err := m.HandleRequest(context.Writer, context.Request); err != nil {
|
|
log.Println(err)
|
|
}
|
|
})
|
|
|
|
m.HandleConnect(func(session *melody.Session) {
|
|
ch := session.Request.URL.Query().Get("channel")
|
|
if err := session.Subscribe(ch); err != nil {
|
|
log.Println(err)
|
|
}
|
|
})
|
|
|
|
m.HandleMessage(func(session *melody.Session, msg []byte) {
|
|
if err := session.Publish(msg); err != nil {
|
|
log.Println(err)
|
|
}
|
|
})
|
|
|
|
m.HandleSentMessage(func(session *melody.Session, bytes []byte) {
|
|
log.Printf("%+v", session.Channel().Online())
|
|
log.Printf("%+v", string(bytes))
|
|
})
|
|
router.Run(":8080")
|
|
}
|