Question

我想用python的asyncio模块跟踪守护进程的重启过程。所以我需要运行shell命令tail -f -n 0 /var/log/daemon.log并分析它的输出，让我们说，service daemon restart在后台执行。守护进程继续写入日志服务重启命令完成它的执行并报告它的内部检查。跟踪进程读取检查信息并根据内部逻辑报告重启是否成功。

import asyncio
from asyncio.subprocess import PIPE, STDOUT

async def track():
    output = []
    process = await asyncio.create_subprocess_shell(
        'tail -f -n0 ~/daemon.log',
        stdin=PIPE, stdout=PIPE, stderr=STDOUT
    )
    while True:
        line = await process.stdout.readline()
        if line.decode() == 'reboot starts\n':
            output.append(line)
            break
    while True:
        line = await process.stdout.readline()
        if line.decode() == '1st check completed\n':
            output.append(line)
            break
    return output

async def reboot():
    lines = [
        '...',
        '...',
        'reboot starts',
        '...',
        '1st check completed',
        '...',
    ]
    p = await asyncio.create_subprocess_shell(
        (
            'echo "rebooting"; '
            'for line in {}; '
                'do echo $line >> ~/daemon.log; sleep 1; '
            'done; '
            'echo "rebooted";'
        ).format(' '.join('"{}"'.format(l) for l in lines)),
        stdin=PIPE, stdout=PIPE, stderr=STDOUT
    )
    return (await p.communicate())[0].splitlines()

if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    loop.run_until_complete(asyncio.gather(
        asyncio.ensure_future(track()),
        asyncio.ensure_future(reboot())
    ))
    loop.close()

这段代码是我发现并行运行两个协程的唯一方法。但是如何在track()之前严格运行reboot以不丢失日志中的任何可能输出？以及如何检索两个协同程序的返回值？

Answer 1

但是如何在重启之前严格运行track（）以不丢失日志中的任何可能输出？

在运行第二个子进程之前，您可以await创建第一个子进程。

如何检索两个协同程序的返回值？

asyncio.gather返回汇总结果。

示例：

async def main():
    process_a = await asyncio.create_subprocess_shell([...])
    process_b = await asyncio.create_subprocess_shell([...])
    return await asyncio.gather(monitor_a(process_a), monitor_b(process_b))

loop = asyncio.get_event_loop()
result_a, result_b = loop.run_until_complete(main())

如何使用asyncio复杂地管理shell进程？

1 个答案: