message-server/main.go

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

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

	"github.com/joho/godotenv"
)

func main() {
	// 加载 .env 文件
	if err := godotenv.Load(); err != nil {
		panic("没有找到 .env 文件，将使用系统环境变量和默认值")
	}
	
	var addr = flag.String("addr", ":8080", "Addr of the App")
	flag.Parse()

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

	// 初始化Redis连接
	// 从环境变量获取Redis配置，如果没有设置则使用默认值
	redisAddr := os.Getenv("REDIS_ADDR")
	if redisAddr == "" {
		redisAddr = "localhost:6379" // 默认Redis地址
	}
	
	redisPassword := os.Getenv("REDIS_PASSWORD") // 默认为空
	
	redisDB := 0 // 默认数据库
	if dbStr := os.Getenv("REDIS_DB"); dbStr != "" {
		if db, err := strconv.Atoi(dbStr); err == nil {
			redisDB = db
		}
	}

	// 初始化Redis连接
	if err := wsClient.InitRedis(redisAddr, redisPassword, redisDB); err != nil {
		log.Printf("Redis初始化失败: %v", err)
		log.Println("将继续运行，但不会存储消息到Redis")
	} else {
		// 确保程序退出时关闭Redis连接
		defer wsClient.CloseRedis()
	}

	// 设置静态文件服务器，处理 CSS、JS 等静态资源
	// StripPrefix 会移除 URL 前缀 "/static/"，然后从 "static" 目录提供文件
	http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static"))))
	
	// 主页路由，渲染 index.html 模板
	http.Handle("/", &templates.TemplateHandler{Filename: "index.html"})

	// 聊天页面路由，渲染 chat.html 模板
	http.Handle("/chat", &templates.TemplateHandler{Filename: "chat.html"})

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

		// 解析房间参数
		roomParams, err := wsClient.ParseWebSocketProtocols(protocols)
		if err != nil {
			log.Printf("解析WebSocket协议参数失败: %v", err)
			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)

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

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