limepotato/jormungandr-bite
1
0
Fork 0
mirror of https://iceshrimp.dev/limepotato/jormungandr-bite.git synced 2024-11-22 01:47:39 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

[backend] Dynamically import redisClient in postgre.ts and dependencies

Browse source 
This fixes unnecessary redisClient instances being created during migration runs, which previously prevented the revertmigration command from exiting by itself due to a TypeORM inconsistency between migration:run and migration:revert.
This commit is contained in:
Laura Hausmann 2023-11-07 22:50:31 +01:00
parent 1488529228
commit 2baa787b72
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
3 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
packages/backend/src/db/postgre.ts
View file

@ -75,7 +75,6 @@ import { NoteEdit } from "@/models/entities/note-edit.js";
import { entities as charts } from "@/services/chart/entities.js"; import { entities as charts } from "@/services/chart/entities.js";
import { envOption } from "../env.js"; import { envOption } from "../env.js";
import { dbLogger } from "./logger.js"; import { dbLogger } from "./logger.js";
import { redisClient } from "./redis.js";
import { OAuthApp } from "@/models/entities/oauth-app.js"; import { OAuthApp } from "@/models/entities/oauth-app.js";
import { OAuthToken } from "@/models/entities/oauth-token.js"; import { OAuthToken } from "@/models/entities/oauth-token.js";
@ -238,6 +237,7 @@ export async function initDb(force = false) {
export async function resetDb() { export async function resetDb() {
const reset = async () => { const reset = async () => {
const { redisClient } = await import("./redis.js");
await redisClient.flushdb(); await redisClient.flushdb();
const tables = await db.query(`SELECT relname AS "table" const tables = await db.query(`SELECT relname AS "table"
FROM pg_class C LEFT JOIN pg_namespace N ON (N.oid = C.relnamespace) FROM pg_class C LEFT JOIN pg_namespace N ON (N.oid = C.relnamespace)

4
packages/backend/src/migration/1695749948350-move-antenna-to-cache.ts
View file

@ -1,5 +1,4 @@
import { MigrationInterface, QueryRunner } from "typeorm"; import { MigrationInterface, QueryRunner } from "typeorm";
import { redisClient } from "../db/redis.js";
export class MoveAntennaToCache1695749948350 implements MigrationInterface { export class MoveAntennaToCache1695749948350 implements MigrationInterface {
name = "MoveAntennaToCache1695749948350"; name = "MoveAntennaToCache1695749948350";
@ -19,6 +18,7 @@ export class MoveAntennaToCache1695749948350 implements MigrationInterface {
console.log('ANTENNA_MIGRATION_SKIP = true, skipping antenna note migration'); console.log('ANTENNA_MIGRATION_SKIP = true, skipping antenna note migration');
} }
else { else {
const { redisClient } = await import("../db/redis.js");
const total = await queryRunner.query(`SELECT COUNT(1) FROM "antenna_note"`) const total = await queryRunner.query(`SELECT COUNT(1) FROM "antenna_note"`)
.then(p => p[0]['count']); .then(p => p[0]['count']);
@ -41,6 +41,8 @@ export class MoveAntennaToCache1695749948350 implements MigrationInterface {
query = `SELECT "id", "noteId", "antennaId" FROM "antenna_note" WHERE "id" > '${res.at(-1).id}' ORDER BY "id" ASC LIMIT ${Math.min(readLimit, remaining)}`; query = `SELECT "id", "noteId", "antennaId" FROM "antenna_note" WHERE "id" > '${res.at(-1).id}' ORDER BY "id" ASC LIMIT ${Math.min(readLimit, remaining)}`;
} }
redisClient.quit();
} }
await queryRunner.query(`DROP TABLE IF EXISTS "antenna_note"`); await queryRunner.query(`DROP TABLE IF EXISTS "antenna_note"`);

2
packages/backend/src/services/chart/core.ts
View file

@ -15,7 +15,6 @@ import {
subtractTime, subtractTime,
addTime, addTime,
} from "@/prelude/time.js"; } from "@/prelude/time.js";
import { getChartInsertLock } from "@/misc/app-lock.js";
import { db } from "@/db/postgre.js"; import { db } from "@/db/postgre.js";
import promiseLimit from "promise-limit"; import promiseLimit from "promise-limit";
@ -430,6 +429,7 @@ export default abstract class Chart<T extends Schema> {
? `${this.name}:${date}:${span}:${group}` ? `${this.name}:${date}:${span}:${group}`
: `${this.name}:${date}:${span}`; : `${this.name}:${date}:${span}`;
const { getChartInsertLock } = await import("@/misc/app-lock.js");
const unlock = await getChartInsertLock(lockKey); const unlock = await getChartInsertLock(lockKey);
try { try {
// ロック内でもう1回チェックする // ロック内でもう1回チェックする