正在显示
8 个修改的文件
包含
43 行增加
和
163 行删除
@@ -54,6 +54,8 @@ spec: | @@ -54,6 +54,8 @@ spec: | ||
54 | env: | 54 | env: |
55 | - name: LOG_LEVEL | 55 | - name: LOG_LEVEL |
56 | value: "debug" | 56 | value: "debug" |
57 | + - name: LOG_FRAMEWORK | ||
58 | + value: "beego" | ||
57 | - name: LOG_TYPE | 59 | - name: LOG_TYPE |
58 | value: "file" | 60 | value: "file" |
59 | - name: ERROR_BASE_CODE | 61 | - name: ERROR_BASE_CODE |
@@ -61,7 +63,7 @@ spec: | @@ -61,7 +63,7 @@ spec: | ||
61 | - name: ERROR_BASE_CODE_MULTIPLE | 63 | - name: ERROR_BASE_CODE_MULTIPLE |
62 | value: "1000" | 64 | value: "1000" |
63 | - name: ENABLE_KAFKA_LOG | 65 | - name: ENABLE_KAFKA_LOG |
64 | - value: "true" | 66 | + value: "false" |
65 | - name: USER_MODULE_HOST | 67 | - name: USER_MODULE_HOST |
66 | value: "http://allied-creation-user-dev.fjmaimaimai.com" | 68 | value: "http://allied-creation-user-dev.fjmaimaimai.com" |
67 | - name: REDIS_HOST | 69 | - name: REDIS_HOST |
1 | package main | 1 | package main |
2 | 2 | ||
3 | import ( | 3 | import ( |
4 | + "fmt" | ||
4 | "github.com/beego/beego/v2/server/web" | 5 | "github.com/beego/beego/v2/server/web" |
6 | + "github.com/linmadan/egglib-go/log/logrus" | ||
7 | + "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/constant" | ||
5 | _ "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/constant" | 8 | _ "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/constant" |
6 | _ "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/pg" | 9 | _ "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/pg" |
10 | + "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/log" | ||
7 | _ "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/log" | 11 | _ "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/log" |
8 | _ "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/port/beego" | 12 | _ "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/port/beego" |
9 | ) | 13 | ) |
10 | 14 | ||
11 | func main() { | 15 | func main() { |
12 | // web主服务 | 16 | // web主服务 |
13 | - //if constant.ENABLE_KAFKA_LOG { | ||
14 | - // w, _ := logrus.NewKafkaWriter(constant.KAFKA_HOSTS, constant.TOPIC_LOG_STASH, false) | ||
15 | - // log.Logger.AddHook(w) | ||
16 | - //} | 17 | + if constant.ENABLE_KAFKA_LOG { |
18 | + w, err := logrus.NewKafkaWriter(constant.KAFKA_HOSTS, constant.TOPIC_LOG_STASH, false) | ||
19 | + if err != nil { | ||
20 | + log.Logger.Error(err.Error()) | ||
21 | + } | ||
22 | + log.Logger.AddHook(w) | ||
23 | + } | ||
24 | + | ||
25 | + fmt.Printf("Server running at:%d\n", web.BConfig.Listen.HTTPPort) | ||
26 | + | ||
17 | web.Run() | 27 | web.Run() |
18 | } | 28 | } |
@@ -5,7 +5,8 @@ import "os" | @@ -5,7 +5,8 @@ import "os" | ||
5 | const SERVICE_NAME = "allied-creation-cooperation" | 5 | const SERVICE_NAME = "allied-creation-cooperation" |
6 | 6 | ||
7 | // LOG_TYPE 日志相关设置 | 7 | // LOG_TYPE 日志相关设置 |
8 | -var LOG_TYPE = "terminal" // file, terminal, es | 8 | +var LOG_TYPE = "terminal" // file, terminal, es |
9 | +var LOG_FRAMEWORK = "beego" // beego logrus | ||
9 | var LOG_LEVEL = "debug" | 10 | var LOG_LEVEL = "debug" |
10 | var LOG_FILE = "app.log" | 11 | var LOG_FILE = "app.log" |
11 | var LOG_PREFIX = "[allied-creation-cooperation]" | 12 | var LOG_PREFIX = "[allied-creation-cooperation]" |
@@ -23,6 +24,9 @@ func init() { | @@ -23,6 +24,9 @@ func init() { | ||
23 | if os.Getenv("LOG_LEVEL") != "" { | 24 | if os.Getenv("LOG_LEVEL") != "" { |
24 | LOG_LEVEL = os.Getenv("LOG_LEVEL") | 25 | LOG_LEVEL = os.Getenv("LOG_LEVEL") |
25 | } | 26 | } |
27 | + if os.Getenv("LOG_FRAMEWORK") != "" { | ||
28 | + LOG_FRAMEWORK = os.Getenv("LOG_FRAMEWORK") | ||
29 | + } | ||
26 | if os.Getenv("LOG_FILE") != "" { | 30 | if os.Getenv("LOG_FILE") != "" { |
27 | LOG_FILE = os.Getenv("LOG_FILE") | 31 | LOG_FILE = os.Getenv("LOG_FILE") |
28 | } | 32 | } |
@@ -2,23 +2,27 @@ package constant | @@ -2,23 +2,27 @@ package constant | ||
2 | 2 | ||
3 | import "os" | 3 | import "os" |
4 | 4 | ||
5 | -// kafka 地址 | 5 | +// KAFKA_HOSTS kafka 地址 |
6 | var KAFKA_HOSTS = "192.168.0.250:9092,192.168.0.251:9092,192.168.0.252:9092" | 6 | var KAFKA_HOSTS = "192.168.0.250:9092,192.168.0.251:9092,192.168.0.252:9092" |
7 | 7 | ||
8 | -// kafka topic log stash | 8 | +// TOPIC_LOG_STASH kafka topic log stash |
9 | var TOPIC_LOG_STASH = "go_stash_dev" | 9 | var TOPIC_LOG_STASH = "go_stash_dev" |
10 | 10 | ||
11 | -// 是否启用日志收集 (本地不启用) | 11 | +// ENABLE_KAFKA_LOG 是否启用日志收集 (本地不启用) |
12 | var ENABLE_KAFKA_LOG = false | 12 | var ENABLE_KAFKA_LOG = false |
13 | 13 | ||
14 | func init() { | 14 | func init() { |
15 | - //if os.Getenv("KAFKA_HOSTS") != "" { | ||
16 | - // KAFKA_HOSTS = os.Getenv("KAFKA_HOSTS") | ||
17 | - //} | ||
18 | - //if os.Getenv("TOPIC_LOG_STASH") != "" { | ||
19 | - // POSTGRESQL_PORT = os.Getenv("TOPIC_LOG_STASH") | ||
20 | - //} | 15 | + if os.Getenv("KAFKA_HOSTS") != "" { |
16 | + KAFKA_HOSTS = os.Getenv("KAFKA_HOSTS") | ||
17 | + } | ||
18 | + if os.Getenv("TOPIC_LOG_STASH") != "" { | ||
19 | + POSTGRESQL_PORT = os.Getenv("TOPIC_LOG_STASH") | ||
20 | + } | ||
21 | if os.Getenv("ENABLE_KAFKA_LOG") != "" { | 21 | if os.Getenv("ENABLE_KAFKA_LOG") != "" { |
22 | - ENABLE_KAFKA_LOG = true | 22 | + if os.Getenv("ENABLE_KAFKA_LOG") == "true" { |
23 | + ENABLE_KAFKA_LOG = true | ||
24 | + } else if os.Getenv("ENABLE_KAFKA_LOG") == "false" { | ||
25 | + ENABLE_KAFKA_LOG = false | ||
26 | + } | ||
23 | } | 27 | } |
24 | } | 28 | } |
@@ -248,7 +248,7 @@ func (repository *CooperationApplicationRepository) Find(queryOptions map[string | @@ -248,7 +248,7 @@ func (repository *CooperationApplicationRepository) Find(queryOptions map[string | ||
248 | if userBaseId, ok := queryOptions["userBaseId"]; ok && userBaseId.(int64) != 0 { | 248 | if userBaseId, ok := queryOptions["userBaseId"]; ok && userBaseId.(int64) != 0 { |
249 | query.Where(`(cooperation_application.cooperation_application_applicant->>'userBaseId' ='?')`, userBaseId) | 249 | query.Where(`(cooperation_application.cooperation_application_applicant->>'userBaseId' ='?')`, userBaseId) |
250 | } | 250 | } |
251 | - if isCanceled, ok := queryOptions["isCanceled"]; ok && isCanceled.(int32) != 3 { | 251 | + if isCanceled, ok := queryOptions["isCanceled"]; ok && isCanceled.(int32) != 0 { |
252 | query.Where("is_canceled = ?", isCanceled) | 252 | query.Where("is_canceled = ?", isCanceled) |
253 | } | 253 | } |
254 | if companyId, ok := queryOptions["companyId"]; ok && companyId.(int64) != 0 { | 254 | if companyId, ok := queryOptions["companyId"]; ok && companyId.(int64) != 0 { |
@@ -26,7 +26,6 @@ type BeegoLog struct { | @@ -26,7 +26,6 @@ type BeegoLog struct { | ||
26 | } | 26 | } |
27 | 27 | ||
28 | func (logger *BeegoLog) AddHook(write io.Writer) { | 28 | func (logger *BeegoLog) AddHook(write io.Writer) { |
29 | - panic("implement me") | ||
30 | } | 29 | } |
31 | 30 | ||
32 | func (logger *BeegoLog) SetServiceName(serviceName string) { | 31 | func (logger *BeegoLog) SetServiceName(serviceName string) { |
@@ -2,25 +2,22 @@ package log | @@ -2,25 +2,22 @@ package log | ||
2 | 2 | ||
3 | import ( | 3 | import ( |
4 | "github.com/linmadan/egglib-go/log" | 4 | "github.com/linmadan/egglib-go/log" |
5 | + "github.com/linmadan/egglib-go/log/logrus" | ||
5 | "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/constant" | 6 | "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/constant" |
6 | ) | 7 | ) |
7 | 8 | ||
8 | var Logger log.Logger | 9 | var Logger log.Logger |
9 | 10 | ||
10 | func init() { | 11 | func init() { |
11 | - //Logger = logrus.NewLogrusLogger() | ||
12 | - //Logger.SetServiceName(constant.SERVICE_NAME) | ||
13 | - //Logger.SetLevel(constant.LOG_LEVEL) | ||
14 | - | ||
15 | - if constant.LOG_TYPE == "es" { | ||
16 | - Logger = NewServerLog() | ||
17 | - Logger.SetServiceName(constant.SERVICE_NAME) | ||
18 | - Logger.SetLevel(constant.LOG_LEVEL) | ||
19 | - } else { | 12 | + if constant.LOG_FRAMEWORK == "beego" { |
20 | Logger = NewBeegoLog(LoggerConfig{ | 13 | Logger = NewBeegoLog(LoggerConfig{ |
21 | Filename: constant.LOG_FILE, | 14 | Filename: constant.LOG_FILE, |
22 | Level: 7, | 15 | Level: 7, |
23 | MaxSize: 1024 * 1024 * 2, | 16 | MaxSize: 1024 * 1024 * 2, |
24 | }) | 17 | }) |
18 | + } else if constant.LOG_FRAMEWORK == "logrus" { | ||
19 | + Logger = logrus.NewLogrusLogger() | ||
20 | + Logger.SetServiceName(constant.SERVICE_NAME) | ||
21 | + Logger.SetLevel(constant.LOG_LEVEL) | ||
25 | } | 22 | } |
26 | } | 23 | } |
pkg/log/logrus.go
已删除
100644 → 0
1 | -package log | ||
2 | - | ||
3 | -import ( | ||
4 | - "fmt" | ||
5 | - "github.com/olivere/elastic/v7" | ||
6 | - "github.com/sirupsen/logrus" | ||
7 | - "gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/constant" | ||
8 | - "gopkg.in/sohlich/elogrus.v7" | ||
9 | - "io" | ||
10 | - "os" | ||
11 | -) | ||
12 | - | ||
13 | -type ServerLog struct { | ||
14 | - serviceName string | ||
15 | - logrus *logrus.Logger | ||
16 | -} | ||
17 | - | ||
18 | -func (logger *ServerLog) AddHook(write io.Writer) { | ||
19 | - panic("implement me") | ||
20 | -} | ||
21 | - | ||
22 | -func (logger *ServerLog) SetServiceName(serviceName string) { | ||
23 | - logger.serviceName = serviceName | ||
24 | -} | ||
25 | - | ||
26 | -func (logger *ServerLog) SetLevel(level string) { | ||
27 | - switch level { | ||
28 | - case "trace": | ||
29 | - logger.logrus.Level = logrus.TraceLevel | ||
30 | - case "debug": | ||
31 | - logger.logrus.Level = logrus.DebugLevel | ||
32 | - case "info": | ||
33 | - logger.logrus.Level = logrus.InfoLevel | ||
34 | - case "warn": | ||
35 | - logger.logrus.Level = logrus.WarnLevel | ||
36 | - case "error": | ||
37 | - logger.logrus.Level = logrus.ErrorLevel | ||
38 | - case "fatal": | ||
39 | - logger.logrus.Level = logrus.FatalLevel | ||
40 | - case "panic": | ||
41 | - logger.logrus.Level = logrus.PanicLevel | ||
42 | - default: | ||
43 | - logger.logrus.Level = logrus.DebugLevel | ||
44 | - } | ||
45 | -} | ||
46 | - | ||
47 | -func (logger *ServerLog) Trace(msg string, appends ...map[string]interface{}) { | ||
48 | - contextLogger := logger.logrus.WithFields(logrus.Fields{"serviceName": logger.serviceName}) | ||
49 | - for _, append := range appends { | ||
50 | - contextLogger = contextLogger.WithFields(append) | ||
51 | - } | ||
52 | - contextLogger.Trace(msg) | ||
53 | -} | ||
54 | - | ||
55 | -func (logger *ServerLog) Debug(msg string, appends ...map[string]interface{}) { | ||
56 | - contextLogger := logger.logrus.WithFields(logrus.Fields{"serviceName": logger.serviceName}) | ||
57 | - for _, append := range appends { | ||
58 | - contextLogger = contextLogger.WithFields(append) | ||
59 | - } | ||
60 | - contextLogger.Debug(msg) | ||
61 | -} | ||
62 | - | ||
63 | -func (logger *ServerLog) Info(msg string, appends ...map[string]interface{}) { | ||
64 | - contextLogger := logger.logrus.WithFields(logrus.Fields{"serviceName": logger.serviceName}) | ||
65 | - for _, append := range appends { | ||
66 | - contextLogger = contextLogger.WithFields(append) | ||
67 | - } | ||
68 | - contextLogger.Info(msg) | ||
69 | -} | ||
70 | - | ||
71 | -func (logger *ServerLog) Warn(msg string, appends ...map[string]interface{}) { | ||
72 | - contextLogger := logger.logrus.WithFields(logrus.Fields{"serviceName": logger.serviceName}) | ||
73 | - for _, append := range appends { | ||
74 | - contextLogger = contextLogger.WithFields(append) | ||
75 | - } | ||
76 | - contextLogger.Warn(msg) | ||
77 | -} | ||
78 | - | ||
79 | -func (logger *ServerLog) Error(msg string, appends ...map[string]interface{}) { | ||
80 | - contextLogger := logger.logrus.WithFields(logrus.Fields{"serviceName": logger.serviceName}) | ||
81 | - for _, append := range appends { | ||
82 | - contextLogger = contextLogger.WithFields(append) | ||
83 | - } | ||
84 | - contextLogger.Error(msg) | ||
85 | -} | ||
86 | - | ||
87 | -func (logger *ServerLog) Fatal(msg string, appends ...map[string]interface{}) { | ||
88 | - contextLogger := logger.logrus.WithFields(logrus.Fields{"serviceName": logger.serviceName}) | ||
89 | - for _, append := range appends { | ||
90 | - contextLogger = contextLogger.WithFields(append) | ||
91 | - } | ||
92 | - contextLogger.Fatal(msg) | ||
93 | -} | ||
94 | - | ||
95 | -func (logger *ServerLog) Panic(msg string, appends ...map[string]interface{}) { | ||
96 | - contextLogger := logger.logrus.WithFields(logrus.Fields{"serviceName": logger.serviceName}) | ||
97 | - for _, append := range appends { | ||
98 | - contextLogger = contextLogger.WithFields(append) | ||
99 | - } | ||
100 | - contextLogger.Panic(msg) | ||
101 | -} | ||
102 | - | ||
103 | -func NewServerLog() *ServerLog { | ||
104 | - logger := logrus.New() | ||
105 | - logger.Formatter = &logrus.JSONFormatter{ | ||
106 | - TimestampFormat: "2006-01-02 15:04:05", | ||
107 | - PrettyPrint: true, | ||
108 | - } | ||
109 | - if constant.LOG_TYPE == "file" { | ||
110 | - file, err := os.OpenFile("logrus.log", os.O_CREATE|os.O_WRONLY, 0666) | ||
111 | - if err == nil { | ||
112 | - _ = file.Truncate(1024 * 1024 * 2) | ||
113 | - logger.Out = file | ||
114 | - } else { | ||
115 | - logger.Info("Failed to log to file, using default stderr") | ||
116 | - } | ||
117 | - } else if constant.LOG_TYPE == "es" { | ||
118 | - url := fmt.Sprintf("http://%s:%s", "es.chenzhiying.asia", "80") | ||
119 | - client, err := elastic.NewClient(elastic.SetURL(url), elastic.SetSniff(false)) | ||
120 | - if err != nil { | ||
121 | - //log.Panic(err) | ||
122 | - panic(err) | ||
123 | - } | ||
124 | - hook, err := elogrus.NewAsyncElasticHook(client, "es.chenzhiying.asia", logrus.DebugLevel, "boss-appointment") | ||
125 | - if err != nil { | ||
126 | - //log.Panic(err) | ||
127 | - panic(err) | ||
128 | - } | ||
129 | - logger.Hooks.Add(hook) | ||
130 | - } else { | ||
131 | - logger.Out = os.Stdout | ||
132 | - } | ||
133 | - return &ServerLog{ | ||
134 | - logrus: logger, | ||
135 | - } | ||
136 | -} |
-
请 注册 或 登录 后发表评论