使用Python實現多線程、多進程、非同步IO的socket通信

来源:https://www.cnblogs.com/engpj/archive/2022/11/28/16933904.html
-Advertisement-
Play Games

多線程實現socket通信伺服器端代碼 import socket import threading class MyServer(object): def __init__(self): # 初始化socket self.server = socket.socket(socket.AF_INET, ...


多線程實現socket通信伺服器端代碼

import socket
import threading


class MyServer(object):
    def __init__(self):
        # 初始化socket
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # 設置伺服器IP地址
        host = '192.168.152.1'
        # 設置伺服器埠號
        port = 4321
        # 綁定IP地址和埠
        self.server.bind((host, port))
        # 設置最大監聽數
        self.server.listen(5)
        # 設置一個字典，用來保存每一個客戶端的連接和身份信息
        self.socket_mapping = {}
        # 設置接收的最大位元組數
        self.maxSize = 1024

    def run(self):
        while True:
            socket, addr = self.server.accept()
            # 發送信息，提示客戶端已成功連接
            socket.send('success！'.encode('utf-8'))
            # 將客戶端socket等信息存入字典
            self.socket_mapping[socket] = addr
            # 創建線程，負責獲取鍵盤輸入併發送給客戶端
            threading.Thread(target=self.send_to_client, args=(socket,)).start()
            # 創建線程，負責接收客戶端信息並轉發給其他客戶端
            threading.Thread(target=self.recv_from_client, args=(socket,)).start()

    def send_to_client(self, socket):
        """
        獲取鍵盤輸入併發送給客戶端
        :param socket:
        :return:
        """
        while True:
            info = input()
            if info == "quit":
                socket.close()
            for socket in self.socket_mapping.keys():
                socket.send(info.encode("utf-8"))

    def recv_from_client(self, socket):
        """
        接收客戶端信息並轉發給其他客戶端
        :param socket:
        :return:
        """
        while True:
            recv_info = socket.recv(self.maxSize).decode('utf-8')
            print('client{} say: '.format(self.socket_mapping[socket]), recv_info)
            for i_socket in self.socket_mapping.keys():
                if i_socket != socket:
                    i_socket.send(recv_info.encode("utf-8"))


my_server = MyServer()
my_server.run()

多進程實現socket通信伺服器端代碼

存在的問題：在與客戶端連通後，需要伺服器先發送兩條消息，之後才能正常通信。

import os
import socket
import sys
from multiprocessing import Process, Manager


class MyServer(object):
    def __init__(self):
        # 初始化socket
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # 設置伺服器IP地址
        host = '192.168.152.1'
        # 設置伺服器埠號
        port = 4321
        # 綁定IP地址和埠
        self.server.bind((host, port))
        # 設置最大監聽數
        self.server.listen(5)
        # 設置一個字典，用來保存每一個客戶端的連接和身份信息
        self.socket_mapping = Manager().dict()
        # 設置接收的最大位元組數
        self.maxSize = 1024
        # 進程鎖
        self.share_lock = Manager().Lock()

    def run(self):
        fn = sys.stdin.fileno()
        while True:
            socket, addr = self.server.accept()
            # 發送信息，提示客戶端已成功連接
            socket.send('success！'.encode('utf-8'))
            # 將客戶端socket等信息存入字典
            self.modify_mapping(self.socket_mapping, addr, socket, self.share_lock)
            # 創建進程，負責獲取鍵盤輸入併發送給客戶端
            Process(target=self.send_to_client, args=(addr, fn)).start()
            # 創建進程，負責接收客戶端信息並轉發給其他客戶端
            Process(target=self.recv_from_client, args=(addr,)).start()

    def send_to_client(self, addr, fn):
        """
        獲取鍵盤輸入併發送給客戶端
        :param addr:
        :return:
        """
        sys.stdin = os.fdopen(fn)
        while True:
            info = sys.stdin.readline()
            if info == "quit":
                self.socket_mapping[addr].close()
            for socket in self.socket_mapping.values():
                socket.send(info.encode("utf-8"))

    def recv_from_client(self, addr):
        """
        接收客戶端信息並轉發給其他客戶端
        :param addr:
        :return:
        """
        while True:
            recv_info = self.socket_mapping.get(addr).recv(self.maxSize).decode('utf-8')
            print('client{} say: '.format(addr), recv_info)
            for i_addr in self.socket_mapping.keys():
                if i_addr != addr:
                    self.socket_mapping.get(i_addr).send(recv_info.encode("utf-8"))

    @staticmethod
    def modify_mapping(share_var, share_key, share_value, share_lock):
        # 獲取鎖
        share_lock.acquire()
        # 修改數據
        share_var[share_key] = share_value
        # 釋放鎖
        share_lock.release()


if __name__ == "__main__":
    my_server = MyServer()
    my_server.run()

非同步IO實現socket通信伺服器端代碼

存在的問題：通信時需要相互發送幾次消息後，各自才會收到之前的消息並列印。

import socket
import asyncio
import select


class MyServer(object):
    def __init__(self):
        # 初始化socket
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # 設置伺服器IP地址
        host = '192.168.152.1'
        # 設置伺服器埠號
        port = 4321
        # 綁定IP地址和埠
        self.server.bind((host, port))
        # 設置最大監聽數
        self.server.listen(5)
        # 設置一個字典，用來保存每一個客戶端的連接和身份信息
        self.socket_mapping = {self.server: None}  # 這裡存入self.server是為了充當select.select參數
        # 設置接收的最大位元組數
        self.maxSize = 1024
        # 進入事件迴圈
        self.loop = asyncio.get_event_loop()

    def run(self):
        while True:
            # select監聽請求對象
            rret, _, _ = select.select(self.socket_mapping.keys(), [], [])
            for r_socket in rret:
                if r_socket is self.server:
                    socket, addr = r_socket.accept()
                    # 發送信息，提示客戶端已成功連接
                    socket.send('success！'.encode('utf-8'))
                    # 將客戶端socket等信息存入字典
                    self.socket_mapping[socket] = addr
                else:
                    task = [self.send_to_client(r_socket), self.recv_from_client(r_socket)]
                    self.loop.run_until_complete(asyncio.gather(*task))

    async def send_to_client(self, socket):
        """
        獲取鍵盤輸入併發送給客戶端
        :param socket:
        :return:
        """
        info = input()
        if info == "quit":
            socket.close()
        for socket in self.socket_mapping.keys():
            if socket != self.server:
                socket.send(info.encode("utf-8"))

    async def recv_from_client(self, socket):
        """
        接收客戶端信息並轉發給其他客戶端
        :param socket:
        :return:
        """
        recv_info = socket.recv(self.maxSize).decode('utf-8')
        print('client{} say: '.format(self.socket_mapping[socket]), recv_info)
        for i_socket in self.socket_mapping.keys():
            if i_socket != socket and i_socket != self.server:
                i_socket.send(recv_info.encode("utf-8"))


my_server = MyServer()
my_server.run()

客戶端代碼（使用多線程）

import socket
import threading


class MyClient(object):
    def __init__(self):
        # 初始化socket
        self.client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # 設置伺服器IP地址
        self.host = '192.168.152.1'
        # 設置伺服器埠號
        self.port = 4321
        # 設置接收的最大位元組數
        self.max_size = 1024

    def run(self):
        # 與伺服器建立連接
        self.client.connect((self.host, self.port))
        # 創建線程，負責獲取鍵盤輸入併發送給伺服器
        threading.Thread(target=self.sned_to_server).start()
        # 創建線程，接收伺服器信息
        threading.Thread(target=self.recv_from_server).start()

    def sned_to_server(self):
        """
        獲取鍵盤輸入併發送給伺服器
        """
        while True:
            send_msg = input()
            self.client.send(send_msg.encode('utf-8'))
            if send_msg == 'quit':
                break
        self.client.close()

    def recv_from_server(self):
        """
        接收伺服器信息
        """
        while True:
            recv_info = self.client.recv(self.max_size).decode('utf-8')
            print('server{} say: '.format((self.host, self.port)), recv_info)


my_client = MyClient()
my_client.run()

您的分享是我們最大的動力!

-Advertisement-
Play Games
更多相關文章
  • 優先順序反轉那些事兒
    文章主要闡述了優先順序反轉的一些概念和解決思路，並結合iOS平臺的幾種鎖進行了詳細的調研。通過深入的理解，可以去規避一些不必要的優先順序反轉，從而進一步避免卡死異常。 ...
  • Android Studio 自定義模板內容（註釋、代碼）
    ...
  • 淺拷貝和深拷貝
    前言 由於工作需要，要抓取tb上某個介面的文案資源，用來分析借鑒。 本來想著，無非就是驗證一下當前用戶信息之類的，但在研究過後，發現並沒那麼簡單。 1，查看請求相關參數 比如下圖，發現請求中攜帶的參數不少，然後其中隨時變化的就有t、sign、data， 很明顯，這就是tb判斷請求是否合法的參數，其中 ...
  • 記錄--uniapp微信小程式引入threeJs並導入模型
    這裡給大家分享我在網上總結出來的一些知識，希望對大家有所幫助 前言 我的需求是使用uniapp寫微信小程式，在小程式中使用threeJs就行了，目前暫不考慮相容app什麼的。 1.引入小程式版的threejs庫實現 2.使用webview實現(推薦) 重點 我的建議是使用這個庫https://git ...
  • vue3 第二天vue響應式原理以及ref和reactive區別
    前言： 前天我們學了 ref 和 reactive ，提到了響應式數據和 Proxy ，那我們今天就來瞭解一下，vue3 的響應式 在瞭解之前，先複習一下之前 vue2 的響應式原理 vue2 的響應式： 原理： 對象類型：通過 Object.defineProperty（） 對象的讀取，修改進行攔 ...
  • 生產問題處理，可別頭疼醫頭腳疼醫腳
    當遇到生產問題，你是怎麼處理的，斬草除根？or 頭疼醫頭/腳疼醫腳？ 下麵幾個，看看是否中招le！你也可以說，我肯定沒有這麼草率 1 ​併發導致數據表裡出現了同樣的數據 -加唯一索引 2 程式fullGC頻繁 -修改程式啟動參數，加大堆記憶體 3 mq消息堵塞 -重啟服務…重啟服務不奏效？清空隊列 4 ...
  • (Java)設計模式：行為型
    前言 此篇博文內容續接的是 UML建模語言、設計原則、創建型設計模式 的內容，有興趣的可以點前面的鏈接去看一下 3.2、行為型 這類設計模式是專門用於：對象間的高效溝通和職責委派 * 3.2.1、責任鏈模式 定義：責任鏈模式又名職責鏈模式，指的是：對某個請求的所有處理構成一條鏈，如果鏈上的某一處理者 ...
  • python中的面向對象
    # 面向對象是非常重要的！ # 抽象，是個思想，結構 # 小明 小紅 小雨 都是人 # 海爾洗衣機 海東洗衣機 海西洗衣機 都是洗衣機 # 貓 狗 熊貓 都是動物 # 藍圖 # # class WashingMachine: # 類名一般是大駝峰 # pass # # 特征是屬性 # age = 2 ...
