Fix getEmbeds()

pinchbv · May 24, 2024 · 0a3c42c · 0a3c42c
1 parent d25e530
commit 0a3c42c
Showing 1 changed file with 13 additions and 18 deletions.
diff --git a/floor_generator/lib/processor/database_processor.dart b/floor_generator/lib/processor/database_processor.dart
@@ -110,24 +110,19 @@ class DatabaseProcessor extends Processor<Database> {
     final ClassElement databaseClassElement,
     final Set<TypeConverter> typeConverters,
   ) {
-    final entities = _classElement
-        .getAnnotation(annotations.Database)
-        ?.getField(AnnotationField.databaseEmbeds)
-        ?.toListValue()
-        ?.mapNotNull((object) => object.toTypeValue()?.element)
-        .whereType<ClassElement>()
-        .where(_isEmbed)
-        .map((classElement) => EmbedProcessor(
-              classElement,
-              typeConverters,
-            ).process())
-        .toSet();
-
-    if (entities == null || entities.isEmpty) {
-      throw _processorError.noEntitiesDefined;
-    }
-
-    return entities;
+    return _classElement
+            .getAnnotation(annotations.Database)
+            ?.getField(AnnotationField.databaseEmbeds)
+            ?.toListValue()
+            ?.mapNotNull((object) => object.toTypeValue()?.element)
+            .whereType<ClassElement>()
+            .where(_isEmbed)
+            .map((classElement) => EmbedProcessor(
+                  classElement,
+                  typeConverters,
+                ).process())
+            .toSet() ??
+        {};
   }
 
   List<Entity> _getEntities(