[🚀] v0.12.28

pkg/upload/tus.go

@@ -5,6 +5,12 @@ import (
 	"crypto/sha256"
 	"errors"
 	"fmt"
+	"net/http"
+	"os"
+	"strings"
+	"sync"
+	"time"
+
 	"gitea.bvbej.com/bvbej/base-golang/pkg/color"
 	"gitea.bvbej.com/bvbej/base-golang/pkg/ticker"
 	"gitea.bvbej.com/bvbej/base-golang/pkg/token"
@@ -12,11 +18,7 @@ import (
 	"github.com/tus/tusd/pkg/filestore"
 	tus "github.com/tus/tusd/pkg/handler"
 	"go.uber.org/zap"
-	"net/http"
-	"os"
-	"strings"
-	"sync"
-	"time"
+	"go.uber.org/zap/zapcore"
 )
 
 var _ Server = (*server)(nil)
@@ -36,6 +38,7 @@ type server struct {
 	token          token.Token
 	store          filestore.FileStore
 	logger         *zap.Logger
+	logLevel       zapcore.Level
 	httpServer     *http.Server
 	ctx            context.Context
 	done           context.CancelFunc
@@ -52,13 +55,14 @@ type Config struct {
 	Debug           bool
 }
 
-func New(conf Config, logger *zap.Logger) Server {
+func New(conf Config, logger *zap.Logger, logLevel zapcore.Level) Server {
 	ctx, cancelFunc := context.WithCancel(context.Background())
 	return &server{
 		config:         conf,
 		uploading:      sync.Map{},
 		headerTokenKey: "Authorization",
 		logger:         logger,
+		logLevel:       logLevel,
 		token:          token.New(conf.Secret),
 		ctx:            ctx,
 		done:           cancelFunc,
@@ -93,10 +97,11 @@ func (s *server) Start(completedEvent func(sha256, param string, info tus.FileIn
 	s.store = filestore.New(s.config.Dir)
 	s.store.UseIn(composer)
 
+	newLogger, _ := zap.NewStdLogAt(s.logger, s.logLevel)
 	handler, err := tus.NewHandler(tus.Config{
 		StoreComposer:           composer,
 		BasePath:                s.config.Path,
-		Logger:                  zap.NewStdLog(s.logger),
+		Logger:                  newLogger,
 		NotifyCompleteUploads:   true,
 		NotifyTerminatedUploads: true,
 		DisableTermination:      true,