我想编写一个python脚本,它可以启动并终止多个子进程(每个子进程实现一个小TCP服务器),以及在它们运行时从它们中检索输出。是否有针对此的特定模块,还是应该继续使用子流程?
答案 0 :(得分:0)
好像你正在寻找multiprocessing module。它的界面类似于threading module。
您可以launch each process manually或manage them with a Pool。
他们可以使用a Pipe or a Queue进行沟通,也可以使用a Manager object进行沟通。
答案 1 :(得分:0)
从它们检索输出可以通过读取子进程写的文本文件来完成。或者你可以使用基本相同的Pickle
..
使用示例:
child_count=pickle.load( open("child_count.p", "rb"))
child_count-=counter
pickle.dump( child_count, open("child_count.p", "wb"))
对于你的tcp服务器..我认为这个想法是有一个监听器,然后“启动”一个新进程,当进程完成时,它将再次被杀死。
def reap():
#Collect child processes
while 1:
try:
result = os.waitpid(-1, os.WNOHANG)
if not result[0]: break
except:
break
print "Reaped child process %d" % result[0]
host = ''
port = 4200
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind((host, port))
s.listen(1)
print "Parent at %d listening for connections" % os.getpid()
while 1:
try:
clientsock, clientaddr = s.accept();clientsock.settimeout(90)
print "Accepting connection"
except KeyboardInterrupt:
raise
except:
traceback.print_exc()
continue
reap()
#Clone/fork
try:
pid = os.fork()
except:
print "SOMETHING BAD HAPPENED.. fork failed!"
clientsock.close()
continue
if pid:
clientsock.close()
#Cleanup old children
continue
else:
print "New child", os.getpid()
s.close()
#try:
print "Got connection frommm: ", clientsock.getpeername()
data = clientsock.recv(1024)
<ENTER your code for the process here>
这可能是您服务器的核心。
它做什么,它接受一个连接,然后分叉自己,继续侦听其他传入连接,而分叉/克隆部分将处理该进程。
当其他进程完成后,当新连接进入时,进程将被识别为僵尸进程并将被收集。