最近写一个tcp连接测试的脚本,用到了paramiko模块,遇到了一些让自己非常头疼的问题,在此记录。
常规的paramiko初始化和连接等操作网上已经有很多文章,不再赘述,下面主要说明自己遇到的问题和解决方法。
需求:从跳板机连接机器池,任意一台做启server,任选数十台启client,client不停向server发送包,测试连接是否中断
问题1:因为client端需要不断向server传送数据,paramiko执行的命令没有终止,所以直接用stdout和stderr的read() 方法读不到脚本的返回值(read()方法只能读取已经执行完毕的命令的结果)
解决方法:用stdout(stderr)的readline() 方法,可以逐行读取脚本返回值(建议多读paramiko官方文档):
问题2:用readline()可以读取stderr和stdout的实时打印值,但是因为脚本的执行可能有出现stderr,也有正常的stdout,当你监测的buffer file没有值得时候,线程会阻塞,会一直等待监测的buffer file,直到有值为止,而一般情况是,这条线程不会执行其他程序,永久阻塞了
解决方法:将脚本中的error错误通过try...except...转化成正常输出,当监测到错误后重新执行循环,重新执行脚本核心部分;print时最好加上flush,增加方法2和3有不同
其他的解决思路:可以再起两个线程,一个监测stderr,一个监测stdout
补充:获取paramiko exec_command(cmd)执行命令的状态
当通过ssh_client连接到远端服务器后,需要调用exec_command() 来执行远端命令,执行会返回三个FIle Descriptor(文件描述符):
stdin,stdout,stderr = ssh.exec_command(cmd)
运行cmd的返回值(或者pytho脚本等的print值)会被写入stdout,执行cmd的错误信息会写入stderr,我们通过FD的方法read()或者readline()等等(可见官方文档http://docs.paramiko.org/en/2.4/api/file.html)读取命令的返回值,另外一种表示,如果cmd是非延时的,可以通过channel = stdout.channel ===>status = channel.recv_exit_status()获取cmd命令的状态码