websocket_utils

package

v1.0.21 Latest Latest Go to latest Published: Jul 3, 2025 License: MIT Imports: 5 Imported by: 0

Details

This section is empty.

View Source

var WSServer = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {

		return true
	},
}

WSServer 全局WebSocket升级器

This section is empty.

type WebSocket struct {
	// contains filtered or unexported fields
}

WebSocket 封装WebSocket连接

func NewWebSocketByDial(url string, requestHeader http.Header) (*WebSocket, error)

NewWebSocketByDial 主动建立WebSocket连接

func NewWebSocketByUpgrade(w http.ResponseWriter, r *http.Request, responseHeader http.Header) (*WebSocket, error)

NewWebSocketByUpgrade 升级HTTP连接为WebSocket

func (ws *WebSocket) Close()

Close 安全关闭连接

func (ws *WebSocket) Done() <-chan struct{}

Done 获取关闭通知通道

func (ws *WebSocket) OnMessage(handleFunc func(messageType int, payload []byte))

OnMessage 持续处理消息（阻塞式）

func (ws *WebSocket) SendMessage(messageType int, payload []byte) error

SendMessage 安全发送消息