package core import ( "github.com/sirupsen/logrus" "os" "path" "path/filepath" "vote-api/config" "vote-api/global" ) // Logrus 日志记录到文件 func Logrus() *logrus.Logger { // 日志文件 fileName := path.Join(config.C.Log.FilePath, config.C.Log.FileName) // 获取文件夹路径 dirPath := filepath.Dir(fileName) // 创建文件夹(如果不存在) err := os.MkdirAll(dirPath, os.ModePerm) if err != nil { panic("初始化日志文件失败") } // 写入文件 src, err := os.OpenFile(fileName, os.O_CREATE|os.O_APPEND|os.O_RDWR, os.ModePerm) if err != nil { panic("初始化日志文件失败") } global.Logger = logrus.New() // 设置输出 global.Logger.Out = src // 设置日志级别 global.Logger.SetLevel(logrus.DebugLevel) // 设置日志格式 global.Logger.SetFormatter(&logrus.TextFormatter{ TimestampFormat: "2006-01-02 15:04:05", }) return global.Logger }