☰

通过管道传输到多个 awk 命令

tag-icon • tag-icon bash awk pipe stdout tee
通过管道传输到多个 awk 命令

我正在尝试将一个命令的输出通过管道传输到两个不同的awk命令。下列的这个帖子我正在使用tee和处理替换。但是，我看不到替换过程的输出。

nvidia-smi | tee >(awk '/ C / {print $6}') | awk '/ C / {print $3}' | xargs -r ps -o user

这应该显示所有 GPU 进程的用户和内存使用情况。内存使用情况和 PID 分别由和提取nvidia-smi，然后将后者通过管道传输到。但输出仅包含用户。awk '/ C / {print $6}'awk '/ C / {print $3}'ps -o user

我想要的是

<memory-of-process1> <name-of-user-running-process1>
<memory-of-process2> <name-of-user-running-process2>
<memory-of-process3> <name-of-user-running-process3> 
etc

我得到的是

<name-of-user-running-process1>
<name-of-user-running-process2>
<name-of-user-running-process3>
etc

我尝试按照建议将fflush()or添加stdbuf -o0到第一个 awk 命令这里。

答案1

使用 shell 脚本的更紧凑的解决方案：

nvidia-smi | grep ' C ' | while read _ _ pid _ _ mem; do
    user="$( ps -o user "${pid}" | tail -n +2 )"
    printf '%8s  %6d  %s\n' "${mem}" "${pid}" "${user}"
done

请注意，这会为每个 pid 运行三个额外的进程。

原始解决方案使用awk：

#! /bin/bash

function nvidia-smi { cat <<'[][]'
A C 937 D E 1.7MB
A C 1232 D E 0.25MB
A E 6112 D E 13MB
A C 2008 D E 437KB
A C 2024 D E 314157
[][]
}

AWK='
/ C / { Pid[NR] = $3; xPid[$3] = NR; Mem[NR] = $6; }
END {
    for (j in Pid) pp = pp "," Pid[j];
    cmd = "ps 2>&1 -o pid,user -p " substr (pp, 2);
    while (cmd | getline) 
        if ($1 in xPid) User[xPid[$1]] = $2;
    close (cmd);
    fmt = "%8s  %6d  %s\n";
    for (j = 1; j <= NR; ++j) 
        if (j in Mem)
            printf (fmt, Mem[j], Pid[j], User[j]);
}
'
    nvidia-smi | awk "${AWK}"

函数 nvidia-smi 只是提供一些测试数据 - 丢弃它。您需要 AWK 变量（单引号之间的多行常量中的 12 行）及其下面的简短管道。

一个测试。我包括了正在使用内存的 pid：

paul $ ./nVid
   1.7MB     937  syslog
  0.25MB    1232  root
   437KB    2008  postfix
  314157    2024  paul

相关内容