fix null case

gql-dart · Feb 4, 2020 · 0c00314 · 0c00314
1 parent 551abfb
commit 0c00314
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gql_code_builder/lib/src/operation/data.dart b/gql_code_builder/lib/src/operation/data.dart
@@ -42,9 +42,9 @@ String _operationType(
   DocumentNode schema,
   OperationDefinitionNode op,
 ) {
-  final schemaDef = schema.definitions.whereType<SchemaDefinitionNode>().first;
-  if (schemaDef == null) return defaultRootTypes[op.type];
-  return schemaDef.operationTypes
+  final schemaDefs = schema.definitions.whereType<SchemaDefinitionNode>();
+  if (schemaDefs.isEmpty) return defaultRootTypes[op.type];
+  return schemaDefs.first.operationTypes
       .firstWhere((opType) => opType.operation == op.type)
       .type
       .name