private EntityType getEntityTypeForQuery(DataFetchingEnvironment graphEnv) { String entityName = graphEnv.getExecutionStepInfo().getType().getName(); if (entityName == null) { entityName = graphEnv.getExecutionStepInfo().getType().getChildren().get(0).getName(); } EntityType et = env.getDefinitionsSet().getFirstEntityTypeByInterfaceName(namespace + ".model." + entityName); requireNonNull(et, "EntityType must exist"); return et; }