extendSchema: Use consistent naming + inline 'getExtendedType' func

Author: IvanGoncharov

src/utilities/extendSchema.js

@@ -185,7 +185,7 @@ export function extendSchema(
       const typeName = typeRef.name.value;
       const existingType = schema.getType(typeName);
       if (existingType) {
-        return getExtendedType(existingType);
+        return extendNamedType(existingType);
       }
 
       throw new GraphQLError(
@@ -203,12 +203,12 @@ export function extendSchema(
   const existingMutationType = schema.getMutationType();
   const existingSubscriptionType = schema.getSubscriptionType();
   const operationTypes = {
-    query: existingQueryType ? getExtendedType(existingQueryType) : null,
+    query: existingQueryType ? extendNamedType(existingQueryType) : null,
     mutation: existingMutationType
-      ? getExtendedType(existingMutationType)
+      ? extendNamedType(existingMutationType)
       : null,
     subscription: existingSubscriptionType
-      ? getExtendedType(existingSubscriptionType)
+      ? extendNamedType(existingSubscriptionType)
       : null,
   };
 
@@ -228,7 +228,7 @@ export function extendSchema(
   const types = [
     // Iterate through all types, getting the type definition for each, ensuring
     // that any type not directly referenced by a field will get created.
-    ...objectValues(schema.getTypeMap()).map(type => getExtendedType(type)),
+    ...objectValues(schema.getTypeMap()).map(type => extendNamedType(type)),
     // Do the same with new types.
     ...astBuilder.buildTypes(objectValues(typeDefinitionMap)),
   ];
@@ -265,30 +265,25 @@ export function extendSchema(
     );
   }
 
-  function getExtendedType<T: GraphQLNamedType>(type: T): T {
-    if (!extendTypeCache[type.name]) {
-      extendTypeCache[type.name] = extendType(type);
-    }
-    return (extendTypeCache[type.name]: any);
-  }
-
-  // To be called at most once per type. Only getExtendedType should call this.
-  function extendType(type) {
+  function extendNamedType<T: GraphQLNamedType>(type: T): T {
     if (isIntrospectionType(type)) {
       // Introspection types are not extended.
       return type;
     }
-    if (isObjectType(type)) {
-      return extendObjectType(type);
-    }
-    if (isInterfaceType(type)) {
-      return extendInterfaceType(type);
-    }
-    if (isUnionType(type)) {
-      return extendUnionType(type);
+
+    if (!extendTypeCache[type.name]) {
+      if (isObjectType(type)) {
+        extendTypeCache[type.name] = extendObjectType(type);
+      } else if (isInterfaceType(type)) {
+        extendTypeCache[type.name] = extendInterfaceType(type);
+      } else if (isUnionType(type)) {
+        extendTypeCache[type.name] = extendUnionType(type);
+      } else {
+        // This type is not yet extendable.
+        extendTypeCache[type.name] = type;
+      }
     }
-    // This type is not yet extendable.
-    return type;
+    return (extendTypeCache[type.name]: any);
   }
 
   function extendObjectType(type: GraphQLObjectType): GraphQLObjectType {
@@ -332,7 +327,7 @@ export function extendSchema(
     return new GraphQLUnionType({
       name: type.name,
       description: type.description,
-      types: type.getTypes().map(getExtendedType),
+      types: type.getTypes().map(extendNamedType),
       astNode: type.astNode,
       resolveType: type.resolveType,
     });
@@ -341,7 +336,7 @@ export function extendSchema(
   function extendImplementedInterfaces(
     type: GraphQLObjectType,
   ): Array<GraphQLInterfaceType> {
-    const interfaces = type.getInterfaces().map(getExtendedType);
+    const interfaces = type.getInterfaces().map(extendNamedType);
 
     // If there are any extensions to the interfaces, apply those here.
     const extensions = typeExtensionsMap[type.name];
@@ -367,7 +362,7 @@ export function extendSchema(
       newFieldMap[fieldName] = {
         description: field.description,
         deprecationReason: field.deprecationReason,
-        type: extendFieldType(field.type),
+        type: extendType(field.type),
         args: keyMap(field.args, arg => arg.name),
         astNode: field.astNode,
         resolve: field.resolve,
@@ -395,14 +390,14 @@ export function extendSchema(
     return newFieldMap;
   }
 
-  function extendFieldType<T: GraphQLType>(typeDef: T): T {
+  function extendType<T: GraphQLType>(typeDef: T): T {
     if (isListType(typeDef)) {
-      return (GraphQLList(extendFieldType(typeDef.ofType)): any);
+      return (GraphQLList(extendType(typeDef.ofType)): any);
     }
     if (isNonNullType(typeDef)) {
-      return (GraphQLNonNull(extendFieldType(typeDef.ofType)): any);
+      return (GraphQLNonNull(extendType(typeDef.ofType)): any);
     }
-    return getExtendedType(typeDef);
+    return extendNamedType(typeDef);
   }
 }