110 lines
2.1 KiB
Go
110 lines
2.1 KiB
Go
package loger
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"path/filepath"
|
||
"runtime"
|
||
|
||
"github.com/IceWhaleTech/CasaOS/pkg/config"
|
||
file2 "github.com/IceWhaleTech/CasaOS/pkg/utils/file"
|
||
)
|
||
|
||
//定义一个int的别名
|
||
type Level int
|
||
|
||
type OLog interface {
|
||
Debug(v ...interface{})
|
||
Info(v ...interface{})
|
||
Warn(v ...interface{})
|
||
Error(v ...interface{})
|
||
Fatal(v ...interface{})
|
||
Path() string
|
||
}
|
||
|
||
type oLog struct {
|
||
}
|
||
|
||
var (
|
||
F *os.File
|
||
DefaultPrefix = ""
|
||
DefaultCallerDepth = 2
|
||
logger *log.Logger
|
||
logPrefix = ""
|
||
levelFlags = []string{"DEBUG", "INFO", "WARN", "ERROR", "FATAL"}
|
||
)
|
||
|
||
//iota在const关键字出现时将被重置为0(const内部的第一行之前),const中每新增一行常量声明将使iota计数一次(iota可理解为const语句块中的行索引)。
|
||
const (
|
||
DEBUG Level = iota
|
||
INFO
|
||
WARN
|
||
ERROR
|
||
FATAL
|
||
)
|
||
|
||
//日志初始化
|
||
func LogSetup() {
|
||
var err error
|
||
filePath := fmt.Sprintf("%s", config.AppInfo.LogSavePath)
|
||
fileName := fmt.Sprintf("%s.%s",
|
||
config.AppInfo.LogSaveName,
|
||
config.AppInfo.LogFileExt,
|
||
)
|
||
F, err = file2.MustOpen(fileName, filePath)
|
||
if err != nil {
|
||
log.Fatalf("logging.Setup err: %v", err)
|
||
}
|
||
|
||
logger = log.New(F, DefaultPrefix, log.LstdFlags)
|
||
|
||
}
|
||
func (o *oLog) Path() string {
|
||
filePath := fmt.Sprintf("%s", config.AppInfo.LogSavePath)
|
||
fileName := fmt.Sprintf("%s.%s",
|
||
config.AppInfo.LogSaveName,
|
||
config.AppInfo.LogFileExt,
|
||
)
|
||
return filePath + fileName
|
||
}
|
||
func (o *oLog) Debug(v ...interface{}) {
|
||
setPrefix(DEBUG)
|
||
logger.Println(v)
|
||
}
|
||
|
||
func (o *oLog) Info(v ...interface{}) {
|
||
setPrefix(INFO)
|
||
logger.Println(v)
|
||
}
|
||
|
||
func (o *oLog) Warn(v ...interface{}) {
|
||
setPrefix(WARN)
|
||
logger.Println(v)
|
||
}
|
||
|
||
func (o *oLog) Error(v ...interface{}) {
|
||
setPrefix(ERROR)
|
||
logger.Println(v)
|
||
}
|
||
|
||
func (o *oLog) Fatal(v ...interface{}) {
|
||
setPrefix(FATAL)
|
||
logger.Println(v)
|
||
}
|
||
|
||
func setPrefix(level Level) {
|
||
_, file, line, ok := runtime.Caller(DefaultCallerDepth)
|
||
if ok {
|
||
logPrefix = fmt.Sprintf("[%s][%s:%d]", levelFlags[level], filepath.Base(file), line)
|
||
} else {
|
||
logPrefix = fmt.Sprintf("[%s]", levelFlags[level])
|
||
}
|
||
|
||
logger.SetPrefix(logPrefix)
|
||
}
|
||
|
||
func NewOLoger() OLog {
|
||
return &oLog{}
|
||
}
|