domain-extensions/l2-l3/src/main/java/org/opendaylight/groupbasedpolicy/domain_extension/l2_l3/L2L3NetworkDomainAugmentor.java

   1 /*
   2  * Copyright (c) 2015 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.domain_extension.l2_l3;
  10
  11 import java.util.AbstractMap.SimpleImmutableEntry;
  12 import java.util.Map.Entry;
  13
  14 import org.opendaylight.groupbasedpolicy.api.NetworkDomainAugmentor;
  15 import org.opendaylight.groupbasedpolicy.api.NetworkDomainAugmentorRegistry;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.l2_l3.rev160427.SubnetAugmentForwarding;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.l2_l3.rev160427.SubnetAugmentRenderer;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.l2_l3.rev160427.SubnetAugmentRendererBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.rev160427.forwarding.forwarding.by.tenant.NetworkDomain;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.renderer.policy.configuration.renderer.forwarding.renderer.forwarding.by.tenant.RendererNetworkDomain;
  21 import org.opendaylight.yangtools.yang.binding.Augmentation;
  22
  23 import com.google.common.base.Preconditions;
  24
  25 public class L2L3NetworkDomainAugmentor implements NetworkDomainAugmentor, AutoCloseable {
  26
  27     private final NetworkDomainAugmentorRegistry networkDomainAugmentorRegistry;
  28
  29     public L2L3NetworkDomainAugmentor(NetworkDomainAugmentorRegistry networkDomainAugmentorRegistry) {
  30         this.networkDomainAugmentorRegistry = Preconditions.checkNotNull(networkDomainAugmentorRegistry);
  31         this.networkDomainAugmentorRegistry.register(this);
  32     }
  33
  34     @Override
  35     public Entry<Class<? extends Augmentation<RendererNetworkDomain>>, Augmentation<RendererNetworkDomain>> buildRendererNetworkDomainAugmentation(
  36             NetworkDomain input) {
  37         SubnetAugmentForwarding subnetForwarding = input.getAugmentation(SubnetAugmentForwarding.class);
  38         if (subnetForwarding == null) {
  39             return null;
  40         }
  41         return new SimpleImmutableEntry<>(SubnetAugmentRenderer.class,
  42                 new SubnetAugmentRendererBuilder(subnetForwarding).build());
  43     }
  44
  45     @Override
  46     public void close() throws Exception {
  47         networkDomainAugmentorRegistry.unregister(this);
  48     }
  49
  50 }