From 886965f5cd77b06719ea4ddcbbfc11d385bf21b5 Mon Sep 17 00:00:00 2001
From: liulei <liulei@xiaozhu.com>
Date: Wed, 5 Jan 2022 15:04:43 +0800
Subject: [PATCH] add logger interface defination#153

---
 dtmcli/logger/log.go         | 18 ++++++++++++++++--
 dtmcli/logger/logger_test.go | 12 ++++++++++++
 2 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/dtmcli/logger/log.go b/dtmcli/logger/log.go
index 7d3650a13..81466b3af 100644
--- a/dtmcli/logger/log.go
+++ b/dtmcli/logger/log.go
@@ -8,13 +8,27 @@ import (
 	"go.uber.org/zap/zapcore"
 )
 
-var logger *zap.SugaredLogger = nil
+//var logger *zap.SugaredLogger = nil
+
+var logger Logger = nil
 
 func init() {
 	InitLog("info")
 }
 
-// InitLog is a initialization for a logger
+// Logger logger interface
+type Logger interface {
+	Debugf(format string, args ...interface{})
+	Infof(format string, args ...interface{})
+	Warnf(format string, args ...interface{})
+	Errorf(format string, args ...interface{})
+}
+
+func WithLogger(log Logger) {
+	logger = log
+}
+
+// InitLog is an initialization for a logger
 // level can be: debug info warn error
 func InitLog(level string) {
 	config := zap.NewProductionConfig()
diff --git a/dtmcli/logger/logger_test.go b/dtmcli/logger/logger_test.go
index 669a39fdd..6320b58ee 100644
--- a/dtmcli/logger/logger_test.go
+++ b/dtmcli/logger/logger_test.go
@@ -1,6 +1,7 @@
 package logger
 
 import (
+	"github.com/sirupsen/logrus"
 	"os"
 	"testing"
 )
@@ -15,3 +16,14 @@ func TestInitLog(t *testing.T) {
 	FatalfIf(false, "nothing")
 	FatalIfError(nil)
 }
+
+func TestWithLogger(t *testing.T) {
+	logger := logrus.New()
+	WithLogger(logger)
+	Debugf("a debug msg")
+	Infof("a info msg")
+	Warnf("a warn msg")
+	Errorf("a error msg")
+	FatalfIf(false, "nothing")
+	FatalIfError(nil)
+}
\ No newline at end of file