Check in a place where Entity might be nil now.

Khan · Mar 28, 2020 · 55e0f0d · 55e0f0d
1 parent 1ecd074
commit 55e0f0d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugin/federation/federation.go b/plugin/federation/federation.go
@@ -181,7 +181,9 @@ func (e *Entity) allFieldsAreExternal() bool {
 
 func (f *federation) GenerateCode(data *codegen.Data) error {
 	if len(f.Entities) > 0 {
-		data.Objects.ByName("Entity").Root = true
+		if data.Objects.ByName("Entity") != nil {
+			data.Objects.ByName("Entity").Root = true
+		}
 		for _, e := range f.Entities {
 			obj := data.Objects.ByName(e.Def.Name)
 			for _, field := range obj.Fields {