refactor(logger): add Writer method

mgjules · Apr 11, 2022 · b228da3 · b228da3
1 parent c75f265
commit b228da3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/http/server.go b/http/server.go
@@ -16,7 +16,6 @@ import (
 	"github.com/gin-gonic/gin"
 	"go.opentelemetry.io/contrib/instrumentation/github.com/gin-gonic/gin/otelgin"
 	"go.uber.org/fx"
-	"go.uber.org/zap"
 )
 
 const (
@@ -57,10 +56,9 @@ func NewServer(
 		gin.SetMode(gin.ReleaseMode)
 	}
 
-	writer := zap.NewStdLog(logger.Desugar().Logger).Writer()
-
-	gin.DefaultWriter = writer
-	gin.DefaultErrorWriter = writer
+	w := logger.Writer()
+	gin.DefaultWriter = w
+	gin.DefaultErrorWriter = w
 
 	s := Server{
 		router: gin.Default(),

diff --git a/logger/logger.go b/logger/logger.go
@@ -3,6 +3,7 @@ package logger
 import (
 	"context"
 	"fmt"
+	"io"
 
 	"github.com/JulesMike/spoty/config"
 	"github.com/uptrace/opentelemetry-go-extra/otelzap"
@@ -46,3 +47,8 @@ func New(lc fx.Lifecycle, cfg *config.Config) (*Logger, error) {
 
 	return &Logger{otellogger.Sugar()}, nil
 }
+
+// Writer returns the logger's io.Writer.
+func (l *Logger) Writer() io.Writer {
+	return zap.NewStdLog(l.Desugar().Logger).Writer()
+}