Skip to content

Commit 90d7166

Browse files
committed
fix: replace deprecated dataloader constructor
1 parent b6d24bc commit 90d7166

File tree

1 file changed

+14
-5
lines changed

1 file changed

+14
-5
lines changed

request-scoped-dataloader/src/main/java/graphql/servlet/examples/dataloader/requestscope/CustomGraphQLContextBuilder.java

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,14 @@
44

55
import graphql.kickstart.execution.context.GraphQLKickstartContext;
66
import graphql.kickstart.servlet.context.GraphQLServletContextBuilder;
7+
import java.util.HashMap;
8+
import java.util.Map;
79
import javax.servlet.http.HttpServletRequest;
810
import javax.servlet.http.HttpServletResponse;
911
import javax.websocket.Session;
1012
import javax.websocket.server.HandshakeRequest;
1113
import org.dataloader.DataLoader;
14+
import org.dataloader.DataLoaderFactory;
1215
import org.dataloader.DataLoaderRegistry;
1316
import org.springframework.stereotype.Component;
1417

@@ -22,8 +25,11 @@ public CustomGraphQLContextBuilder(CustomerRepository customerRepository) {
2225
}
2326

2427
@Override
25-
public GraphQLKickstartContext build(HttpServletRequest req, HttpServletResponse response) {
26-
return GraphQLKickstartContext.of(buildDataLoaderRegistry());
28+
public GraphQLKickstartContext build(HttpServletRequest request, HttpServletResponse response) {
29+
Map<Object, Object> map = new HashMap<>();
30+
map.put(HttpServletRequest.class, request);
31+
map.put(HttpServletResponse.class, response);
32+
return GraphQLKickstartContext.of(buildDataLoaderRegistry(), map);
2733
}
2834

2935
@Override
@@ -32,14 +38,17 @@ public GraphQLKickstartContext build() {
3238
}
3339

3440
@Override
35-
public GraphQLKickstartContext build(Session session, HandshakeRequest request) {
36-
return GraphQLKickstartContext.of(buildDataLoaderRegistry());
41+
public GraphQLKickstartContext build(Session session, HandshakeRequest handshakeRequest) {
42+
Map<Object, Object> map = new HashMap<>();
43+
map.put(Session.class, session);
44+
map.put(HandshakeRequest.class, handshakeRequest);
45+
return GraphQLKickstartContext.of(buildDataLoaderRegistry(), map);
3746
}
3847

3948
private DataLoaderRegistry buildDataLoaderRegistry() {
4049
DataLoaderRegistry dataLoaderRegistry = new DataLoaderRegistry();
4150
DataLoader<Integer, String> customerLoader =
42-
new DataLoader<>(
51+
DataLoaderFactory.newDataLoader(
4352
customerIds -> supplyAsync(() -> customerRepository.getUserNamesForIds(customerIds)));
4453
dataLoaderRegistry.register("customerDataLoader", customerLoader);
4554
return dataLoaderRegistry;

0 commit comments

Comments
 (0)