python實現socket+threading處理多連接的方法

更新時間：2019年07月23日 09:56:25 作者：狡猾的皮球

今天小編就為大家分享一篇python實現socket+threading處理多連接的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

一.運行效果

先說兩句，之前我在網上找的相關文章標題上寫的是處理多連接，尼瑪，全是假的。網上那些，根本不能異步處理多連接，不能主動給客戶端發(fā)消息。

在服務端控制臺輸入1，查看在線人數。

給指定客戶端發(fā)送消息。

二.開發(fā)思路

以下說的是服務端開發(fā)思路，客戶端比較簡單就不說了。

首先，需要明白的是socket的accept和recv這兩個方法是阻塞線程的。這就意味著我們需要新開線程來處理這兩個方法。

具體的程序流程大概是這樣的：

1.新開一個線程用于接收新的連接（socket.accept()）

2.當有新的連接時，再新開一個線程，用于接收這個連接的消息（socket.recv()）

3.主線程做為控制臺，接收用戶的輸入，進行其他操作

也就是說，服務端需要為每一個連接創(chuàng)建一個線程。

三.服務端代碼

全局部分的代碼：

import socket # 導入 socket 模塊
from threading import Thread
 
ADDRESS = ('127.0.0.1', 8712) # 綁定地址
 
g_socket_server = None # 負責監(jiān)聽的socket
 
g_conn_pool = [] # 連接池

其中，g_conn_pool是用來存放每一個客戶端socket的。

初始化服務端：

def init():
  """
  初始化服務端
  """
  global g_socket_server
  g_socket_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 創(chuàng)建 socket 對象
  g_socket_server.bind(ADDRESS)
  g_socket_server.listen(5) # 最大等待數（有很多人理解為最大連接數，其實是錯誤的）
  print("服務端已啟動，等待客戶端連接...")

socket.listen()的參數意義不要理解錯了哦~

接收客戶端連接和處理客戶端消息：

def accept_client():
  """
  接收新連接
  """
  while True:
    client, _ = g_socket_server.accept() # 阻塞，等待客戶端連接
    # 加入連接池
    g_conn_pool.append(client)
    # 給每個客戶端創(chuàng)建一個獨立的線程進行管理
    thread = Thread(target=message_handle, args=(client,))
    # 設置成守護線程
    thread.setDaemon(True)
    thread.start()
 
 
def message_handle(client):
  """
  消息處理
  """
  client.sendall("連接服務器成功!".encode(encoding='utf8'))
  while True:
    bytes = client.recv(1024)
    print("客戶端消息:", bytes.decode(encoding='utf8'))
    if len(bytes) == 0:
      client.close()
      # 刪除連接
      g_conn_pool.remove(client)
      print("有一個客戶端下線了。")
      break

設置成守護線程的目的是為了防止主線程退出之后，程序進程不退出。

消息處理這一塊，如果接收到一個長度為0的數據包，那么說明客戶端掉線了。實際開發(fā)中，我們一般采用心跳包機制判斷客戶端是否在線。

最后，調用這些方法，讓服務端啟動：

if __name__ == '__main__':
  init()
  # 新開一個線程，用于接收新連接
  thread = Thread(target=accept_client)
  thread.setDaemon(True)
  thread.start()
  # 主線程邏輯
  while True:
    cmd = input("""--------------------------
輸入1:查看當前在線人數
輸入2:給指定客戶端發(fā)送消息
輸入3:關閉服務端
""")
    if cmd == '1':
      print("--------------------------")
      print("當前在線人數：", len(g_conn_pool))
    elif cmd == '2':
      print("--------------------------")
      index, msg = input("請輸入“索引,消息”的形式：").split(",")
      g_conn_pool[int(index)].sendall(msg.encode(encoding='utf8'))
    elif cmd == '3':
      exit()

如果對流程不清楚的話，可以再回到上面，看看文章的第二部分。

四.客戶端代碼

客戶端這部分設計的很簡單。

import socket # 導入 socket 模塊
 
s = socket.socket() # 創(chuàng)建 socket 對象
s.connect(('127.0.0.1', 8712))
print(s.recv(1024).decode(encoding='utf8'))
s.send("連接了".encode('utf8'))
print(s.recv(1024).decode(encoding='utf8'))
input("")

最后的input是為了阻塞線程，防止程序接收到消息之后就退出了。

以上這篇python實現socket+threading處理多連接的方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: