向H2数据库中的所有表添加列而不手动列出表

Question

我有一个包含大约 70 个表的 PostgreSQL 数据库，我想为每个表添加两列，created_by 和 updated_by。由于我们使用的是 Liquibase，因此我编写了一个在 PostgreSQL 中运行良好的更新脚本：

DO $$
DECLARE
  tables CURSOR FOR
    SELECT tablename
    FROM pg_tables
    WHERE schemaname = 'myschema'
    ORDER BY tablename;
BEGIN
  FOR table_record IN tables LOOP
    EXECUTE
      'ALTER TABLE ' || table_record.tablename || ' ADD COLUMN created_by BIGINT; ' ||
      'ALTER TABLE ' || table_record.tablename || ' ADD COLUMN updated_by BIGINT;';
  END LOOP;
END$$;

但是，对于集成测试，我们使用的是 H2 内存数据库，它不接受此 SQL，即使在 PostgreSQL 兼容模式下也是如此。 From this question，我知道如何选择所有表：

SELECT * FROM INFORMATION_SCHEMA.TABLES

但这对myschema 也有效吗？有没有办法存储结果并循环它们以添加列？如答案中所述，我也不明白如何使用或编写准备好的语句。

基本问题是：如何在不手动列出所有表的情况下为 H2 数据库编写等效的更新脚本？

Answer 1

我还没有找到使用 H2 语法的方法。相反，我使用 SQL 使用 PostgreSQL 语法创建适当的查询：

SELECT 'ALTER TABLE ' || tablename || ' ADD COLUMN created_by BIGINT; ' || 
       'ALTER TABLE ' || tablename || ' ADD COLUMN updated_by BIGINT;'
        FROM pg_tables
        WHERE schemaname = 'myschema'
        ORDER BY tablename;

这将返回所有表的ALTER TABLE 查询列表，例如：

ALTER TABLE first_table ADD COLUMN created_by BIGINT;
ALTER TABLE first_table ADD COLUMN updated_by BIGINT;

这不适用于一组变化的表，但我只需要为一组固定的表执行此操作，因此它适用于我的情况。

（我在连接到我的 PostgreSQL 数据库的 SQL Workbench 中执行了查询。结果视图列出了已创建的查询，然后可以将它们复制到在测试期间执行的脚本文件中。）

Answer 2

试试这样的:-)

DECLARE @Schema VARCHAR(20),
    @Table  VARCHAR(50),
    @Query  VARCHAR(4000),
    @Column VARCHAR(100) = 'Column',
    @Column_props VARCHAR(1000) = 'INT NOT NULL DEFAULT 0'
DECLARE c_cursor CURSOR FAST_FORWARD LOCAL FOR
SELECT TABLE_SCHEMA, TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE             
TABLE_TYPE='BASE TABLE'
OPEN c_cursor
FETCH NEXT FROM c_cursor INTO @Schema, @Table
WHILE @@FETCH_STATUS = 0 BEGIN
    SET @Query = '
    IF COL_LENGTH('''+@Schema+'.'+@Table+''', '''+@Column+''') IS NULL  BEGIN
      ALTER TABLE '+@Schema+'.'+@Table+'
      ADD ['+@Column+'] '+@Column_props+'
    END'
    EXEC(@Query)
    FETCH NEXT FROM c_cursor INTO @Schema, @Table
END
CLOSE c_cursor
DEALLOCATE c_cursor