增加链路追踪配置,优化服务退出流程,gf版本升级到v2.4.2

This commit is contained in:
孟帅
2023-06-09 19:13:26 +08:00
parent 48f8c20d9c
commit 05e85f52e7
16 changed files with 94 additions and 33 deletions

View File

@@ -11,8 +11,8 @@ import (
"github.com/Shopify/sarama"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/os/gproc"
"os"
"hotgo/internal/consts"
"hotgo/utility/simple"
"time"
)
@@ -109,7 +109,7 @@ func (r *KafkaMq) ListenReceiveMsgDo(topic string, receiveDo func(mqMsg MqMsg))
<-consumer.ready
Logger().Debug(ctx, "kafka consumer up and running!...")
gproc.AddSigHandlerShutdown(func(sig os.Signal) {
simple.Event().Register(consts.EventServerClose, func(ctx context.Context, args ...interface{}) {
Logger().Debug(ctx, "kafka consumer close...")
cancel()
if err = r.consumerIns.Close(); err != nil {
@@ -202,7 +202,7 @@ func doRegisterKafkaProducer(connOpt KafkaConfig, mqIns *KafkaMq) (err error) {
return
}
gproc.AddSigHandlerShutdown(func(sig os.Signal) {
simple.Event().Register(consts.EventServerClose, func(ctx context.Context, args ...interface{}) {
g.Log().Debug(ctx, "kafka producer AsyncClose...")
mqIns.producerIns.AsyncClose()
})