Java tutorial
/* * Copyright 2014-2015 Groupon, Inc * Copyright 2014-2015 The Billing Project, LLC * * The Billing Project licenses this file to you under the Apache License, version 2.0 * (the "License"); you may not use this file except in compliance with the * License. You may obtain a copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the * License for the specific language governing permissions and limitations * under the License. */ package org.killbill.billing.catalog.override; import java.util.List; import java.util.concurrent.atomic.AtomicLong; import java.util.regex.Pattern; import javax.annotation.Nullable; import org.joda.time.DateTime; import org.killbill.billing.ErrorCode; import org.killbill.billing.callcontext.InternalCallContext; import org.killbill.billing.callcontext.InternalTenantContext; import org.killbill.billing.catalog.DefaultPlan; import org.killbill.billing.catalog.DefaultPlanPhase; import org.killbill.billing.catalog.DefaultPlanPhasePriceOverride; import org.killbill.billing.catalog.StandaloneCatalog; import org.killbill.billing.catalog.StandaloneCatalogWithPriceOverride; import org.killbill.billing.catalog.api.CatalogApiException; import org.killbill.billing.catalog.api.Plan; import org.killbill.billing.catalog.api.PlanPhase; import org.killbill.billing.catalog.api.PlanPhasePriceOverride; import org.killbill.billing.catalog.api.PlanPhaseSpecifier; import org.killbill.billing.catalog.api.StaticCatalog; import org.killbill.billing.catalog.caching.OverriddenPlanCache; import org.killbill.billing.catalog.dao.CatalogOverrideDao; import org.killbill.billing.catalog.dao.CatalogOverridePlanDefinitionModelDao; import com.google.common.base.Predicate; import com.google.common.collect.Iterables; import com.google.inject.Inject; public class DefaultPriceOverride implements PriceOverride { private static final AtomicLong DRY_RUN_PLAN_IDX = new AtomicLong(0); public static final Pattern CUSTOM_PLAN_NAME_PATTERN = Pattern.compile("(.*)-(\\d+)$"); private final CatalogOverrideDao overrideDao; private final OverriddenPlanCache overriddenPlanCache; @Inject public DefaultPriceOverride(final CatalogOverrideDao overrideDao, final OverriddenPlanCache overriddenPlanCache) { this.overrideDao = overrideDao; this.overriddenPlanCache = overriddenPlanCache; } @Override public DefaultPlan getOrCreateOverriddenPlan(final StandaloneCatalog standaloneCatalog, final Plan parentPlan, final DateTime catalogEffectiveDate, final List<PlanPhasePriceOverride> overrides, @Nullable final InternalCallContext context) throws CatalogApiException { final PlanPhasePriceOverride[] resolvedOverride = new PlanPhasePriceOverride[parentPlan .getAllPhases().length]; int index = 0; for (final PlanPhase curPhase : parentPlan.getAllPhases()) { final PlanPhasePriceOverride curOverride = Iterables .tryFind(overrides, new Predicate<PlanPhasePriceOverride>() { @Override public boolean apply(final PlanPhasePriceOverride input) { if (input.getPhaseName() != null) { return input.getPhaseName().equals(curPhase.getName()); } // If the phaseName was not passed, we infer by matching the phaseType. This obviously would not work in a case where // a plan is defined with multiple phases of the same type. final PlanPhaseSpecifier curPlanPhaseSpecifier = input.getPlanPhaseSpecifier(); if (curPlanPhaseSpecifier.getPhaseType().equals(curPhase.getPhaseType())) { return true; } return false; } }).orNull(); resolvedOverride[index++] = curOverride != null ? new DefaultPlanPhasePriceOverride(curPhase.getName(), curOverride.getCurrency(), curOverride.getFixedPrice(), curOverride.getRecurringPrice()) : null; } for (int i = 0; i < resolvedOverride.length; i++) { final PlanPhasePriceOverride curOverride = resolvedOverride[i]; if (curOverride != null) { final DefaultPlanPhase curPhase = (DefaultPlanPhase) parentPlan.getAllPhases()[i]; if (curPhase.getFixed() == null && curOverride.getFixedPrice() != null) { final String error = String.format("There is no existing fixed price for the phase %s", curPhase.getName()); throw new CatalogApiException(ErrorCode.CAT_INVALID_INVALID_PRICE_OVERRIDE, parentPlan.getName(), error); } if (curPhase.getRecurring() == null && curOverride.getRecurringPrice() != null) { final String error = String.format("There is no existing recurring price for the phase %s", curPhase.getName()); throw new CatalogApiException(ErrorCode.CAT_INVALID_INVALID_PRICE_OVERRIDE, parentPlan.getName(), error); } } } final String planName; if (context != null) { final CatalogOverridePlanDefinitionModelDao overriddenPlan = overrideDao .getOrCreateOverridePlanDefinition(parentPlan.getName(), catalogEffectiveDate, resolvedOverride, context); planName = new StringBuffer(parentPlan.getName()).append("-").append(overriddenPlan.getRecordId()) .toString(); } else { planName = new StringBuffer(parentPlan.getName()).append("-dryrun-") .append(DRY_RUN_PLAN_IDX.incrementAndGet()).toString(); } final DefaultPlan result = new DefaultPlan(planName, (DefaultPlan) parentPlan, resolvedOverride); result.initialize(standaloneCatalog, standaloneCatalog.getCatalogURI()); if (context == null) { overriddenPlanCache.addDryRunPlan(planName, result); } return result; } @Override public DefaultPlan getOverriddenPlan(final String planName, final StaticCatalog catalog, final InternalTenantContext context) throws CatalogApiException { return overriddenPlanCache.getOverriddenPlan(planName, catalog, context); } }