TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Fix] Avoid queueing same fileID again

Browse Source
This commit is contained in:
Neeraj Gupta
2024-08-16 18:00:05 +05:30
parent b03b5806db
commit 4e7e506e4d
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mobile/lib/utils/ml_util.dart
View File

@@ -73,6 +73,7 @@ Future<List<FileMLInstruction>> getFilesForMlIndexing() async {
await MLDataDB.instance.getIndexedFileIds();
final Map<int, int> clipIndexedFileIDs =
await MLDataDB.instance.clipIndexedFileWithVersion();
final Set<int> queuedFiledIDs = {};
// Get all regular files and all hidden files
final enteFiles = await SearchService.instance.getAllFiles();
@@ -86,6 +87,11 @@ Future<List<FileMLInstruction>> getFilesForMlIndexing() async {
if (_skipAnalysisEnteFile(enteFile)) {
continue;
}
if (queuedFiledIDs.contains(enteFile.uploadedFileID)) {
continue;
}
queuedFiledIDs.add(enteFile.uploadedFileID!);
final shouldRunFaces =
_shouldRunIndexing(enteFile, faceIndexedFileIDs, faceMlVersion);
final shouldRunClip =
@@ -108,6 +114,10 @@ Future<List<FileMLInstruction>> getFilesForMlIndexing() async {
if (_skipAnalysisEnteFile(enteFile)) {
continue;
}
if (queuedFiledIDs.contains(enteFile.uploadedFileID)) {
continue;
}
queuedFiledIDs.add(enteFile.uploadedFileID!);
final shouldRunFaces =
_shouldRunIndexing(enteFile, faceIndexedFileIDs, faceMlVersion);
final shouldRunClip =