在上一節(jié)中如果并行的客戶端連接數(shù)超過了默認(rèn)開啟進(jìn)程的數(shù)量，那么后來的客戶端請(qǐng)求將會(huì)阻塞，為了不阻塞新的客戶端，我們可以將進(jìn)程的單線程改成多線程即可。
?

服務(wù)端代碼：

import json
import struct
import socket
import threading
import multiprocessingdef handle_conn(conn, ip, handlers):print("{} connect ...".format(ip))# 循環(huán)讀寫while True:length_prefix = conn.recv(4)  # 請(qǐng)求長(zhǎng)度前綴if not length_prefix:         # 連接關(guān)閉了conn.close()print("{} close ...".format(ip))break  # 退出循環(huán)，處理下一個(gè)連接length, = struct.unpack("I", length_prefix)body = conn.recv(length)  # 請(qǐng)求消息體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]  # 查找請(qǐng)求處理器handler(conn, client_parameter)  # 處理請(qǐng)求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})  # 響應(yīng)消息體length_prefix = struct.pack("I", len(res))  # 響應(yīng)長(zhǎ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)建一個(gè) TCP 套接字s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # 打開 reuse addr 選項(xiàng)s.bind(("localhost", 8080))  # 綁定端口s.listen(1)  # 監(jiān)聽客戶端連接handlers = {"func1": func1    # 注冊(cè)請(qǐng)求處理器}# process_request(s, handlers)  # 進(jìn)入服務(wù)循環(huán)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 個(gè)進(jìn)程，每個(gè)進(jìn)程中又開了 2 個(gè)線程，那么此時(shí)最多可以同時(shí)開啟 4 個(gè)客戶端，超過 4 個(gè)客戶端后多余的請(qǐng)求會(huì)被阻塞處理。
?

?

總結(jié)

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

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