41 lines
1.0 KiB
Go
41 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"git.asio.asia/nixevol/NixMsg/test/load"
|
|
)
|
|
|
|
// 压测客户端骨架：连上 N 个 MQTT 连接并打印连接数。完整 1000 连接 / 10 分钟压测留给 Q3。
|
|
func main() {
|
|
addr := flag.String("addr", "", "MQTT broker host:port（必填，不要写死业务端口）")
|
|
n := flag.Int("n", 10, "连接数")
|
|
prefix := flag.String("prefix", "q-bench-", "client id 前缀")
|
|
hold := flag.Duration("hold", 3*time.Second, "保持连接时长")
|
|
flag.Parse()
|
|
if *addr == "" {
|
|
fmt.Fprintln(os.Stderr, "usage: mqttbench -addr host:port [-n 10]")
|
|
os.Exit(2)
|
|
}
|
|
|
|
b := &load.MQTTBench{Addr: *addr, ClientIDPrefix: *prefix, Timeout: 5 * time.Second}
|
|
defer b.Close()
|
|
if err := b.ConnectN(*n, os.Stdout); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
fmt.Printf("held %d connections for %s\n", b.Alive(), hold.String())
|
|
|
|
ch := make(chan os.Signal, 1)
|
|
signal.Notify(ch, os.Interrupt, syscall.SIGTERM)
|
|
select {
|
|
case <-time.After(*hold):
|
|
case <-ch:
|
|
}
|
|
}
|