103 lines
3 KiB
Dart
103 lines
3 KiB
Dart
|
import 'dart:io';
|
||
|
|
|
||
|
|
import 'package:logging/logging.dart';
|
||
|
|
import 'package:path/path.dart';
|
||
|
|
import 'package:path_provider/path_provider.dart';
|
||
|
|
import 'package:sqflite/sqflite.dart';
|
||
|
|
|
||
|
|
// Keeps track of localIDs which should be not uploaded to ente without
|
||
|
|
// user's intervention.
|
||
|
|
// Common use case:
|
||
|
|
// when a user deletes a file just from ente on current or different device.
|
||
|
|
class FilesMigrationDB {
|
||
|
|
static final _databaseName = "ente.files_migration.db";
|
||
|
|
static final _databaseVersion = 1;
|
||
|
|
static final Logger _logger = Logger((FilesMigrationDB).toString());
|
||
|
|
static final tableName = 're_upload_tracker';
|
||
|
|
|
||
|
|
static final columnLocalID = 'local_id';
|
||
|
|
|
||
|
|
Future _onCreate(Database db, int version) async {
|
||
|
|
await db.execute('''
|
||
|
|
CREATE TABLE $tableName (
|
||
|
|
$columnLocalID TEXT NOT NULL,
|
||
|
|
UNIQUE($columnLocalID)
|
||
|
|
);
|
||
|
|
''');
|
||
|
|
}
|
||
|
|
|
||
|
|
FilesMigrationDB._privateConstructor();
|
||
|
|
|
||
|
|
static final FilesMigrationDB instance =
|
||
|
|
FilesMigrationDB._privateConstructor();
|
||
|
|
|
||
|
|
// only have a single app-wide reference to the database
|
||
|
|
static Future<Database> _dbFuture;
|
||
|
|
|
||
|
|
Future<Database> get database async {
|
||
|
|
// lazily instantiate the db the first time it is accessed
|
||
|
|
_dbFuture ??= _initDatabase();
|
||
|
|
return _dbFuture;
|
||
|
|
}
|
||
|
|
|
||
|
|
// this opens the database (and creates it if it doesn't exist)
|
||
|
|
Future<Database> _initDatabase() async {
|
||
|
|
Directory documentsDirectory = await getApplicationDocumentsDirectory();
|
||
|
|
String path = join(documentsDirectory.path, _databaseName);
|
||
|
|
return await openDatabase(
|
||
|
|
path,
|
||
|
|
version: _databaseVersion,
|
||
|
|
onCreate: _onCreate,
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
Future<void> clearTable() async {
|
||
|
|
final db = await instance.database;
|
||
|
|
await db.delete(tableName);
|
||
|
|
}
|
||
|
|
|
||
|
|
Future<void> insertMultiple(List<String> fileLocalIDs) async {
|
||
|
|
final startTime = DateTime.now();
|
||
|
|
final db = await instance.database;
|
||
|
|
var batch = db.batch();
|
||
|
|
int batchCounter = 0;
|
||
|
|
for (String localID in fileLocalIDs) {
|
||
|
|
if (batchCounter == 400) {
|
||
|
|
await batch.commit(noResult: true);
|
||
|
|
batch = db.batch();
|
||
|
|
batchCounter = 0;
|
||
|
|
}
|
||
|
|
batch.insert(
|
||
|
|
tableName,
|
||
|
|
_getRowForReUploadTable(localID),
|
||
|
|
conflictAlgorithm: ConflictAlgorithm.replace,
|
||
|
|
);
|
||
|
|
batchCounter++;
|
||
|
|
}
|
||
|
|
await batch.commit(noResult: true);
|
||
|
|
final endTime = DateTime.now();
|
||
|
|
final duration = Duration(
|
||
|
|
microseconds:
|
||
|
|
endTime.microsecondsSinceEpoch - startTime.microsecondsSinceEpoch);
|
||
|
|
_logger.info("Batch insert of ${fileLocalIDs.length} "
|
||
|
|
"took ${duration.inMilliseconds} ms.");
|
||
|
|
}
|
||
|
|
|
||
|
|
Future<List<String>> getLocalIDsForReUpload() async {
|
||
|
|
final db = await instance.database;
|
||
|
|
final rows = await db.query(tableName);
|
||
|
|
final result = <String>[];
|
||
|
|
for (final row in rows) {
|
||
|
|
result.add(row[columnLocalID]);
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
Map<String, dynamic> _getRowForReUploadTable(String localID) {
|
||
|
|
assert(localID != null);
|
||
|
|
final row = <String, dynamic>{};
|
||
|
|
row[columnLocalID] = localID;
|
||
|
|
return row;
|
||
|
|
}
|
||
|
|
}
|