python用epoll方式实现客户端对服务端发起大量连接（长连接）

最新推荐文章于 2024-05-07 17:10:47 发布

原创

最新推荐文章于 2024-05-07 17:10:47 发布 · 1.6k 阅读

标签

#IO复用 #epoll #python #thrift #select

收录于

本文探讨如何在Python中利用epoll实现客户端对服务端的长连接压测，详细介绍了在客户端发起连接时如何运用IO复用技术。

我们知道的大多数io复用方式都是在服务端接受连接的时候进行，

但如果要进行压测，则需要客户端发起连接的时候用，那么要怎么用呢，看下文

长连接情况下

def epoll_run(self):
    used= 0
    trans_dict ={}
    start_time_dict={}
    epoll = select.epoll()
    for i in range(self.socket_num_per_thread):
        trans = self.trans_init()
        if trans == "err":
            continue
        start_time_dict[trans.sockfd.fileno()]=[]
        trans_dict[trans.sockfd.fileno()]=trans
        # 对该sock进行注册
        epoll.register(trans.sockfd.fileno(), select.EPOLLIN |select.EPOLLOUT |select.EPOLLERR )

    while used < self.run_time:
        error = ''
        used = time.time() - self.start_time
        start = self.default_timer()
        timeout=0.001
        events = epoll.poll(timeout)
        if not events:
            continue
        for fd, event