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

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

当前无限循环的直接问题是每个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

如果您break循环while，则该过程将一直等到所有ClientThreads退出，然后终止。

这将起作用，因为客户端线程是非守护进程。有关详细信息，请参阅threading.Thread.daemon。

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

2 个答案: