engine/database/sql/logger.go

64 lines
1.3 KiB
Go
Raw Permalink Normal View History

2024-02-11 01:31:01 +08:00
package sql
import (
"fmt"
"io"
"log"
"os"
"path"
"path/filepath"
"time"
"git.apinb.com/bsm-sdk/engine/utils"
"git.apinb.com/bsm-sdk/engine/vars"
"gorm.io/gorm/logger"
)
var (
F *os.File
)
func setLogger(serviceName string, stdout bool) logger.Interface {
// create log dir
absPath := path.Join(utils.GetCurrentPath(), vars.SqlLogDir, serviceName)
if !utils.PathExists(absPath) {
os.MkdirAll(absPath, os.ModePerm)
}
// open log file
fullPath := getLogFileFullPath(absPath, getLogFileName(serviceName))
F = openLogFile(fullPath)
// write log
var multiOutput io.Writer
if stdout {
multiOutput = io.MultiWriter(os.Stdout, F)
} else {
multiOutput = io.MultiWriter(F)
}
return logger.New(log.New(multiOutput, "\r\n", log.LstdFlags), logger.Config{
SlowThreshold: 200 * time.Millisecond,
IgnoreRecordNotFoundError: false,
Colorful: false,
})
}
func getLogFileName(serviceName string) string {
return fmt.Sprintf("sql_%s.%s", time.Now().Format(vars.YYYYMMDD), vars.SqlLogFileExt)
}
func getLogFileFullPath(dir, fn string) string {
return filepath.Join(dir, fn)
}
func openLogFile(filePath string) *os.File {
handle, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
log.Printf("Fail to OpenFile :%v", err)
}
return handle
}