KeyMapping is immutable
[bgpcep.git] / pcep / topology / topology-provider / src / main / java / org / opendaylight / bgpcep / pcep / topology / provider / config / PCEPTopologyConfiguration.java
1 /*
2  * Copyright (c) 2017 AT&T Intellectual Property. 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.bgpcep.pcep.topology.provider.config;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.collect.ImmutableMap;
13 import com.google.common.net.InetAddresses;
14 import java.net.InetAddress;
15 import java.net.InetSocketAddress;
16 import java.util.HashMap;
17 import java.util.Map;
18 import org.eclipse.jdt.annotation.NonNull;
19 import org.eclipse.jdt.annotation.Nullable;
20 import org.opendaylight.protocol.concepts.KeyMapping;
21 import org.opendaylight.protocol.pcep.SpeakerIdMapping;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IetfInetUtil;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressNoZone;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.config.rev200120.pcep.config.SessionConfig;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.config.rev181109.PcepNodeConfig;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.sync.optimizations.config.rev181109.PcepNodeSyncConfig;
28 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
29 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
34 import org.opendaylight.yangtools.concepts.Immutable;
35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
36 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
37
38 public final class PCEPTopologyConfiguration implements Immutable {
39     private final @NonNull InetSocketAddress address;
40     private final @NonNull KeyMapping keys;
41     private final short rpcTimeout;
42     private final @NonNull SpeakerIdMapping speakerIds;
43     private final @NonNull KeyedInstanceIdentifier<Topology, TopologyKey> topology;
44
45     public PCEPTopologyConfiguration(final @NonNull SessionConfig config, final @NonNull Topology topology) {
46         this.topology = InstanceIdentifier.create(NetworkTopology.class).child(Topology.class, topology.key());
47         address = getInetSocketAddress(config.getListenAddress(), config.getListenPort());
48         keys = constructKeys(topology.getNode());
49         speakerIds = contructSpeakersId(topology.getNode());
50         rpcTimeout = config.getRpcTimeout();
51     }
52
53     public @NonNull TopologyId getTopologyId() {
54         return topology.getKey().getTopologyId();
55     }
56
57     public @NonNull KeyedInstanceIdentifier<Topology, TopologyKey> getTopology() {
58         return topology;
59     }
60
61     public short getRpcTimeout() {
62         return rpcTimeout;
63     }
64
65     public @NonNull InetSocketAddress getAddress() {
66         return address;
67     }
68
69     public @NonNull KeyMapping getKeys() {
70         return keys;
71     }
72
73     public @NonNull SpeakerIdMapping getSpeakerIds() {
74         return speakerIds;
75     }
76
77     private static @NonNull KeyMapping constructKeys(final @Nullable Map<NodeKey, Node> nodes) {
78         if (nodes == null) {
79             return KeyMapping.of();
80         }
81
82         final var passwords = new HashMap<InetAddress, String>();
83         for (var node : nodes.values()) {
84             if (node != null) {
85                 final var nodeConfig = node.augmentation(PcepNodeConfig.class);
86                 if (nodeConfig != null) {
87                     final var sessionConfig = nodeConfig.getSessionConfig();
88                     if (sessionConfig != null) {
89                         final var rfc2385KeyPassword = sessionConfig.getPassword();
90                         if (rfc2385KeyPassword != null) {
91                             final String password = rfc2385KeyPassword.getValue();
92                             if (!password.isEmpty()) {
93                                 passwords.put(nodeAddress(node), password);
94                             }
95                         }
96                     }
97                 }
98             }
99         }
100
101         return KeyMapping.of(passwords);
102     }
103
104     private static @NonNull SpeakerIdMapping contructSpeakersId(final @Nullable Map<NodeKey, Node> nodes) {
105         if (nodes == null) {
106             return SpeakerIdMapping.of();
107         }
108
109         final var builder = ImmutableMap.<InetAddress, byte[]>builder();
110         for (var node : nodes.values()) {
111             if (node != null) {
112                 final var nodeConfig = node.augmentation(PcepNodeConfig.class);
113                 if (nodeConfig != null) {
114                     final var sessionConfig = nodeConfig.getSessionConfig();
115                     if (sessionConfig != null) {
116                         final var nodeSyncConfig = sessionConfig.augmentation(PcepNodeSyncConfig.class);
117                         if (nodeSyncConfig != null) {
118                             final var speakerEntityId = nodeSyncConfig.getSpeakerEntityIdValue();
119                             if (speakerEntityId != null) {
120                                 builder.put(nodeAddress(node), speakerEntityId);
121                             }
122                         }
123                     }
124                 }
125             }
126         }
127
128         return SpeakerIdMapping.copyOf(builder.build());
129     }
130
131     private static InetAddress nodeAddress(final Node node) {
132         return InetAddresses.forString(node.getNodeId().getValue());
133     }
134
135     private static @NonNull InetSocketAddress getInetSocketAddress(final IpAddressNoZone address,
136             final PortNumber port) {
137         return new InetSocketAddress(IetfInetUtil.INSTANCE.inetAddressForNoZone(requireNonNull(address)),
138             port.getValue().toJava());
139     }
140 }