用Python实现一个简单的多线程TCP服务器的教程

Z时代
2024-01-10
分类：综合

最近看《python核心编程》，书中实现了一个简单的1对1的TCPserver，但是在实际使用中1对1的形势明显是不行的，所以研究了一下如何在server端通过启动不同的线程（进程）来实现每个链接一个线程。

其实python在类的设计上已经考虑到了这一方面的需求，我们只要在自己的server上继承一下SocketServer.BaseRequestHandler就可以了。

server端代码如下：


#!/usr/bin/env python 
import SocketServer 
from time import ctime 
HOST = '' 
PORT = 21567 
ADDR = (HOST, PORT) 
class MyRequestHandler(SocketServer.BaseRequestHandler): 
  def handle(self): 
    print '...connected from:', self.client_address 
    while True: 
      self.request.sendall('[%s] %s' % (ctime(),self.request.recv(1024))) 
tcpServ = SocketServer.ThreadingTCPServer(ADDR, MyRequestHandler) 
print 'waiting for connection...' 
tcpServ.serve_forever()

客户端代码如下(基本和书中一模一样，只是把循环中的关闭链接注释掉了)：


#!/usr/bin/env python 
from socket import * 
HOST = 'localhost' 
PORT = 21567 
BUFSIZ = 1024 
ADDR = (HOST, PORT) 
while True: 
  tcpCliSock = socket(AF_INET, SOCK_STREAM) 
  tcpCliSock.connect(ADDR) 
  data = raw_input('> ') 
  if not data: 
    break 
  tcpCliSock.send('%s\r\n' % data) 
  data = tcpCliSock.recv(BUFSIZ) 
  if not data: 
    break 
  print data.strip() 
#tcpCliSock.close()