add log info switch

libi · Dec 12, 2022 · 745603e · 745603e
1 parent c239ca5
commit 745603e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/dcron.go b/dcron.go
@@ -27,7 +27,8 @@ type Dcron struct {
 	nodePool   *NodePool
 	running    int32
 
-	logger interface{ Printf(string, ...interface{}) }
+	logger  interface{ Printf(string, ...interface{}) }
+	logInfo bool
 
 	nodeUpdateDuration time.Duration
 	hashReplicas       int
@@ -80,7 +81,9 @@ func (d *Dcron) GetLogger() interface{ Printf(string, ...interface{}) } {
 }
 
 func (d *Dcron) info(format string, v ...interface{}) {
-	d.logger.Printf("INFO: "+format, v...)
+	if d.logInfo {
+		d.logger.Printf("INFO: "+format, v...)
+	}
 }
 func (d *Dcron) err(format string, v ...interface{}) {
 	d.logger.Printf("ERR: "+format, v...)

diff --git a/option.go b/option.go
@@ -19,6 +19,13 @@ func WithLogger(logger interface{ Printf(string, ...interface{}) }) Option {
 	}
 }
 
+// PrintLogInfo set log info level
+func WithPrintLogInfo() Option {
+	return func(dcron *Dcron) {
+		dcron.logInfo = true
+	}
+}
+
 // WithNodeUpdateDuration set node update duration
 func WithNodeUpdateDuration(d time.Duration) Option {
 	return func(dcron *Dcron) {