72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
package websocketUtil
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"sync"
|
||
)
|
||
|
||
var (
|
||
wsmap sync.Map
|
||
errLog func(format string, args ...interface{})
|
||
debugLog func(format string, args ...interface{})
|
||
)
|
||
|
||
// SetLog
|
||
// @Description:设置日志回调函数信息
|
||
// @param errLogFun 错误日志回调函数
|
||
// @param debugLogFun debug日志回调函数
|
||
func SetLog(errLogFun, debugLogFun func(format string, args ...interface{})) {
|
||
errLog = errLogFun
|
||
debugLog = debugLogFun
|
||
}
|
||
|
||
// Send
|
||
// @Description: 发送数据
|
||
// @param wsurl 发送地址
|
||
// @param st 发送类型，使用websocket下的定义
|
||
// @param data 发送数据
|
||
// @return error
|
||
func Send(wsurl string, st int, data []byte) error {
|
||
if debugLog == nil || errLog == nil {
|
||
return errors.New("websocketUtil未设置errLog,debugLog回调函数，请调用websocketUtil.Setlog函数设置相关信息")
|
||
}
|
||
|
||
m, err := getOrAdd(wsurl)
|
||
if err != nil {
|
||
errLog(fmt.Sprintf("websocketUtil.Send error:%s", err))
|
||
return err
|
||
}
|
||
|
||
m.send(st, data)
|
||
|
||
return nil
|
||
}
|
||
|
||
// getOrAdd
|
||
// @Description:获取或者添加对象
|
||
// @param url string
|
||
// @return *model
|
||
// @return error
|
||
func getOrAdd(url string) (*model, error) {
|
||
var m *model
|
||
var err error
|
||
|
||
v, isOk := wsmap.Load(url)
|
||
if isOk {
|
||
m = v.(*model)
|
||
return m, nil
|
||
}
|
||
|
||
// 不存在，创建对应的socket连接
|
||
m, err = newModel(url)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 保存到map
|
||
wsmap.Store(url, m)
|
||
|
||
return m, err
|
||
}
|