tcp_server_协程gevent版本

zuiaiqun 2020-01-23

#!/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()

相关推荐

CandyGL / 0评论 2016-03-02