v0.0.18.5 Добавление логирование для ws и stream
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
parent
93f3a68de9
commit
7cbf071532
@ -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
|
||||
}
|
||||
|
||||
@ -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)
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user