溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

python實(shí)現(xiàn)socket+threading處理多連接的方法

發(fā)布時(shí)間:2020-09-07 08:38:58 來源:腳本之家 閱讀:176 作者:狡猾的皮球 欄目:開發(fā)技術(shù)

一.運(yùn)行效果

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

python實(shí)現(xiàn)socket+threading處理多連接的方法

在服務(wù)端控制臺(tái)輸入1,查看在線人數(shù)。

python實(shí)現(xiàn)socket+threading處理多連接的方法

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

python實(shí)現(xiàn)socket+threading處理多連接的方法

二.開發(fā)思路

以下說的是服務(wù)端開發(fā)思路,客戶端比較簡(jiǎn)單就不說了。

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

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

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

2.當(dāng)有新的連接時(shí),再新開一個(gè)線程,用于接收這個(gè)連接的消息(socket.recv())

3.主線程做為控制臺(tái),接收用戶的輸入,進(jìn)行其他操作

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

三.服務(wù)端代碼

全局部分的代碼:

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

其中,g_conn_pool是用來存放每一個(gè)客戶端socket的。

初始化服務(wù)端:

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

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

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

def accept_client():
  """
  接收新連接
  """
  while True:
    client, _ = g_socket_server.accept() # 阻塞,等待客戶端連接
    # 加入連接池
    g_conn_pool.append(client)
    # 給每個(gè)客戶端創(chuàng)建一個(gè)獨(dú)立的線程進(jìn)行管理
    thread = Thread(target=message_handle, args=(client,))
    # 設(shè)置成守護(hù)線程
    thread.setDaemon(True)
    thread.start()
 
 
def message_handle(client):
  """
  消息處理
  """
  client.sendall("連接服務(wù)器成功!".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("有一個(gè)客戶端下線了。")
      break

設(shè)置成守護(hù)線程的目的是為了防止主線程退出之后,程序進(jìn)程不退出。

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

最后,調(diào)用這些方法,讓服務(wù)端啟動(dòng):

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

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

四.客戶端代碼

客戶端這部分設(shè)計(jì)的很簡(jiǎn)單。

import socket # 導(dǎo)入 socket 模塊
 
s = socket.socket() # 創(chuàng)建 socket 對(duì)象
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實(shí)現(xiàn)socket+threading處理多連接的方法就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI