From 124c00c871fffbfa20d1e2ea1aa522ed1b887916 Mon Sep 17 00:00:00 2001 From: Utkarsh Shukla Date: Wed, 4 Dec 2024 10:16:55 +0530 Subject: [PATCH] Minor edit --- app/client/main.go | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/app/client/main.go b/app/client/main.go index 3d87f54..69da384 100644 --- a/app/client/main.go +++ b/app/client/main.go @@ -51,7 +51,6 @@ import ( "google.golang.org/grpc/keepalive" "google.golang.org/grpc/metadata" "gopkg.in/yaml.v3" - "gorm.io/gorm/logger" ) const ( @@ -124,11 +123,11 @@ func getHeaderContext(ctx context.Context, timeout time.Duration) (context.Conte } func waitForRequest(ctx context.Context, c pb.TunnelServiceClient) error { + ctx, logger := loggerFromContext(ctx) + ctx, cancel := getHeaderContext(ctx, 0) logger.Info("Entered waitForRequest") logger.Infof("Entered waitForRequest") logger.Infow("Entered waitForRequest") - ctx, logger := loggerFromContext(ctx) - ctx, cancel := getHeaderContext(ctx, 0) defer cancel() stream, err := c.WaitForRequest(ctx, &pb.WaitForRequestArgs{}) if err != nil {