Rename the directory of 'log' to 'initlog'

QMPlusServer/cmd/windows.go

@@ -3,7 +3,7 @@ package cmd
 import (
 	"fmt"
 	"gin-vue-admin/config"
-	"gin-vue-admin/init/log"
+	"gin-vue-admin/init/initlog"
 	"github.com/gin-gonic/gin"
 	"net/http"
 	"time"

QMPlusServer/init/initRedis/init_redis.go

@@ -2,7 +2,7 @@ package initRedis
 
 import (
 	"gin-vue-admin/config"
-	"gin-vue-admin/init/log"
+	"gin-vue-admin/init/initlog"
 	"github.com/go-redis/redis"
 )
 

QMPlusServer/init/initRouter/init_router.go

@@ -2,7 +2,7 @@ package initRouter
 
 import (
 	_ "gin-vue-admin/docs"
-	"gin-vue-admin/init/log"
+	"gin-vue-admin/init/initlog"
 	"gin-vue-admin/middleware"
 	"gin-vue-admin/router"
 	"github.com/gin-gonic/gin"

QMPlusServer/init/initlog/log.go

+package log
+
+// Custom Logger
+type Logger interface {
+	Debug(v ...interface{})
+	Info(v ...interface{})
+	Warning(v ...interface{})
+	Error(v ...interface{})
+	Critical(v ...interface{})
+	Fatal(v ...interface{})
+}
+
+var L Logger
+
+func SetLogger(logger Logger) {
+	L = logger
+}

QMPlusServer/init/initlog/qmlog/log.go

+package qmlog
+
+// Register logger
+import (
+	"errors"
+	"fmt"
+	"gin-vue-admin/config"
+	"gin-vue-admin/init/initlog"
+	"gin-vue-admin/tools"
+	rotatelogs "github.com/lestrrat/go-file-rotatelogs"
+	oplogging "github.com/op/go-logging"
+	"io"
+	"os"
+	"strings"
+	"time"
+)
+
+const (
+	logDir = "log"
+	logSoftLink = "api.log"
+	module = "gin-vue-admin"
+)
+
+var (
+	configNotFound = errors.New("logger config not found")
+
+	defaultFormatter = `%{time:2006/01/02 - 15:04:05.000} %{longfile} %{color:bold}▶ [%{level:.6s}] %{message}%{color:reset}`
+)
+
+type Logger struct{
+	logger *oplogging.Logger
+}
+
+func NewLogger() (log.Logger, error) {
+	c := config.GinVueAdminconfig.Log
+	logger := oplogging.MustGetLogger(module)
+	var backends []oplogging.Backend
+	backends = registerStdout(c, backends)
+	backends = registerFile(c, backends)
+
+	oplogging.SetBackend(backends...)
+	log.SetLogger(logger)
+	return logger, nil
+}
+
+func registerStdout(c config.Log, backends []oplogging.Backend) []oplogging.Backend {
+	for _, v := range c.Stdout {
+		level, err := oplogging.LogLevel(v)
+		if err != nil {
+			fmt.Println(err)
+			continue
+		}
+		backends = append(backends, createBackend(os.Stdout, c, level))
+	}
+
+	return backends
+}
+
+func registerFile(c config.Log, backends []oplogging.Backend) []oplogging.Backend {
+	if len(c.File) > 0 {
+		if ok, _ := tools.PathExists(logDir); !ok {
+			// directory not exist
+			fmt.Println("create log directory")
+			_ = os.Mkdir(logDir, os.ModePerm)
+		}
+		apiLogPath := logDir + string(os.PathSeparator) + logSoftLink
+		fileWriter, err := rotatelogs.New(
+			apiLogPath+".%Y-%m-%d-%H-%M.log",
+			// generate soft link, point to latest log file
+			rotatelogs.WithLinkName(apiLogPath),
+			// maximum time to save log files
+			rotatelogs.WithMaxAge(7*24*time.Hour),
+			// time period of log file switching
+			rotatelogs.WithRotationTime(24*time.Hour),
+		)
+		if err != nil {
+			fmt.Println(err)
+			return backends
+		}
+		for _, v := range c.File {
+			level, err := oplogging.LogLevel(v)
+			if err != nil {
+				fmt.Println(err)
+				continue
+			}
+			backends = append(backends, createBackend(fileWriter, c, level))
+		}
+	}
+
+	return backends
+}
+
+func createBackend(w io.Writer, c config.Log, level oplogging.Level) oplogging.Backend {
+	backend := oplogging.NewLogBackend(w, c.Prefix, 0)
+	stdoutWriter := false
+	if w == os.Stdout {
+		stdoutWriter = true
+	}
+	format := getLogFormatter(c, stdoutWriter)
+	backendLeveled := oplogging.AddModuleLevel(oplogging.NewBackendFormatter(backend, format))
+	backendLeveled.SetLevel(level, module)
+	return backendLeveled
+}
+
+func getLogFormatter(c config.Log, stdoutWriter bool) oplogging.Formatter {
+	pattern := defaultFormatter
+	if !stdoutWriter {
+		// Color is only required for console output
+		// Other writers don't need %{color} tag
+		pattern = strings.Replace(pattern, "%{color:bold}", "", -1)
+		pattern = strings.Replace(pattern, "%{color:reset}", "", -1)
+	}
+	if !c.LogFile {
+		// Remove %{logfile} tag
+		pattern = strings.Replace(pattern, "%{longfile}", "", -1)
+	}
+	return oplogging.MustStringFormatter(pattern)
+}
+
+func (l Logger) Debug(v ...interface{}) {
+	l.logger.Debug(v)
+}
+
+func (l Logger) Info(v ...interface{}) {
+	l.logger.Info(v)
+}
+
+func (l Logger) Warning(v ...interface{}) {
+	l.logger.Warning(v)
+}
+
+func (l Logger) Error(v ...interface{}) {
+	l.logger.Error(v)
+}
+
+func (l Logger) Critical(v ...interface{}) {
+	l.logger.Critical(v)
+}
+
+func (l Logger) Fatal(v ...interface{}) {
+	l.logger.Fatal(v)
+}
+

QMPlusServer/init/qmsql/initMysql.go

@@ -2,7 +2,7 @@ package qmsql
 
 import (
 	"gin-vue-admin/config"
-	"gin-vue-admin/init/log"
+	"gin-vue-admin/init/initlog"
 	"github.com/jinzhu/gorm"
 	_ "github.com/jinzhu/gorm/dialects/mysql"
 )

QMPlusServer/init/registTable/regist_table.go

 package registTable
 
 import (
-	"gin-vue-admin/init/log"
+	"gin-vue-admin/init/initlog"
 	"gin-vue-admin/model/dbModel"
 	"gin-vue-admin/model/sysModel"
 	"github.com/jinzhu/gorm"

QMPlusServer/main.go

@@ -6,7 +6,7 @@ import (
 	"gin-vue-admin/config"
 	"gin-vue-admin/init/initRedis"
 	"gin-vue-admin/init/initRouter"
-	"gin-vue-admin/init/log/qmlog"
+	"gin-vue-admin/init/initlog/qmlog"
 	"gin-vue-admin/init/qmsql"
 	"gin-vue-admin/init/registTable"
 	//"runtime"

QMPlusServer/middleware/logger.go

@@ -2,7 +2,7 @@ package middleware
 
 import (
 	"bytes"
-	"gin-vue-admin/init/log"
+	"gin-vue-admin/init/initlog"
 	"net/http/httputil"
 	"strings"
 	"time"