mirror of
https://iceshrimp.dev/iceshrimp/iceshrimp
synced 2024-12-03 01:08:05 +09:00
85 lines
3.1 KiB
JavaScript
85 lines
3.1 KiB
JavaScript
import { redisClient } from "../built/db/redis.js";
|
|
|
|
export class MoveAntennaToCache1695749948350 {
|
|
name = "MoveAntennaToCache1695749948350";
|
|
async up(queryRunner) {
|
|
const tableExists = await queryRunner.query(`SELECT EXISTS ( SELECT 1 FROM pg_tables WHERE tablename = 'antenna_note' ) AS table_existence`)
|
|
.then(p => !!p[0]['table_existence']);
|
|
|
|
if (!tableExists) {
|
|
console.log('Skipping migration ("antenna_note" table does not exist)');
|
|
return;
|
|
}
|
|
|
|
const skipCopy = process.env.ANTENNA_MIGRATION_SKIP === 'true';
|
|
let readLimit = parseInt(process.env.ANTENNA_MIGRATION_READ_LIMIT ?? "10000", 10) ?? 10000;
|
|
|
|
if (skipCopy) {
|
|
console.log('ANTENNA_MIGRATION_SKIP = true, skipping antenna note migration');
|
|
}
|
|
else {
|
|
const total = await queryRunner.query(`SELECT COUNT(1) FROM "antenna_note"`)
|
|
.then(p => p[0]['count']);
|
|
|
|
console.log(`Copying ${total} entries in "antenna_note", please hang tight!`);
|
|
|
|
let remaining = total;
|
|
|
|
let query = `SELECT "id", "noteId", "antennaId" FROM "antenna_note" ORDER BY "id" ASC LIMIT ${readLimit}`;
|
|
|
|
while (remaining > 0) {
|
|
let res = await queryRunner.query(query);
|
|
if (res.length === 0) break;
|
|
remaining -= res.length;
|
|
|
|
for (const hit of res) {
|
|
redisClient.xadd(`antennaTimeline:${hit.antennaId}`, "MAXLEN", "~", "200", "*", "note", hit.noteId);
|
|
}
|
|
|
|
console.log(`Copied ${total-remaining}/${total} notes to cache.`);
|
|
|
|
query = `SELECT "id", "noteId", "antennaId" FROM "antenna_note" WHERE "id" > '${res.at(-1).id}' ORDER BY "id" ASC LIMIT ${Math.min(readLimit, remaining)}`;
|
|
}
|
|
}
|
|
|
|
await queryRunner.query(`DROP TABLE IF EXISTS "antenna_note"`);
|
|
}
|
|
|
|
async down(queryRunner) {
|
|
await queryRunner.query(
|
|
`CREATE TABLE IF NOT EXISTS "antenna_note" ("id" character varying(32) NOT NULL, "noteId" character varying(32) NOT NULL, "antennaId" character varying(32) NOT NULL, CONSTRAINT "PK_fb28d94d0989a3872df19fd6ef8" PRIMARY KEY ("id"))`
|
|
);
|
|
await queryRunner.query(
|
|
`CREATE INDEX "IDX_bd0397be22147e17210940e125" ON "antenna_note" ("noteId") `,
|
|
);
|
|
await queryRunner.query(
|
|
`CREATE INDEX "IDX_0d775946662d2575dfd2068a5f" ON "antenna_note" ("antennaId") `,
|
|
);
|
|
await queryRunner.query(
|
|
`CREATE UNIQUE INDEX "IDX_335a0bf3f904406f9ef3dd51c2" ON "antenna_note" ("noteId", "antennaId") `,
|
|
);
|
|
await queryRunner.query(
|
|
`ALTER TABLE "antenna_note" ADD CONSTRAINT "FK_bd0397be22147e17210940e125b" FOREIGN KEY ("noteId") REFERENCES "note"("id") ON DELETE CASCADE ON UPDATE NO ACTION`,
|
|
undefined,
|
|
);
|
|
await queryRunner.query(
|
|
`ALTER TABLE "antenna_note" ADD CONSTRAINT "FK_0d775946662d2575dfd2068a5f5" FOREIGN KEY ("antennaId") REFERENCES "antenna"("id") ON DELETE CASCADE ON UPDATE NO ACTION`,
|
|
undefined,
|
|
);
|
|
await queryRunner.query(
|
|
`ALTER TABLE "antenna_note" ADD "read" boolean NOT NULL DEFAULT false`,
|
|
undefined,
|
|
);
|
|
await queryRunner.query(
|
|
`CREATE INDEX "IDX_9937ea48d7ae97ffb4f3f063a4" ON "antenna_note" ("read") `,
|
|
undefined,
|
|
);
|
|
await queryRunner.query(
|
|
`COMMENT ON COLUMN "antenna_note"."noteId" IS 'The note ID.'`,
|
|
);
|
|
await queryRunner.query(
|
|
`COMMENT ON COLUMN "antenna_note"."antennaId" IS 'The antenna ID.'`,
|
|
);
|
|
}
|
|
}
|