renderers/faas/src/main/java/org/opendaylight/groupbasedpolicy/renderer/faas/FaasEndpointAug.java

   1 /*
   2  * Copyright (c) 2015 Huawei Technologies 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 package org.opendaylight.groupbasedpolicy.renderer.faas;
   9
  10 import java.util.AbstractMap.SimpleImmutableEntry;
  11 import java.util.Map;
  12
  13 import org.opendaylight.groupbasedpolicy.api.EpRendererAugmentation;
  14 import org.opendaylight.groupbasedpolicy.api.EpRendererAugmentationRegistry;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.faas.faas.endpoint.rev151009.FaasEndpointContext;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.faas.faas.endpoint.rev151009.FaasEndpointContextBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.faas.faas.endpoint.rev151009.FaasEndpointContextInput;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.RegisterEndpointInput;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.RegisterL3PrefixEndpointInput;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.Endpoint;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3Prefix;
  23 import org.opendaylight.yangtools.yang.binding.Augmentation;
  24
  25 public class FaasEndpointAug implements EpRendererAugmentation, AutoCloseable {
  26
  27     private EpRendererAugmentationRegistry epRendererAugmentationRegistry;
  28
  29     public FaasEndpointAug(EpRendererAugmentationRegistry epRendererAugmentationRegistry) {
  30
  31         this.epRendererAugmentationRegistry = epRendererAugmentationRegistry;
  32         epRendererAugmentationRegistry.register(this);
  33     }
  34
  35     @Override
  36     public Map.Entry<Class<? extends Augmentation<Endpoint>>, Augmentation<Endpoint>> buildEndpointAugmentation(
  37             RegisterEndpointInput input) {
  38         FaasEndpointContextBuilder pb;
  39         FaasEndpointContextInput pix = input.getAugmentation(FaasEndpointContextInput.class);
  40         if (pix == null) {
  41             pb = new FaasEndpointContextBuilder();
  42         } else {
  43             pb = new FaasEndpointContextBuilder(pix);
  44         }
  45         return new SimpleImmutableEntry<Class<? extends Augmentation<Endpoint>>, Augmentation<Endpoint>>(
  46                 FaasEndpointContext.class, pb.build());
  47
  48     }
  49
  50     @Override
  51     public Map.Entry<Class<? extends Augmentation<EndpointL3>>, Augmentation<EndpointL3>> buildEndpointL3Augmentation(
  52             RegisterEndpointInput input) {
  53         return null;
  54     }
  55
  56     @Override
  57     public Map.Entry<Class<? extends Augmentation<EndpointL3Prefix>>, Augmentation<EndpointL3Prefix>> buildL3PrefixEndpointAugmentation(
  58             RegisterL3PrefixEndpointInput input) {
  59         return null;
  60     }
  61
  62     @Override
  63     public void close() throws Exception {
  64         epRendererAugmentationRegistry.unregister(this);
  65     }
  66 }