DataFetcher to get a Company from a Map

CompanyQuery.java

public class CompanyQuery implements DataFetcher<Company> {
    private final Map<String, Company> companies;

    public CompanyFetcher(Map<String, Company> companies) {
        this.companies = companies;
    }

    @Override
    public Company get(DataFetchingEnvironment environment) {
        return companies.get(environment.getArgument("name"));
    }
}