欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Python subprocess.Popen子进程管道阻塞

程序员文章站 2022-03-11 11:48:58
...

问题产生描述

使用子进程处理一个大的日志文件,并对文件进行分析查询,需要等待子进程执行的输出结果,进行下一步处理。

出问题的代码

# 启用子进程执行外部shell命令
def __subprocess(self,cmd):
   try:
       # 执行外部shell命令, 输出结果输出管道
       p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)  
       p.wait()

       # 从标准输出读出shell命令的输出结果
       #rt = p.stdout.read().decode()

       # 以换行符拆分数据,并去掉换行符号存入列表
       rt_list = rt.strip().split('\n')

   except Exception as e:
       if(DEBUG):
           print(traceback.format_exc())

   return rt_list

问题分析

子进程产生一些数据,他们会被buffer起来,当buffer满了,会写到子进程的标准输出和标准错误输出,这些东西通过管道发送给父进程。当管道满了之后,子进程就停止写入,于是就卡住了,及时取走管道的输出就不会出现阻塞了

但是本人此处采取的是临时文件接收子进程输出,由于临时文件是建立在磁盘上的,没有size的限制,并且文件被close后,相应的磁盘上的空间也会被释放掉。

已改进的代码

import tempfile
# 启用子进程执行外部shell命令
def __subprocess(self,cmd):
    try:
        # 得到一个临时文件对象, 调用close后,此文件从磁盘删除
        out_temp = tempfile.TemporaryFile(mode='w+')
        # 获取临时文件的文件号
        fileno = out_temp.fileno()

        # 执行外部shell命令, 输出结果存入临时文件中
        p = subprocess.Popen(cmd, shell=True, stdout=fileno, stderr=fileno) 
        p.wait()

        # 从临时文件读出shell命令的输出结果
        out_temp.seek(0)
        rt = out_temp.read()

        # 以换行符拆分数据,并去掉换行符号存入列表
        rt_list = rt.strip().split('\n')

    except Exception as e:
        if(DEBUG):
            print(traceback.format_exc())

    finally:
        if out_temp:
            out_temp.close()

    return rt_list