update logger

grpcutil/middlewares/gerr/gerr_middleware.go

@@ -15,8 +15,7 @@ import (
 	"google.golang.org/grpc/status"
 )
 
-type ErrorMiddleware struct {
-}
+type ErrorMiddleware struct{}
 
 func NewErrorMiddleware() *ErrorMiddleware {
 	return &ErrorMiddleware{}
@@ -50,6 +49,10 @@ func (m *ErrorMiddleware) handleError(ctx context.Context, handlerErr error) err
 	}
 
 	log := zerolog.Ctx(ctx)
+	if errorLogger := errors.Logger(handlerErr); errorLogger != nil {
+		log = errorLogger
+	}
+
 	if log == nil {
 		fmt.Printf("ERROR - ZEROLOG LOGGER MISSING FROM CONTEXT: %v\n", handlerErr)
 		return status.New(codes.Internal, "internal logging error, please contact the administrator").Err()
@@ -70,11 +73,6 @@ func (m *ErrorMiddleware) handleError(ctx context.Context, handlerErr error) err
 		asertoErr = errors.ErrUnknown
 	}
 
-	errorLogger := errors.Logger(handlerErr)
-	if errorLogger != nil {
-		log = errorLogger
-	}
-
 	asertoErr = asertoErr.Int(errors.HTTPStatusErrorMetadata, asertoErr.HTTPCode)
 
 	log.Warn().Stack().Err(handlerErr).