refactor: init log

cmd/hub/main.go

@@ -1,25 +1,4 @@
 package main
 
-import (
-	"log"
-	"net"
-
-	"github.com/lorsanstand/HomeOps-Hub/api/gen/homeops"
-	grpcserver "github.com/lorsanstand/HomeOps-Hub/internal/hub/grpc"
-	"google.golang.org/grpc"
-)
-
 func main() {
-	lis, err := net.Listen("tcp", ":6756")
-	if err != nil {
-		return
-	}
-
-	grpcServer := grpc.NewServer()
-
-	srv := &grpcserver.Server{}
-	homeops.RegisterHubServer(grpcServer, srv)
-
-	log.Println("Start serve")
-	grpcServer.Serve(lis)
 }

internal/hub/app/app.go

@@ -0,0 +1,11 @@
+package app
+
+import (
+	"github.com/lorsanstand/HomeOps-Hub/internal/shared/config"
+	"github.com/rs/zerolog"
+)
+
+type App struct {
+	cfg *config.Config
+	log *zerolog.Logger
+}

internal/hub/grpc/server.go

@@ -2,7 +2,6 @@ package grpc
 
 import (
 	"context"
-	"log"
 
 	"github.com/lorsanstand/HomeOps-Hub/api/gen/homeops"
 	"google.golang.org/protobuf/types/known/emptypb"
@@ -13,7 +12,6 @@ type Server struct {
 }
 
 func (s *Server) Ping(ctx context.Context, _ *emptypb.Empty) (*homeops.PongResponse, error) {
-	log.Println("Answer")
 	return &homeops.PongResponse{Pong: "Huiiii"}, nil
 
 }

internal/shared/config/config.go

@@ -1,4 +1,4 @@
-package cfg
+package config
 
 import (
 	"fmt"

internal/shared/log/init.go

@@ -1,11 +1,11 @@
 package log
 
 import (
+	"io"
 	"os"
 	"time"
 
 	"github.com/rs/zerolog"
-	"github.com/rs/zerolog/log"
 )
 
 type cfgLogStore interface {
@@ -13,15 +13,17 @@ type cfgLogStore interface {
 	GetMode() string
 }
 
-func Init(cfg cfgLogStore) {
+func NewLogger(cfg cfgLogStore) zerolog.Logger {
-	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
+	var output io.Writer = os.Stdout
 
 	if cfg.GetMode() != "PROD" {
-		log.Logger = log.Output(zerolog.ConsoleWriter{
+		output = zerolog.ConsoleWriter{
 			Out:        os.Stdout,
 			TimeFormat: time.Kitchen,
-		})
+		}
-	} else {
-		zerolog.SetGlobalLevel(zerolog.InfoLevel)
 	}
+
+	level := cfg.GetLogLevel()
+
+	return zerolog.New(output).Level(level).With().Timestamp().Logger()
 }