2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.protocol.bgp.evpn.spi.pojo;
11 import com.google.common.base.Preconditions;
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.rev171213.NlriType;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev171213.evpn.EvpnChoice;
21 import org.opendaylight.yangtools.yang.binding.DataContainer;
22 import org.opendaylight.yangtools.yang.common.QName;
23 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
24 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
25 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
27 public final class SimpleEvpnNlriRegistry implements EvpnRegistry {
28 private static final SimpleEvpnNlriRegistry SINGLETON = new SimpleEvpnNlriRegistry();
29 private final HandlerRegistry<DataContainer, EvpnParser, EvpnSerializer> handlers = new HandlerRegistry<>();
30 private final MultiRegistry<NodeIdentifier, EvpnSerializer> modelHandlers = new MultiRegistry<>();
32 private SimpleEvpnNlriRegistry() {
36 private interface SerializerInterface {
37 EvpnChoice check(EvpnSerializer serializer, ContainerNode cont);
40 public static SimpleEvpnNlriRegistry getInstance() {
44 public AutoCloseable registerNlriParser(final NlriType esiType, final EvpnParser parser) {
45 return this.handlers.registerParser(esiType.getIntValue(), parser);
48 public AutoCloseable registerNlriSerializer(final Class<? extends EvpnChoice> evpnClass,
49 final EvpnSerializer serializer) {
50 return this.handlers.registerSerializer(evpnClass, serializer);
53 public AutoCloseable registerNlriModelSerializer(final QName qname, final EvpnSerializer serializer) {
54 return this.modelHandlers.register(new NodeIdentifier(qname), serializer);
58 public EvpnChoice parseEvpn(final NlriType type, final ByteBuf buffer) {
59 Preconditions.checkArgument(buffer != null && buffer.isReadable(),
60 "Array of bytes is mandatory. Can't be null or empty.");
61 final EvpnParser parser = this.handlers.getParser(type.getIntValue());
65 return parser.parseEvpn(buffer);
69 public ByteBuf serializeEvpn(final EvpnChoice evpn, final ByteBuf common) {
70 final EvpnSerializer serializer = this.handlers.getSerializer(evpn.getImplementedInterface());
71 if (serializer == null) {
74 return serializer.serializeEvpn(evpn, common);
78 public EvpnChoice serializeEvpnModel(final ChoiceNode evpnChoice) {
79 return getEvpnCase(evpnChoice, EvpnSerializer::serializeEvpnModel);
83 public EvpnChoice serializeEvpnRouteKey(final ChoiceNode evpnChoice) {
84 return getEvpnCase(evpnChoice, EvpnSerializer::createRouteKey);
87 private EvpnChoice getEvpnCase(final ChoiceNode evpnChoice, final SerializerInterface serializerInterface) {
88 Preconditions.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) {
95 return serializerInterface.check(serializer, cont);