cbae35b5164e29c4ee3aef4ddf8ec619d2580194
[bgpcep.git] / bgp / rib-impl / src / main / java / org / opendaylight / protocol / bgp / rib / impl / BGPSessionProposalImpl.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.rib.impl;
9
10 import com.google.common.collect.Lists;
11 import java.util.ArrayList;
12 import java.util.List;
13 import java.util.Map;
14 import java.util.Map.Entry;
15 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionPreferences;
16 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionProposal;
17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.BgpParametersBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.OptionalCapabilities;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.OptionalCapabilitiesBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.As4BytesCaseBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.as4.bytes._case.As4BytesCapabilityBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.GracefulRestartCaseBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.MultiprotocolCaseBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.graceful.restart._case.GracefulRestartCapabilityBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.multiprotocol._case.MultiprotocolCapabilityBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
30
31 /**
32  * Basic implementation of BGP Session Proposal. The values are taken from conf-bgp.
33  */
34 public final class BGPSessionProposalImpl implements BGPSessionProposal {
35
36     private final short holdTimer;
37
38     private final AsNumber as;
39
40     private final Ipv4Address bgpId;
41
42     private final BGPSessionPreferences prefs;
43
44     public BGPSessionProposalImpl(final short holdTimer, final AsNumber as, final Ipv4Address bgpId,
45             final Map<Class<? extends AddressFamily>, Class<? extends SubsequentAddressFamily>> tables) {
46         this.holdTimer = holdTimer;
47         this.as = as;
48         this.bgpId = bgpId;
49         final List<OptionalCapabilities> caps = new ArrayList<>();
50
51         for (final Entry<Class<? extends AddressFamily>, Class<? extends SubsequentAddressFamily>> e : tables.entrySet()) {
52             caps.add(new OptionalCapabilitiesBuilder().setCParameters(
53                     new MultiprotocolCaseBuilder().setMultiprotocolCapability(
54                             new MultiprotocolCapabilityBuilder().setAfi(e.getKey()).setSafi(e.getValue()).build()).build()).build());
55         }
56         caps.add(new OptionalCapabilitiesBuilder().setCParameters(
57                 new As4BytesCaseBuilder().setAs4BytesCapability(new As4BytesCapabilityBuilder().setAsNumber(as).build()).build()).build());
58         caps.add(new OptionalCapabilitiesBuilder().setCParameters(
59                 new GracefulRestartCaseBuilder().setGracefulRestartCapability(
60                     new GracefulRestartCapabilityBuilder().build()).build()).build());
61         this.prefs = new BGPSessionPreferences(as, holdTimer, bgpId, Lists.newArrayList(new BgpParametersBuilder().setOptionalCapabilities(caps).build()));
62     }
63
64     @Override
65     public BGPSessionPreferences getProposal() {
66         return this.prefs;
67     }
68
69     /**
70      * @return the holdTimer
71      */
72     public short getHoldTimer() {
73         return this.holdTimer;
74     }
75
76     /**
77      * @return the as
78      */
79     public AsNumber getAs() {
80         return this.as;
81     }
82
83     /**
84      * @return the bgpId
85      */
86     public Ipv4Address getBgpId() {
87         return this.bgpId;
88     }
89 }