shenlong-tanwen
34 lines
1.1 KiB
Dart
34 lines
1.1 KiB
Dart
import 'package:drift/drift.dart';
|
|
import 'package:immich_mobile/domain/models/asset/base_asset.model.dart';
|
|
import 'package:immich_mobile/infrastructure/entities/local_asset.entity.dart';
|
|
import 'package:immich_mobile/infrastructure/entities/local_asset.entity.drift.dart';
|
|
import 'package:immich_mobile/infrastructure/repositories/db.repository.dart';
|
|
|
|
class DriftLocalAssetRepository extends DriftDatabaseRepository {
|
|
final Drift _db;
|
|
const DriftLocalAssetRepository(this._db) : super(_db);
|
|
|
|
Stream<LocalAsset?> watchAsset(String id) {
|
|
final query = _db.localAssetEntity.select()
|
|
..where((row) => row.id.equals(id));
|
|
|
|
return query.map((row) => row.toDto()).watchSingleOrNull();
|
|
}
|
|
|
|
Future<void> updateHashes(Iterable<LocalAsset> hashes) {
|
|
if (hashes.isEmpty) {
|
|
return Future.value();
|
|
}
|
|
|
|
return _db.batch((batch) async {
|
|
for (final asset in hashes) {
|
|
batch.update(
|
|
_db.localAssetEntity,
|
|
LocalAssetEntityCompanion(checksum: Value(asset.checksum)),
|
|
where: (e) => e.id.equals(asset.id),
|
|
);
|
|
}
|
|
});
|
|
}
|
|
}
|