message-server/main.go

// WebSocket 聊天应用主程序
// 提供基本的 HTTP 服务器和 WebSocket 连接处理
package main

import (
	"chatapp/logger"
	"chatapp/wsClient"
	"flag"
	"fmt"
	"net/http"
	"os"
	"strconv"

	"github.com/joho/godotenv"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
)

func main() {
	// 加载 .env 文件
	if err := godotenv.Load(); err != nil {
		panic("没有找到 .env 文件，将使用系统环境变量和默认值")
	}

	logMode := os.Getenv("LOG_MODE")
	if logMode == "" {
		logMode = "color" // 默认开发环境
	}
	//logLevel, err := strconv.ParseInt(os.Getenv("LOG_LEVEL"), 10, 8)
	logLevel, err := strconv.Atoi(os.Getenv("LOG_LEVEL"))
	if err != nil {
		log.Error().Err(err).Msg("获取 log level 失败，使用默认值 0")
		logLevel = 0 // 默认 debug log level.
	}
	logger.InitLogger(logMode, zerolog.Level(logLevel))

	var addr = flag.String("addr", ":8080", "Addr of the App")
	flag.Parse()

	// 检查是否有服务器地址的环境变量设置
	if serverAddr := os.Getenv("SERVER_ADDR"); serverAddr != "" && *addr == ":8080" {
		*addr = serverAddr
	}

	// 初始化Redis连接
	if err := wsClient.InitializeRedisFromEnv(); err != nil {
		log.Error().Err(err).Msg("Redis初始化失败")
		log.Info().Msg("将继续运行,但不会存储消息到Redis")
	} else {
		// 确保程序退出时关闭Redis连接
		defer wsClient.CloseRedis()
	}

	// Handle all websocket connections for chat rooms dynamically
	http.HandleFunc("/room", func(w http.ResponseWriter, r *http.Request) {
		log.Info().Msg("--------------------------------")
		log.Info().Msg("Received request for room")
		// 从WebSocket子协议中预解析房间名称
		protocols := r.Header["Sec-Websocket-Protocol"]
		if len(protocols) == 0 {
			log.Warn().Msg("No protocols found")
			return
		}

		// 解析房间参数
		roomParams, err := wsClient.ParseWebSocketProtocols(protocols)
		if err != nil {
			log.Error().Err(err).Msg("解析WebSocket协议参数失败")
			http.Error(w, "Invalid WebSocket protocols", http.StatusBadRequest)
			return
		}

		// 组合房间名称
		roomName := fmt.Sprintf("custom_id-%dANDshop_id-%d", roomParams.CustomId, roomParams.ShopId)
		if roomName == "" {
			http.Error(w, "Room name required in WebSocket protocols or URL parameter", http.StatusBadRequest)
		}
		roomParams.RoomName = roomName
		// log.Printf("---- 房间名称: %s\n\n", roomName)
		log.Info().Str("roomName", roomName).Msg("---- 房间名称")

		// 获取(或创建)指定名称的房间，并启动房间的消息处理协程
		room := wsClient.GetRoomAndRun(roomName)
		// 将 WebSocket 连接交给房间处理，使用解析好的参数避免重复解析
		room.HttpServe(w, r, roomParams)
	})

	// 启动 HTTP 服务器
	log.Info().Str("addr", *addr).Msg("正在启动 Web 服务器")
	// 开始监听并处理 HTTP 请求，如果启动失败则退出程序
	if err := http.ListenAndServe(*addr, nil); err != nil {
		log.Fatal().Err(err).Msg("服务器启动失败")
	}
}