test different context

This commit is contained in:
Pascal Fischer 2024-08-08 18:03:17 +02:00
parent a10023f7e1
commit d98df551c0

View File

@ -6,7 +6,6 @@ import (
"io" "io"
"time" "time"
"github.com/netbirdio/signal-dispatcher/dispatcher"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"go.opentelemetry.io/otel/attribute" "go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/metric" "go.opentelemetry.io/otel/metric"
@ -14,6 +13,8 @@ import (
"google.golang.org/grpc/metadata" "google.golang.org/grpc/metadata"
"google.golang.org/grpc/status" "google.golang.org/grpc/status"
"github.com/netbirdio/signal-dispatcher/dispatcher"
"github.com/netbirdio/netbird/signal/metrics" "github.com/netbirdio/netbird/signal/metrics"
"github.com/netbirdio/netbird/signal/peer" "github.com/netbirdio/netbird/signal/peer"
"github.com/netbirdio/netbird/signal/proto" "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 &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 // ConnectStream connects to the exchange stream