feat(0.7.x): 日只优化

Author: huaqi666

jd_test.go

@@ -8,6 +8,7 @@ import (
 	"io/ioutil"
 	"os"
 	"testing"
+	"time"
 )
 
 type LocalConfig struct {
@@ -148,6 +149,7 @@ func TestGoodsServiceImpl_GoodsGigfieldQuery(t *testing.T) {
 func TestGoodsServiceImpl_GoodsJingfenQuery(t *testing.T) {
 	config := GetConfig()
 	service := NewJosService(config.AppKey, config.SecretKey, "")
+	log.FileLog("logs/", "jd-go.log", 7, 1*time.Hour, 24*time.Minute)
 	goodsService := service.GetGoodsService()
 	res, err := goodsService.GoodsJingfenQuery(&GoodsJingfenQueryRequest{
 		EliteId: 33,

log/log.go

@@ -9,6 +9,7 @@ import (
 	"github.com/sirupsen/logrus"
 	"os"
 	"path"
+	"runtime"
 	"strings"
 	"time"
 )
@@ -95,6 +96,7 @@ func (l *LoggerImpl) Trace(args ...interface{}) {
 
 func (l *LoggerImpl) Logf(level Level, format string, args ...interface{}) {
 	if l.checkLevel(level) {
+		l.Line()
 		n := strings.Count(format, "%s")
 		le := len(args)
 		for i := 0; i < (le - n); i++ {
@@ -119,6 +121,27 @@ func (l *LoggerImpl) checkLevel(level Level) bool {
 	return uint64(l.Level) >= uint64(level)
 }
 
+func (l LoggerImpl) Line() {
+	_, fullFile, line, ok := runtime.Caller(5) // 5 skip: 指的是跳过多少个引用
+	file := fullFile
+	if file != "" {
+		// Truncate file name at last file name separator.
+		if index := strings.LastIndex(file, "/"); index >= 0 {
+			file = file[index+1:]
+		} else if index = strings.LastIndex(file, "\\"); index >= 0 {
+			file = file[index+1:]
+		}
+	} else {
+		file = "???"
+	}
+	if ok {
+		buf := new(strings.Builder)
+
+		_, _ = fmt.Fprintf(buf, "\033[35m[jd-go]\033[0m: %s:%d", fullFile, line)
+		l.Logger.Log(l.Level, buf.String())
+	}
+}
+
 // record output file log
 func (l *LoggerImpl) FileLog(logPath string, logFileName string, maxNum uint, maxAge, rotationTime time.Duration) {