feat: 初始化多机器人服务端项目框架
基于Go语言构建多机器人服务端框架，包含配置管理、事件总线、依赖注入等核心模块 添加项目基础结构、README、gitignore和初始代码实现
This commit is contained in:
165
pkg/net/server.go
Normal file
165
pkg/net/server.go
Normal file
@@ -0,0 +1,165 @@
|
||||
package net
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
|
||||
"cellbot/internal/engine"
|
||||
"cellbot/internal/protocol"
|
||||
"github.com/valyala/fasthttp"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
// Server HTTP服务器
|
||||
type Server struct {
|
||||
host string
|
||||
port int
|
||||
logger *zap.Logger
|
||||
botManager *protocol.BotManager
|
||||
eventBus *engine.EventBus
|
||||
server *fasthttp.Server
|
||||
}
|
||||
|
||||
// NewServer 创建HTTP服务器
|
||||
func NewServer(host string, port int, logger *zap.Logger, botManager *protocol.BotManager, eventBus *engine.EventBus) *Server {
|
||||
return &Server{
|
||||
host: host,
|
||||
port: port,
|
||||
logger: logger.Named("server"),
|
||||
botManager: botManager,
|
||||
eventBus: eventBus,
|
||||
}
|
||||
}
|
||||
|
||||
// Start 启动服务器
|
||||
func (s *Server) Start() error {
|
||||
s.server = &fasthttp.Server{
|
||||
Handler: s.requestHandler,
|
||||
MaxConnsPerIP: 1000,
|
||||
MaxRequestsPerConn: 1000,
|
||||
ReadTimeout: 0,
|
||||
WriteTimeout: 0,
|
||||
IdleTimeout: 0,
|
||||
DisableKeepalive: false,
|
||||
}
|
||||
|
||||
addr := net.JoinHostPort(s.host, strconv.Itoa(s.port))
|
||||
s.logger.Info("Starting HTTP server", zap.String("address", addr))
|
||||
|
||||
go func() {
|
||||
if err := s.server.ListenAndServe(addr); err != nil {
|
||||
s.logger.Error("Server error", zap.Error(err))
|
||||
}
|
||||
}()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Stop 停止服务器
|
||||
func (s *Server) Stop() error {
|
||||
if s.server != nil {
|
||||
s.logger.Info("Stopping HTTP server")
|
||||
return s.server.Shutdown()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// requestHandler 请求处理器
|
||||
func (s *Server) requestHandler(ctx *fasthttp.RequestCtx) {
|
||||
path := string(ctx.Path())
|
||||
method := string(ctx.Method())
|
||||
|
||||
s.logger.Debug("Received request",
|
||||
zap.String("path", path),
|
||||
zap.String("method", method))
|
||||
|
||||
// 路由处理
|
||||
switch path {
|
||||
case "/":
|
||||
s.handleRoot(ctx)
|
||||
case "/health":
|
||||
s.handleHealth(ctx)
|
||||
case "/bots":
|
||||
s.handleBots(ctx)
|
||||
case "/bots/create":
|
||||
s.handleCreateBot(ctx)
|
||||
case "/events/publish":
|
||||
s.handlePublishEvent(ctx)
|
||||
case "/events/subscribe":
|
||||
s.handleSubscribeEvent(ctx)
|
||||
default:
|
||||
ctx.Error("Not Found", fasthttp.StatusNotFound)
|
||||
}
|
||||
}
|
||||
|
||||
// handleRoot 根路径处理
|
||||
func (s *Server) handleRoot(ctx *fasthttp.RequestCtx) {
|
||||
ctx.SetContentType("application/json")
|
||||
ctx.SetBodyString(`{"message":"CellBot Server","version":"1.0.0"}`)
|
||||
}
|
||||
|
||||
// handleHealth 健康检查
|
||||
func (s *Server) handleHealth(ctx *fasthttp.RequestCtx) {
|
||||
ctx.SetContentType("application/json")
|
||||
ctx.SetBodyString(`{"status":"ok"}`)
|
||||
}
|
||||
|
||||
// handleBots 获取机器人列表
|
||||
func (s *Server) handleBots(ctx *fasthttp.RequestCtx) {
|
||||
bots := s.botManager.GetAll()
|
||||
ctx.SetContentType("application/json")
|
||||
|
||||
if len(bots) == 0 {
|
||||
ctx.SetBodyString(`{"bots":[]}`)
|
||||
return
|
||||
}
|
||||
|
||||
// 简化实现,实际应该序列化完整信息
|
||||
response := `{"bots":[`
|
||||
for i, bot := range bots {
|
||||
if i > 0 {
|
||||
response += ","
|
||||
}
|
||||
response += fmt.Sprintf(`{"id":"%s","name":"%s","status":"%s"}`,
|
||||
bot.GetID(), bot.Name(), bot.GetStatus())
|
||||
}
|
||||
response += `]}`
|
||||
|
||||
ctx.SetBodyString(response)
|
||||
}
|
||||
|
||||
// handleCreateBot 创建机器人
|
||||
func (s *Server) handleCreateBot(ctx *fasthttp.RequestCtx) {
|
||||
if string(ctx.Method()) != "POST" {
|
||||
ctx.Error("Method Not Allowed", fasthttp.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
ctx.SetContentType("application/json")
|
||||
ctx.SetBodyString(`{"message":"Bot creation not implemented yet"}`)
|
||||
}
|
||||
|
||||
// handlePublishEvent 发布事件
|
||||
func (s *Server) handlePublishEvent(ctx *fasthttp.RequestCtx) {
|
||||
if string(ctx.Method()) != "POST" {
|
||||
ctx.Error("Method Not Allowed", fasthttp.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
// 解析请求体并发布事件
|
||||
// 这里简化实现,实际应该解析JSON并创建Event
|
||||
ctx.SetContentType("application/json")
|
||||
ctx.SetBodyString(`{"message":"Event published"}`)
|
||||
}
|
||||
|
||||
// handleSubscribeEvent 订阅事件
|
||||
func (s *Server) handleSubscribeEvent(ctx *fasthttp.RequestCtx) {
|
||||
if string(ctx.Method()) != "GET" {
|
||||
ctx.Error("Method Not Allowed", fasthttp.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
ctx.SetContentType("application/json")
|
||||
ctx.SetBodyString(`{"message":"Event subscription not implemented yet"}`)
|
||||
}
|
||||
244
pkg/net/websocket.go
Normal file
244
pkg/net/websocket.go
Normal file
@@ -0,0 +1,244 @@
|
||||
package net
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net/url"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"cellbot/internal/engine"
|
||||
"cellbot/internal/protocol"
|
||||
|
||||
"github.com/fasthttp/websocket"
|
||||
"github.com/valyala/fasthttp"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
// WebSocketManager WebSocket连接管理器
|
||||
type WebSocketManager struct {
|
||||
connections map[string]*WebSocketConnection
|
||||
logger *zap.Logger
|
||||
eventBus *engine.EventBus
|
||||
mu sync.RWMutex
|
||||
upgrader *websocket.FastHTTPUpgrader
|
||||
}
|
||||
|
||||
// NewWebSocketManager 创建WebSocket管理器
|
||||
func NewWebSocketManager(logger *zap.Logger, eventBus *engine.EventBus) *WebSocketManager {
|
||||
return &WebSocketManager{
|
||||
connections: make(map[string]*WebSocketConnection),
|
||||
logger: logger.Named("websocket"),
|
||||
eventBus: eventBus,
|
||||
upgrader: &websocket.FastHTTPUpgrader{
|
||||
ReadBufferSize: 1024,
|
||||
WriteBufferSize: 1024,
|
||||
CheckOrigin: func(ctx *fasthttp.RequestCtx) bool {
|
||||
return true // 允许所有来源，生产环境应加强检查
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// WebSocketConnection WebSocket连接
|
||||
type WebSocketConnection struct {
|
||||
ID string
|
||||
Conn *websocket.Conn
|
||||
BotID string
|
||||
Logger *zap.Logger
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
}
|
||||
|
||||
// NewWebSocketConnection 创建WebSocket连接
|
||||
func NewWebSocketConnection(conn *websocket.Conn, botID string, logger *zap.Logger) *WebSocketConnection {
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
connID := generateConnID()
|
||||
return &WebSocketConnection{
|
||||
ID: connID,
|
||||
Conn: conn,
|
||||
BotID: botID,
|
||||
Logger: logger.With(zap.String("conn_id", connID)),
|
||||
ctx: ctx,
|
||||
cancel: cancel,
|
||||
}
|
||||
}
|
||||
|
||||
// UpgradeWebSocket 升级HTTP连接为WebSocket
|
||||
func (wsm *WebSocketManager) UpgradeWebSocket(ctx *fasthttp.RequestCtx) (*WebSocketConnection, error) {
|
||||
// 获取查询参数
|
||||
botID := string(ctx.QueryArgs().Peek("bot_id"))
|
||||
if botID == "" {
|
||||
return nil, fmt.Errorf("missing bot_id parameter")
|
||||
}
|
||||
|
||||
// 创建通道用于传递连接
|
||||
connChan := make(chan *websocket.Conn, 1)
|
||||
|
||||
// 升级连接
|
||||
err := wsm.upgrader.Upgrade(ctx, func(conn *websocket.Conn) {
|
||||
connChan <- conn
|
||||
})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to upgrade connection: %w", err)
|
||||
}
|
||||
|
||||
// 等待连接在回调中建立
|
||||
conn := <-connChan
|
||||
|
||||
// 创建连接对象
|
||||
wsConn := NewWebSocketConnection(conn, botID, wsm.logger)
|
||||
|
||||
// 存储连接
|
||||
wsm.mu.Lock()
|
||||
wsm.connections[wsConn.ID] = wsConn
|
||||
wsm.mu.Unlock()
|
||||
|
||||
wsm.logger.Info("WebSocket connection established",
|
||||
zap.String("conn_id", wsConn.ID),
|
||||
zap.String("bot_id", botID))
|
||||
|
||||
// 启动读取循环
|
||||
go wsConn.readLoop(wsm.eventBus)
|
||||
|
||||
return wsConn, nil
|
||||
}
|
||||
|
||||
// readLoop 读取循环
|
||||
func (wsc *WebSocketConnection) readLoop(eventBus *engine.EventBus) {
|
||||
defer wsc.close()
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-wsc.ctx.Done():
|
||||
return
|
||||
default:
|
||||
messageType, message, err := wsc.Conn.ReadMessage()
|
||||
if err != nil {
|
||||
wsc.Logger.Error("Failed to read message", zap.Error(err))
|
||||
return
|
||||
}
|
||||
|
||||
// 处理消息
|
||||
wsc.handleMessage(message, eventBus)
|
||||
// messageType 可用于区分文本或二进制消息
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// handleMessage 处理消息
|
||||
func (wsc *WebSocketConnection) handleMessage(data []byte, eventBus *engine.EventBus) {
|
||||
wsc.Logger.Debug("Received message", zap.ByteString("data", data))
|
||||
|
||||
// TODO: 解析消息为Event对象
|
||||
// 这里简化实现,实际应该根据协议解析
|
||||
event := &protocol.BaseEvent{
|
||||
Type: protocol.EventTypeMessage,
|
||||
DetailType: "private",
|
||||
Timestamp: time.Now().Unix(),
|
||||
SelfID: wsc.BotID,
|
||||
Data: make(map[string]interface{}),
|
||||
}
|
||||
|
||||
// 发布到事件总线
|
||||
eventBus.Publish(event)
|
||||
}
|
||||
|
||||
// SendMessage 发送消息
|
||||
func (wsc *WebSocketConnection) SendMessage(data []byte) error {
|
||||
wsc.Logger.Debug("Sending message", zap.ByteString("data", data))
|
||||
|
||||
err := wsc.Conn.WriteMessage(websocket.TextMessage, data)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to send message: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// close 关闭连接
|
||||
func (wsc *WebSocketConnection) close() {
|
||||
wsc.cancel()
|
||||
if err := wsc.Conn.Close(); err != nil {
|
||||
wsc.Logger.Error("Failed to close connection", zap.Error(err))
|
||||
}
|
||||
}
|
||||
|
||||
// RemoveConnection 移除连接
|
||||
func (wsm *WebSocketManager) RemoveConnection(connID string) {
|
||||
wsm.mu.Lock()
|
||||
defer wsm.mu.Unlock()
|
||||
|
||||
if conn, ok := wsm.connections[connID]; ok {
|
||||
conn.close()
|
||||
delete(wsm.connections, connID)
|
||||
wsm.logger.Info("WebSocket connection removed", zap.String("conn_id", connID))
|
||||
}
|
||||
}
|
||||
|
||||
// GetConnection 获取连接
|
||||
func (wsm *WebSocketManager) GetConnection(connID string) (*WebSocketConnection, bool) {
|
||||
wsm.mu.RLock()
|
||||
defer wsm.mu.RUnlock()
|
||||
conn, ok := wsm.connections[connID]
|
||||
return conn, ok
|
||||
}
|
||||
|
||||
// GetConnectionByBotID 根据BotID获取连接
|
||||
func (wsm *WebSocketManager) GetConnectionByBotID(botID string) []*WebSocketConnection {
|
||||
wsm.mu.RLock()
|
||||
defer wsm.mu.RUnlock()
|
||||
|
||||
connections := make([]*WebSocketConnection, 0)
|
||||
for _, conn := range wsm.connections {
|
||||
if conn.BotID == botID {
|
||||
connections = append(connections, conn)
|
||||
}
|
||||
}
|
||||
return connections
|
||||
}
|
||||
|
||||
// BroadcastToBot 向指定Bot的所有连接广播消息
|
||||
func (wsm *WebSocketManager) BroadcastToBot(botID string, data []byte) {
|
||||
connections := wsm.GetConnectionByBotID(botID)
|
||||
for _, conn := range connections {
|
||||
if err := conn.SendMessage(data); err != nil {
|
||||
wsm.logger.Error("Failed to send message to connection",
|
||||
zap.String("conn_id", conn.ID),
|
||||
zap.Error(err))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Dial 建立WebSocket客户端连接
|
||||
func (wsm *WebSocketManager) Dial(addr string, botID string) (*WebSocketConnection, error) {
|
||||
u, err := url.Parse(addr)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("invalid URL: %w", err)
|
||||
}
|
||||
|
||||
conn, _, err := websocket.DefaultDialer.Dial(addr, nil)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to dial: %w", err)
|
||||
}
|
||||
|
||||
wsConn := NewWebSocketConnection(conn, botID, wsm.logger)
|
||||
|
||||
wsm.mu.Lock()
|
||||
wsm.connections[wsConn.ID] = wsConn
|
||||
wsm.mu.Unlock()
|
||||
|
||||
wsm.logger.Info("WebSocket client connected",
|
||||
zap.String("conn_id", wsConn.ID),
|
||||
zap.String("bot_id", botID),
|
||||
zap.String("addr", addr))
|
||||
|
||||
// 启动读取循环
|
||||
go wsConn.readLoop(wsm.eventBus)
|
||||
|
||||
return wsConn, nil
|
||||
}
|
||||
|
||||
// generateConnID 生成连接ID
|
||||
func generateConnID() string {
|
||||
return fmt.Sprintf("conn-%d", time.Now().UnixNano())
|
||||
}
|
||||
Reference in New Issue
Block a user