self-hosted/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2024-11-28 09:33:27 +02:00
Code Issues Releases Activity

simplify

Browse Source
This commit is contained in:
mertalev 2024-07-03 19:54:08 -04:00
parent 90709b4788
commit 8322af0baf
No known key found for this signature in database
GPG Key ID: 9181CD92C0A1C5E3
1 changed files with 8 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
server/src/migrations/1718486162779-AddFaceSearchRelation.ts
View File

@ -9,26 +9,22 @@ export class AddFaceSearchRelation1718486162779 implements MigrationInterface {
await queryRunner.query(`SET vectors.pgvector_compatibility=on`); await queryRunner.query(`SET vectors.pgvector_compatibility=on`);
} }
await queryRunner.query(`ALTER TABLE asset_faces ADD COLUMN IF NOT EXISTS embedding vector(512)`);
await queryRunner.query(`ALTER TABLE smart_search ADD COLUMN IF NOT EXISTS embedding vector(512) NOT NULL`);
await queryRunner.query(` await queryRunner.query(`
CREATE TABLE face_search ( CREATE TABLE face_search (
"faceId" uuid PRIMARY KEY REFERENCES asset_faces(id) ON DELETE CASCADE, "faceId" uuid PRIMARY KEY REFERENCES asset_faces(id) ON DELETE CASCADE,
embedding vector(512) NOT NULL )`); embedding vector(512) NOT NULL )`);
await queryRunner.query(`ALTER TABLE face_search ALTER COLUMN embedding SET STORAGE EXTERNAL`); await queryRunner.query(`ALTER TABLE face_search ALTER COLUMN embedding SET STORAGE EXTERNAL`);
await queryRunner.query(`ALTER TABLE smart_search ADD COLUMN IF NOT EXISTS embedding vector(512)`);
await queryRunner.query(`ALTER TABLE smart_search ALTER COLUMN embedding SET STORAGE EXTERNAL`); await queryRunner.query(`ALTER TABLE smart_search ALTER COLUMN embedding SET STORAGE EXTERNAL`);
const assetFacesColumns = await queryRunner.query( await queryRunner.query(`
`SELECT column_name as name INSERT INTO face_search("faceId", embedding)
FROM information_schema.columns SELECT id, embedding
WHERE table_name = 'asset_faces'`); FROM asset_faces faces
const hasFaceEmbeddings = assetFacesColumns.some((column: { name: string }) => column.name === 'embedding'); WHERE faces.embedding IS NOT NULL`);
if (hasFaceEmbeddings) {
await queryRunner.query(`
INSERT INTO face_search("faceId", embedding)
SELECT id, embedding
FROM asset_faces faces`);
}
await queryRunner.query(`ALTER TABLE asset_faces DROP COLUMN IF EXISTS embedding`); await queryRunner.query(`ALTER TABLE asset_faces DROP COLUMN IF EXISTS embedding`);