上一章节，我们采用多线程的技术去进行服务器端口的扫描，遗留了一些问题待优化，今天，我们采用协程的方式去尝试一下是否解决这个问题。
协程是一种轻量级的线程，协程拥有自己的寄存器上下文和栈。协程调度切换时，将寄存器上下文和栈保存到其他地方，在切回来的时候，恢复先前保存的寄存器上下文和栈。因此：协程能保留上一次调用时的状态，每次过程重入时，就相当于进入上一次调用的状态。也就是说同一线程下的一段代码执行着执行着就可以中断，然后跳去执行另一段代码，当再次回来执行代码块的时候，接着从之前中断的地方开始执行。
协程的优点：
1、执行效率高，尤其是在线程数较多的情况下，与多线程对比的优势更明显
2、不需要多线程的锁机制，因为只有一个线程，也不存在同时写变量冲突，在协程中控制共享资源不加锁，只需要判断状态就好，因为执行效率比多线程高很多。
缺点：
1、无法利用多核资源：协程的本质是个单线程,它不能同时将 单个CPU 的多个核用上,协程需要和进程配合才能运行在多CPU上.当然我们日常所编写的绝大部分应用都没有这个必要，除非是cpu密集型应用。
2、进行阻塞（Blocking）操作（如IO时）会阻塞掉整个程序
接下来，让我们通过一段代码来看一下运行的效果：
import gevent from gevent import monkey import time def fun1(): for num in range(3): print('fun1方法正在运行') #time.sleep(1) def fun2(): for num in range(3): print('fun2方法正在运行') #time.sleep(1) # 创建协程对象 t1 = gevent.spawn(fun1) t2 = gevent.spawn(fun2) monkey.patch_all() gevent.joinall([t1, t2])
以上代码执行的时候，输出结果如下：
fun1方法正在运行 fun1方法正在运行 fun2方法正在运行 fun2方法正在运行 fun2方法正在运行
是不是跟预想的不一样呢，是的，因为程序执行没有阻塞/中断，所以打印结果没有交叉打印 ，把time.sleep(1)放开后，再执行：
import gevent from gevent import monkey import time def fun1(): for num in range(3): print('fun1方法正在运行') time.sleep(1) # 协程遇到耗时操作后会自动切换其他协程运行 def fun2(): for num in range(3): print('fun2方法正在运行') time.sleep(1) # 协程遇到耗时操作后会自动切换其他协程运行 # 创建协程对象 t1 = gevent.spawn(fun1) t2 = gevent.spawn(fun2) monkey.patch_all() gevent.joinall([t1, t2])
打印结果如下：
fun1方法正在运行 fun2方法正在运行 fun1方法正在运行 fun2方法正在运行 fun1方法正在运行 fun2方法正在运行
代码说明:
本次采用gevent库实现协程的相关操作，在使用之前需要先安装该插件。
安装命令：pip install gevent
gevent.spawn()函数：创建协程对象
gevnet.joinall([传入携程对象列表])：会等待所有协程对象运行结束后再退出
接下来改造端口扫描的代码，采用协程的方式实现：
import socket import time import gevent from gevent import monkey from gevent.pool import Pool monkey.patch_all() def scan_port(host, port): sk = socket.socket() sk.settimeout(0.5) conn_result = sk.connect_ex((host, port)) if conn_result == 0: print(f'服务器{host}的{port}端口已开放') sk.close() def gevent_scan_host(host): # 8.129.162.225 start_time = time.time() run_list = [] g = Pool(200) # 限制协程并发数量，单线程的，不要设置太大 for port in range(0, 65536): run_list.append(g.spawn(scan_port, host, port)) gevent.joinall(run_list) end_time = time.time() print(f'耗时:{end_time-start_time}') host = input('请输入服务器ip地址:') gevent_scan_host(host)
相关文章都是边学习边整理的笔记，如有错误，欢迎指正。如需进学习交流群，或者互相交流，文章催更，商务合作等，可添加微信xiaobotester ，记得备注一下。
