Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-05-24 01:12:58 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: lower disk usage during migration (#18440)

Browse Source 
feat: less disk usage during migration
This commit is contained in:
Mert 2025-05-22 03:41:10 -04:00 committed by GitHub
parent 0b8fc7b493
commit bc8e08f5e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server/src/repositories/database.repository.ts
View File

@ -247,8 +247,8 @@ export class DatabaseRepository {
return; return;
} }
const dimSize = await this.getDimensionSize(table); const dimSize = await this.getDimensionSize(table);
await sql`DROP INDEX IF EXISTS ${sql.raw(indexName)}`.execute(this.db);
await this.db.transaction().execute(async (tx) => { await this.db.transaction().execute(async (tx) => {
await sql`DROP INDEX IF EXISTS ${sql.raw(indexName)}`.execute(tx);
if (!rows.some((row) => row.columnName === 'embedding')) { if (!rows.some((row) => row.columnName === 'embedding')) {
this.logger.warn(`Column 'embedding' does not exist in table '${table}', truncating and adding column.`); this.logger.warn(`Column 'embedding' does not exist in table '${table}', truncating and adding column.`);
await sql`TRUNCATE TABLE ${sql.raw(table)}`.execute(tx); await sql`TRUNCATE TABLE ${sql.raw(table)}`.execute(tx);
@ -262,6 +262,11 @@ export class DatabaseRepository {
SET DATA TYPE ${sql.raw(schema)}vector(${sql.raw(String(dimSize))})`.execute(tx); SET DATA TYPE ${sql.raw(schema)}vector(${sql.raw(String(dimSize))})`.execute(tx);
await sql.raw(vectorIndexQuery({ vectorExtension, table, indexName, lists })).execute(tx); await sql.raw(vectorIndexQuery({ vectorExtension, table, indexName, lists })).execute(tx);
}); });
try {
await sql`VACUUM ANALYZE ${sql.raw(table)}`.execute(this.db);
} catch (error: any) {
this.logger.warn(`Failed to vacuum table '${table}'. The DB will temporarily use more disk space: ${error}`);
}
this.logger.log(`Reindexed ${indexName}`); this.logger.log(`Reindexed ${indexName}`);
} }