You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
2.1 KiB
Go
110 lines
2.1 KiB
Go
3 years ago
|
/**
|
||
|
|
* @Author: jager
|
||
|
|
* @Email: lhj168os@gmail.com
|
||
|
|
* @File: service
|
||
|
|
* @Date: 2021/7/8 5:30 下午
|
||
|
|
* @package: service
|
||
|
|
* @Version: v1.0.0
|
||
|
|
*
|
||
|
|
* @Description:
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
package ws
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/jageros/hawox/contextx"
|
||
|
|
"github.com/jageros/hawox/errcode"
|
||
|
|
"github.com/jageros/hawox/httpx"
|
||
|
|
"github.com/jageros/hawox/logx"
|
||
|
|
"gopkg.in/olahol/melody.v1"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
"wechat/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
seq int64
|
||
|
|
mux sync.Mutex
|
||
|
|
names = map[string]string{
|
||
|
|
"13160676597": "杰",
|
||
|
|
"13612225480": "文",
|
||
|
|
"13750043941": "哲",
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
var ss *service
|
||
|
|
|
||
|
|
type service struct {
|
||
|
|
ctx contextx.Context
|
||
|
|
m *melody.Melody
|
||
|
|
callTimeout time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
func Init(ctx contextx.Context, r *gin.RouterGroup, relativePath string) {
|
||
|
|
ss = &service{
|
||
|
|
ctx: ctx,
|
||
|
|
m: melody.New(),
|
||
|
|
callTimeout: time.Second * 5,
|
||
|
|
}
|
||
|
|
ss.m.HandleMessageBinary(ss.handleMessage)
|
||
|
|
ss.m.HandleConnect(ss.onConnect)
|
||
|
|
ss.m.HandleDisconnect(ss.onDisconnect)
|
||
|
|
r.GET(relativePath, ss.handler)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *service) handler(c *gin.Context) {
|
||
|
|
uid := c.GetHeader("uid")
|
||
|
|
if _, ok := names[uid]; !ok {
|
||
|
|
httpx.ErrInterrupt(c, errcode.InvalidParam)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
err := s.m.HandleRequestWithKeys(c.Writer, c.Request, map[string]interface{}{"uid": uid})
|
||
|
|
if err != nil {
|
||
|
|
httpx.ErrInterrupt(c, errcode.WithErrcode(-1, err))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *service) onConnect(session *melody.Session) {
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *service) onDisconnect(session *melody.Session) {
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *service) handleMessage(session *melody.Session, bytes []byte) {
|
||
|
|
start := time.Now()
|
||
|
|
uid, exist := session.Get("uid")
|
||
|
|
if !exist {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
name := names[uid.(string)]
|
||
|
|
|
||
|
|
mux.Lock()
|
||
|
|
defer mux.Unlock()
|
||
|
|
seq += 1
|
||
|
|
|
||
|
|
msg := fmt.Sprintf("[%d]%s(%s): %s\n", seq, name, time.Now().Format("15:04:05"), string(bytes))
|
||
|
|
|
||
|
|
var resp = &types.Msg{
|
||
|
|
MsgID: 1,
|
||
|
|
Msg: msg,
|
||
|
|
}
|
||
|
|
bty, err := json.Marshal(resp)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
err = s.m.Broadcast(bty)
|
||
|
|
if err != nil {
|
||
|
|
logx.Error(err)
|
||
|
|
}
|
||
|
|
take := time.Now().Sub(start)
|
||
|
|
if take > time.Millisecond*100 {
|
||
|
|
logx.Warnf("send msg take: %s", take.String())
|
||
|
|
}
|
||
|
|
}
|