Browse Source

Merge pull request #13 from granty1/master

Add check for log directory.
main
蒋吉兆 5 years ago
committed by GitHub
parent
commit
854b96ffa2
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
  1. 6
      QMPlusServer/init/qmlog/qmlog.go
  2. 14
      QMPlusServer/tools/directory.go

6
QMPlusServer/init/qmlog/qmlog.go

@ -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{
} }
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",

14
QMPlusServer/tools/directory.go

@ -0,0 +1,14 @@
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
}
Loading…
Cancel
Save