improve log system

Author: lienze

src/file/file.go

@@ -61,7 +61,7 @@ func SaveFile(filename string, content string) error {
 }
 
 func AddContent2File(filename string, content string) error {
-	fmt.Println("Start Add Content 2 File")
+	//fmt.Println("Start Add Content 2 File")
 	fileObj, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
 	if err != nil {
 		return err

src/gtime/gtime.go

@@ -8,6 +8,9 @@ const (
 	BASIC_NANO = 3
 	BASIC_FULL = 4
 )
+const (
+	STYLE1 = 1
+)
 
 func GetCurTime(iType int8) string {
 	switch iType {
@@ -24,3 +27,12 @@ func GetCurTime(iType int8) string {
 	}
 	return time.Now().Format("2006-1-2 15:04:05")
 }
+
+func GetCurDate(iType int8) string {
+	switch iType {
+	case STYLE1:
+		return time.Now().Format("20060102")
+	default:
+		return time.Now().Format("20060102")
+	}
+}

src/log/log.go

@@ -11,8 +11,9 @@ import (
 var (
 	bEnableLog     bool = false
 	logchan        chan string
-	bShowInConsole bool = false
-	arrLogType          = [...]string{"normal", "warning", "error"}
+	bShowInConsole bool           = false
+	arrLogType                    = [...]string{"normal", "warning", "error"}
+	mapLogPath     map[int]string = make(map[int]string)
 	logPath        string
 )
 
@@ -23,15 +24,20 @@ func InitLog() {
 	bShowInConsole = config.GConfig.LogCfg.ShowInConsole
 	fmt.Println("Log path from config file:" + config.GConfig.LogCfg.LogPath)
 	logPath = config.GConfig.LogCfg.LogPath
-	//check if the logPath exists
-	if err := file.FolderExists(logPath); err != nil {
-		if err = file.CreateFolder(logPath); err != nil {
-			panic(err)
-		}
+	for key, val := range arrLogType {
+		checkAndCreateFolder(logPath+"/"+val+"/", key)
 	}
 	go Listen4Log()
 }
 
+func checkAndCreateFolder(path string, pos int) {
+	if err := file.FolderExists(path); err != nil {
+		if err = file.CreateFolder(path); err != nil {
+			panic(err)
+		}
+	}
+	mapLogPath[pos] = path
+}
 func wlog(info string) {
 	logchan <- info
 }
@@ -43,16 +49,17 @@ func Listen4Log() {
 	select {
 	case recvStr = <-logchan:
 		logType := recvStr[0:1]
+		rawContent := recvStr[1:]
 		if iType, err = strconv.Atoi(logType); err != nil {
 			iType = 0
 			fmt.Printf("error logType in Listen4Log [%s]\n", logType)
 		}
 		if bShowInConsole {
 			fmt.Printf("[%s][%s] %s\n",
-				gtime.GetCurTime(gtime.BASIC_MILL), arrLogType[iType], recvStr[1:])
+				gtime.GetCurTime(gtime.BASIC_MILL), arrLogType[iType], rawContent)
 		}
-		//file.SaveFile(logPath+"firstFile", "Hello World")
-		file.AddContent2File(logPath+"firstFile", "Hello World")
+		filePath := mapLogPath[iType] + gtime.GetCurDate(gtime.STYLE1)
+		file.AddContent2File(filePath, rawContent+"\n")
 	}
 }