You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

58 lines
1.3 KiB
Go

package main
import (
"flag"
"fmt"
"os"
log "github.com/fangdingjun/go-log"
"github.com/fangdingjun/go-log/formatters"
"github.com/fangdingjun/go-log/writers"
)
var _config *conf
func main() {
var configfile string
var logfile string
var loglevel string
var logFileSize int64
var logKeepCount int
flag.StringVar(&configfile, "c", "config.yaml", "config file")
flag.StringVar(&logfile, "log_file", "", "log file path, default to stdout")
flag.StringVar(&loglevel, "log_level", "INFO", "log level, levels are: \nOFF, FATAL, PANIC, ERROR, WARN, INFO, DEBUG")
flag.Int64Var(&logFileSize, "log_file_size", 10, "max log file size, MB")
flag.IntVar(&logKeepCount, "log_count", 10, "max count of log file to keep")
flag.Parse()
if logfile != "" {
log.Default.Out = &writers.FixedSizeFileWriter{
Name: logfile,
MaxSize: logFileSize * 1024 * 1024,
MaxCount: logKeepCount,
}
}
if loglevel != "" {
lvname, err := log.ParseLevel(loglevel)
if err != nil {
fmt.Fprintf(os.Stderr, "invalid level %s", loglevel)
os.Exit(1)
}
log.Default.Level = lvname
}
log.Default.Formatter = &formatters.TextFormatter{
TimeFormat: "2006-01-02 15:04:05.000",
}
cfg, err := loadConfig(configfile)
if err != nil {
log.Fatalf("load config file error: %s", err)
}
_config = cfg
log.Debugf("config: %+v", _config)
initHandler()
select {}
}