输出顺序差异：如何理解实时与缓冲输出？

Linux

2024-03-11 03:56:08

输出顺序差异：理解实时对缓冲输出

问题：重定向时输出顺序改变

当你使用 > 将程序输出重定向到文件时，输出顺序可能会与直接在终端中执行程序时不同。这是因为 shell 在命令完成前会缓冲输出。

原因：终端与文件缓冲

在没有重定向的情况下，输出会立即显示在终端中，因为终端充当缓冲区。但是，当重定向到文件时，shell 会将输出暂时存储在缓冲区中，直到命令完成才将所有输出写入文件。

实时获取输出：`subprocess.Popen`

为了以正确格式实时获取输出，我们可以使用 subprocess.Popen 模块。它允许我们创建子进程并监视其输出。

缓冲输出的处理

在需要缓冲输出的情况下（例如重定向到文件），我们可以通过以下方法处理输出：

使用 p.communicate() ：等到子进程完成，一次性获取所有输出。
循环遍历 p.stdout ：逐行获取输出，但需要考虑终端或文件缓冲的差异。

根据运行情况定制输出处理

为了根据运行情况（终端或文件）定制输出处理，我们可以使用 p.stdout.isatty() 检查输出是否直接流向终端。

代码示例

以下是如何使用 subprocess.Popen 获取实时输出的示例：

import subprocess

# 创建子进程
p = subprocess.Popen(['./test_1.py'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

# 根据运行情况处理输出
if p.stdout.isatty():
    # 终端运行，立即显示输出
    for line in p.stdout:
        print(line.decode('utf-8').rstrip())
else:
    # 重定向到文件，缓冲输出直到子进程完成
    output, _ = p.communicate()
    print(output.decode('utf-8'))