线程完成时退出的多线程 Python 服务器

Question

假设您有一个相当基本的客户端/服务器代码，其中每个客户端创建三个线程并且多个客户端可以同时连接。我希望服务器等待传入连接，一旦它开始获取连接，就运行直到没有更多线程在运行，然后退出。代码类似于下面。 （即，而不是服务器“永远服务”，我希望它在所有线程完成后退出）。

编辑：我希望服务器等待传入连接。一旦连接开始，它应该继续接受连接，直到没有线程运行，然后退出。这些联系会有些零散。

import socket
import threading

# Our thread class:
class ClientThread ( threading.Thread ):

   # Override Thread's __init__ method to accept the parameters needed:
   def __init__ ( self, channel, details ):

      self.channel = channel
      self.details = details
      threading.Thread.__init__ ( self )

   def run ( self ):

      print 'Received connection:', self.details [ 0 ]
      self.channel.send ( 'hello from server' )
      for x in xrange ( 10 ):
         print self.channel.recv ( 1024 )
      self.channel.close()
      print 'Closed connection:', self.details [ 0 ]

# Set up the server:
server = socket.socket ( socket.AF_INET, socket.SOCK_STREAM )
server.bind ( ( '', 2727 ) )
server.listen ( 5 )

# Have the server serve "forever":
while True:
   channel, details = server.accept()
   ClientThread ( channel, details ).start()

Answer 1

根据您的 cmets，您正在寻找的是在第一个连接到服务器后开始计算连接，并在没有更多现有连接时终止服务器。

当前无限 while 循环的一个直接问题是每个 accept() 上的块。所以无论如何，它总是在等待另一个连接。您将不得不从其他线程中断它以使其脱离该循环。但另一种解决方案是让你的事件循环更大，接受新连接的行为只是其中的一部分。循环还应该检查退出条件。

这个例子只是一种可能的方式。它使用Queue.Queue 来协调工作计数器。

import socket
import threading
import select
from Queue import Queue

class ClientThread ( threading.Thread ):

   def __init__ ( self, channel, details, queue=None ):
      self.channel = channel
      self.details = details
      self.queue = queue
      threading.Thread.__init__ ( self )

   def run ( self ):

      if self.queue:
         self.queue.put(1)

      print 'Received connection:', self.details [ 0 ]
      self.channel.send ( 'hello from server' )
      for x in xrange ( 10 ):
         print self.channel.recv ( 1024 )
      self.channel.close()
      print 'Closed connection:', self.details [ 0 ]

      if self.queue:
         self.queue.get_nowait()

# Set up the server:
server = socket.socket ( socket.AF_INET, socket.SOCK_STREAM )
server.bind ( ( '', 2727 ) )
server.listen ( 5 )

rlist = [server]
work_queue = Queue()

def accept_client():
   channel, details = server.accept()
   ClientThread ( channel, details, work_queue ).start() 

accept_client()

while not work_queue.empty():
   server_ready, _, _ = select.select(rlist,[],[], .25)
   if server in server_ready:
      accept_client()

print "Shutting down"
server.close()
print "Exiting"

我们使用select.select 作为一种简单的方法来检测服务器套接字上的活动，但也有超时。如果服务器准备好了，那么我们接受新的连接。如果它达到.25 秒超时，我们只需再次循环并等待。

您会看到我们创建了一个队列并不断检查它是否为空。队列被传递到每个线程。当一个线程启动时，它会将一些工作记录到队列中。数据是任意的。只是一面旗帜。当线程完成时，它将从队列中清除该项目。结果是收到第一个连接后，队列不再为空，循环会一直运行。如果在任何时候队列变为空（因为所有当前线程都已完成），循环将中断，服务器将关闭。

Answer 2

如果你从while循环中break，进程会一直等到所有ClientThreads退出，然后才会终止。

这会起作用，因为客户端线程是非守护进程。 请参阅threading.Thread.daemon 了解更多详情。