feat: consumer add support custom logger (#293)

Author: Asutorufa

consumer/config.go

@@ -4,6 +4,7 @@ import (
 	"net/http"
 
 	sls "github.com/aliyun/aliyun-log-go-sdk"
+	"github.com/go-kit/kit/log"
 )
 
 type LogHubConfig struct {
@@ -65,6 +66,7 @@ type LogHubConfig struct {
 	MaxFetchLogGroupCount     int
 	CursorStartTime           int64 // Unix time stamp; Units are seconds.
 	InOrder                   bool
+	Logger                    log.Logger
 	AllowLogLevel             string
 	LogFileName               string
 	IsJsonType                bool

consumer/worker.go

@@ -1,8 +1,8 @@
 package consumerLibrary
 
 import (
-	"os"
 	"io"
+	"os"
 	"sync"
 	"time"
 
@@ -53,7 +53,11 @@ func InitConsumerWorkerWithCheckpointTracker(option LogHubConfig, do func(int, *
 // InitConsumerWorkerWithProcessor
 // you need save checkpoint by yourself and can do something after consumer shutdown
 func InitConsumerWorkerWithProcessor(option LogHubConfig, processor Processor) *ConsumerWorker {
-	logger := logConfig(option)
+	logger := option.Logger
+	if logger == nil {
+		logger = logConfig(option)
+	}
+
 	consumerClient := initConsumerClient(option, logger)
 	consumerHeatBeat := initConsumerHeatBeat(consumerClient, logger)
 	consumerWorker := &ConsumerWorker{