[FIX]优化配置读取
This commit is contained in:
+26
-28
@@ -2,43 +2,41 @@ package logging
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"github.com/go-pripro/shop/pkg/file"
|
||||
"github.com/go-pripro/shop/pkg/setting"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
LogSavePath = "runtime/logs/"
|
||||
LogSaveName = "log"
|
||||
LogFileExt = "log"
|
||||
TimeFormat = "20060102"
|
||||
)
|
||||
func getLogFilePath() string {
|
||||
return fmt.Sprintf("%s", LogSavePath)
|
||||
return fmt.Sprintf("%s%s", setting.AppSetting.RuntimeRootPath, setting.AppSetting.LogSavePath)
|
||||
}
|
||||
func getLogFileFullPath() string {
|
||||
prefixPath := getLogFilePath()
|
||||
suffixPath := fmt.Sprintf("%s%s.%s", LogSaveName, time.Now().Format(TimeFormat), LogFileExt)
|
||||
return fmt.Sprintf("%s%s", prefixPath, suffixPath)
|
||||
|
||||
func getLogFileName() string {
|
||||
return fmt.Sprintf("%s%s.%s",
|
||||
setting.AppSetting.LogSavename,
|
||||
time.Now().Format(setting.AppSetting.TimeFormat),
|
||||
setting.AppSetting.LogFileExt,
|
||||
)
|
||||
}
|
||||
func openLogFile(filePath string) *os.File {
|
||||
_, err := os.Stat(filePath)
|
||||
switch {
|
||||
case os.IsNotExist(err):
|
||||
mkDir()
|
||||
case os.IsPermission(err):
|
||||
log.Fatalf("Permission :%v", err)
|
||||
}
|
||||
handle, err := os.OpenFile(filePath, os.O_APPEND | os.O_CREATE | os.O_WRONLY, 0644)
|
||||
|
||||
func openLogFile(fileName, filePath string) (*os.File, error) {
|
||||
dir, err := os.Getwd()
|
||||
if err != nil {
|
||||
log.Fatalf("Fail to OpenFile :%v", err)
|
||||
return nil, fmt.Errorf("os.Getwd err: %v", err)
|
||||
}
|
||||
return handle
|
||||
}
|
||||
func mkDir() {
|
||||
dir, _ := os.Getwd()
|
||||
err := os.MkdirAll(dir + "/" + getLogFilePath(), os.ModePerm)
|
||||
src := dir + "/" + filePath
|
||||
perm := file.CheckPermission(src)
|
||||
if perm == true {
|
||||
return nil, fmt.Errorf("file.CheckPermission Permission denied src: %s", src)
|
||||
}
|
||||
err = file.IsNotExistMkDir(src)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
return nil, fmt.Errorf("file.IsNotExistMkDir src: %s, err: %v", src, err)
|
||||
}
|
||||
f, err := file.Open(src + fileName, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Fail to OpenFile :%v", err)
|
||||
}
|
||||
return f, nil
|
||||
}
|
||||
+10
-3
@@ -27,9 +27,16 @@ const (
|
||||
FATAL
|
||||
)
|
||||
|
||||
func init() {
|
||||
filePath := getLogFileFullPath()
|
||||
F = openLogFile(filePath)
|
||||
func Setup() {
|
||||
var err error
|
||||
|
||||
filePath := getLogFilePath()
|
||||
fileName := getLogFileName()
|
||||
F, err = openLogFile(fileName, filePath)
|
||||
if err != nil {
|
||||
log.Fatalf("logging.Setup err: %v", err)
|
||||
}
|
||||
|
||||
logger = log.New(F, DefaultPrefix, log.LstdFlags)
|
||||
}
|
||||
func Debug(v ...interface{}) {
|
||||
|
||||
Reference in New Issue
Block a user