websocket_utils

package

v1.0.14 Latest Latest Go to latest Published: Mar 26, 2025 License: MIT Imports: 4 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 {
	Conn *websocket.Conn // 底层 WebSocket 连接对象
	Done chan 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 关闭 WebSocket 连接

func (ws *WebSocket) OnMessage(handleFunc func([]byte))

OnMessage 注册一个处理消息的回调函数

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

SendMessage 发送一条消息