當前位置:
首頁 >
RPC 笔记(08)— socket 通信(多进程多线程服务器)
發布時間:2023/11/28
41
豆豆
生活随笔
收集整理的這篇文章主要介紹了
RPC 笔记(08)— socket 通信(多进程多线程服务器)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
在上一節中如果并行的客戶端連接數超過了默認開啟進程的數量,那么后來的客戶端請求將會阻塞,為了不阻塞新的客戶端,我們可以將進程的單線程改成多線程即可。
?
服務端代碼:
import json
import struct
import socket
import threading
import multiprocessingdef handle_conn(conn, ip, handlers):print("{} connect ...".format(ip))# 循環讀寫while True:length_prefix = conn.recv(4) # 請求長度前綴if not length_prefix: # 連接關閉了conn.close()print("{} close ...".format(ip))break # 退出循環,處理下一個連接length, = struct.unpack("I", length_prefix)body = conn.recv(length) # 請求消息體request = json.loads(body)client_method = request['client']client_parameter = request['params']print("client request method is {}, params is {}".format(client_method, client_parameter))handler = handlers[client_method] # 查找請求處理器handler(conn, client_parameter) # 處理請求def process_request(sock, handlers):while True:conn, host_ip = sock.accept() # 接收連接handle_conn(conn, host_ip, handlers) # 處理連接def multi_thread(sock, handlers):threads = []thread_nums = 2for i in range(thread_nums):t = threading.Thread(target=process_request, args=(sock, handlers))threads.append(t)for i in range(thread_nums):threads[i].start()for i in range(thread_nums):threads[i].join()def func1(conn, params):res = json.dumps({"response": "OK", "result": params}) # 響應消息體length_prefix = struct.pack("I", len(res)) # 響應長度前綴conn.sendall(length_prefix)# conn.sendall(response) # python2conn.sendall(str.encode(res)) # python3def main():s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 創建一個 TCP 套接字s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) # 打開 reuse addr 選項s.bind(("localhost", 8080)) # 綁定端口s.listen(1) # 監聽客戶端連接handlers = {"func1": func1 # 注冊請求處理器}# process_request(s, handlers) # 進入服務循環process = []process_nums = 2for i in range(process_nums):p = multiprocessing.Process(target=multi_thread, args=(s, handlers))process.append(p)for i in range(process_nums):process[i].start()for i in range(process_nums):process[i].join()if __name__ == '__main__':main()
示例代碼中開啟 2 個進程,每個進程中又開了 2 個線程,那么此時最多可以同時開啟 4 個客戶端,超過 4 個客戶端后多余的請求會被阻塞處理。
?
?
總結
以上是生活随笔為你收集整理的RPC 笔记(08)— socket 通信(多进程多线程服务器)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 2022-2028年中国数字乡村建设深度
- 下一篇: Pyinstaller 打包 torch