Merge pull request #421 from berty/dev/moul/chizap

feat: switch to chizap to log user agent
berty · Jun 25, 2021 · 7168e96 · 7168e96
2 parents f4e5b2f + 96a6581
commit 7168e96
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 37 deletions.
diff --git a/go.mod b/go.mod
diff --git a/go.sum b/go.sum
diff --git a/go/pkg/yolosvc/server.go b/go/pkg/yolosvc/server.go
@@ -24,11 +24,11 @@ import (
 	"github.com/rs/cors"
 	"github.com/stretchr/signature"
 	"github.com/tevino/abool"
-	chilogger "github.com/treastech/logger"
 	"go.uber.org/zap"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
+	"moul.io/chizap"
 	"moul.io/u"
 )
 
@@ -123,7 +123,7 @@ func NewServer(ctx context.Context, svc Service, opts ServerOpts) (*Server, erro
 		})
 		r.Use(cors.Handler)
 	}
-	r.Use(chilogger.Logger(srv.logger))
+	r.Use(chizap.New(srv.logger, &chizap.Opts{WithUserAgent: true, WithReferer: true}))
 	r.Use(middleware.Timeout(opts.RequestTimeout))
 	r.Use(middleware.Recoverer)