性能测试 - Locust dubbo client

歪小Y_02

Locust dubbo客户端实现

1. # _*_ coding:utf-8 _*_
   
2. 
   
3. '''
   
4. Locust dubbo client
   
5. Max.Bai
   
6. 2018-08
   
7. '''
   
8. 
   
9. 
   
10. import time
    
11. import random
    
12. import json
    
13. import socket
    
14. import dubbo_telnet
    
15. import requests
    
16. from locust import Locust, TaskSet, events, task
    
17. 
    
18. 
    
19. class DubboClient(object):
    
20.     def __init__(self, host, port):
    
21.         self.conn = dubbo_telnet.connect(host, port)
    
22.         self.conn.set_connect_timeout(10)
    
23. 
    
24.     def do(self,msg):
    
25.         start_time = time.time()
    
26.         try:
    
27.             self.conn.do(msg)
    
28.         except Exception as e:
    
29.             total_time = int((time.time() - start_time) * 1000)
    
30.             events.request_failure.fire(request_type="dubbo", name="start", response_time=total_time, exception=e)
    
31.         else:
    
32.             total_time = int((time.time() - start_time) * 1000)
    
33.             events.request_success.fire(request_type="dubbo", name="start", response_time=total_time, response_length=0)
    
34. 
    
35. 
    
36. 
    
37. class DubboLocust(Locust):
    
38.     """
    
39.     This is the abstract Locust class which should be subclassed. It provides an dubbo client
    
40.     that can be used to make dubbo requests that will be tracked in Locust's statistics.
    
41.     """
    
42.     def __init__(self, *args, **kwargs):
    
43.         super(DubboLocust, self).__init__(*args, **kwargs)
    
44.         self.client = DubboClient(self.host, self.port)
    
45. 
    
46. 
    
47. 
    
48. class TestUser(DubboLocust):
    
49. 
    
50.     host = "119.29.67.113"
    
51.     port = 20904
    
52.     min_wait = 2000
    
53.     max_wait = 2000
    
54. 
    
55.     class task_set(TaskSet):
    
56.         @task
    
57.         def send_data1(self):
    
58.             msg = 'invoke com.abc.com.base.service.faceattendance.abcService.action("parm1", "parm2", "parm3")'
    
59.             response = self.client.do(msg)
    
60.             print "recv:%s"%str(response)
    
61. 
    
62. 
    
63. if __name__ == "__main__":
    
64.     user = TestUser()
    
65.     user.run()
    

复制代码

qqq911

感谢分享

Miss_love

感谢分享