在Python中控制多个子进程?

时间:2013-06-10 07:20:36

标签: python subprocess

我想编写一个python脚本,它可以启动并终止多个子进程(每个子进程实现一个小TCP服务器),以及在它们运行时从它们中检索输出。是否有针对此的特定模块,还是应该继续使用子流程?

2 个答案:

答案 0 :(得分:0)

好像你正在寻找multiprocessing module。它的界面类似于threading module

您可以launch each process manuallymanage 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>

这可能是您服务器的核心。

它做什么,它接受一个连接,然后分叉自己,继续侦听其他传入连接,而分叉/克隆部分将处理该进程。

当其他进程完成后,当新连接进入时,进程将被识别为僵尸进程并将被收集。