sxp-integration/sxp-ep-provider/src/main/java/org/opendaylight/groupbasedpolicy/sxp/ep/provider/impl/EPPolicyTemplateProviderRegistryImpl.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.groupbasedpolicy.sxp.ep.provider.impl;
  10
  11 import com.google.common.collect.Sets;
  12 import java.util.Set;
  13 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.api.EPPolicyTemplateProvider;
  14 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.api.EPPolicyTemplateProviderRegistry;
  15 import org.opendaylight.groupbasedpolicy.sxp.ep.provider.api.TemplateProviderDistributionTarget;
  16 import org.opendaylight.yangtools.concepts.AbstractObjectRegistration;
  17 import org.opendaylight.yangtools.concepts.ObjectRegistration;
  18
  19 /**
  20  * Purpose: provide registration and internal distribution for obtained {@link EPPolicyTemplateProvider}
  21  */
  22 public class EPPolicyTemplateProviderRegistryImpl implements EPPolicyTemplateProviderRegistry {
  23     private EPPolicyTemplateProvider templateProvider;
  24     private Set<TemplateProviderDistributionTarget<EPPolicyTemplateProvider>> distributionTargets = Sets.newConcurrentHashSet();
  25
  26     @Override
  27     public ObjectRegistration<EPPolicyTemplateProvider> registerTemplateProvider(final EPPolicyTemplateProvider templateProvider) {
  28         distributeTemplateProvider(templateProvider);
  29         return new AbstractObjectRegistration<EPPolicyTemplateProvider>(templateProvider) {
  30             @Override
  31             protected void removeRegistration() {
  32                 distributeTemplateProvider(null);
  33             }
  34         };
  35     }
  36
  37     @Override
  38     public void addDistributionTarget(final TemplateProviderDistributionTarget<EPPolicyTemplateProvider> templateProviderDistributionTarget) {
  39         distributionTargets.add(templateProviderDistributionTarget);
  40         if (templateProvider != null) {
  41             templateProviderDistributionTarget.setTemplateProvider(templateProvider);
  42         }
  43     }
  44
  45     /**
  46      * inject given templateProvider into available targets
  47      * @param templateProvider current provider
  48      */
  49     private void distributeTemplateProvider(final EPPolicyTemplateProvider templateProvider) {
  50         this.templateProvider = templateProvider;
  51         for (TemplateProviderDistributionTarget<EPPolicyTemplateProvider> distributionTarget : distributionTargets) {
  52             distributionTarget.setTemplateProvider(templateProvider);
  53         }
  54     }
  55
  56     @Override
  57     public void close() {
  58         distributeTemplateProvider(null);
  59         distributionTargets.clear();
  60     }
  61 }