一周排行
    -Advertisement-
    Play Games
  • UOS 開啟 VisualStudio 遠程調試 .NET 應用之旅
    前言 在我們開發過程中基本上不可或缺的用到一些敏感機密數據，比如SQL伺服器的連接串或者是OAuth2的Secret等，這些敏感數據在代碼中是不太安全的，我們不應該在源代碼中存儲密碼和其他的敏感數據，一種推薦的方式是通過Asp.Net Core的機密管理器。 機密管理器 在 ASP.NET Core ...
  • C# - 能否讓 SortedSet.RemoveWhere 內傳入的委托非同步執行
    新改進提供的Taurus Rpc 功能，可以簡化微服務間的調用，同時可以不用再手動輸出模塊名稱，或調用路徑，包括負載均衡，這一切，由框架實現並提供了。新的Taurus Rpc 功能，將使得服務間的調用，更加輕鬆、簡約、高效。 ...
  • 順序棧的介面程式
    順序棧的介面程式 目錄順序棧的介面程式頭文件創建順序棧入棧出棧利用棧將10進位轉16進位數驗證 頭文件 #include <stdio.h> #include <stdbool.h> #include <stdlib.h> 創建順序棧 // 指的是順序棧中的元素的數據類型，用戶可以根據需要進行修改 ...
  • mORMot2 獲取數據集1
    前言 整理這個官方翻譯的系列，原因是網上大部分的 tomcat 版本比較舊，此版本為 v11 最新的版本。 開源項目 從零手寫實現 tomcat minicat 別稱【嗅虎】心有猛虎，輕嗅薔薇。 系列文章 web server apache tomcat11-01-官方文檔入門介紹 web serv ...
  • 萬字長文總結與剖析C語言關鍵字 -- <<C語言深度解剖>>
    C總結與剖析:關鍵字篇 -- <<C語言深度解剖>> 目錄C總結與剖析:關鍵字篇 -- <<C語言深度解剖>>程式的本質:二進位文件變數1.變數:記憶體上的某個位置開闢的空間2.變數的初始化3.為什麼要有變數4.局部變數與全局變數5.變數的大小由類型決定6.任何一個變數,記憶體賦值都是從低地址開始往高地 ...
  • 【Flink入門修煉】2-3 Flink Checkpoint 原理機制
    如果讓你來做一個有狀態流式應用的故障恢復，你會如何來做呢？ 單機和多機會遇到什麼不同的問題？ Flink Checkpoint 是做什麼用的？原理是什麼？ ...
  • C++ 多級繼承與多重繼承：代碼組織與靈活性的平衡
    C++ 多級繼承 多級繼承是一種面向對象編程（OOP）特性，允許一個類從多個基類繼承屬性和方法。它使代碼更易於組織和維護，並促進代碼重用。 多級繼承的語法 在 C++ 中，使用 : 符號來指定繼承關係。多級繼承的語法如下： class DerivedClass : public BaseClass1 ...
  • 圖文結合手把手教你創建SpringCloud項目
    前言 什麼是SpringCloud？ Spring Cloud 是一系列框架的有序集合，它利用 Spring Boot 的開發便利性簡化了分散式系統的開發，比如服務註冊、服務發現、網關、路由、鏈路追蹤等。Spring Cloud 並不是重覆造輪子，而是將市面上開發得比較好的模塊集成進去，進行封裝，從 ...
  • 類模板的常見用法
    class_template 類模板和函數模板的定義和使用類似，我們已經進行了介紹。有時，有兩個或多個類，其功能是相同的，僅僅是數據類型不同。類模板用於實現類所需數據的類型參數化 template<class NameType, class AgeType> class Person { publi ...
  • system-v IPC共用記憶體通信
    目錄system v IPC簡介共用記憶體需要用到的函數介面shmget函數--獲取對象IDshmat函數--獲得映射空間shmctl函數--釋放資源共用記憶體實現思路註意 system v IPC簡介 消息隊列、共用記憶體和信號量統稱為system v IPC(進程間通信機制)，V是羅馬數字5，是UNI ...
所有分類