@clickup/pg-microsharding/src/internal/changeTableOwners.ts

Microshards support for PostgreSQL

import groupBy from "lodash/groupBy";
import { ident, join, sql } from "./quote";
import { runSql } from "./runSql";

export async function changeTableOwners({
  toDsn,
  tables,
  comment,
}: {
  toDsn: string;
  tables: Array<{ schema: string; name: string; owner: string }>;
  comment?: string;
}): Promise<void> {
  if (tables.length === 0) {
    return;
  }

  const queries = Object.entries(groupBy(tables, (table) => table.schema)).map(
    ([schema, tables]) =>
      join(
        tables.map(
          ({ name, owner }) =>
            sql`ALTER TABLE ONLY ${ident(schema)}.${ident(name)} OWNER TO ${ident(owner)};`,
        ),
        " ",
      ),
  );
  await runSql(toDsn, join(queries, "\n"), comment);
}