v0.0.18.5 Добавление логирование для ws и stream
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
madipo2611 2025-08-17 22:14:15 +03:00
parent 93f3a68de9
commit 7cbf071532
3 changed files with 7 additions and 0 deletions

View File

@ -3,6 +3,7 @@ package graph
import (
"context"
"fmt"
"log"
"tailly_back_v2/internal/domain"
"tailly_back_v2/proto"
"time"
@ -128,15 +129,18 @@ func (r *subscriptionResolver) MessageStream(ctx context.Context, userID int) (<
stream, err := r.MessageClient.StreamMessages(ctx, &proto.StreamMessagesRequest{
UserId: int32(userID),
})
log.Printf("stream messages: %v", stream)
if err != nil {
return nil, fmt.Errorf("failed to stream messages: %w", err)
}
messageChan := make(chan *domain.Message)
log.Printf("messageChan: %d", messageChan)
go func() {
defer close(messageChan)
for {
msg, err := stream.Recv()
log.Printf("recv message: %v", msg)
if err != nil {
return
}

View File

@ -2,6 +2,7 @@ package middleware
import (
"context"
"log"
"net/http"
"tailly_back_v2/pkg/auth"
)
@ -14,6 +15,7 @@ func WSAuthMiddleware(tokenAuth *auth.TokenAuth) func(http.Handler) http.Handler
if r.Header.Get("Upgrade") == "websocket" {
// Извлекаем токен из query параметров или заголовков
token := extractTokenFromRequest(r)
log.Printf("token ws: %s", token)
if token != "" {
userID, err := tokenAuth.ValidateAccessToken(token)

View File

@ -76,6 +76,7 @@ func (s *Server) configureRouter() {
Upgrader: websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
origin := r.Header.Get("Origin")
log.Printf("origin: %s", origin)
for _, allowed := range allowedOrigins {
if origin == allowed {
return true