11package soham .spring .graphqljava .datafetcher ;
22
3- import graphql .execution .DataFetcherResult ;
43import graphql .schema .DataFetcher ;
54import org .dataloader .DataLoader ;
65import org .springframework .beans .factory .annotation .Autowired ;
98import soham .spring .graphqljava .entity .Service ;
109import soham .spring .graphqljava .service .ServicesService ;
1110
11+ import java .util .List ;
12+ import java .util .concurrent .CompletableFuture ;
13+
1214@ Component
1315public class ServiceDataFetcher {
1416
1517 @ Autowired
1618 ServicesService servicesService ;
1719
1820
19- public DataFetcher getAllServices () {
20- return dataFetchingEnvironment -> {
21- return DataFetcherResult .newResult ().data (servicesService .findAll ()).build ();
22- };
21+ public DataFetcher <List <Service >> getAllServices () {
22+ return dataFetchingEnvironment -> servicesService .findAll ();
2323 }
2424
25- public DataFetcher getServiceById () {
25+ public DataFetcher < Service > getServiceById () {
2626 return dataFetchingEnvironment -> {
2727 String serviceId = dataFetchingEnvironment .getArgument ("id" );
28- return DataFetcherResult . newResult (). data ( servicesService .findById (serviceId )). build ( );
28+ return servicesService .findById (serviceId );
2929 };
3030 }
3131
32- public DataFetcher getProviderForService () {
32+ public DataFetcher < CompletableFuture < Provider >> getProviderForService () {
3333 return dataFetchingEnvironment -> {
3434 Service service = dataFetchingEnvironment .getSource ();
3535
@@ -38,20 +38,20 @@ public DataFetcher getProviderForService() {
3838 };
3939 }
4040
41- public DataFetcher getServicesForProvider () {
41+ public DataFetcher < List < Service >> getServicesForProvider () {
4242 return dataFetchingEnvironment -> {
4343 Provider provider = dataFetchingEnvironment .getSource ();
44- return DataFetcherResult . newResult (). data ( servicesService .findAllByProviderId (provider .getId ())). build ( );
44+ return servicesService .findAllByProviderId (provider .getId ());
4545 };
4646 }
4747
48- public DataFetcher addService () {
48+ public DataFetcher < Service > addService () {
4949 return dataFetchingEnvironment -> {
5050 String name = dataFetchingEnvironment .getArgument ("name" );
5151 String description = dataFetchingEnvironment .getArgument ("description" );
5252 String providerId = dataFetchingEnvironment .getArgument ("providerId" );
5353
54- return DataFetcherResult . newResult (). data ( servicesService .addService (name , description , providerId )). build ( );
54+ return servicesService .addService (name , description , providerId );
5555 };
5656 }
5757}
0 commit comments