|
|
@@ -1,5 +1,6 @@
|
|
|
package com.remy.graphql.builder;
|
|
|
|
|
|
+import com.remy.common.module.CommentDTO;
|
|
|
import com.remy.common.module.GoodsDTO;
|
|
|
import com.remy.common.module.OrderDTO;
|
|
|
import com.remy.common.module.UserDTO;
|
|
|
@@ -62,6 +63,7 @@ public class GraphqlContextBuilder implements GraphQLServletContextBuilder {
|
|
|
dataLoaderRegistry.register("orderGoodsDataLoader", orderGoodsDataLoader());
|
|
|
dataLoaderRegistry.register("orderUserDataLoader", orderUserDataLoader());
|
|
|
dataLoaderRegistry.register("userAttentionsDataLoader", userAttentionsDataLoader());
|
|
|
+ dataLoaderRegistry.register("orderCommentsDataLoader", orderCommentsDataLoader());
|
|
|
return dataLoaderRegistry;
|
|
|
}
|
|
|
|
|
|
@@ -127,4 +129,14 @@ public class GraphqlContextBuilder implements GraphQLServletContextBuilder {
|
|
|
}, StringUtils.join(attentionIdList, ",")).getBody())
|
|
|
.collect(Collectors.toList())));
|
|
|
}
|
|
|
+
|
|
|
+ private DataLoader<String, CommentDTO> orderCommentsDataLoader() {
|
|
|
+ return new DataLoader<>(
|
|
|
+ orderNos -> supplyAsync(
|
|
|
+ () -> orderNos.stream().map(orderNo ->
|
|
|
+ new RestTemplate().getForObject(apiEnv.getCommentsEndpoint() + orderNo, CommentDTO.class))
|
|
|
+ .collect(Collectors.toList())
|
|
|
+ )
|
|
|
+ );
|
|
|
+ }
|
|
|
}
|