feat: add log control

producer/logger.go

@@ -4,28 +4,34 @@ import (
 	"github.com/go-kit/kit/log"
 	"github.com/go-kit/kit/log/level"
 	"gopkg.in/natefinch/lumberjack.v2"
-	"os"
 	"io"
+	"os"
 )
 
 func logConfig(producerConfig *ProducerConfig) log.Logger {
 	var writer io.Writer
-	if producerConfig.LogFileName == "" {
-		writer = log.NewSyncWriter(os.Stdout)
-	} else {
-		if producerConfig.LogMaxSize == 0 {
-			producerConfig.LogMaxSize = 10
-		}
-		if producerConfig.LogMaxBackups == 0 {
-			producerConfig.LogMaxBackups = 10
-		}
-		writer = &lumberjack.Logger{
-			Filename:   producerConfig.LogFileName,
-			MaxSize:    producerConfig.LogMaxSize,
-			MaxBackups: producerConfig.LogMaxBackups,
-			Compress:   producerConfig.LogCompress,
+
+	if producerConfig.Logger {
+		if producerConfig.LogFileName == "" {
+			writer = log.NewSyncWriter(os.Stdout)
+		} else {
+			if producerConfig.LogMaxSize == 0 {
+				producerConfig.LogMaxSize = 10
+			}
+			if producerConfig.LogMaxBackups == 0 {
+				producerConfig.LogMaxBackups = 10
+			}
+			writer = &lumberjack.Logger{
+				Filename:   producerConfig.LogFileName,
+				MaxSize:    producerConfig.LogMaxSize,
+				MaxBackups: producerConfig.LogMaxBackups,
+				Compress:   producerConfig.LogCompress,
+			}
 		}
+	} else {
+		writer = NullWriter{}
 	}
+
 	var logger log.Logger
 	if producerConfig.IsJsonType {
 		logger = log.NewJSONLogger(writer)

producer/producer_config.go

@@ -53,6 +53,8 @@ type ProducerConfig struct {
 	Region           string
 	AuthVersion      sls.AuthVersionType
 	CompressType     int // only work for logstore now
+
+	Logger bool // print the log
 }
 
 func GetDefaultProducerConfig() *ProducerConfig {
@@ -71,5 +73,6 @@ func GetDefaultProducerConfig() *ProducerConfig {
 		MaxBatchCount:         4096,
 		NoRetryStatusCodeList: []int{400, 404},
 		CompressType:          sls.Compress_LZ4,
+		Logger:                true,
 	}
 }

producer/writer.go

@@ -0,0 +1,8 @@
+package producer
+
+type NullWriter struct {
+}
+
+func (u NullWriter) Write(p []byte) (n int, err error) {
+	return 0, err
+}