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

栏目: Go · 发布时间: 5年前

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

最近公司项目需要优化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!!")
}

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

缺失的终结

缺失的终结

【美】克尔•哈里斯 / 艾博 / 中国人民大学出版社 / 2017-6-15 / 39.00元

《缺失的终结》是一本反思互联网时代对普通人日常生活影响的书。作者认为,我们都曾渴望有一段远离现实生活烦扰的“瓦尔登湖”时光,悠闲地沉入自己的内心世界,从飞速变化的漩涡中暂时“缺失”。然而,互联网生活让我们每时每刻被互联网化,手环监测着我们的心跳和睡眠,微信、脸书、推特让我们人在床上就与世界链接起来,跑步了吗?走路了吗?英语单词背了吗?早餐的卡路里是否超标?坐在办公室,无数亟待处理的信息狂风暴雨般地......一起来看看 《缺失的终结》 这本书的介绍吧!

JS 压缩/解压工具
JS 压缩/解压工具

在线压缩/解压 JS 代码

UNIX 时间戳转换
UNIX 时间戳转换

UNIX 时间戳转换

HEX CMYK 转换工具
HEX CMYK 转换工具

HEX CMYK 互转工具