内容简介:最近公司项目需要优化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!!") }
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持 码农网
猜你喜欢:- 一个Web浏览器压力测试工具
- Apache JMeter 3.3 发布,压力测试工具
- Apache JMeter 5.3 发布,压力测试工具
- Apache JMeter 5.4 发布,压力测试工具
- Apache JMeter 5.2 发布,压力测试工具
- API 测试工具 Hitchhiker 0.6 发布,改进压力测试
本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们。
Linux程序设计
Neil Matthew、Richard Stones / 陈健、宋健建 / 人民邮电出版社 / 201005 / 99.00元
时至今日,Linux系统已经从一个个人作品发展为可以用于各种关键任务的成熟、高效和稳定的操作系统,因为具备跨平台、开源、支持众多应用软件和网络协议等优点,它得到了各大主流软硬件厂商的支持,也成为广大程序设计人员理想的开发平台。 本书是Linux程序设计领域的经典名著,以简单易懂、内容全面和示例丰富而受到广泛好评。中文版前两版出版后,在国内的Linux爱好者和程序员中也引起了强烈反响,这一热潮......一起来看看 《Linux程序设计》 这本书的介绍吧!