From d98df551c0ec0423a4ff231a9a900a6bd4e473a4 Mon Sep 17 00:00:00 2001 From: Pascal Fischer Date: Thu, 8 Aug 2024 18:03:17 +0200 Subject: [PATCH] test different context --- signal/server/signal.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/signal/server/signal.go b/signal/server/signal.go index a58ef436c..ee67fb62c 100644 --- a/signal/server/signal.go +++ b/signal/server/signal.go @@ -6,7 +6,6 @@ import ( "io" "time" - "github.com/netbirdio/signal-dispatcher/dispatcher" log "github.com/sirupsen/logrus" "go.opentelemetry.io/otel/attribute" "go.opentelemetry.io/otel/metric" @@ -14,6 +13,8 @@ import ( "google.golang.org/grpc/metadata" "google.golang.org/grpc/status" + "github.com/netbirdio/signal-dispatcher/dispatcher" + "github.com/netbirdio/netbird/signal/metrics" "github.com/netbirdio/netbird/signal/peer" "github.com/netbirdio/netbird/signal/proto" @@ -75,7 +76,12 @@ func (s *Server) Send(ctx context.Context, msg *proto.EncryptedMessage) (*proto. return &proto.EncryptedMessage{}, nil } - return s.dispatcher.SendMessage(ctx, msg) + // if _, found := s.registry.Get(msg.RemoteKey); found { + // s.forwardMessageToPeer(ctx, msg) + // return &proto.EncryptedMessage{}, nil + // } + + return s.dispatcher.SendMessage(context.Background(), msg) } // ConnectStream connects to the exchange stream