bgp/extensions/evpn/src/main/java/org/opendaylight/protocol/bgp/evpn/spi/pojo/SimpleEvpnNlriRegistry.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 package org.opendaylight.protocol.bgp.evpn.spi.pojo;
   9
  10 import static com.google.common.base.Preconditions.checkArgument;
  11
  12 import com.google.common.collect.Iterables;
  13 import io.netty.buffer.ByteBuf;
  14 import org.opendaylight.protocol.bgp.evpn.spi.EvpnParser;
  15 import org.opendaylight.protocol.bgp.evpn.spi.EvpnRegistry;
  16 import org.opendaylight.protocol.bgp.evpn.spi.EvpnSerializer;
  17 import org.opendaylight.protocol.concepts.HandlerRegistry;
  18 import org.opendaylight.protocol.concepts.MultiRegistry;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.NlriType;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.evpn.EvpnChoice;
  21 import org.opendaylight.yangtools.concepts.Registration;
  22 import org.opendaylight.yangtools.yang.binding.DataContainer;
  23 import org.opendaylight.yangtools.yang.common.QName;
  24 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  25 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  26 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  27
  28 public final class SimpleEvpnNlriRegistry implements EvpnRegistry {
  29     private static final SimpleEvpnNlriRegistry SINGLETON = new SimpleEvpnNlriRegistry();
  30     private final HandlerRegistry<DataContainer, EvpnParser, EvpnSerializer> handlers = new HandlerRegistry<>();
  31     private final MultiRegistry<NodeIdentifier, EvpnSerializer> modelHandlers = new MultiRegistry<>();
  32
  33     private SimpleEvpnNlriRegistry() {
  34     }
  35
  36     @FunctionalInterface
  37     private interface SerializerInterface {
  38         EvpnChoice check(EvpnSerializer serializer, ContainerNode cont);
  39     }
  40
  41     public static SimpleEvpnNlriRegistry getInstance() {
  42         return SINGLETON;
  43     }
  44
  45     public Registration registerNlriParser(final NlriType esiType, final EvpnParser parser) {
  46         return this.handlers.registerParser(esiType.getIntValue(), parser);
  47     }
  48
  49     public Registration registerNlriSerializer(final Class<? extends EvpnChoice> evpnClass,
  50             final EvpnSerializer serializer) {
  51         return this.handlers.registerSerializer(evpnClass, serializer);
  52     }
  53
  54     public Registration registerNlriModelSerializer(final QName qname, final EvpnSerializer serializer) {
  55         return this.modelHandlers.register(new NodeIdentifier(qname), serializer);
  56     }
  57
  58     @Override
  59     public EvpnChoice parseEvpn(final NlriType type, final ByteBuf buffer) {
  60         checkArgument(buffer != null && buffer.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
  61         final EvpnParser parser = this.handlers.getParser(type.getIntValue());
  62         if (parser == null) {
  63             return null;
  64         }
  65         return parser.parseEvpn(buffer);
  66     }
  67
  68     @Override
  69     public ByteBuf serializeEvpn(final EvpnChoice evpn, final ByteBuf common) {
  70         final EvpnSerializer serializer = this.handlers.getSerializer(evpn.implementedInterface());
  71         if (serializer == null) {
  72             return common;
  73         }
  74         return serializer.serializeEvpn(evpn, common);
  75     }
  76
  77     @Override
  78     public EvpnChoice serializeEvpnModel(final ChoiceNode evpnChoice) {
  79         return getEvpnCase(evpnChoice, EvpnSerializer::serializeEvpnModel);
  80     }
  81
  82     @Override
  83     public EvpnChoice serializeEvpnRouteKey(final ChoiceNode evpnChoice) {
  84         return getEvpnCase(evpnChoice, EvpnSerializer::createRouteKey);
  85     }
  86
  87     private EvpnChoice getEvpnCase(final ChoiceNode evpnChoice, final SerializerInterface serializerInterface) {
  88         checkArgument(evpnChoice != null && !evpnChoice.getValue().isEmpty(),
  89                 "Evpn case is mandatory. Can't be null or empty.");
  90         final ContainerNode cont = (ContainerNode) Iterables.getOnlyElement(evpnChoice.getValue());
  91         final EvpnSerializer serializer = this.modelHandlers.get(cont.getIdentifier());
  92         if (serializer == null) {
  93             return null;
  94         }
  95         return serializerInterface.check(serializer, cont);
  96     }
  97 }