4bb881ef67b21c7cd8faff9cb06415f87f32d0a3
[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.message.BgpParametersBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilities;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.c.parameters.As4BytesCapabilityBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1Builder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.GracefulRestartCapabilityBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.optional.capabilities.c.parameters.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         final AsNumber remoteAs) {
47         this.holdTimer = holdTimer;
48         this.as = as;
49         this.bgpId = bgpId;
50         final List<OptionalCapabilities> caps = new ArrayList<>();
51
52         for (final Entry<Class<? extends AddressFamily>, Class<? extends SubsequentAddressFamily>> e : tables.entrySet()) {
53             caps.add(new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder().addAugmentation(
54                 CParameters1.class, new CParameters1Builder().setMultiprotocolCapability(new MultiprotocolCapabilityBuilder()
55                     .setAfi(e.getKey()).setSafi(e.getValue()).build()).build()).build()).build());
56         }
57         caps.add(new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder().setAs4BytesCapability(
58             new As4BytesCapabilityBuilder().setAsNumber(as).build()).build()).build());
59         caps.add(new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder().addAugmentation(
60             CParameters1.class, new CParameters1Builder().setGracefulRestartCapability(
61                 new GracefulRestartCapabilityBuilder().build()).build()).build()).build());
62         this.prefs = new BGPSessionPreferences(as, holdTimer, bgpId, remoteAs, Lists.newArrayList(
63             new BgpParametersBuilder().setOptionalCapabilities(caps).build()));
64     }
65
66     @Override
67     public BGPSessionPreferences getProposal() {
68         return this.prefs;
69     }
70
71     /**
72      * @return the holdTimer
73      */
74     public short getHoldTimer() {
75         return this.holdTimer;
76     }
77
78     /**
79      * @return the as
80      */
81     public AsNumber getAs() {
82         return this.as;
83     }
84
85     /**
86      * @return the bgpId
87      */
88     public Ipv4Address getBgpId() {
89         return this.bgpId;
90     }
91 }