Start using Codecs and CodecsRegistry instead of RIBSupportContext.
[bgpcep.git] / bgp / rib-impl / src / main / java / org / opendaylight / protocol / bgp / rib / impl / RIBSupportContextImpl.java
1 /*
2  * Copyright (c) 2015 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.rib.impl;
9
10 import com.google.common.base.Preconditions;
11 import com.google.common.base.Verify;
12 import java.util.Map.Entry;
13 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
14 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
15 import org.opendaylight.protocol.bgp.rib.impl.spi.CodecsRegistry;
16 import org.opendaylight.protocol.bgp.rib.impl.spi.RIBSupportContext;
17 import org.opendaylight.protocol.bgp.rib.spi.RIBSupport;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.Attributes;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpReachNlri;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpUnreachNlri;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.tables.Routes;
22 import org.opendaylight.yangtools.yang.common.QName;
23 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
24 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
25 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
26 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
27 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
28 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
29 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
30
31 class RIBSupportContextImpl extends RIBSupportContext {
32
33     private static final ContainerNode EMPTY_TABLE_ATTRIBUTES = ImmutableNodes.containerNode(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.tables.Attributes.QNAME);
34
35     private final RIBSupport ribSupport;
36     private final Codecs codecs;
37
38     public RIBSupportContextImpl(final RIBSupport ribSupport, final CodecsRegistry codecs) {
39         this.ribSupport = Preconditions.checkNotNull(ribSupport);
40         this.codecs = codecs.getCodecs(this.ribSupport);
41     }
42
43     @Override
44     public void writeRoutes(final DOMDataWriteTransaction tx, final YangInstanceIdentifier tableId, final MpReachNlri nlri,
45             final Attributes attributes) {
46         final ContainerNode domNlri = this.codecs.serializeReachNlri(nlri);
47         final ContainerNode routeAttributes = this.codecs.serializeAttributes(attributes);
48         this.ribSupport.putRoutes(tx, tableId, domNlri, routeAttributes);
49     }
50
51     @Override
52     public void clearTable(final DOMDataWriteTransaction tx, final YangInstanceIdentifier tableId) {
53         final DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> tb = ImmutableNodes.mapEntryBuilder();
54         tb.withNodeIdentifier((NodeIdentifierWithPredicates)tableId.getLastPathArgument());
55         tb.withChild(EMPTY_TABLE_ATTRIBUTES);
56
57         // tableId is keyed, but that fact is not directly visible from YangInstanceIdentifier, see BUG-2796
58         final NodeIdentifierWithPredicates tableKey = (NodeIdentifierWithPredicates) tableId.getLastPathArgument();
59         for (final Entry<QName, Object> e : tableKey.getKeyValues().entrySet()) {
60             tb.withChild(ImmutableNodes.leafNode(e.getKey(), e.getValue()));
61         }
62
63         final ChoiceNode routes = this.ribSupport.emptyRoutes();
64         Verify.verifyNotNull(routes, "Null empty routes in %s", this.ribSupport);
65         Verify.verify(Routes.QNAME.equals(routes.getNodeType()), "Empty routes have unexpected identifier %s, expected %s", routes.getNodeType(), Routes.QNAME);
66
67         tx.put(LogicalDatastoreType.OPERATIONAL, tableId, tb.withChild(routes).build());
68     }
69
70     @Override
71     public void deleteRoutes(final DOMDataWriteTransaction tx, final YangInstanceIdentifier tableId, final MpUnreachNlri nlri) {
72         this.ribSupport.deleteRoutes(tx, tableId, this.codecs.serializeUnreachNlri(nlri));
73     }
74
75     @Override
76     public RIBSupport getRibSupport() {
77         return this.ribSupport;
78     }
79 }