From 63cb6a9ac3f6cceee243b9375ca5a868a0906130 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期五, 31 五月 2019 13:03:03 +0800
Subject: [PATCH] add log level
---
log/log.go | 10 ++++++++++
server.go | 4 ++++
2 files changed, 14 insertions(+), 0 deletions(-)
diff --git a/log/log.go b/log/log.go
index e5ca557..929dd33 100644
--- a/log/log.go
+++ b/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
diff --git a/server.go b/server.go
index 0caff95..7aa34e2 100644
--- a/server.go
+++ b/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)
--
Gitblit v1.8.0