com.commercetools.sync.services.impl.CategoryServiceImpl Maven / Gradle / Ivy

Go to download
package com.commercetools.sync.services.impl;

import static com.commercetools.sync.commons.utils.SyncUtils.batchElements;

import com.commercetools.api.client.ByProjectKeyCategoriesGet;
import com.commercetools.api.client.ByProjectKeyCategoriesKeyByKeyGet;
import com.commercetools.api.client.ByProjectKeyCategoriesPost;
import com.commercetools.api.models.category.Category;
import com.commercetools.api.models.category.CategoryDraft;
import com.commercetools.api.models.category.CategoryPagedQueryResponse;
import com.commercetools.api.models.category.CategoryUpdateAction;
import com.commercetools.api.models.category.CategoryUpdateBuilder;
import com.commercetools.api.predicates.query.category.CategoryQueryBuilderDsl;
import com.commercetools.sync.categories.CategorySyncOptions;
import com.commercetools.sync.commons.models.GraphQlQueryResource;
import com.commercetools.sync.services.CategoryService;
import io.vrap.rmf.base.client.ApiHttpResponse;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import java.util.function.Function;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

/** Implementation of CategoryService interface. */
public final class CategoryServiceImpl
    extends BaseService<
        CategorySyncOptions,
        Category,
        CategoryDraft,
        ByProjectKeyCategoriesGet,
        CategoryPagedQueryResponse,
        ByProjectKeyCategoriesKeyByKeyGet,
        Category,
        CategoryQueryBuilderDsl,
        ByProjectKeyCategoriesPost>
    implements CategoryService {

  public CategoryServiceImpl(@Nonnull final CategorySyncOptions syncOptions) {
    super(syncOptions);
  }

  @Nonnull
  @Override
  public CompletionStage> cacheKeysToIds(
      @Nonnull final Set categoryKeys) {
    return super.cacheKeysToIdsUsingGraphQl(categoryKeys, GraphQlQueryResource.CATEGORIES);
  }

  @Nonnull
  @Override
  public CompletionStage> fetchMatchingCategoriesByKeys(
      @Nonnull final Set keys) {
    return fetchMatchingResources(
        keys,
        categories -> categories.getKey(),
        (keysNotCached) ->
            syncOptions
                .getCtpClient()
                .categories()
                .get()
                .withWhere("key in :keys")
                .withPredicateVar("keys", keysNotCached));
  }

  @Nonnull
  public CompletionStage> fetchCategory(@Nullable final String key) {
    ByProjectKeyCategoriesKeyByKeyGet byProjectKeyCategoriesKeyByKeyGet =
        syncOptions.getCtpClient().categories().withKey(key).get();
    return super.fetchResource(key, byProjectKeyCategoriesKeyByKeyGet);
  }

  @Nonnull
  @Override
  public CompletionStage> fetchCachedCategoryId(@Nonnull final String key) {
    ByProjectKeyCategoriesGet query =
        syncOptions
            .getCtpClient()
            .categories()
            .get()
            .withWhere("key in :keys")
            .withPredicateVar("keys", Collections.singletonList(key));

    return this.fetchCachedResourceId(key, query);
  }

  @Nonnull
  CompletionStage> fetchCachedResourceId(
      @Nullable final String key, @Nonnull final ByProjectKeyCategoriesGet query) {
    return fetchCachedResourceId(key, resource -> resource.getKey(), query);
  }

  @Nonnull
  @Override
  public CompletionStage> createCategory(
      @Nonnull final CategoryDraft categoryDraft) {
    return super.createResource(
        categoryDraft,
        CategoryDraft::getKey,
        Category::getId,
        Function.identity(),
        () -> syncOptions.getCtpClient().categories().post(categoryDraft));
  }

  @Nonnull
  @Override
  public CompletionStage updateCategory(
      @Nonnull final Category category, @Nonnull final List updateActions) {
    final List> actionBatches =
        batchElements(updateActions, MAXIMUM_ALLOWED_UPDATE_ACTIONS);

    CompletionStage> resultStage =
        CompletableFuture.completedFuture(new ApiHttpResponse<>(200, null, category));

    for (final List batch : actionBatches) {
      resultStage =
          resultStage
              .thenApply(ApiHttpResponse::getBody)
              .thenCompose(
                  updatedCategory ->
                      syncOptions
                          .getCtpClient()
                          .categories()
                          .withId(updatedCategory.getId())
                          .post(
                              CategoryUpdateBuilder.of()
                                  .actions(batch)
                                  .version(updatedCategory.getVersion())
                                  .build())
                          .execute());
    }

    return resultStage.thenApply(ApiHttpResponse::getBody);
  }
}