dockerfile修改

cmd/server/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"ChatRoom/internal/rabbitmq"
+	"ChatRoom/internal/redis"
 	"ChatRoom/internal/ws"
 	"log"
 	"net/http"
@@ -14,16 +15,27 @@ func main() {
 	rabbitmqURL := os.Getenv("RABBITMQ_URL")
 	if rabbitmqURL == "" {
 		rabbitmqURL = "amqp://guest:guest@localhost:5672/"
-	} 
+	}
 	rmq, err := rabbitmq.NewClient(rabbitmqURL)
 	if err != nil {
 		log.Fatalf("RabbitMQ 连接失败: %v", err)
 	}
 	defer rmq.Close()
 
+	// 1.5 初始化 Redis
+	redisAddr := os.Getenv("REDIS_ADDR")
+	if redisAddr == "" {
+		redisAddr = "localhost:6379"
+	}
+	redisClient, err := redis.NewClient(redisAddr, "", 0)
+	if err != nil {
+		log.Fatalf("Redis 连接失败: %v", err)
+	}
+	defer redisClient.Close()
+
 	// 2. WebSocket 路由
 	http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
-		ws.NewConnection(w, r, rmq) // 传入 RabbitMQ 客户端
+		ws.NewConnection(w, r, rmq, redisClient) // 传入 RabbitMQ 和 Redis 客户端
 	})
 
 	// 3. 静态文件服务（用于测试前端）