java - 使用 graphql-java-kickstart 库的 GraphQL 和数据加载器
问题描述
我正在尝试使用 graphql-java-kickstart 库中的 DataLoader 功能:
https://github.com/graphql-java-kickstart
我的应用程序是使用 2.3.0.RELEASE 的 Spring Boot 应用程序。我使用的是 graphql-spring-boot-starter 库的 7.0.1 版。
该库非常易于使用,当我不使用数据加载器时它可以工作。但是,我受到 N+1 SQL 问题的困扰,因此需要使用数据加载器来帮助缓解这个问题。当我执行一个请求时,我最终得到这个:
Can't resolve value (/findAccountById[0]/customers) : type mismatch error, expected type LIST got class com.daluga.api.account.domain.Customer
我确定我在配置中遗漏了一些东西,但真的不知道那是什么。
这是我的 graphql 架构:
type Account {
id: ID!
accountNumber: String!
customers: [Customer]
}
type Customer {
id: ID!
fullName: String
}
我创建了一个 CustomGraphQLContextBuilder:
@Component
public class CustomGraphQLContextBuilder implements GraphQLServletContextBuilder {
private final CustomerRepository customerRepository;
public CustomGraphQLContextBuilder(CustomerRepository customerRepository) {
this.customerRepository = customerRepository;
}
@Override
public GraphQLContext build(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse) {
return DefaultGraphQLServletContext.createServletContext(buildDataLoaderRegistry(), null).with(httpServletRequest).with(httpServletResponse).build();
}
@Override
public GraphQLContext build(Session session, HandshakeRequest handshakeRequest) {
return DefaultGraphQLWebSocketContext.createWebSocketContext(buildDataLoaderRegistry(), null).with(session).with(handshakeRequest).build();
}
@Override
public GraphQLContext build() {
return new DefaultGraphQLContext(buildDataLoaderRegistry(), null);
}
private DataLoaderRegistry buildDataLoaderRegistry() {
DataLoaderRegistry dataLoaderRegistry = new DataLoaderRegistry();
dataLoaderRegistry.register("customerDataLoader",
new DataLoader<Long, Customer>(accountIds ->
CompletableFuture.supplyAsync(() ->
customerRepository.findCustomersByAccountIds(accountIds), new SyncTaskExecutor())));
return dataLoaderRegistry;
}
}
我还创建了一个 AccountResolver:
public CompletableFuture<List<Customer>> customers(Account account, DataFetchingEnvironment dfe) {
final DataLoader<Long, List<Customer>> dataloader = ((GraphQLContext) dfe.getContext())
.getDataLoaderRegistry().get()
.getDataLoader("customerDataLoader");
return dataloader.load(account.getId());
}
这是客户存储库:
public List<Customer> findCustomersByAccountIds(List<Long> accountIds) {
Instant begin = Instant.now();
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
String inClause = getInClauseParamFromList(accountIds, namedParameters);
String sql = StringUtils.replace(SQL_FIND_CUSTOMERS_BY_ACCOUNT_IDS,"__ACCOUNT_IDS__", inClause);
List<Customer> customers = jdbcTemplate.query(sql, namedParameters, new CustomerRowMapper());
Instant end = Instant.now();
LOGGER.info("Total Time in Millis to Execute findCustomersByAccountIds: " + Duration.between(begin, end).toMillis());
return customers;
}
我可以在 Customer Repository 中放置一个断点,然后查看 SQL 执行并返回一个 Customer 对象列表。您还可以看到架构需要一组客户。如果我删除上面的代码并放入解析器以逐个获取客户......它可以工作......但真的很慢。
我在配置中缺少什么会导致这种情况?
Can't resolve value (/findAccountById[0]/customers) : type mismatch error, expected type LIST got class com.daluga.api.account.domain.Customer
谢谢你的帮助!
担
解决方案
谢谢,@Bms bharadwaj!问题在于我理解数据如何在数据加载器中返回。我最终使用 MappedBatchLoader 将数据放入地图中。地图中的键是 accountId。
private DataLoader<Long, List<Customer>> getCustomerDataLoader() {
MappedBatchLoader<Long, List<Customer>> customerMappedBatchLoader = accountIds -> CompletableFuture.supplyAsync(() -> {
List<Customer> customers = customerRepository.findCustomersByAccountId(accountIds);
Map<Long, List<Customer>> groupByAccountId = customers.stream().collect(Collectors.groupingBy(cust -> cust.getAccountId()));
return groupByAaccountId;
});
// }, new SyncTaskExecutor());
return DataLoader.newMappedDataLoader(customerMappedBatchLoader);
}
这似乎成功了,因为在我发出数百条 SQL 语句之前,现在减少到 2 条(一条用于驱动程序 SQL...帐户,一条用于客户)。
推荐阅读
- html - 为什么我的锚中不显示 Font Awesome 图标?
- rest - 具有相同 REST 资源的多种身份验证方法
- django-rest-framework - 创建具有两级父表的django rest框架ViewSet
- javascript - 检查来自firebase的子字符串并检索数据
- json - JQ转换为数字,从shell变量生成新json时转换为布尔值
- android - 带有图标和微调器的 Android Recycler+CardView - 排序问题
- c# - Xamarin Forms 4.11.0 错误:未找到 UWP 的属性、可绑定属性或事件
- c# - WPF 文件关联相对路径
- ios - 是否可以重新初始化静态库?
- css - 使用 QSS 设置 QAbstractSpinBox 样式时,从 QColorDialog 中的 QSpinBoxes 中排除样式