AMT-Cheif
/
drift
mirror of https://github.com/AMT-Cheif/drift.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Support primaryKey override with inheritance (#467)

This commit is contained in:
Simon Binder 2020-04-03 22:02:01 +02:00
parent eac461486e
commit c692c56d43
No known key found for this signature in database
GPG Key ID: 7891917E4147B8C0
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
moor_generator/lib/src/analyzer/dart/table_parser.dart
View File

@ -74,7 +74,8 @@ class TableParser {
Future<Set<MoorColumn>> _readPrimaryKey(
ClassElement element, List<MoorColumn> columns) async {
final primaryKeyGetter = element.getGetter('primaryKey');
final primaryKeyGetter =
element.lookUpGetter('primaryKey', element.library);
if (primaryKeyGetter == null) {
return null;
}

6
moor_generator/test/analyzer/dart/table_parser_test.dart
View File

@ -58,7 +58,10 @@ void main() {
class Socks extends Table {
TextColumn get name => text()();
IntColumn get id => integer().autoIncrement()();
IntColumn get id => integer()();
@override
Set<Column> get primaryKey => {id};
}
class ArchivedSocks extends Socks {
@ -209,6 +212,7 @@ void main() {
expect(archivedSocks.columns, hasLength(4));
expect(archivedSocks.columns.map((c) => c.name.name),
['name', 'id', 'archived_by', 'archived_on']);
expect(archivedSocks.primaryKey.map((e) => e.name.name), ['id']);
});
});
}