Return empty list of empty list query result (#32)

pinchbv · Feb 8, 2019 · 33b2cdb · 33b2cdb
1 parent ecbca12
commit 33b2cdb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/floor_generator/lib/writer/query_method_writer.dart b/floor_generator/lib/writer/query_method_writer.dart
@@ -54,6 +54,9 @@ class QueryMethodWriter implements Writer {
       return 'return rows.map((row) => $constructorCall).toList();';
     } else {
       return '''
+      if (rows.isEmpty) {
+        return null;
+      }
       final row = rows.first;
       return $constructorCall;
       ''';
@@ -87,9 +90,6 @@ class QueryMethodWriter implements Writer {
 
     return '''
     final rows = await database.rawQuery('${queryMethod.query}');
-    if (rows.isEmpty) {
-      return null;
-    }
     $mapping
     ''';
   }