Python实现的HTTP并发测试完整示例

时间:2022-10-09 08:30:34

可修改变量thread_count指定最大的并发数量,即线程的数量。

完成之后,打印输出失败的次数,以及开始时间和结束时间,单位是毫秒。

主要是学习一下Python,仅供参考。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#!/usr/bin/python3
 
import sys, time, json, _thread
import http.client, urllib.parse
 
thread_count = 100  #并发数量
now_count = 0
error_count = 0
begin_time = ''
 
lock_obj = _thread.allocate()
 
def test_http_engine():
  global now_count
  global error_count
  global thread_count
  global begin_time
  conn = None
  if now_count == 0:
    begin_time = int(round(time.time() * 1000))
  try:
    conn = http.client.HTTPConnection("192.168.1.1", 80)
    conn.request('GET', '/')
 
    response = conn.getresponse()
    data = response.read()
    print (data)
 
    if json.dumps(response.status) != '200':
      error_count += 1;
      print ('error count: ' + str(error_count))
 
    sys.stdout.flush()
    now_count += 1
    if now_count == thread_count:
      print ('### error count: ' + str(error_count) + ' ###')
      print ('### begin time : ' + str(begin_time))
      print ('### end time  : ' + str(int(round(time.time() * 1000))))
 
  except Exception as e:
    print (e)
  finally:
    if conn:
      conn.close()
 
def test_thread_func():
  global now_count
  global lock_obj
  cnt = 0
 
  lock_obj.acquire()
  print ('')
  print ('=== Request: ' + str(now_count) + ' ===')
 
  cnt += 1
  test_http_engine()
  sys.stdout.flush()
  lock_obj.release()
 
 
def test_main():
  global thread_count
  for i in range(thread_count):
    _thread.start_new_thread(test_thread_func, ())
 
if __name__=='__main__':
  test_main()
  while True:
    time.sleep(5)