From dba3c37e2ecb149f86fa73d9ea6df73240576c9a Mon Sep 17 00:00:00 2001
From: yedf2 <120050102@qq.com>
Date: Sun, 5 Dec 2021 09:40:05 +0800
Subject: [PATCH] log use DTM_DEBUG

---
 dtmcli/dtmimp/utils.go | 4 ++--
 test/main_test.go      | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/dtmcli/dtmimp/utils.go b/dtmcli/dtmimp/utils.go
index b78c901..3b01261 100644
--- a/dtmcli/dtmimp/utils.go
+++ b/dtmcli/dtmimp/utils.go
@@ -26,7 +26,7 @@ import (
 
 // AsError wrap a panic value as an error
 func AsError(x interface{}) error {
-	LogRedf("panic to error: '%v'", x)
+	LogRedf("panic wrapped to error: '%v'", x)
 	if e, ok := x.(error); ok {
 		return e
 	}
@@ -124,7 +124,7 @@ var logger *zap.SugaredLogger = nil
 func init() {
 	config := zap.NewProductionConfig()
 	config.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
-	if os.Getenv("DTM_PRODUCTION") == "" {
+	if os.Getenv("DTM_DEBUG") != "" {
 		config.Encoding = "console"
 		config.EncoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder
 	}
diff --git a/test/main_test.go b/test/main_test.go
index 251d4e5..658831e 100644
--- a/test/main_test.go
+++ b/test/main_test.go
@@ -7,6 +7,7 @@
 package test
 
 import (
+	"os"
 	"testing"
 	"time"
 
@@ -19,6 +20,7 @@ import (
 func TestMain(m *testing.M) {
 	common.MustLoadConfig()
 	dtmcli.SetCurrentDBType(common.DtmConfig.DB["driver"])
+	os.Setenv("DTM_DEBUG", "1")
 	dtmsvr.TransProcessedTestChan = make(chan string, 1)
 	dtmsvr.NowForwardDuration = 0 * time.Second
 	dtmsvr.CronForwardDuration = 180 * time.Second