tcp_server_协程gevent版本

Posted 2021-03-14 liuyanygz

 

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Time : 2020/1/23 1:50
# @Author : liuyan
# @File : test5_tcp_server_5协程gevent版本.py
# @Software: PyCharm

import gevent
from gevent import monkey; monkey.patch_all()

#使用此方法，会将代码中检查一遍，如有time.sleep()等延时方法，会自动转换为gevent.sleep()
#在执行时动态替换,通常是在startup的时候.
#把标准库中的thread/socket等给替换掉.这样我们在后面使用socket的时候能够跟寻常一样使用,无需改动
# 调用此函数就不需要将time.sleep()手动换为gevent.sleep()，socket也不需要手动替换

# gevent的优势不仅仅是在代码中调用方便，厉害的是它拥有的monkey机制。
# 假设你不愿意修改原来已经写好的python代码，但是又想充分利用gevent机制，那么你就可以用monkey来做到这一点。
# 你所要做的就是在文件开头打一个patch，那么它就会自动替换你原来的thread、socket、time、multiprocessing等代码，全部变成gevent框架。
# 这一切都是由gevent自动完成的。注意这个patch是在所有module都import了之后再打，否则没有效果。
# 甚至在编写的Web App代码的时候，不需要引入gevent的包，也不需要改任何代码，仅仅在部署的时候，用一个支持gevent的WSGI服务器，就可以获得数倍的性能提升
#
# 链接：https://www.jianshu.com/p/bb6c7f9aa1ae


import time
import socket  # 使用gevent，原代码不需要修改，socket和time.sleep不需要手动替换，调用monkey.patch_all()自动替换

def each_do(clientfd):
    while True:
        recvdata = clientfd.recv(1024)  # gevent遇到这种耗时操作，自动切换动其它协程
        if len(recvdata) > 0:
            clientfd.send(recvdata)
            print("recv=%s, echo back" % recvdata.decode("utf8"))
        else:
            print("closed")
            clientfd.close()
            break


def tcp_server():
    fd_listen = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    fd_listen.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # 地址重用
    fd_listen.bind((‘‘, 7777))
    fd_listen.listen(3)  # 可以不指定最大允许量, 指定表示："正在排队等待accept的半连接个数"

    while True:
        clientfd, clientInfo = fd_listen.accept()  # gevent遇到这种耗时操作，自动切换动其它协程
        print("accept from %s" % (clientInfo,))
        gevent.spawn(each_do, clientfd)


if __name__ == ‘__main__‘:
    tcp_server()