config
This commit is contained in:
parent
20d0a2bddd
commit
d596cab3fd
2
.gitignore
vendored
2
.gitignore
vendored
@ -4,7 +4,6 @@
|
|||||||
!.vscode/launch.json
|
!.vscode/launch.json
|
||||||
!.vscode/extensions.json
|
!.vscode/extensions.json
|
||||||
!.vscode/*.code-snippets
|
!.vscode/*.code-snippets
|
||||||
!main.go
|
|
||||||
|
|
||||||
# Local History for Visual Studio Code
|
# Local History for Visual Studio Code
|
||||||
.history/
|
.history/
|
||||||
@ -19,7 +18,6 @@ log/*
|
|||||||
*.dll
|
*.dll
|
||||||
*.so
|
*.so
|
||||||
*.dylib
|
*.dylib
|
||||||
apisirius
|
|
||||||
|
|
||||||
# Test binary, built with `go test -c`
|
# Test binary, built with `go test -c`
|
||||||
*.test
|
*.test
|
||||||
|
37
cmd/apisirius/config.go
Normal file
37
cmd/apisirius/config.go
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
|
||||||
|
"github.com/spf13/viper"
|
||||||
|
"gitstore.ru/tolikproh/sirius/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
configPath string
|
||||||
|
configFile string = "config"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
flag.StringVar(&configPath, "conf-path", "configs", "path to config file 'config.yml'")
|
||||||
|
flag.StringVar(&configFile, "conf-file", "config", "name file to config")
|
||||||
|
}
|
||||||
|
|
||||||
|
func initConfig() (*model.Config, error) {
|
||||||
|
cnf := model.NewConfig()
|
||||||
|
|
||||||
|
viper.SetConfigName(configFile)
|
||||||
|
viper.SetConfigType("yaml")
|
||||||
|
viper.AddConfigPath(configPath)
|
||||||
|
viper.AddConfigPath("./")
|
||||||
|
|
||||||
|
if err := viper.ReadInConfig(); err != nil {
|
||||||
|
return cnf, err
|
||||||
|
} else {
|
||||||
|
if err := viper.Unmarshal(&cnf); err != nil {
|
||||||
|
return cnf, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return cnf, nil
|
||||||
|
}
|
46
cmd/apisirius/logger.go
Normal file
46
cmd/apisirius/logger.go
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io"
|
||||||
|
|
||||||
|
"github.com/sirupsen/logrus"
|
||||||
|
"github.com/snowzach/rotatefilehook"
|
||||||
|
"gitstore.ru/tolikproh/sirius/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewLogger(cfg *model.Config) *logrus.Logger {
|
||||||
|
|
||||||
|
logFile := "sirius_server.log"
|
||||||
|
|
||||||
|
log := logrus.New()
|
||||||
|
|
||||||
|
logLevel := model.ValidLogLevel(cfg)
|
||||||
|
|
||||||
|
log.SetLevel(logLevel)
|
||||||
|
|
||||||
|
rotateFileHook, err := rotatefilehook.NewRotateFileHook(rotatefilehook.RotateFileConfig{
|
||||||
|
Filename: cfg.Srv.LogPath + "/" + logFile,
|
||||||
|
MaxSize: 1, // megabytes
|
||||||
|
MaxBackups: 7, // amouts
|
||||||
|
MaxAge: 1, //days
|
||||||
|
Level: logLevel,
|
||||||
|
Formatter: &logrus.JSONFormatter{},
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
logrus.Fatalf("Failed to initialize file rotate hook: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
//log.SetOutput(colorable.NewColorableStdout())
|
||||||
|
log.SetOutput(io.Discard)
|
||||||
|
// log.SetFormatter(&logrus.TextFormatter{
|
||||||
|
// PadLevelText: true,
|
||||||
|
// ForceColors: true,
|
||||||
|
// FullTimestamp: true,
|
||||||
|
// TimestampFormat: "2006-01-02 15:04:05",
|
||||||
|
// })
|
||||||
|
|
||||||
|
log.AddHook(rotateFileHook)
|
||||||
|
|
||||||
|
return log
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user