BUG-6647 Increase code coverage and clean up III
[bgpcep.git] / bgp / openconfig-impl / src / main / java / org / opendaylight / protocol / bgp / openconfig / impl / BGPOpenConfigMappingServiceImpl.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
9 package org.opendaylight.protocol.bgp.openconfig.impl;
10
11 import static org.opendaylight.protocol.bgp.openconfig.impl.util.OpenConfigUtil.APPLICATION_PEER_GROUP_NAME;
12 import static org.opendaylight.protocol.bgp.openconfig.impl.util.OpenConfigUtil.toPeerType;
13
14 import com.google.common.base.Optional;
15 import java.math.BigDecimal;
16 import java.util.ArrayList;
17 import java.util.Collections;
18 import java.util.HashMap;
19 import java.util.List;
20 import java.util.Map;
21 import java.util.stream.Collectors;
22 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
23 import org.opendaylight.protocol.bgp.mode.impl.add.all.paths.AllPathSelection;
24 import org.opendaylight.protocol.bgp.mode.impl.add.n.paths.AddPathBestNPathSelection;
25 import org.opendaylight.protocol.bgp.openconfig.impl.util.OpenConfigUtil;
26 import org.opendaylight.protocol.bgp.openconfig.spi.BGPOpenConfigMappingService;
27 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
28 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.BgpNeighborAddPathsConfig;
29 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.AfiSafisBuilder;
30 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base.ConfigBuilder;
31 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.RouteReflectorBuilder;
32 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TimersBuilder;
33 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TransportBuilder;
34 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.Neighbor;
35 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.NeighborBuilder;
36 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.NeighborKey;
37 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.peer.group.PeerGroup;
38 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.peer.group.PeerGroupBuilder;
39 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.peer.group.PeerGroupKey;
40 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.Bgp;
41 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.BgpBuilder;
42 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Global;
43 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.GlobalBuilder;
44 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.NeighborsBuilder;
45 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.PeerGroups;
46 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.PeerGroupsBuilder;
47 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.CommunityType;
48 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.PeerType;
49 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.Protocol;
50 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolBuilder;
51 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolKey;
52 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
53 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
54 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
55 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
56 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.SendReceive;
59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.add.path.capability.AddressFamilies;
60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi1;
62 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.AfiSafi2;
63 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Config2;
64 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Config2Builder;
65 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Protocol1;
66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Protocol1Builder;
67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.ApplicationRibId;
68 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
69 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId;
70 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.SimpleRoutingPolicy;
71 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
72 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
73 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.ClusterIdentifier;
74 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.rfc2385.cfg.rev160324.Rfc2385Key;
75
76 public final class BGPOpenConfigMappingServiceImpl implements BGPOpenConfigMappingService {
77
78     private static final PeerGroup APP_PEER_GROUP = new PeerGroupBuilder().setPeerGroupName(APPLICATION_PEER_GROUP_NAME)
79             .setKey(new PeerGroupKey(APPLICATION_PEER_GROUP_NAME)).build();
80     private static final PeerGroups PEER_GROUPS = new PeerGroupsBuilder().setPeerGroup(Collections.singletonList(APP_PEER_GROUP)).build();
81     private static final BigDecimal DEFAULT_KEEP_ALIVE = BigDecimal.valueOf(30);
82     private static final BigDecimal DEFAULT_MINIMUM_ADV_INTERVAL = BigDecimal.valueOf(30);
83
84     @Override
85     public List<BgpTableType> toTableTypes(final List<AfiSafi> afiSafis) {
86         return afiSafis.stream()
87                 .map(afiSafi -> OpenConfigUtil.toBgpTableType(afiSafi.getAfiSafiName()))
88                 .filter(Optional::isPresent)
89                 .map(Optional::get)
90                 .collect(Collectors.toList());
91     }
92
93     @Override
94     public Map<BgpTableType, PathSelectionMode> toPathSelectionMode(final List<AfiSafi> afiSafis) {
95         final Map<BgpTableType, PathSelectionMode> pathSelectionModes = new HashMap<>();
96         for (final AfiSafi afiSafi : afiSafis) {
97             final BgpNeighborAddPathsConfig afiSafi2 = afiSafi.getAugmentation(AfiSafi2.class);
98             final Optional<BgpTableType> bgpTableType = OpenConfigUtil.toBgpTableType(afiSafi.getAfiSafiName());
99             if (afiSafi2 != null && bgpTableType.isPresent()) {
100                 final Short sendMax = afiSafi2.getSendMax();
101                 final PathSelectionMode selectionMode;
102                 if (sendMax > 1) {
103                     selectionMode = new AddPathBestNPathSelection(sendMax.longValue());
104                 } else {
105                     selectionMode = new AllPathSelection();
106                 }
107                 pathSelectionModes.put(bgpTableType.get(), selectionMode);
108             }
109         }
110         return pathSelectionModes;
111     }
112
113     @Override
114     public boolean isApplicationPeer(final Neighbor neighbor) {
115         return OpenConfigUtil.isAppNeighbor(neighbor);
116     }
117
118     @Override
119     public PeerRole toPeerRole(final Neighbor neighbor) {
120         return OpenConfigUtil.toPeerRole(neighbor);
121     }
122
123     @Override
124     public List<AddressFamilies> toAddPathCapability(final List<AfiSafi> afiSafis) {
125         final List<AddressFamilies> addPathCapability = new ArrayList<>();
126         for (final AfiSafi afiSafi : afiSafis) {
127             final BgpNeighborAddPathsConfig afiSafi1 = afiSafi.getAugmentation(AfiSafi1.class);
128             final Optional<BgpTableType> bgpTableType = OpenConfigUtil.toBgpTableType(afiSafi.getAfiSafiName());
129             if (afiSafi1 != null && bgpTableType.isPresent()) {
130                 final AddressFamiliesBuilder builder = new AddressFamiliesBuilder(bgpTableType.get());
131                 builder.setSendReceive(toSendReceiveMode(afiSafi1));
132                 addPathCapability.add(builder.build());
133             }
134         }
135         return addPathCapability;
136     }
137
138     private static SendReceive toSendReceiveMode(final BgpNeighborAddPathsConfig addPath) {
139         if (addPath.isReceive() && addPath.getSendMax() != null) {
140             return SendReceive.Both;
141         }
142         if (addPath.getSendMax() != null) {
143             return SendReceive.Send;
144         }
145         return SendReceive.Receive;
146     }
147
148     @Override
149     public Protocol fromRib(final BgpId bgpId, final ClusterIdentifier clusterIdentifier, final RibId ribId,
150             final AsNumber localAs, final List<BgpTableType> localTables,
151             final Map<TablesKey, PathSelectionMode> pathSelectionStrategies) {
152         final Bgp bgp = toGlobalConfiguration(bgpId, clusterIdentifier, localAs, localTables, pathSelectionStrategies);
153         final ProtocolBuilder protocolBuilder = new ProtocolBuilder();
154         protocolBuilder.setIdentifier(BGP.class);
155         protocolBuilder.setName(ribId.getValue());
156         protocolBuilder.setKey(new ProtocolKey(protocolBuilder.getIdentifier(), protocolBuilder.getName()));
157         return protocolBuilder.addAugmentation(Protocol1.class, new Protocol1Builder().setBgp(bgp).build()).build();
158     }
159
160     private static Bgp toGlobalConfiguration(final BgpId bgpId, final ClusterIdentifier clusterIdentifier,
161             final AsNumber localAs, final List<BgpTableType> localTables,
162             final Map<TablesKey, PathSelectionMode> pathSelectionStrategies) {
163         final BgpBuilder bgpBuilder = new BgpBuilder();
164         bgpBuilder.setNeighbors(new NeighborsBuilder().build());
165         bgpBuilder.setPeerGroups(PEER_GROUPS);
166         final Global global = new GlobalBuilder().setAfiSafis(new AfiSafisBuilder().setAfiSafi(OpenConfigUtil.toAfiSafis(localTables,
167                 (afiSafi, tableType) -> OpenConfigUtil.toGlobalAfiSafiAddPath(afiSafi, tableType, pathSelectionStrategies))).build())
168                 .setConfig(new ConfigBuilder().setAs(localAs).setRouterId(bgpId).build()).build();
169         bgpBuilder.setGlobal(global);
170         return bgpBuilder.build();
171     }
172
173     @Override
174     public Neighbor fromBgpPeer(final List<AddressFamilies> addPathCapabilities,
175             final List<BgpTableType> advertisedTables, final Integer holdTimer, final IpAddress ipAddress,
176             final Boolean isActive, final Rfc2385Key password, final PortNumber portNumber, final Integer retryTimer,
177             final AsNumber remoteAs, final PeerRole peerRole, final SimpleRoutingPolicy simpleRoutingPolicy) {
178         final NeighborBuilder neighborBuilder = new NeighborBuilder();
179         neighborBuilder.setNeighborAddress(ipAddress);
180         neighborBuilder.setKey(new NeighborKey(ipAddress));
181         neighborBuilder.setAfiSafis(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafisBuilder().setAfiSafi(OpenConfigUtil.toAfiSafis(advertisedTables,
182                 (afiSafi, tableType) -> OpenConfigUtil.toNeighborAfiSafiAddPath(afiSafi, tableType, addPathCapabilities))).build());
183         neighborBuilder.setTransport(new TransportBuilder().setConfig(
184                 new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.transport.ConfigBuilder()
185                 .setPassiveMode(!isActive)
186                 .setMtuDiscovery(Boolean.FALSE)
187                 .addAugmentation(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Config1.class,
188                         new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev160614.Config1Builder()
189                 .setRemotePort(portNumber).build())
190                 .build()).build());
191         neighborBuilder.setConfig(
192                 new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.ConfigBuilder()
193                 .setAuthPassword(password != null ? password.getValue() : null)
194                 .setPeerAs(remoteAs)
195                 .setPeerType(toPeerType(peerRole))
196                 .setSendCommunity(CommunityType.NONE)
197                 .setRouteFlapDamping(Boolean.FALSE)
198                 .build());
199         neighborBuilder.setTimers(new TimersBuilder().setConfig(
200                 new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.timers.ConfigBuilder()
201                 .setHoldTime(BigDecimal.valueOf(holdTimer))
202                 .setConnectRetry(BigDecimal.valueOf(retryTimer))
203                 .setKeepaliveInterval(DEFAULT_KEEP_ALIVE)
204                 .setMinimumAdvertisementInterval(DEFAULT_MINIMUM_ADV_INTERVAL)
205                 .build()).build());
206         neighborBuilder.setRouteReflector(new RouteReflectorBuilder().setConfig(
207                 new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.route.reflector.ConfigBuilder()
208                 .setRouteReflectorClient(peerRole == PeerRole.RrClient).build()).build());
209         return neighborBuilder.build();
210     }
211
212     @Override
213     public Neighbor fromApplicationPeer(final ApplicationRibId applicationRibId, final BgpId bgpId) {
214         final NeighborBuilder neighborBuilder = new NeighborBuilder();
215         neighborBuilder.setNeighborAddress(new IpAddress(new Ipv4Address(bgpId.getValue())));
216         neighborBuilder.setKey(new NeighborKey(neighborBuilder.getNeighborAddress()));
217         neighborBuilder.setConfig(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.ConfigBuilder()
218         .setDescription(applicationRibId.getValue())
219         .addAugmentation(Config2.class, new Config2Builder().setPeerGroup(APPLICATION_PEER_GROUP_NAME).build())
220         .build());
221
222         return neighborBuilder.build();
223     }
224
225 }