movies and everything union

andimarek · andimarek · commit d33fe9db78ef · 2019-12-05T13:24:36.000+11:00
diff --git a/book-details/src/main/java/com/graphqljava/tutorial/bookdetails/GraphQLDataFetchers.java b/book-details/src/main/java/com/graphqljava/tutorial/bookdetails/GraphQLDataFetchers.java
@@ -2,8 +2,11 @@
 
 import com.google.common.collect.ImmutableMap;
 import graphql.schema.DataFetcher;
+import graphql.schema.GraphQLObjectType;
+import graphql.schema.TypeResolver;
 import org.springframework.stereotype.Component;
 
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
@@ -26,6 +29,11 @@ public class GraphQLDataFetchers {
                     "authorId", "author-3")
     );
 
+    private static List<Map<String, String>> movies = Arrays.asList(
+            ImmutableMap.of("id", "movie-1",
+                    "name", "Lost in translation"
+            ));
+
     private static List<Map<String, String>> authors = Arrays.asList(
             ImmutableMap.of("id", "author-1",
                     "firstName", "Joanne",
@@ -61,9 +69,47 @@ public DataFetcher getAuthorDataFetcher() {
         };
     }
 
+    public DataFetcher getBookTitleDataFetcher() {
+        return dataFetchingEnvironment -> {
+            Map<String, String> book = dataFetchingEnvironment.getSource();
+            return book.get("name");
+        };
+    }
+
     public DataFetcher helloDataFetcher() {
         return dataFetchingEnvironment -> {
             return "world";
         };
     }
+
+    public DataFetcher getMovieByIdDataFetcher() {
+        return dataFetchingEnvironment -> {
+            String movieId = dataFetchingEnvironment.getArgument("id");
+            return movies
+                    .stream()
+                    .filter(book -> book.get("id").equals(movieId))
+                    .findFirst()
+                    .orElse(null);
+        };
+    }
+
+    public DataFetcher everythingDataFetcher() {
+        return env -> {
+            List<Object> result = new ArrayList<>();
+            result.addAll(movies);
+            result.addAll(books);
+            return result;
+        };
+    }
+
+    public TypeResolver everythingTypeResolver = env -> {
+        Map<String, Object> bookOrMovie = env.getObject();
+        String id = (String) bookOrMovie.get("id");
+        if (id.startsWith("movie")) {
+            return (GraphQLObjectType) env.getSchema().getType("Movie");
+        } else {
+            return (GraphQLObjectType) env.getSchema().getType("Book");
+        }
+
+    };
 }
diff --git a/book-details/src/main/java/com/graphqljava/tutorial/bookdetails/GraphQLProvider.java b/book-details/src/main/java/com/graphqljava/tutorial/bookdetails/GraphQLProvider.java
@@ -46,10 +46,15 @@ private RuntimeWiring buildWiring() {
         return RuntimeWiring.newRuntimeWiring()
                 .type(newTypeWiring("Query")
                         .dataFetcher("bookById", graphQLDataFetchers.getBookByIdDataFetcher())
+                        .dataFetcher("movieById", graphQLDataFetchers.getMovieByIdDataFetcher())
                         .dataFetcher("hello", graphQLDataFetchers.helloDataFetcher())
+                        .dataFetcher("everything", graphQLDataFetchers.everythingDataFetcher())
                 )
                 .type(newTypeWiring("Book")
-                        .dataFetcher("author", graphQLDataFetchers.getAuthorDataFetcher()))
+                        .dataFetcher("author", graphQLDataFetchers.getAuthorDataFetcher())
+                        .dataFetcher("title", graphQLDataFetchers.getBookTitleDataFetcher()))
+                .type(newTypeWiring("Thing")
+                        .typeResolver(graphQLDataFetchers.everythingTypeResolver))
                 .build();
     }
 
diff --git a/book-details/src/main/resources/schema.graphqls b/book-details/src/main/resources/schema.graphqls
@@ -1,11 +1,20 @@
 type Query {
     hello: String
     bookById(id: ID): Book
+    movieById(id: ID): Movie
+    everything: [Thing]
 }
 
-type Book {
+union Thing = Movie | Book
+
+type Movie {
     id: ID
     name: String
+}
+
+type Book {
+    id: ID
+    title: String
     pageCount: Int
     author: Author
 }
