From 9d5f4871108f19b7b5f0ab19a05a3a994747d906 Mon Sep 17 00:00:00 2001
From: waze <87033883@qq.com>
Date: Mon, 12 Apr 2021 14:38:13 +0800
Subject: [PATCH] =?UTF-8?q?=E6=97=A5=E5=BF=97=E6=96=87=E4=BB=B6=E5=90=8D?=
 =?UTF-8?q?=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cluster/cluster.go |  4 ++++
 log/log.go         | 10 ++++++----
 node/node.go       | 16 +++++++++-------
 3 files changed, 19 insertions(+), 11 deletions(-)

diff --git a/cluster/cluster.go b/cluster/cluster.go
index 7156b68..d300785 100644
--- a/cluster/cluster.go
+++ b/cluster/cluster.go
@@ -231,3 +231,7 @@ func (cls *Cluster) triggerRpcEvent (bConnect bool,nodeId int) {
 	}
 }
 
+func (cls *Cluster) GetLocalNodeInfo() *NodeInfo {
+	return &cls.localNodeInfo
+}
+
diff --git a/log/log.go b/log/log.go
index 8d2b542..03d9db5 100644
--- a/log/log.go
+++ b/log/log.go
@@ -34,6 +34,7 @@ const (
 
 type Logger struct {
 	filePath   string
+	filepre 	string
 	logTime    time.Time
 	level      int
 	stdLogger   *log.Logger
@@ -42,7 +43,7 @@ type Logger struct {
 	flag       int
 }
 
-func New(strLevel string, pathname string, flag int) (*Logger, error) {
+func New(strLevel string, pathname string, filepre string, flag int) (*Logger, error) {
 	// level
 	var level int
 	switch strings.ToLower(strLevel) {
@@ -75,7 +76,7 @@ func New(strLevel string, pathname string, flag int) (*Logger, error) {
 			now.Minute(),
 			now.Second())
 
-		file, err := os.Create(path.Join(pathname, filename))
+		file, err := os.Create(path.Join(pathname, filepre+filename))
 		if err != nil {
 			return nil, err
 		}
@@ -95,6 +96,7 @@ func New(strLevel string, pathname string, flag int) (*Logger, error) {
 	logger.logTime = now
 	logger.filePath = pathname
 	logger.flag = flag
+	logger.filepre = filepre
 
 	return logger, nil
 }
@@ -128,7 +130,7 @@ func (logger *Logger) doPrintf(level int, printLevel string, format string, a ..
 				now.Minute(),
 				now.Second())
 
-			file, err := os.Create(path.Join(logger.filePath, filename))
+			file, err := os.Create(path.Join(logger.filePath, logger.filepre+filename))
 			if err == nil {
 				logger.baseFile.Close()
 				logger.baseLogger = log.New(file, "", logger.flag)
@@ -172,7 +174,7 @@ func (logger *Logger) Fatal(format string, a ...interface{}) {
 	logger.doPrintf(fatalLevel, printFatalLevel, format, a...)
 }
 
-var gLogger, _ = New("debug", "", log.LstdFlags|log.Lshortfile)
+var gLogger, _ = New("debug", "", "", log.LstdFlags|log.Lshortfile)
 
 // It's dangerous to call the method on logging
 func Export(logger *Logger) {
diff --git a/node/node.go b/node/node.go
index 127dbc1..45ffc38 100644
--- a/node/node.go
+++ b/node/node.go
@@ -134,6 +134,11 @@ func initNode(id int){
 		log.Fatal("read system config is error %+v",err)
 	}
 
+	err = initLog()
+	if err != nil{
+		return
+	}
+
 	//2.setup service
 	for _,s := range preSetupService {
 		//是否配置的service
@@ -156,7 +161,9 @@ func initLog() error{
 		setLogPath("./log")
 	}
 
-	logger,err := log.New(logLevel,logPath,slog.LstdFlags|slog.Lshortfile)
+	localnodeinfo := cluster.GetCluster().GetLocalNodeInfo()
+	filepre := fmt.Sprintf("%s[%d]", localnodeinfo.NodeName, localnodeinfo.NodeId)
+	logger,err := log.New(logLevel,logPath,filepre,slog.LstdFlags|slog.Lshortfile)
 	if err != nil {
 		fmt.Printf("cannot create log file!\n")
 		return err
@@ -220,11 +227,6 @@ func startNode(args interface{}) error{
 		return fmt.Errorf("invalid option %s",param)
 	}
 
-	err = initLog()
-	if err != nil {
-		return err
-	}
-
 	timer.StartTimer(10*time.Millisecond,100000)
 	log.Release("Start running server.")
 	//2.初始化node
@@ -285,7 +287,7 @@ func GetConfigDir() string {
 }
 
 func SetSysLog(strLevel string, pathname string, flag int){
-	logs,_:= log.New(strLevel,pathname,flag)
+	logs,_:= log.New(strLevel,pathname, "", flag)
 	log.Export(logs)
 }