add log level

liuxiaolong

2019-05-31 63cb6a9ac3f6cceee243b9375ca5a868a0906130

add log level

 log/log.go

@@ -16,6 +16,16 @@
    Log.SetLevel(logrus.ErrorLevel)
}

func SetLogLevel(level string){
    if level == "ErrorLevel" {
        log.Log.SetLevel(logrus.ErrorLevel)
    } else if level == "InfoLevel" {
        log.Log.SetLevel(logrus.InfoLevel)
    } else {
        log.Log.SetLevel(logrus.InfoLevel)
    }
}

func NewLogger() *logrus.Logger {
    if Log != nil {
        return Log

 server.go

@@ -18,9 +18,13 @@
var port = flag.Int("port", 6000, "The port to listen on; ")
var sec = flag.Int("sec", 10, "the second for query data. ")

var Level = flag.String("level","ErrorLevel","log level")

func main() {
    flag.Parse()

    log.SetLogLevel(*Level)

    fmt.Println(*port)
    src := *addr + ":" + strconv.Itoa(*port)
    listener, err := net.Listen("tcp", src)