测试系统中有一项记录ssh登录日志,需要对此进行并发压力测试。
于是用多线程进行python并发记录
因为需要安装的一些依赖和模块比较麻烦,脚本完成后再用pyinstaller打成exe包分发给其他测试人员一起使用。
1.脚本编写
# -*- coding: utf-8 -*-
import paramiko
import threading
import time
lt = []
def ssh(a,xh,sp):
count = 0
for i in range(0,xh):
try:
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('ip地址',22,'用户名', '密码')
ssh.close()
print u"线程[%s]第[%s]次登录"%(a,i)
if sp != 0:
time.sleep(sp)
count += 1
except:
print u"线程异常,已处理"
lt.append(count)
if __name__ == "__main__":
figlet = '''
_____ _____ _
| ___| | _ \ | |
| |__ | |_| | | |
| __| |