上節(jié)我們完成了一個簡單的多線程服務(wù)器，可以并發(fā)處理多個客戶端連接。但是 Python 由于全局解釋器鎖 GIL 的存在，致使多個線程只能占滿一個 CPU 核心，多線程并不能充分利用多核的優(yōu)勢。所以多數(shù) Python 服務(wù)器推薦使用多進程模型。
?

服務(wù)端代碼

import json
import struct
import socket
import multiprocessingdef handle_conn(conn, ip, handlers):print("{} connect ...".format(ip))# 循環(huán)讀寫while True:length_prefix = conn.recv(4)  # 請求長度前綴if not length_prefix:         # 連接關(guān)閉了conn.close()print("{} close ...".format(ip))break  # 退出循環(huán)，處理下一個連接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 func1(conn, params):res = json.dumps({"response": "OK", "result": params})  # 響應(yīng)消息體length_prefix = struct.pack("I", len(res))  # 響應(yīng)長度前綴conn.sendall(length_prefix)# conn.sendall(response)  # python2conn.sendall(str.encode(res))  # python3def main():s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 創(chuàng)建一個 TCP 套接字s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # 打開 reuse addr 選項s.bind(("localhost", 8080))  # 綁定端口s.listen(1)  # 監(jiān)聽客戶端連接handlers = {"func1": func1    # 注冊請求處理器}# process_request(s, handlers)  # 進入服務(wù)循環(huán)process = []process_nums = 3for i in range(process_nums):p = multiprocessing.Process(target=process_request, 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()

示例代碼中開啟了 3 個進程，如果客戶端請求數(shù)量大于 3 那么多余的請求就會阻塞，直到之前的客戶端請求關(guān)閉后，新的請求才會處理。

總結(jié)

以上是生活随笔為你收集整理的RPC 笔记（07）— socket 通信（多进程服务器）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。