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.
44 lines
714 B
44 lines
714 B
package main
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-fsnotify/fsnotify"
|
|
"github.com/olahol/melody"
|
|
"io/ioutil"
|
|
"net/http"
|
|
)
|
|
|
|
func main() {
|
|
file := "file.txt"
|
|
|
|
r := gin.Default()
|
|
m := melody.New()
|
|
w, _ := fsnotify.NewWatcher()
|
|
|
|
r.GET("/", func(c *gin.Context) {
|
|
http.ServeFile(c.Writer, c.Request, "index.html")
|
|
})
|
|
|
|
r.GET("/ws", func(c *gin.Context) {
|
|
m.HandleRequest(c.Writer, c.Request)
|
|
})
|
|
|
|
m.HandleConnect(func(s *melody.Session) {
|
|
content, _ := ioutil.ReadFile(file)
|
|
s.Write(content)
|
|
})
|
|
|
|
go func() {
|
|
for {
|
|
ev := <-w.Events
|
|
if ev.Op == fsnotify.Write {
|
|
content, _ := ioutil.ReadFile(ev.Name)
|
|
m.Broadcast(content)
|
|
}
|
|
}
|
|
}()
|
|
|
|
w.Add(file)
|
|
|
|
r.Run(":5000")
|
|
}
|