使用golang开发mqtt服务压力测试工具

最近公司项目需要优化mqtt服务,为了更全面的测试mqtt服务性能模拟正式环境,这里使用golang的写了一个测试工具

代码如下:

package main
import (
 "flag"
 "fmt"
 "sync"
 "time"
 //导入mqtt包
 MQTT "github.com/eclipse/paho.mqtt.golang"
)
var f MQTT.MessageHandler = func(client MQTT.Client, msg MQTT.Message) {
 fmt.Printf("TOPIC: %s\n", msg.Topic())
 fmt.Printf("MSG: %s\n", msg.Payload())
}
var fail_nums int = 0
func main() {
 //生成连接的客户端数
 c := flag.Uint64("n", 3000, "client nums")
 flag.Parse()
 nums := int(*c)
 wg := sync.WaitGroup{}
 for i := 0; i < nums; i++ {
 wg.Add(1)
 time.Sleep(5 * time.Millisecond)
 go createTask(i, &wg)
 }
 wg.Wait()
}
func createTask(taskId int, wg *sync.WaitGroup) {
 defer wg.Done()
 opts := MQTT.NewClientOptions().AddBroker("tcp://192.168.1.110:1883").SetUsername("test").SetPassword("test")
 opts.SetClientID(fmt.Sprintf("go-simple-client:%d-%d", taskId, time.Now().Unix()))
 opts.SetDefaultPublishHandler(f)
 opts.SetConnectTimeout(time.Duration(60) * time.Second)
 //创建连接
 c := MQTT.NewClient(opts)
 if token := c.Connect(); token.WaitTimeout(time.Duration(60)*time.Second) && token.Wait() && token.Error() != nil {
 fail_nums++
 fmt.Printf("taskId:%d,fail_nums:%d,error:%s \n", taskId, fail_nums, token.Error())
 return
 }
 //每隔5秒向topic发送一条消息
 i := 0
 for {
 i++
 time.Sleep(time.Duration(5) * time.Second)
 text := fmt.Sprintf("this is msg #%d! from task:%d", i, taskId)
 token := c.Publish("test", 1, false, text)
 token.Wait()
 }
 c.Disconnect(250)
 fmt.Println("task ok!!")
}

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传