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