From 00d17dedf72093fdcb5a8179c3d91f4f936517c4 Mon Sep 17 00:00:00 2001 From: "qiyunfanbo126.com" <815699> Date: Fri, 17 Jan 2025 15:37:01 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E9=85=8D=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/handlers/mq/quenue.go | 2 +- app/http/tcppool/single.go | 2 +- config/config.go | 2 ++ main.go | 10 +++++++++- 4 files changed, 13 insertions(+), 3 deletions(-) diff --git a/app/handlers/mq/quenue.go b/app/handlers/mq/quenue.go index 99374f7..12c3db9 100644 --- a/app/handlers/mq/quenue.go +++ b/app/handlers/mq/quenue.go @@ -26,7 +26,7 @@ func StartQunueServer() error { if config.GetConf().StartQunue == 1 { for i := 0; i < 1; i++ { fmt.Println("对列" + strconv.Itoa(i)) - startQunue(common.ORDER_RESEND_TOPICAL, OrderCharge, common.MQ_KFK_V2, 0, "", i) //单聊 + startQunue(config.GetConf().Topical, OrderCharge, common.MQ_KFK_V2, 0, "", i) //单聊 } } select {} diff --git a/app/http/tcppool/single.go b/app/http/tcppool/single.go index f4091be..f4ab20a 100644 --- a/app/http/tcppool/single.go +++ b/app/http/tcppool/single.go @@ -24,7 +24,7 @@ type TcpHelper struct { func (t *TcpHelper) Init(port string) *TcpHelper { lockSingle.Do(func() { - var conn, err = net.DialTimeout("tcp", "192.168.110.50:"+port, 10*time.Second) + var conn, err = net.DialTimeout("tcp", config.GetConf().Url+":"+port, 10*time.Second) if err == nil { t.client = conn t.watch(t.client) diff --git a/config/config.go b/config/config.go index c8b35e2..ee59235 100644 --- a/config/config.go +++ b/config/config.go @@ -33,6 +33,8 @@ type Config struct { KafkaGroup string `toml:"KafkaGroup"` OrderPort string `toml:"OrderPort"` Num int `toml:"Num"` + Url string `toml:"Url"` + Topical string `toml:"Topical"` } func newConfig() *Config { diff --git a/main.go b/main.go index a8ca9c4..7ad45a2 100644 --- a/main.go +++ b/main.go @@ -64,13 +64,18 @@ var ( port string kafka string num int + url string + group string + topic string ) func initConfig() { flag.StringVar(&kafka, "kafka", "", "kafka url") flag.StringVar(&port, "port", "9502", "tcp port") flag.IntVar(&num, "num", 10, "进程数量") - + flag.StringVar(&url, "url", "", "内网地址") + flag.StringVar(&group, "group", "", "消费分组") + flag.StringVar(&topic, "topic", "", "消费topic") } func main() { //解析启动命令 @@ -121,6 +126,9 @@ func startServer(opts *config.Options) (err error) { config.GetConf().KafkaUrl = kafka config.GetConf().OrderPort = port config.GetConf().Num = num + config.GetConf().Url = url + config.GetConf().KafkaGroup = group + config.GetConf().Topical = topic fmt.Println(port, kafka, num) //根据启动命令行参数,决定启动哪种服务模式 switch opts.App {