@@ -57,40 +57,41 @@ public Document createSchemaDefinition(Map<String, Object> introspectionResult)
5757
5858 Map <String , Object > queryType = (Map <String , Object >) schema .get ("queryType" );
5959 assertNotNull (queryType , "queryType expected" );
60- TypeName query = new TypeName (( String ) queryType .get ("name" ));
60+ TypeName query = TypeName . newTypeName (). name (( String ) queryType .get ("name" )). build ( );
6161 boolean nonDefaultQueryName = !"Query" .equals (query .getName ());
6262
63- SchemaDefinition schemaDefinition = SchemaDefinition .newSchemaDefinition ().operationTypeDefinition (new OperationTypeDefinition ("query" , query )).build ();
63+ SchemaDefinition .Builder schemaDefinition = SchemaDefinition .newSchemaDefinition ();
64+ schemaDefinition .operationTypeDefinition (OperationTypeDefinition .newOperationTypeDefinition ().name ("query" ).typeName (query ).build ());
6465
6566 Map <String , Object > mutationType = (Map <String , Object >) schema .get ("mutationType" );
6667 boolean nonDefaultMutationName = false ;
6768 if (mutationType != null ) {
68- TypeName mutation = new TypeName (( String ) mutationType .get ("name" ));
69+ TypeName mutation = TypeName . newTypeName (). name (( String ) mutationType .get ("name" )). build ( );
6970 nonDefaultMutationName = !"Mutation" .equals (mutation .getName ());
70- schemaDefinition .getOperationTypeDefinitions (). add ( new OperationTypeDefinition ( "mutation" , mutation ));
71+ schemaDefinition .operationTypeDefinition ( OperationTypeDefinition . newOperationTypeDefinition (). name ( "mutation" ). typeName ( mutation ). build ( ));
7172 }
7273
7374 Map <String , Object > subscriptionType = (Map <String , Object >) schema .get ("subscriptionType" );
7475 boolean nonDefaultSubscriptionName = false ;
7576 if (subscriptionType != null ) {
76- TypeName subscription = new TypeName (( String ) subscriptionType .get ("name" ));
77+ TypeName subscription = TypeName . newTypeName (). name ((( String ) subscriptionType .get ("name" ))). build ( );
7778 nonDefaultSubscriptionName = !"Subscription" .equals (subscription .getName ());
78- schemaDefinition .getOperationTypeDefinitions (). add ( new OperationTypeDefinition ( "subscription" , subscription ));
79+ schemaDefinition .operationTypeDefinition ( OperationTypeDefinition . newOperationTypeDefinition (). name ( "subscription" ). typeName ( subscription ). build ( ));
7980 }
8081
81- Document document = new Document ( Lists . newArrayList () );
82+ Document . Builder document = Document . newDocument ( );
8283 if (nonDefaultQueryName || nonDefaultMutationName || nonDefaultSubscriptionName ) {
83- document .getDefinitions (). add ( schemaDefinition );
84+ document .definition ( schemaDefinition . build () );
8485 }
8586
8687 List <Map <String , Object >> types = (List <Map <String , Object >>) schema .get ("types" );
8788 for (Map <String , Object > type : types ) {
8889 TypeDefinition typeDefinition = createTypeDefinition (type );
8990 if (typeDefinition == null ) continue ;
90- document .getDefinitions (). add (typeDefinition );
91+ document .definition (typeDefinition );
9192 }
9293
93- return document ;
94+ return document . build () ;
9495 }
9596
9697 private TypeDefinition createTypeDefinition (Map <String , Object > type ) {
@@ -272,11 +273,11 @@ private Type createTypeIndirection(Map<String, Object> type) {
272273 case "ENUM" :
273274 case "INPUT_OBJECT" :
274275 case "SCALAR" :
275- return new TypeName (( String ) type .get ("name" ));
276+ return TypeName . newTypeName (). name (( String ) type .get ("name" )). build ( );
276277 case "NON_NULL" :
277- return new NonNullType ( createTypeIndirection ((Map <String , Object >) type .get ("ofType" )));
278+ return NonNullType . newNonNullType (). type ( createTypeIndirection ((Map <String , Object >) type .get ("ofType" ))). build ( );
278279 case "LIST" :
279- return new ListType ( createTypeIndirection ((Map <String , Object >) type .get ("ofType" )));
280+ return ListType . newListType (). type ( createTypeIndirection ((Map <String , Object >) type .get ("ofType" ))). build ( );
280281 default :
281282 return assertShouldNeverHappen ("Unknown kind %s" , kind );
282283 }
0 commit comments