Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-06-02 13:14:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(server): clean face tables after delete (#12375)

Browse Source 
clean face tables after delete
This commit is contained in:
Mert 2024-09-05 19:09:19 -04:00 committed by GitHub
parent 649897f737
commit eb7777639d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
server/src/repositories/person.repository.ts
View File

@ -6,6 +6,7 @@ import { AssetFaceEntity } from 'src/entities/asset-face.entity';
import { AssetJobStatusEntity } from 'src/entities/asset-job-status.entity'; import { AssetJobStatusEntity } from 'src/entities/asset-job-status.entity';
import { AssetEntity } from 'src/entities/asset.entity'; import { AssetEntity } from 'src/entities/asset.entity';
import { PersonEntity } from 'src/entities/person.entity'; import { PersonEntity } from 'src/entities/person.entity';
import { SourceType } from 'src/enum';
import { import {
AssetFaceId, AssetFaceId,
DeleteAllFacesOptions, DeleteAllFacesOptions,
@ -53,16 +54,20 @@ export class PersonRepository implements IPersonRepository {
} }
async deleteAllFaces({ sourceType }: DeleteAllFacesOptions): Promise<void> { async deleteAllFaces({ sourceType }: DeleteAllFacesOptions): Promise<void> {
if (sourceType) { if (!sourceType) {
return this.assetFaceRepository.query('TRUNCATE TABLE asset_faces CASCADE');
}
await this.assetFaceRepository await this.assetFaceRepository
.createQueryBuilder('asset_faces') .createQueryBuilder('asset_faces')
.delete() .delete()
.andWhere('sourceType = :sourceType', { sourceType }) .andWhere('sourceType = :sourceType', { sourceType })
.execute(); .execute();
return;
}
await this.assetFaceRepository.query('TRUNCATE TABLE asset_faces CASCADE'); await this.assetFaceRepository.query('VACUUM ANALYZE asset_faces, face_search');
if (sourceType === SourceType.MACHINE_LEARNING) {
await this.assetFaceRepository.query('REINDEX INDEX face_index');
}
} }
getAllFaces( getAllFaces(