提交 77a1ff1d 编写于 作者: G Grant Yin

Add check for log directory.

上级 a801bf7a
...@@ -6,6 +6,7 @@ import ( ...@@ -6,6 +6,7 @@ import (
rotatelogs "github.com/lestrrat/go-file-rotatelogs" rotatelogs "github.com/lestrrat/go-file-rotatelogs"
"github.com/rifflock/lfshook" "github.com/rifflock/lfshook"
"github.com/sirupsen/logrus" "github.com/sirupsen/logrus"
"main/tools"
"os" "os"
"time" "time"
) )
...@@ -19,6 +20,11 @@ func InitLog() *logrus.Logger{ ...@@ -19,6 +20,11 @@ func InitLog() *logrus.Logger{
} }
QMLog.Out = src QMLog.Out = src
QMLog.SetLevel(logrus.DebugLevel) QMLog.SetLevel(logrus.DebugLevel)
if ok, _ := tools.PathExists("./log"); !ok {
// Directory not exist
fmt.Println("Create log.")
os.Mkdir("log", os.ModePerm)
}
apiLogPath := "./log/api.log" apiLogPath := "./log/api.log"
logWriter, err := rotatelogs.New( logWriter, err := rotatelogs.New(
apiLogPath+".%Y-%m-%d-%H-%M.log", apiLogPath+".%Y-%m-%d-%H-%M.log",
......
package tools
import "os"
func PathExists(path string) (bool, error) {
_, err := os.Stat(path)
if err == nil {
return true, nil
}
if os.IsNotExist(err) {
return false, nil
}
return false, err
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册