[ADD]first commit
This commit is contained in:
@@ -0,0 +1,44 @@
|
||||
package logging
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
LogSavePath = "runtime/logs/"
|
||||
LogSaveName = "log"
|
||||
LogFileExt = "log"
|
||||
TimeFormat = "20060102"
|
||||
)
|
||||
func getLogFilePath() string {
|
||||
return fmt.Sprintf("%s", 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 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)
|
||||
if err != nil {
|
||||
log.Fatalf("Fail to OpenFile :%v", err)
|
||||
}
|
||||
return handle
|
||||
}
|
||||
func mkDir() {
|
||||
dir, _ := os.Getwd()
|
||||
err := os.MkdirAll(dir + "/" + getLogFilePath(), os.ModePerm)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,63 @@
|
||||
package logging
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
)
|
||||
|
||||
type Level int
|
||||
|
||||
var (
|
||||
F *os.File
|
||||
DefaultPrefix = ""
|
||||
DefaultCallerDepth = 2
|
||||
logger *log.Logger
|
||||
logPrefix = ""
|
||||
levelFlags = []string{"DEBUG", "INFO", "WARN", "ERROR", "FATAL"}
|
||||
)
|
||||
|
||||
const (
|
||||
DEBUG Level = iota
|
||||
INFO
|
||||
WARNING
|
||||
ERROR
|
||||
FATAL
|
||||
)
|
||||
|
||||
func init() {
|
||||
filePath := getLogFileFullPath()
|
||||
F = openLogFile(filePath)
|
||||
logger = log.New(F, DefaultPrefix, log.LstdFlags)
|
||||
}
|
||||
func Debug(v ...interface{}) {
|
||||
setPrefix(DEBUG)
|
||||
logger.Println(v)
|
||||
}
|
||||
func Info(v ...interface{}) {
|
||||
setPrefix(INFO)
|
||||
logger.Println(v)
|
||||
}
|
||||
func Warn(v ...interface{}) {
|
||||
setPrefix(WARNING)
|
||||
logger.Println(v)
|
||||
}
|
||||
func Error(v ...interface{}) {
|
||||
setPrefix(ERROR)
|
||||
logger.Println(v)
|
||||
}
|
||||
func Fatal(v ...interface{}) {
|
||||
setPrefix(FATAL)
|
||||
logger.Fatalln(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)
|
||||
}
|
||||
Reference in New Issue
Block a user