forked from ProjectWIND/ProjectWIND
141 lines
2.9 KiB
Go
141 lines
2.9 KiB
Go
package protocol
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/url"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
var gProtocolAddr string
|
|
|
|
// WebSocketHandler 接收WebSocket连接处的消息并处理
|
|
func WebSocketHandler(protocolAddr string) error {
|
|
// 保存全局变量
|
|
gProtocolAddr = protocolAddr
|
|
// 解析连接URL
|
|
u, err := url.Parse(protocolAddr)
|
|
if err != nil {
|
|
log.Println("[ERROR] Parse URL error:", err)
|
|
return err
|
|
}
|
|
|
|
conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
|
|
if err != nil {
|
|
log.Println("[ERROR] Dial error:", err)
|
|
return err
|
|
}
|
|
defer func(conn *websocket.Conn) {
|
|
err := conn.Close()
|
|
if err != nil {
|
|
log.Println("[ERROR] Close error:", err)
|
|
}
|
|
}(conn)
|
|
|
|
log.Println("[INFO] New connection established.")
|
|
|
|
// 定义通道,缓存消息和消息类型，防止消息处理阻塞
|
|
messageChan := make(chan []byte, 32)
|
|
messageTypeChan := make(chan int, 32)
|
|
|
|
for {
|
|
// 接收消息并放入通道
|
|
messageType, message, err := conn.ReadMessage()
|
|
if err != nil {
|
|
log.Println("[ERROR] Read error:", err)
|
|
return err
|
|
}
|
|
messageChan <- message
|
|
messageTypeChan <- messageType
|
|
|
|
// 启动一个新的goroutine来处理消息
|
|
go func() {
|
|
defer func() {
|
|
// 处理完成后从通道中移除消息
|
|
<-messageChan
|
|
<-messageTypeChan
|
|
}()
|
|
processMessage(messageType, message)
|
|
}()
|
|
}
|
|
}
|
|
|
|
// processMessage 处理接收到的消息
|
|
func processMessage(messageType int, message []byte) {
|
|
if messageType != websocket.TextMessage {
|
|
log.Println("[INFO] Invalid message type:", messageType)
|
|
return
|
|
}
|
|
//message json解析
|
|
var messageMap map[string]interface{}
|
|
err := json.Unmarshal(message, &messageMap)
|
|
if err != nil {
|
|
log.Println("[ERROR] Unmarshal error:", err)
|
|
return
|
|
}
|
|
// 处理接收到的消息
|
|
messageTypeStr := messageMap["post_type"]
|
|
switch messageTypeStr {
|
|
case "message":
|
|
{
|
|
// 处理message消息
|
|
HandleMessage(message)
|
|
return
|
|
}
|
|
case "notice":
|
|
{
|
|
// 处理notice消息
|
|
HandleNotice(message)
|
|
return
|
|
}
|
|
case "request":
|
|
{
|
|
// 处理request消息
|
|
HandleRequest(message)
|
|
return
|
|
}
|
|
case "meta_event":
|
|
{
|
|
// 处理meta_event消息
|
|
HandleMetaEvent(message)
|
|
return
|
|
}
|
|
default:
|
|
{
|
|
// 打印接收到的消息
|
|
log.Printf("[WARN] Received message: %s", message)
|
|
}
|
|
}
|
|
}
|
|
|
|
// wsSendMessage 向WebSocket服务器发送消息并返回发送状态
|
|
func wsSendMessage(message []byte) error {
|
|
// 解析连接URL
|
|
u, err := url.Parse(fmt.Sprintf("%v/api", gProtocolAddr))
|
|
if err != nil {
|
|
return fmt.Errorf("无效的URL: %v", err)
|
|
}
|
|
|
|
// 建立连接
|
|
conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
|
|
if err != nil {
|
|
return fmt.Errorf("连接失败: %v", err)
|
|
}
|
|
defer func(conn *websocket.Conn) {
|
|
err := conn.Close()
|
|
if err != nil {
|
|
log.Println("[ERROR] Close error:", err)
|
|
}
|
|
}(conn)
|
|
|
|
// 发送消息
|
|
err = conn.WriteMessage(websocket.TextMessage, message)
|
|
if err != nil {
|
|
return fmt.Errorf("发送消息失败: %v", err)
|
|
}
|
|
|
|
return nil
|
|
}
|