mirror of https://github.com/AMT-Cheif/drift.git
63 lines
1.5 KiB
Dart
63 lines
1.5 KiB
Dart
import 'dart:async';
|
|
import 'package:sally_flutter/sally_flutter.dart';
|
|
|
|
part 'database.g.dart';
|
|
|
|
@DataClassName('TodoEntry')
|
|
class Todos extends Table {
|
|
IntColumn get id => integer().autoIncrement()();
|
|
|
|
TextColumn get title => text().withLength(min: 4, max: 16).nullable()();
|
|
|
|
TextColumn get content => text()();
|
|
|
|
DateTimeColumn get targetDate => dateTime().nullable()();
|
|
|
|
IntColumn get category => integer().nullable()();
|
|
}
|
|
|
|
@DataClassName('Category')
|
|
class Categories extends Table {
|
|
IntColumn get id => integer().autoIncrement()();
|
|
|
|
TextColumn get description => text().named('desc')();
|
|
}
|
|
|
|
class CategoryWithCount {
|
|
final Category category;
|
|
final int count; // amount of entries in this category
|
|
|
|
CategoryWithCount(this.category, this.count);
|
|
}
|
|
|
|
@UseSally(tables: [Todos, Categories])
|
|
class Database extends _$Database {
|
|
Database()
|
|
: super(FlutterQueryExecutor.inDatabaseFolder(
|
|
path: 'db.sqlite', logStatements: true));
|
|
|
|
@override
|
|
int get schemaVersion => 1;
|
|
|
|
@override
|
|
MigrationStrategy get migration => MigrationStrategy(onCreate: (Migrator m) {
|
|
return m.createAllTables();
|
|
}, onUpgrade: (Migrator m, int from, int to) async {
|
|
if (from == 1) {
|
|
await m.addColumn(todos, todos.targetDate);
|
|
}
|
|
});
|
|
|
|
Stream<List<TodoEntry>> allEntries() {
|
|
return select(todos).watch();
|
|
}
|
|
|
|
Future addEntry(TodoEntry entry) {
|
|
return into(todos).insert(entry);
|
|
}
|
|
|
|
Future deleteEntry(TodoEntry entry) {
|
|
return (delete(todos)..where((t) => t.id.equals(entry.id))).go();
|
|
}
|
|
}
|