From 7fa706d53d934f6ec540378b4438151b15e68332 Mon Sep 17 00:00:00 2001
From: kenriortega <kenriortega@gmail.com>
Date: Thu, 11 Nov 2021 09:42:16 -0500
Subject: [PATCH] ref: add log formating to grpc cmd

---
 cmd/cli/grpcxy.go    | 15 ++++++++-------
 pkg/logger/logger.go |  5 +----
 2 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/cmd/cli/grpcxy.go b/cmd/cli/grpcxy.go
index afb9ccf..0653af6 100644
--- a/cmd/cli/grpcxy.go
+++ b/cmd/cli/grpcxy.go
@@ -3,13 +3,13 @@ package cli
 import (
 	"context"
 	"fmt"
-	"log"
 	"net"
 	"os"
 	"os/signal"
 	"strings"
 	"time"
 
+	"github.com/kenriortega/ngonx/pkg/errors"
 	"github.com/kenriortega/ngonx/pkg/logger"
 	"github.com/spf13/cobra"
 	"github.com/talos-systems/grpc-proxy/proxy"
@@ -29,7 +29,7 @@ var grpcCmd = &cobra.Command{
 
 		lis, err := net.Listen("tcp", configFromYaml.GrpcProxy.Listener)
 		if err != nil {
-			log.Fatalf("failed to listen: %v", err)
+			logger.LogError(errors.Errorf("grpc: failed to listen: %v", err).Error())
 		}
 
 		logger.LogInfo(fmt.Sprintf("Proxy running at %q\n", configFromYaml.GrpcProxy.Listener))
@@ -43,7 +43,7 @@ var grpcCmd = &cobra.Command{
 						creds, sslErr := credentials.NewClientTLSFromFile(
 							configFromYaml.GrpcClientCert, "")
 						if sslErr != nil {
-							log.Fatalf("Failed to parse credentials: %v", sslErr)
+							logger.LogError(errors.Errorf("grpc: failed to parse credentials: %v", sslErr).Error())
 						}
 						conn, err := grpc.DialContext(
 							ctx,
@@ -94,7 +94,7 @@ var grpcCmd = &cobra.Command{
 				configFromYaml.GrpcSSL.KeyFile,
 			)
 			if sslErr != nil {
-				log.Fatalf("Failed to parse credentials: %v", sslErr)
+				logger.LogError(errors.Errorf("grpc: failed to parse credentials: %v", sslErr).Error())
 				return
 			}
 			opts = append(opts, grpc.Creds(creds))
@@ -104,7 +104,7 @@ var grpcCmd = &cobra.Command{
 
 		go gracefulShutdown(server)
 		if err := server.Serve(lis); err != nil {
-			log.Fatalf("failed to serve: %v", err)
+			logger.LogError(errors.Errorf("grpc: failed to serve: %v", err).Error())
 		}
 	},
 }
@@ -119,11 +119,12 @@ func gracefulShutdown(server *grpc.Server) {
 
 	signal.Notify(quit, os.Interrupt)
 	sig := <-quit
-	logger.LogInfo(fmt.Sprintf("server is shutting down %s", sig.String()))
+	logger.LogWarn(fmt.Sprintf("grpc: server is shutting down %s", sig.String()))
 
 	_, cancel := context.WithTimeout(context.Background(), 30*time.Second)
 	defer cancel()
 
 	server.GracefulStop()
-	logger.LogInfo("server stopped")
+	logger.LogWarn(fmt.Sprintf("grpc: server is shutting down %s", sig.String()))
+
 }
diff --git a/pkg/logger/logger.go b/pkg/logger/logger.go
index d658005..11f0aae 100644
--- a/pkg/logger/logger.go
+++ b/pkg/logger/logger.go
@@ -9,15 +9,12 @@ import (
 var log *zap.Logger
 
 func init() {
-	// var err error
-	// config := zap.NewProductionConfig()
+
 	encoderConfig := zap.NewProductionEncoderConfig()
 	encoderConfig.TimeKey = "timestamp"
 	encoderConfig.StacktraceKey = ""
 	encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
 
-	// lumberjack.Logger is already safe for concurrent use, so we don't need to
-	// lock it.
 	w := zapcore.AddSync(&lumberjack.Logger{
 		Filename:   "./ngonx-log/ngonx.log",
 		MaxSize:    500, // megabytes