bgp/parser-impl/src/main/java/org/opendaylight/protocol/bgp/parser/impl/message/open/SimpleCapabilityRegistry.java

   1 /*
   2  * Copyright (c) 2013 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.parser.impl.message.open;
   9
  10 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
  11 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
  12 import org.opendaylight.protocol.bgp.parser.spi.CapabilityParser;
  13 import org.opendaylight.protocol.bgp.parser.spi.CapabilityRegistry;
  14 import org.opendaylight.protocol.bgp.parser.spi.CapabilitySerializer;
  15 import org.opendaylight.protocol.concepts.HandlerRegistry;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.open.bgp.parameters.CParameters;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.open.bgp.parameters.c.parameters.CAs4Bytes;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130918.open.bgp.parameters.c.parameters.CMultiprotocol;
  19 import org.opendaylight.yangtools.yang.binding.DataContainer;
  20
  21 import com.google.common.base.Preconditions;
  22
  23 public final class SimpleCapabilityRegistry implements CapabilityRegistry {
  24         public static final CapabilityRegistry INSTANCE;
  25
  26         static {
  27                 final SimpleCapabilityRegistry reg = new SimpleCapabilityRegistry();
  28
  29                 final MultiProtocolCapabilityHandler multi =
  30                                 new MultiProtocolCapabilityHandler();
  31                 reg.registerCapabilityParser(MultiProtocolCapabilityHandler.CODE, multi);
  32                 reg.registerCapabilitySerializer(CMultiprotocol.class, multi);
  33
  34                 final As4CapabilityHandler as4 = new As4CapabilityHandler();
  35                 reg.registerCapabilityParser(As4CapabilityHandler.CODE, as4);
  36                 reg.registerCapabilitySerializer(CAs4Bytes.class, as4);
  37
  38                 INSTANCE = reg;
  39         }
  40
  41         private final HandlerRegistry<DataContainer, CapabilityParser, CapabilitySerializer> handlers = new HandlerRegistry<>();
  42
  43         @Override
  44         public AutoCloseable registerCapabilityParser(final int messageType, final CapabilityParser parser) {
  45                 Preconditions.checkArgument(messageType >= 0 && messageType <= 255);
  46                 return handlers.registerParser(messageType, parser);
  47         }
  48
  49         @Override
  50         public AutoCloseable registerCapabilitySerializer(final Class<? extends CParameters> paramClass, final CapabilitySerializer serializer) {
  51                 return handlers.registerSerializer(paramClass, serializer);
  52         }
  53
  54         @Override
  55         public CParameters parseCapability(final int type, final byte[] bytes) throws BGPDocumentedException, BGPParsingException {
  56                 final CapabilityParser parser = handlers.getParser(type);
  57                 if (parser == null) {
  58                         return null;
  59                 }
  60
  61                 return parser.parseCapability(bytes);
  62         }
  63
  64         @Override
  65         public byte[] serializeCapability(final CParameters capability) {
  66                 final CapabilitySerializer serializer = handlers.getSerializer(capability.getImplementedInterface());
  67                 if (serializer == null) {
  68                         return null;
  69                 }
  70
  71                 return serializer.serializeCapability(capability);
  72         }
  73 }