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都可接入  7 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:　　　　　　#事件循環 40     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())