在awk中预先填充关联数组键？

Question

我写了一个munin插件,它使用slurm的sacct来监视HPC集群上的作业状态.我用sh + awk编写了它(而不是我常用的工具,perl).

该脚本有效,但我花了很多时间才弄清楚如何预先填充可能状态的关联数组(一些/大多数可能不存在于sacct输出中,我希望它们默认为零).谷歌没有多大帮助,我能想到的最好的就是在字符串上使用split来生成一个临时数组,然后我迭代了.

我想出了这个:

BEGIN {
    num = split("cancelled completed completing failed nodefail pending running suspended timeout",statenames," ");
    for (i=1;i<=num;i++) {
        states[statenames[i]] = 0
    }
  }

这可行,但与我在perl中的方式相比看起来很笨拙,如下所示:

foreach (qw(cancelled completed completing failed nodefail pending running suspended timeout)) {
    $states{$_} = 0;
}

或这个

%states = map {$_ => 0} qw(cancelled completed completing failed nodefail pending running suspended timeout);

我的问题是:有没有办法在awk中执行此操作,类似于任何一个perl版本？

[编辑]

澄清一下,这是一个我输入awk的sacct输出样本.请注意,此输出中的唯一状态是RUNNING,COMPLETED和CANCELED - 其他状态不存在(因为它们今天没有出现),但我仍然希望它们在我的脚本输出中(以munin可用的形式表示为" statename.value 0").

# sacct -X -P -o 'state' -n
RUNNING
RUNNING
RUNNING
RUNNING
COMPLETED
RUNNING
COMPLETED
RUNNING
COMPLETED
COMPLETED
CANCELLED by 1000
COMPLETED

[再次编辑]

这是我的munin插件的示例输出:

# ./slurm-sacct
suspended.value 0
pending.value 0
nodefail.value 0
failed.value 0
running.value 6
completing.value 0
completed.value 5
timeout.value 0
cancelled.value 1

脚本运行并执行我想要的操作,我只是想知道是否有更好的方法来初始化关联数组.

Answer 1

你可能根本不需要这样做.awk中的变量是动态的,这意味着它们在首次使用时(自动分配或访问)会自动初始化,这也适用于数组元素.

如果在数值上下文中访问变量,则变量将初始化为0,否则将变为空字符串.(至少gawk这样做,虽然我不确定它是否依赖于实现)所以如果你做的事情就像计算每个状态中的作业数量一样,整个程序就像这样简单

{ states[$1]++ }
END {
     for (state in states) print state, states[state]
}

每次states[$1]++执行表达式时,它将检查是否存在states[$1]并将其初始化为0(如果它尚不存在).

编辑:从你的评论我猜你想打印出每个可能状态的一行,无论该状态是否有任何工作.在这种情况下,您需要包含所有可能的状态名称,并且没有像Perl中那样执行此操作的快捷方式.据我所知,你已经发现的东西是干净利落的.(Awk并没有真正考虑到这种用法)

我建议如下:

{ states[$1]++ }
END {
     split("cancelled completed completing failed nodefail pending running suspended timeout",statenames," ");
     for (state in statenames) print state, states[state]+0
}