From 8324f872ef3a4d0c978a9b1d062800c6a1701c12 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 01 十二月 2023 09:58:17 +0800 Subject: [PATCH] fix --- nsq/nsq.go | 21 +++++++++++++++++++++ 1 files changed, 21 insertions(+), 0 deletions(-) diff --git a/nsq/nsq.go b/nsq/nsq.go index 68274b2..c447e8b 100644 --- a/nsq/nsq.go +++ b/nsq/nsq.go @@ -9,6 +9,7 @@ "context" "errors" "fmt" + "github.com/shirou/gopsutil/net" "sync" "sync/atomic" ) @@ -85,10 +86,30 @@ }) } +func (c *consumerManager) ping() bool { + connections, err := net.Connections("inet") + if err != nil { + fmt.Println("Error:", err) + return false + } + for _, conn := range connections { + ipPort := fmt.Sprintf("%s:%d", conn.Raddr.IP, conn.Raddr.Port) + fmt.Println("net.Connections ipPort", ipPort) + if ipPort == conf.Conf.NsqConf.NsqdAddr && conn.Status == "ESTABLISHED" { + return true + } + } + return false +} + func Init() error { return defaultConsumerManager.init() } +func Ping() bool { + return defaultConsumerManager.ping() +} + func Stop() { defaultConsumerManager.stop() StopProducer() -- Gitblit v1.8.0