diff --git a/inlong-tubemq/tubemq-client-twins/tubemq-client-go/client/consumer_impl.go b/inlong-tubemq/tubemq-client-twins/tubemq-client-go/client/consumer_impl.go index 6cc0174835f..4becc04bfc4 100644 --- a/inlong-tubemq/tubemq-client-twins/tubemq-client-go/client/consumer_impl.go +++ b/inlong-tubemq/tubemq-client-twins/tubemq-client-go/client/consumer_impl.go @@ -77,6 +77,10 @@ func NewConsumer(config *config.Config) (Consumer, error) { if err := config.ValidateConsumer(); err != nil { return nil, err } + log.SetLogLevel(config.Log.LogLevel) + log.SetLogPath(config.Log.LogPath) + logConfig := log.GetLogConfig() + log.NewLogger(logConfig) log.Infof("The config of the consumer is %s", config) selector, err := selector.Get("ip") if err != nil { diff --git a/inlong-tubemq/tubemq-client-twins/tubemq-client-go/config/config.go b/inlong-tubemq/tubemq-client-twins/tubemq-client-go/config/config.go index 687990cd8df..723185b01f2 100644 --- a/inlong-tubemq/tubemq-client-twins/tubemq-client-go/config/config.go +++ b/inlong-tubemq/tubemq-client-twins/tubemq-client-go/config/config.go @@ -143,6 +143,11 @@ type Config struct { // AfterFail is the heartbeat timeout after a heartbeat failure. AfterFail time.Duration } + + Log struct { + LogPath string + LogLevel string + } } // NewDefaultConfig returns a default config of the client. @@ -175,6 +180,9 @@ func NewDefaultConfig() *Config { c.Heartbeat.MaxRetryTimes = 5 c.Heartbeat.AfterFail = 60000 * time.Millisecond + c.Log.LogLevel = "warn" + c.Log.LogPath = "../log/tubemq.log" + return c } @@ -620,3 +628,17 @@ func WithConsumePosition(consumePosition int) Option { c.Consumer.ConsumePosition = consumePosition } } + +// WithLogLevel set log level +func WithLogLevel(level string) Option { + return func(c *Config) { + c.Log.LogLevel = level + } +} + +// WithLogPath set log path +func WithLogPath(path string) Option { + return func(c *Config) { + c.Log.LogPath = path + } +} diff --git a/inlong-tubemq/tubemq-client-twins/tubemq-client-go/log/config.go b/inlong-tubemq/tubemq-client-twins/tubemq-client-go/log/config.go index fee8f4a75e3..0258268820d 100644 --- a/inlong-tubemq/tubemq-client-twins/tubemq-client-go/log/config.go +++ b/inlong-tubemq/tubemq-client-twins/tubemq-client-go/log/config.go @@ -39,3 +39,18 @@ var defaultConfig = &OutputConfig{ MaxAge: 3, Level: "warn", } + +// SetLogLevel set log level +func SetLogLevel(level string) { + defaultConfig.Level = level +} + +// SetLogPath set log path +func SetLogPath(path string) { + defaultConfig.LogPath = path +} + +// GetLogConfig get log config +func GetLogConfig() *OutputConfig { + return defaultConfig +}