小言_互联网的博客

python实现socket+threading处理多连接

300人阅读  评论(0)

一.运行效果

    先说两句,之前我在网上找的相关文章标题上写的是处理多连接,尼玛,全是假的。网上那些,根本不能异步处理多连接,不能主动给客户端发消息。

在服务端控制台输入1,查看在线人数。

给指定客户端发送消息。

 

二.开发思路

    以下说的是服务端开发思路,客户端比较简单就不说了。

    首先,需要明白的是socket的accept和recv这两个方法是阻塞线程的。这就意味着我们需要新开线程来处理这两个方法。

程序流程大概是这样的:

1.新开一个线程用于接收新的连接(socket.accept())

2.当有新的连接时,再新开一个线程,用于接收这个连接的消息(socket.recv())

3.主线程做为控制台,接收用户的输入,进行其他操作

也就是说,服务端需要为每一个连接创建一个线程。

 

三.服务端代码

全局部分的代码:


  
  1. import socket # 导入 socket 模块
  2. from threading import Thread
  3. ADDRESS = ( '127.0.0.1', 8712) # 绑定地址
  4. g_socket_server = None # 负责监听的socket
  5. g_conn_pool = [] # 连接池

其中,g_conn_pool是用来存放每一个客户端socket的。

 

初始化服务端:


  
  1. def init():
  2. """
  3. 初始化服务端
  4. """
  5. global g_socket_server
  6. g_socket_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 创建 socket 对象
  7. g_socket_server.bind(ADDRESS)
  8. g_socket_server.listen( 5) # 最大等待数(有很多人理解为最大连接数,其实是错误的)
  9. print( "服务端已启动,等待客户端连接...")

socket.listen()的参数意义不要理解错了哦~

 

接收客户端连接和处理客户端消息:


  
  1. def accept_client():
  2. """
  3. 接收新连接
  4. """
  5. while True:
  6. client, _ = g_socket_server.accept() # 阻塞,等待客户端连接
  7. # 加入连接池
  8. g_conn_pool.append(client)
  9. # 给每个客户端创建一个独立的线程进行管理
  10. thread = Thread(target=message_handle, args=(client,))
  11. # 设置成守护线程
  12. thread.setDaemon( True)
  13. thread.start()
  14. def message_handle(client):
  15. """
  16. 消息处理
  17. """
  18. client.sendall( "连接服务器成功!".encode(encoding= 'utf8'))
  19. while True:
  20. bytes = client.recv( 1024)
  21. print( "客户端消息:", bytes.decode(encoding= 'utf8'))
  22. if len(bytes) == 0:
  23. client.close()
  24. # 删除连接
  25. g_conn_pool.remove(client)
  26. print( "有一个客户端下线了。")
  27. break

设置成守护线程的目的是为了防止主线程退出之后,程序进程不退出。

消息处理这一块,如果接收到一个长度为0的数据包,那么说明客户端掉线了。实际开发中,我们一般采用心跳包机制判断客户端是否在线。

 

最后,调用这些方法,让服务端启动:


  
  1. if __name__ == '__main__':
  2. init()
  3. # 新开一个线程,用于接收新连接
  4. thread = Thread(target=accept_client)
  5. thread.setDaemon( True)
  6. thread.start()
  7. # 主线程逻辑
  8. while True:
  9. cmd = input( """--------------------------
  10. 输入1:查看当前在线人数
  11. 输入2:给指定客户端发送消息
  12. 输入3:关闭服务端
  13. """)
  14. if cmd == '1':
  15. print( "--------------------------")
  16. print( "当前在线人数:", len(g_conn_pool))
  17. elif cmd == '2':
  18. print( "--------------------------")
  19. index, msg = input( "请输入“索引,消息”的形式:").split( ",")
  20. g_conn_pool[int(index)].sendall(msg.encode(encoding= 'utf8'))
  21. elif cmd == '3':
  22. exit()

如果对流程不清楚的话,可以再回到上面,看看文章的第二部分。

 

四.客户端代码

    客户端这部分设计的很简单。

    


  
  1. import socket # 导入 socket 模块
  2. s = socket.socket() # 创建 socket 对象
  3. s.connect(( '127.0.0.1', 8712))
  4. print(s.recv( 1024).decode(encoding= 'utf8'))
  5. s.send( "连接了".encode( 'utf8'))
  6. print(s.recv( 1024).decode(encoding= 'utf8'))
  7. input( "")

最后的input是为了阻塞线程,防止程序接收到消息之后就退出了。


转载:https://blog.csdn.net/qq_39687901/article/details/81531101
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场