☰

複数のawkコマンドによるパイプ

tag-icon • tag-icon bash awk pipe stdout tee
複数のawkコマンドによるパイプ

1つのコマンドの出力を2つの異なるコマンドにパイプしようとしています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

提案されているように、最初のawkコマンドにfflush()または追加しようとしました。stdbuf -o0ここ。

答え1

シェルスクリプトを使用するよりコンパクトなソリューション：

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ごとに3つの追加プロセスを実行することに注意してください。

元のソリューションは以下を使用します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

関連情報