logger.go

/*
*	Description : logger   TODO 测试
*	Author 		: ManGe
*	Mail 		: 2912882908@qq.com
**/

package gathertool

import (
	"bytes"
	"context"
	"fmt"
	"os"
	"runtime"
	"strconv"
	"strings"
	"time"

	gormLogger "gorm.io/gorm/logger"
)

// LogClose 是否关闭日志
var LogClose = true
var std = newStd()

// CloseLog 关闭日志
func CloseLog() {
	LogClose = false
}

type logger struct {
	outFile       bool
	outFileWriter *os.File
}

func newStd() *logger {
	return &logger{}
}

// SetLogFile 设置日志输出到的指定文件
func SetLogFile(name string) {
	std.outFile = true
	std.outFileWriter, _ = os.OpenFile(name+time.Now().Format("-20060102")+".log",
		os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
}

// Level 日志等级
type Level int

var LevelMap = map[Level]string{
	1: "[Info]  ",
	2: "[Debug] ",
	3: "[Warn]  ",
	4: "[Error] ",
	5: "[HTTP/s]",
}

// Log 日志
func (l *logger) Log(level Level, args string, times int) {
	var buffer bytes.Buffer
	buffer.WriteString(time.Now().Format(TimeMilliTemplate))
	buffer.WriteString(LevelMap[level])
	_, file, line, _ := runtime.Caller(times)
	fileList := strings.Split(file, "/")
	// 最多显示两级路径
	if len(fileList) > 3 {
		fileList = fileList[len(fileList)-3:]
	}
	buffer.WriteString(strings.Join(fileList, "/"))
	buffer.WriteString(":")
	buffer.WriteString(strconv.Itoa(line))
	buffer.WriteString(" \t| ")
	buffer.WriteString(args)
	buffer.WriteString("\n")
	out := buffer.Bytes()
	if LogClose {
		_, _ = buffer.WriteTo(os.Stdout)
	}

	// 输出到文件或远程日志服务
	if l.outFile {
		_, _ = l.outFileWriter.Write(out)
	}
}

// Info 日志-信息
func Info(args ...any) {
	std.Log(1, fmt.Sprint(args...), 2)
}

// InfoF 日志-信息
func InfoF(format string, args ...any) {
	std.Log(1, fmt.Sprintf(format, args...), 2)
}

// InfoTimes 日志-信息, 指定日志代码位置的定位调用层级
func InfoTimes(times int, args ...any) {
	std.Log(1, fmt.Sprint(args...), times)
}

// InfoFTimes 日志-信息, 指定日志代码位置的定位调用层级
func InfoFTimes(times int, format string, args ...any) {
	std.Log(1, fmt.Sprintf(format, args...), times)
}

// Debug 日志-调试
func Debug(args ...any) {
	std.Log(2, fmt.Sprint(args...), 2)
}

// DebugF 日志-调试
func DebugF(format string, args ...any) {
	std.Log(2, fmt.Sprintf(format, args...), 2)
}

// DebugTimes 日志-调试, 指定日志代码位置的定位调用层级
func DebugTimes(times int, args ...any) {
	std.Log(1, fmt.Sprint(args...), times)
}

// DebugFTimes 日志-调试, 指定日志代码位置的定位调用层级
func DebugFTimes(format string, times int, args ...any) {
	std.Log(1, fmt.Sprintf(format, args...), times)
}

// Warn 日志-警告
func Warn(args ...any) {
	std.Log(3, fmt.Sprint(args...), 2)
}

// WarnF 日志-警告
func WarnF(format string, args ...any) {
	std.Log(3, fmt.Sprintf(format, args...), 2)
}

// WarnTimes 日志-警告, 指定日志代码位置的定位调用层级
func WarnTimes(times int, args ...any) {
	std.Log(1, fmt.Sprint(args...), times)
}

// WarnFTimes 日志-警告, 指定日志代码位置的定位调用层级
func WarnFTimes(format string, times int, args ...any) {
	std.Log(1, fmt.Sprintf(format, args...), times)
}

// Error 日志-错误
func Error(args ...any) {
	std.Log(4, fmt.Sprint(args...), 2)
}

// ErrorF 日志-错误
func ErrorF(format string, args ...any) {
	std.Log(4, fmt.Sprintf(format, args...), 2)
}

// ErrorTimes 日志-错误, 指定日志代码位置的定位调用层级
func ErrorTimes(times int, args ...any) {
	std.Log(4, fmt.Sprint(args...), times)
}

// ErrorFTimes 日志-错误, 指定日志代码位置的定位调用层级
func ErrorFTimes(format string, times int, args ...any) {
	std.Log(4, fmt.Sprintf(format, args...), times)
}

func Panic(args ...any) {
	panic(args)
}

// HTTPTimes 日志-信息, 指定日志代码位置的定位调用层级
func HTTPTimes(times int, args ...any) {
	std.Log(5, fmt.Sprint(args...), times)
}

// Bar 终端显示的进度条
type Bar struct {
	percent int64  //百分比
	cur     int64  //当前进度位置
	total   int64  //总进度
	rate    string //进度条
	graph   string //显示符号
}

func (bar *Bar) NewOption(start, total int64) {
	bar.cur = start
	bar.total = total
	if bar.graph == "" {
		bar.graph = "█"
	}
	bar.percent = bar.getPercent()
	for i := 0; i < int(bar.percent); i += 2 {
		bar.rate += bar.graph //初始化进度条位置
	}
}

func (bar *Bar) getPercent() int64 {
	return int64(float32(bar.cur) / float32(bar.total) * 100)
}

func (bar *Bar) NewOptionWithGraph(start, total int64, graph string) {
	bar.graph = graph
	bar.NewOption(start, total)
}

func (bar *Bar) Play(cur int64) {
	bar.cur = cur
	last := bar.percent
	bar.percent = bar.getPercent()
	if bar.percent != last && bar.percent%2 == 0 {
		bar.rate += bar.graph
	}
	fmt.Printf("\r[%-50s]%3d%% %8d/%d", bar.rate, bar.percent, bar.cur, bar.total)
}

func (bar *Bar) Finish() {
	fmt.Println()
}

type GormLogger struct {
	SlowThreshold time.Duration
}

var _ gormLogger.Interface = (*GormLogger)(nil)

func NewGormLogger() *GormLogger {
	return &GormLogger{
		SlowThreshold: 200 * time.Millisecond, // 一般超过200毫秒就算慢查所以不使用配置进行更改
	}
}

var _ gormLogger.Interface = (*GormLogger)(nil)

func (l *GormLogger) LogMode(lev gormLogger.LogLevel) gormLogger.Interface {
	return &GormLogger{}
}
func (l *GormLogger) Info(ctx context.Context, msg string, data ...interface{}) {
	InfoF(msg, data)
}
func (l *GormLogger) Warn(ctx context.Context, msg string, data ...interface{}) {
	WarnF(msg, data)
}
func (l *GormLogger) Error(ctx context.Context, msg string, data ...interface{}) {
	ErrorF(msg, data)
}
func (l *GormLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
	elapsed := time.Since(begin)
	sql, rows := fc()
	if err != nil {
		ErrorFTimes("[SQL-Error]\t| err = %v \t| rows= %v \t| %v \t| %v", 5, err, rows, elapsed, sql)
	}
	if l.SlowThreshold != 0 && elapsed > l.SlowThreshold {
		WarnFTimes("[SQL-SlowLog]\t| rows= %v \t| %v \t| %v", 5, rows, elapsed, sql)
	}
	InfoFTimes(5, "[SQL]\t| rows= %v \t| %v \t| %v", rows, elapsed, sql)
}