如何在Chef中可靠检测PostgreSQL数据库并存储检测结果？

我之前刚好处理过类似的Chef + PostgreSQL数据库检测场景，给你一套可行的实现方案，完美适配你的需求：

核心思路 #

通过Ruby代码执行检测命令，把每个数据库的存在状态临时存储到Chef的run_state中（这是本次Chef运行的临时内存存储，不会持久化到节点属性），然后遍历每个数据库，根据存储的状态用only_if/not_if guard控制后续资源的执行。

具体实现步骤 #

1. 定义待检测的数据库列表

先把要检测的3个数据库列出来，方便后续循环处理：

# 替换成你的3个数据库名称
target_databases = ['db_prod', 'db_staging', 'db_test']

2. 执行数据库存在性检测并存储结果

用ruby_block资源执行检测命令，因为Chef的编译阶段和执行阶段是分离的，ruby_block会在执行阶段运行，这样后续资源能拿到实时的检测结果：

ruby_block 'check_postgresql_database_existence' do
  block do
    target_databases.each do |db_name|
      # 执行检测命令，指定postgres用户（默认PostgreSQL管理员）
      check_cmd = Mixlib::ShellOut.new(
        "psql -d #{db_name} -c 'SELECT 1 as check_;' 2>/dev/null",
        user: 'postgres',
        timeout: 10
      )
      check_cmd.run_command

      # 把检测结果存入run_state：true=存在，false=不存在
      node.run_state["pg_db_exists_#{db_name}"] = check_cmd.exitstatus == 0
    end
  end
  action :run
end

注意：这里加了2>/dev/null来屏蔽错误输出，避免Chef run日志被无关信息刷屏；同时指定了超时时间，防止命令卡住。

3. 根据检测结果处理每个数据库的后续步骤

遍历每个数据库，用only_if/not_if来控制创建用户、授权、加载数据等操作：

target_databases.each do |db_name|
  # 示例1：如果数据库不存在，创建它（如果你的需求是需要创建的话）
  postgresql_database db_name do
    owner 'postgres'
    encoding 'UTF8'
    action :create
    not_if { node.run_state["pg_db_exists_#{db_name}"] }
  end

  # 示例2：创建专属数据库用户（可根据需求调整guard规则）
  postgresql_user "app_user_#{db_name}" do
    password 'your_secure_password_here'
    action :create
  end

  # 示例3：仅当数据库存在时，给用户授权
  postgresql_access "grant_full_access_to_#{db_name}" do
    database_name db_name
    role "app_user_#{db_name}"
    privileges [:all]
    action :grant
    only_if { node.run_state["pg_db_exists_#{db_name}"] }
  end

  # 示例4：仅当数据库存在时，加载初始化数据
  execute "load_initial_data_into_#{db_name}" do
    command "psql -d #{db_name} -U postgres -f /path/to/your/init_data.sql"
    only_if { node.run_state["pg_db_exists_#{db_name}"] }
  end
end

关键细节说明 #

• 为什么用run_state？ 它是Chef运行时的临时存储，只在本次run中有效，不会污染节点的持久化属性，非常适合这种一次性的检测结果存储。
• 权限问题：确保执行psql命令的用户（这里用postgres）有足够的权限连接PostgreSQL，如果你用的是其他用户，需要提前配置好对应的pg_hba规则。
• 命令路径：如果psql不在默认PATH里，要写绝对路径（比如/usr/pgsql-14/bin/psql，根据你的PostgreSQL版本调整）。

这样就能实现逐个检测数据库，跳过不存在的库的后续操作，自动处理下一个数据库的逻辑了。

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