v0.0.18.8 Добавлен отдельный хэндлер для вебсокета

internal/http/server.go

@@ -17,6 +17,7 @@ import (
 	"tailly_back_v2/internal/http/middleware"
 	"tailly_back_v2/internal/service"
 	"tailly_back_v2/pkg/auth"
+	"time"
 )
 
 type Server struct {
@@ -64,32 +65,40 @@ func (s *Server) configureRouter() {
 	s.router.Use(middleware.MetricsMiddleware)
 	s.router.Use(middleware.CORS(allowedOrigins))
 
-	s.router.Use(middleware.WSAuthMiddleware(s.tokenAuth))
-	s.router.Use(middleware.AuthMiddleware(s.tokenAuth))
-
+	// Основной GraphQL обработчик
 	resolver := graph.NewResolver(s.services, s.db, s.services.Messages)
 	srv := handler.NewDefaultServer(graph.NewExecutableSchema(graph.Config{
 		Resolvers: resolver,
 	}))
 
-	wsTransport := transport.Websocket{
-		Upgrader: websocket.Upgrader{
-			CheckOrigin: func(r *http.Request) bool {
-				origin := r.Header.Get("Origin")
-				for _, allowed := range allowedOrigins {
-					if origin == allowed {
-						return true
-					}
+	// Настройка WebSocket транспорта
+	wsUpgrader := websocket.Upgrader{
+		CheckOrigin: func(r *http.Request) bool {
+			origin := r.Header.Get("Origin")
+			for _, allowed := range allowedOrigins {
+				if origin == allowed {
+					return true
 				}
-				return false
-			},
+			}
+			return false
 		},
 	}
 
-	srv.AddTransport(&wsTransport)
+	srv.AddTransport(&transport.Websocket{
+		Upgrader:              wsUpgrader,
+		KeepAlivePingInterval: 15 * time.Second,
+	})
 
+	// Раздельные эндпоинты:
+
+	// 1. Обычные HTTP GraphQL запросы (только POST)
+	s.router.With(middleware.AuthMiddleware(s.tokenAuth)).Post("/query", srv.ServeHTTP)
+
+	// 2. WebSocket подписки (отдельный эндпоинт)
+	s.router.With(middleware.WSAuthMiddleware(s.tokenAuth)).Get("/ws", srv.ServeHTTP)
+
+	// 3. Playground и другие обработчики
 	s.router.Handle("/", playground.Handler("GraphQL playground", "/query"))
-	s.router.Handle("/query", srv)
 	s.router.Handle("/uploads/*", http.StripPrefix("/uploads/", http.FileServer(http.Dir("./uploads"))))
 }
 
@@ -105,9 +114,12 @@ func (s *Server) configureMetrics() {
 }
 
 func (s *Server) Run() error {
+	log.Printf("Server running on %s:%s", s.cfg.Server.Host, s.cfg.Server.Port)
+	log.Printf("WebSocket subscriptions at /ws")
 	return http.ListenAndServe(s.cfg.Server.Host+":"+s.cfg.Server.Port, s.router)
 }
 
 func (s *Server) Shutdown(ctx context.Context) error {
+	// Graceful shutdown logic
 	return nil
 }