如何程序化查找运行PSQL服务器的SLURM节点

方法1：通过共享存储记录运行节点 #

启动PostgreSQL时，将当前节点名写入集群所有节点都能访问的共享存储文件（比如你的home目录，多数SLURM集群home是共享的）：

# 启动PSQL前先写入节点信息
echo $(hostname) > ~/pg_running_node.txt
pg_ctl start -D /path/to/your/pg/data

之后在任意节点读取这个文件就能知道PSQL所在节点：

cat ~/pg_running_node.txt

注意：如果重启PSQL，要确保这个文件被更新；可以把写入命令加到你的PSQL启动脚本里，避免手动操作遗漏。

方法2：端口扫描检测开放的PSQL端口 #

PostgreSQL默认监听5432端口，你可以遍历所有登录节点，检查该端口是否开放：

# 获取所有登录节点列表（根据集群节点命名调整grep的匹配规则）
for node in $(sinfo -N -h | grep -E 'login|head' | awk '{print $1}'); do
  # 使用nc检测端口连通性
  if nc -z $node 5432; then
    echo "PostgreSQL is running on $node"
    exit 0
  fi
done
echo "PostgreSQL not found on any login node"

局限性：如果集群防火墙限制了跨节点端口访问，这个方法会失效；另外如果修改了PSQL默认端口，要把5432换成你的自定义端口。

方法3：跨节点检查PostgreSQL进程 #

如果集群允许登录节点之间无密码SSH（多数集群默认支持），可以逐个登录节点检查postgres进程：

# 获取所有登录节点列表
for node in $(sinfo -N -h | grep -E 'login|head' | awk '{print $1}'); do
  # 远程检查postgres进程是否存在
  ssh $node "pgrep -x postgres" > /dev/null 2>&1
  if [ $? -eq 0 ]; then
    echo "PostgreSQL is running on $node"
    exit 0
  fi
done
echo "PostgreSQL not found on any login node"

如果没有无密码SSH，可以自己生成密钥对，将公钥添加到每个登录节点的~/.ssh/authorized_keys（若home目录共享，只需配置一次即可同步到所有节点）。

方法4：利用SLURM节点自定义标记（需权限支持） #

如果集群管理员允许用户修改节点注释，启动PSQL后可以给当前节点打上标记：

# 标记当前节点为运行PSQL的节点
scontrol update NodeName=$(hostname) Comment="pg_active"

之后在任意节点通过sinfo筛选带该标记的节点：

sinfo -N -o "%N %E" | grep pg_active

局限性：部分集群会限制普通用户使用scontrol update命令，先执行测试是否有权限。

内容的提问来源于stack exchange，提问作者Liuka