Merge pull request #1233 from commercetools/SUPPORT-36587/customer-address-custom-type-resolution

SUPPORT-36587: support custom type resolution in customer's address

Author: kafis

src/main/java/com/commercetools/sync/customers/utils/CustomerReferenceResolutionUtils.java

@@ -111,7 +111,7 @@ private static CustomerDraft mapToCustomerDraft(
         .dateOfBirth(customer.getDateOfBirth())
         .isEmailVerified(customer.getIsEmailVerified())
         .vatId(customer.getVatId())
-        .addresses(mapToAddressesDraft(customer.getAddresses()))
+        .addresses(mapToAddressesDraft(customer.getAddresses(), referenceIdToKeyCache))
         .defaultBillingAddress(
             getAddressIndex(customer.getAddresses(), customer.getDefaultBillingAddressId()))
         .billingAddresses(
@@ -143,7 +143,9 @@ private static CustomerGroupResourceIdentifier mapToCustomerGroupResourceIdentif
     return null;
   }
 
-  private static List<BaseAddress> mapToAddressesDraft(@Nonnull List<Address> addresses) {
+  private static List<BaseAddress> mapToAddressesDraft(
+      @Nonnull List<Address> addresses,
+      @Nonnull final ReferenceIdToKeyCache referenceIdToKeyCache) {
     if (addresses.isEmpty()) {
       return emptyList();
     }
@@ -178,8 +180,9 @@ private static List<BaseAddress> mapToAddressesDraft(@Nonnull List<Address> addr
                       .externalId(address.getExternalId());
 
               if (address.getCustom() != null) {
-                builder.custom(
-                    CustomFieldsDraftBuilder.of().fields(address.getCustom().getFields()).build());
+                CustomFieldsDraft draft =
+                    mapToCustomFieldsDraft(address.getCustom(), referenceIdToKeyCache);
+                builder.custom(draft);
               }
               return builder.build();
             })

src/main/java/com/commercetools/sync/customers/utils/CustomerTransformUtils.java

@@ -3,6 +3,7 @@
 import static java.util.stream.Collectors.toSet;
 
 import com.commercetools.api.client.ProjectApiRoot;
+import com.commercetools.api.models.common.Address;
 import com.commercetools.api.models.common.Reference;
 import com.commercetools.api.models.customer.Customer;
 import com.commercetools.api.models.customer.CustomerDraft;
@@ -84,6 +85,17 @@ private CompletableFuture<Void> transformCustomTypeReference(
               .map(Reference::getId)
               .collect(toSet());
 
+      final Set<String> setOfAddressCustomTypeIds =
+          customers.stream()
+              .flatMap(c -> c.getAddresses().stream())
+              .map(Address::getCustom)
+              .filter(Objects::nonNull)
+              .map(CustomFields::getType)
+              .map(Reference::getId)
+              .collect(toSet());
+
+      setOfTypeIds.addAll(setOfAddressCustomTypeIds);
+
       return fetchAndFillReferenceIdToKeyCache(setOfTypeIds, GraphQlQueryResource.TYPES);
     }