如何在Node.js中动态创建PostgreSQL数据库枚举类型

没问题，我来帮你搞定这个动态创建PostgreSQL枚举的需求！你的核心问题在于PostgreSQL的DDL语句（比如创建枚举）并不支持直接用参数化查询来传入枚举值列表——参数占位符只能用于数据操作（比如INSERT/SELECT），不能用于这类语法结构层面的内容。之前的两种方案之所以失效，原因也在这里：

• 方案一：把整个数组当作单个参数传入，PostgreSQL会把它解析成单个枚举值（比如数组转成字符串后会变成'["firstType","secondType"]'），完全不符合你的预期。
• 方案二：手动生成多参数占位符的思路本身有问题，因为ENUM ($1, $2)这种写法在PostgreSQL里是不合法的——参数占位符不能出现在枚举值定义的位置。

下面给你两种安全且可行的实现方式：

方式一：使用pg-format库生成安全的SQL #

pg-format是专门为PostgreSQL设计的格式化工具，它能自动帮你转义字符串，避免SQL注入风险，同时完美处理枚举值列表的拼接。

步骤： #

1. 先安装依赖：

npm install pg-format

2. 实现代码：

const axios = require('axios');
const pg = require('pg');
const format = require('pg-format');
const fs = require('fs');
const path = require('path');

async function setupDb() {
  // 初始化数据库连接池
  const pgPool = new pg.Pool({
    host: 'your-host',
    user: 'your-user',
    password: 'your-password',
    database: 'your-db'
  });

  try {
    // 从接口获取枚举值数组
    const { data: tableTypes } = await axios.get('/tableTypes');
    
    // 校验数组合法性
    if (!Array.isArray(tableTypes) || tableTypes.length === 0) {
      throw new Error('获取到的枚举值数组为空或格式错误');
    }

    // 开启事务
    await pgPool.query('BEGIN');

    // 动态生成创建枚举的SQL：%L会自动转义每个枚举值
    const createEnumSql = format(
      `DO $$ BEGIN 
        IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'tabletype') THEN 
          CREATE TYPE tabletype AS ENUM (%L); 
        END IF; 
      END $$;`,
      tableTypes
    );
    await pgPool.query(createEnumSql);

    // 创建使用该枚举的表
    const createTableSql = `
      CREATE TABLE IF NOT EXISTS new_table(
        id SERIAL PRIMARY KEY,
        enumValue tabletype NOT NULL
      );
    `;
    await pgPool.query(createTableSql);

    // 提交事务
    await pgPool.query('COMMIT');
    console.log('枚举类型和表创建成功！');
  } catch (err) {
    // 出错回滚事务
    await pgPool.query('ROLLBACK');
    console.error('操作失败:', err.message);
    throw err;
  } finally {
    // 关闭连接池
    await pgPool.end();
  }
}

// 执行初始化
setupDb();

方式二：使用PostgreSQL内置的format函数 #

如果你不想引入第三方库，也可以用PostgreSQL自带的format函数结合EXECUTE来动态执行DDL，同样能保证安全性：

// 省略数据库连接和接口请求部分，直接看核心SQL执行代码
const createEnumSql = `
DO $$ BEGIN
  IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'tabletype') THEN
    -- 用format函数转义枚举值，array_to_string把数组转成逗号分隔的字符串
    EXECUTE format('CREATE TYPE tabletype AS ENUM (%L)', array_to_string($1, ','));
  END IF;
END $$;
`;

// 传入枚举值数组作为参数
await pgPool.query(createEnumSql, [tableTypes]);

关键说明： #

两种方式的核心都是安全地转义枚举值，避免SQL注入。绝对不要直接用字符串拼接（比如tableTypes.join(',')），如果枚举值里包含单引号或其他特殊字符，会直接导致SQL语法错误，甚至引入注入风险。

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