python并发（阻塞、非阻塞、epoll）

在Linux系统中

01 阻塞服务端

特征：1对1，阻塞。

 

import socket

server = socket.socket()        #生成套接字对象
server.bind(('0.0.0.0', 8000))    #套接字绑定ip和端口，变为监听套接字
server.listen(5)                    #开始监听

while True:
    conn, addr = server.accept()        #建立连接，生成对等套接字
    print('用户连接:', addr)
    while True:
        try:
            data = conn.recv(1024)
            if data == b'Q' or data == b'q':
                print('用户退出:', addr)
                break
            else:
                print('收到的消息:', data.decode())
                conn.send(data)
        except Exception as e:
            print(e)
            conn.close()
            break

 

---

 

 

02 非阻塞服务端

特征：1对多，轮询，非阻塞，占用资源多。

 

import socket

server = socket.socket()        #创建套接字
server.setblocking(False)       #把套接字设置为非阻塞
server.bind(('0.0.0.0', 8001))  #绑定IP和端口
server.listen(5)                #监听端口


all_connection = []             #保存已经连接的客户
while True:
    #只管连接的事情
    try:
        conn, addr = server.accept()    # 建立连接，没有就抛出异常
        conn.setblocking(False)         #设置非阻塞
        print('用户连接:', addr)
        all_connection.append(conn)
    except Exception as e:
        pass


    #处理已经连接用户的消息
    handle = all_connection.copy()  #完全拷贝了列表
    for connection in handle:
        try:
            recv_data = connection.recv(1024)
            if recv_data:
                print(recv_data.decode())
                connection.send(recv_data)
            else:                               #客户端消息处理完了，已经断开了连接
                print('断开连接', connection)
                connection.close()
                all_connection.remove(connection)       #从客户列表里移除断开连接的客户
        except Exception as e:
            pass

 

---

 

 

03 epoll服务端

特征：1对多，通知机制，非阻塞，占用资源少；
epoll：注册惰性事件回调。

 

import selectors #调用epoll的模块
import socket

epoll = selectors.EpollSelector()   #生成一个epoll
server = socket.socket()            #生成套接字
server.bind(('', 8082))　　#参数1‘’与‘0.0.0.0’等价，表示ip都可接入
server.listen(100)


#回调函数
def create_conneciton(server):
    #百分百有人连接，不会阻塞
    conn, addres = server.accept()  #生成对等连接套接字


    #处理消息的函数注册
    epoll.register(conn, selectors.EVENT_READ, read_data)
    return conn



#回调函数   处理消息
def read_data(conn):

    data = conn.recv(1024)
    if data:
        print(data)
        conn.send(data)
    else:
        epoll.unregister(conn)  #删掉注册事件


#1
#把监听套接字和生成对等套接字的函数注册到read事件(有用户连接)
epoll.register(server, selectors.EVENT_READ, create_conneciton)


#2
while True:　　　　　　#事件循环
    events = epoll.select()  #去操作系统查询

    for key,mask in events:
        sock = key.fileobj      #连接客户端的套接字
        callback = key.data     #回调函数

        #read_data（conn）,   create_conneciton（server）
        callback(sock)      #调用函数

 

---

 

04 客户端

测试服务端。

 

import socket

client = socket.socket()
client.connect(('127.0.0.1', 8082))

while True:
    data = input('输入数据:')
    client.send(data.encode())
    recv_data = client.recv(1024)
    print(recv_data.decode())