正在显示
1 个修改的文件
包含
29 行增加
和
0 行删除
services/websocket/websocket.go
0 → 100644
| 1 | +package websocket | ||
| 2 | + | ||
| 3 | +import ( | ||
| 4 | + "github.com/gorilla/websocket" | ||
| 5 | + "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/mybeego" | ||
| 6 | + ws "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/websocket" | ||
| 7 | + "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/log" | ||
| 8 | + "net/http" | ||
| 9 | + "strconv" | ||
| 10 | +) | ||
| 11 | + | ||
| 12 | +var upgrader = websocket.Upgrader{} | ||
| 13 | + | ||
| 14 | +func Upgrage(w http.ResponseWriter, r *http.Request) { | ||
| 15 | + requestHead := &mybeego.RequestHead{} | ||
| 16 | + requestHead.Uid, _ = strconv.ParseInt(r.Header.Get("uid"), 10, 64) | ||
| 17 | + requestHead.AppId, _ = strconv.Atoi(r.Header.Get("appid")) | ||
| 18 | + requestHead.Token = r.Header.Get("token") | ||
| 19 | + conn, err := upgrader.Upgrade(w, r, nil) | ||
| 20 | + if err != nil { | ||
| 21 | + log.Fatal(err) | ||
| 22 | + } | ||
| 23 | + wsConn := ws.NewWebsocketConnection(conn, requestHead, WsOnReceive) | ||
| 24 | + wsConn.Serve() | ||
| 25 | +} | ||
| 26 | + | ||
| 27 | +func WsOnReceive(data []byte) *mybeego.Message { | ||
| 28 | + return mybeego.NewMessage(0) | ||
| 29 | +} |
-
请 注册 或 登录 后发表评论