Code clean up
[bgpcep.git] / bgp / linkstate / src / main / java / org / opendaylight / protocol / bgp / linkstate / spi / pojo / SimpleBindingSubTlvsRegistry.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
9 package org.opendaylight.protocol.bgp.linkstate.spi.pojo;
10
11 import io.netty.buffer.ByteBuf;
12 import java.util.ArrayList;
13 import java.util.List;
14 import org.opendaylight.protocol.bgp.linkstate.spi.BindingSubTlvsParser;
15 import org.opendaylight.protocol.bgp.linkstate.spi.BindingSubTlvsSerializer;
16 import org.opendaylight.protocol.concepts.HandlerRegistry;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.ProtocolId;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.binding.sid.tlv.BindingSubTlvs;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.binding.sid.tlv.BindingSubTlvsBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.binding.sub.tlvs.BindingSubTlv;
21 import org.opendaylight.yangtools.yang.binding.DataContainer;
22 import org.slf4j.Logger;
23 import org.slf4j.LoggerFactory;
24
25 public final class SimpleBindingSubTlvsRegistry {
26     private static final Logger LOG = LoggerFactory.getLogger(SimpleBindingSubTlvsRegistry.class);
27     private static final SimpleBindingSubTlvsRegistry SINGLETON = new SimpleBindingSubTlvsRegistry();
28     private final HandlerRegistry<DataContainer, BindingSubTlvsParser, BindingSubTlvsSerializer> handlers = new HandlerRegistry<>();
29
30     private SimpleBindingSubTlvsRegistry() {
31     }
32
33
34     public static SimpleBindingSubTlvsRegistry getInstance() {
35         return SINGLETON;
36     }
37
38     public AutoCloseable registerBindingSubTlvsParser(final int bindingSubTlvsType, final BindingSubTlvsParser parser) {
39         return this.handlers.registerParser(bindingSubTlvsType, parser);
40     }
41
42     public AutoCloseable registerBindingSubTlvsSerializer(final Class<? extends BindingSubTlv> esiType, final BindingSubTlvsSerializer serializer) {
43         return this.handlers.registerSerializer(esiType, serializer);
44     }
45
46     public void serializeBindingSubTlvs(final List<BindingSubTlvs> bindingSubTlvs, final ByteBuf aggregator) {
47         for (final BindingSubTlvs subTlv : bindingSubTlvs) {
48             final BindingSubTlv bindingSubTlv = subTlv.getBindingSubTlv();
49             final BindingSubTlvsSerializer serializer = this.handlers.getSerializer(bindingSubTlv.getImplementedInterface());
50             if (serializer == null) {
51                 LOG.info("Unknown binding sub Tlv type {}", subTlv);
52                 return;
53             }
54             serializer.serializeSubTlv(bindingSubTlv, aggregator);
55         }
56     }
57
58     public List<BindingSubTlvs> parseBindingSubTlvs(final ByteBuf buffer, final ProtocolId protocolId) {
59         final List<BindingSubTlvs> subTlvs = new ArrayList<>();
60         if (buffer != null) {
61             while (buffer.isReadable()) {
62                 final int type = buffer.readUnsignedShort();
63                 final int length = buffer.readUnsignedShort();
64                 final ByteBuf slice = buffer.readSlice(length);
65                 final BindingSubTlvsParser parser = this.handlers.getParser(type);
66                 if (parser == null) {
67                     return null;
68                 }
69                 subTlvs.add(new BindingSubTlvsBuilder().setBindingSubTlv(parser.parseSubTlv(slice, protocolId)).build());
70             }
71         }
72         return subTlvs;
73     }
74 }