beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

[mob][photos] Migrate to sqlite_async(11)

Browse source
This commit is contained in:
ashilkn 2024-05-16 19:34:59 +05:30
parent 16d54645bc
commit 2b0fa9bae6
1 changed files with 37 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

67
mobile/lib/db/device_files_db.dart
View file

@ -279,40 +279,47 @@ extension DeviceFiles on FilesDB {
return result; return result;
} }
Future<void> updateDevicePathSyncStatus(Map<String, bool> syncStatus) async { Future<void> updateDevicePathSyncStatus(
final db = await database; Map<String, bool> syncStatus,
var batch = db.batch(); ) async {
final db = await sqliteAsyncDB;
int batchCounter = 0; int batchCounter = 0;
final parameterSets = <List<Object?>>[];
for (MapEntry e in syncStatus.entries) { for (MapEntry e in syncStatus.entries) {
final String pathID = e.key; final String pathID = e.key;
parameterSets.add([e.value ? _sqlBoolTrue : _sqlBoolFalse, pathID]);
batchCounter++;
if (batchCounter == 400) { if (batchCounter == 400) {
await batch.commit(noResult: true); await db.executeBatch(
batch = db.batch(); '''
UPDATE device_collections SET should_backup = ? WHERE id = ?;
''',
parameterSets,
);
parameterSets.clear();
batchCounter = 0; batchCounter = 0;
} }
batch.update(
"device_collections",
{
"should_backup": e.value ? _sqlBoolTrue : _sqlBoolFalse,
},
where: 'id = ?',
whereArgs: [pathID],
);
batchCounter++;
} }
await batch.commit(noResult: true);
await db.executeBatch(
'''
UPDATE device_collections SET should_backup = ? WHERE id = ?;
''',
parameterSets,
);
} }
Future<void> updateDeviceCollection( Future<void> updateDeviceCollection(
String pathID, String pathID,
int collectionID, int collectionID,
) async { ) async {
final db = await database; final db = await sqliteAsyncDB;
await db.update( await db.execute(
"device_collections", '''
{"collection_id": collectionID}, UPDATE device_collections SET collection_id = ? WHERE id = ?;
where: 'id = ?', ''',
whereArgs: [pathID], [collectionID, pathID],
); );
return; return;
} }
@ -325,7 +332,7 @@ extension DeviceFiles on FilesDB {
int? limit, int? limit,
bool? asc, bool? asc,
}) async { }) async {
final db = await database; final db = await sqliteAsyncDB;
final order = (asc ?? false ? 'ASC' : 'DESC'); final order = (asc ?? false ? 'ASC' : 'DESC');
final String rawQuery = ''' final String rawQuery = '''
SELECT * SELECT *
@ -340,7 +347,7 @@ extension DeviceFiles on FilesDB {
ORDER BY ${FilesDB.columnCreationTime} $order , ${FilesDB.columnModificationTime} $order ORDER BY ${FilesDB.columnCreationTime} $order , ${FilesDB.columnModificationTime} $order
''' + ''' +
(limit != null ? ' limit $limit;' : ';'); (limit != null ? ' limit $limit;' : ';');
final results = await db.rawQuery(rawQuery); final results = await db.getAll(rawQuery);
final files = convertToFiles(results); final files = convertToFiles(results);
final dedupe = deduplicateByLocalID(files); final dedupe = deduplicateByLocalID(files);
return FileLoadResult(dedupe, files.length == limit); return FileLoadResult(dedupe, files.length == limit);
@ -350,7 +357,7 @@ extension DeviceFiles on FilesDB {
String pathID, String pathID,
int ownerID, int ownerID,
) async { ) async {
final db = await database; final db = await sqliteAsyncDB;
const String rawQuery = ''' const String rawQuery = '''
SELECT ${FilesDB.columnLocalID}, ${FilesDB.columnUploadedFileID}, SELECT ${FilesDB.columnLocalID}, ${FilesDB.columnUploadedFileID},
${FilesDB.columnFileSize} ${FilesDB.columnFileSize}
@ -362,7 +369,7 @@ extension DeviceFiles on FilesDB {
${FilesDB.columnLocalID} IN ${FilesDB.columnLocalID} IN
(SELECT id FROM device_files where path_id = ?) (SELECT id FROM device_files where path_id = ?)
'''; ''';
final results = await db.rawQuery(rawQuery, [ownerID, pathID]); final results = await db.getAll(rawQuery, [ownerID, pathID]);
final localIDs = <String>{}; final localIDs = <String>{};
final uploadedIDs = <int>{}; final uploadedIDs = <int>{};
int localSize = 0; int localSize = 0;
@ -386,17 +393,17 @@ extension DeviceFiles on FilesDB {
"$includeCoverThumbnail", "$includeCoverThumbnail",
); );
try { try {
final db = await database; final db = await sqliteAsyncDB;
final coverFiles = <EnteFile>[]; final coverFiles = <EnteFile>[];
if (includeCoverThumbnail) { if (includeCoverThumbnail) {
final fileRows = await db.rawQuery( final fileRows = await db.getAll(
'''SELECT * FROM FILES where local_id in (select cover_id from device_collections) group by local_id; '''SELECT * FROM FILES where local_id in (select cover_id from device_collections) group by local_id;
''', ''',
); );
final files = convertToFiles(fileRows); final files = convertToFiles(fileRows);
coverFiles.addAll(files); coverFiles.addAll(files);
} }
final deviceCollectionRows = await db.rawQuery( final deviceCollectionRows = await db.getAll(
'''SELECT * from device_collections''', '''SELECT * from device_collections''',
); );
final List<DeviceCollection> deviceCollections = []; final List<DeviceCollection> deviceCollections = [];
@ -444,8 +451,8 @@ extension DeviceFiles on FilesDB {
Future<EnteFile?> getDeviceCollectionThumbnail(String pathID) async { Future<EnteFile?> getDeviceCollectionThumbnail(String pathID) async {
debugPrint("Call fallback method to get potential thumbnail"); debugPrint("Call fallback method to get potential thumbnail");
final db = await database; final db = await sqliteAsyncDB;
final fileRows = await db.rawQuery( final fileRows = await db.getAll(
'''SELECT * FROM FILES f JOIN device_files df on f.local_id = df.id '''SELECT * FROM FILES f JOIN device_files df on f.local_id = df.id
and df.path_id= ? order by f.creation_time DESC limit 1; and df.path_id= ? order by f.creation_time DESC limit 1;
''', ''',