add dependency check for qosPolicyId to NeutronNetworkInterface
[neutron.git] / transcriber / src / main / java / org / opendaylight / neutron / transcriber / NeutronNetworkInterface.java
1 /*
2  * Copyright (c) 2013, 2015 IBM Corporation 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.neutron.transcriber;
9
10 import com.google.common.collect.ImmutableBiMap;
11 import java.util.ArrayList;
12 import java.util.List;
13 import javax.inject.Inject;
14 import javax.inject.Singleton;
15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
16 import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
17 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
18 import org.opendaylight.neutron.spi.INeutronNetworkCRUD;
19 import org.opendaylight.neutron.spi.NeutronNetwork;
20 import org.opendaylight.neutron.spi.NeutronNetworkSegment;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.l3.ext.rev150712.NetworkL3Extension;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.l3.ext.rev150712.NetworkL3ExtensionBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.NetworkTypeBase;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.NetworkTypeFlat;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.NetworkTypeGre;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.NetworkTypeVlan;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.NetworkTypeVxlan;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.Networks;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.Network;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.NetworkBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.NetworkKey;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.provider.ext.rev150712.NetworkProviderExtension;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.provider.ext.rev150712.NetworkProviderExtensionBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.provider.ext.rev150712.neutron.networks.network.Segments;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.provider.ext.rev150712.neutron.networks.network.SegmentsBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.qos.ext.rev160613.QosNetworkExtension;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.qos.ext.rev160613.QosNetworkExtensionBuilder;
38 import org.ops4j.pax.cdi.api.OsgiServiceProvider;
39
40 @Singleton
41 @OsgiServiceProvider(classes = INeutronNetworkCRUD.class)
42 public final class NeutronNetworkInterface
43         extends AbstractNeutronInterface<Network, Networks, NetworkKey, NeutronNetwork>
44         implements INeutronNetworkCRUD {
45
46     private static final ImmutableBiMap<Class<? extends NetworkTypeBase>,
47             String> NETWORK_MAP = new ImmutableBiMap.Builder<Class<? extends NetworkTypeBase>, String>()
48                     .put(NetworkTypeFlat.class, "flat").put(NetworkTypeGre.class, "gre")
49                     .put(NetworkTypeVlan.class, "vlan").put(NetworkTypeVxlan.class, "vxlan").build();
50
51     private final NeutronQosPolicyInterface qosPolicyInterface;
52
53     @Inject
54     public NeutronNetworkInterface(DataBroker db, NeutronQosPolicyInterface qosPolicyInterface) {
55         super(NetworkBuilder.class, db);
56         this.qosPolicyInterface = qosPolicyInterface;
57     }
58
59     // IfNBNetworkCRUD methods
60     @Override
61     protected List<Network> getDataObjectList(Networks networks) {
62         return networks.getNetwork();
63     }
64
65     @Override
66     protected NeutronNetwork fromMd(Network network) {
67         final NeutronNetwork result = new NeutronNetwork();
68         result.initDefaults();
69         fromMdAdminAttributes(network, result);
70         result.setShared(network.isShared());
71         result.setVlanTransparent(network.isVlanTransparent());
72
73         final NetworkL3Extension l3Extension = network.augmentation(NetworkL3Extension.class);
74         result.setRouterExternal(l3Extension.isExternal());
75
76         final NetworkProviderExtension providerExtension = network.augmentation(NetworkProviderExtension.class);
77         result.setProviderPhysicalNetwork(providerExtension.getPhysicalNetwork());
78         result.setProviderSegmentationID(providerExtension.getSegmentationId());
79         result.setProviderNetworkType(NETWORK_MAP.get(providerExtension.getNetworkType()));
80         final List<NeutronNetworkSegment> segments = new ArrayList<>();
81         if (providerExtension.getSegments() != null) {
82             for (final Segments segment : providerExtension.getSegments()) {
83                 final NeutronNetworkSegment neutronSegment = new NeutronNetworkSegment();
84                 neutronSegment.setProviderPhysicalNetwork(segment.getPhysicalNetwork());
85                 neutronSegment.setProviderSegmentationID(segment.getSegmentationId());
86                 neutronSegment.setProviderNetworkType(NETWORK_MAP.get(segment.getNetworkType()));
87                 segments.add(neutronSegment);
88             }
89         }
90         final QosNetworkExtension qos = network.augmentation(QosNetworkExtension.class);
91         if (qos != null && qos.getQosPolicyId() != null) {
92             result.setQosPolicyId(qos.getQosPolicyId().getValue());
93         }
94         result.setSegments(segments);
95         return result;
96     }
97
98     private void fillExtensions(NetworkBuilder networkBuilder, NeutronNetwork network) {
99         final NetworkL3ExtensionBuilder l3ExtensionBuilder = new NetworkL3ExtensionBuilder();
100         if (network.getRouterExternal() != null) {
101             l3ExtensionBuilder.setExternal(network.getRouterExternal());
102         }
103
104         final NetworkProviderExtensionBuilder providerExtensionBuilder = new NetworkProviderExtensionBuilder();
105         if (network.getProviderPhysicalNetwork() != null) {
106             providerExtensionBuilder.setPhysicalNetwork(network.getProviderPhysicalNetwork());
107         }
108         if (network.getProviderSegmentationID() != null) {
109             providerExtensionBuilder.setSegmentationId(network.getProviderSegmentationID());
110         }
111         if (network.getProviderNetworkType() != null) {
112             final ImmutableBiMap<String, Class<? extends NetworkTypeBase>> mapper = NETWORK_MAP.inverse();
113             providerExtensionBuilder
114                     .setNetworkType(mapper.get(network.getProviderNetworkType()));
115         }
116         if (network.getSegments() != null) {
117             final List<Segments> segments = new ArrayList<>();
118             long count = 0;
119             for (final NeutronNetworkSegment segment : network.getSegments()) {
120                 count++;
121                 final SegmentsBuilder segmentsBuilder = new SegmentsBuilder();
122                 if (segment.getProviderPhysicalNetwork() != null) {
123                     segmentsBuilder.setPhysicalNetwork(segment.getProviderPhysicalNetwork());
124                 }
125                 if (segment.getProviderSegmentationID() != null) {
126                     segmentsBuilder.setSegmentationId(segment.getProviderSegmentationID());
127                 }
128                 if (segment.getProviderNetworkType() != null) {
129                     final ImmutableBiMap<String, Class<? extends NetworkTypeBase>> mapper = NETWORK_MAP.inverse();
130                     segmentsBuilder.setNetworkType(
131                             mapper.get(segment.getProviderNetworkType()));
132                 }
133                 segmentsBuilder.setSegmentationIndex(Long.valueOf(count));
134                 segments.add(segmentsBuilder.build());
135             }
136             providerExtensionBuilder.setSegments(segments);
137         }
138         if (network.getProviderSegmentationID() != null) {
139             providerExtensionBuilder.setSegmentationId(network.getProviderSegmentationID());
140         }
141         if (network.getQosPolicyId() != null) {
142             final QosNetworkExtensionBuilder qosExtensionBuilder = new QosNetworkExtensionBuilder();
143             qosExtensionBuilder.setQosPolicyId(toUuid(network.getQosPolicyId()));
144             networkBuilder.addAugmentation(QosNetworkExtension.class, qosExtensionBuilder.build());
145         }
146         networkBuilder.addAugmentation(NetworkL3Extension.class, l3ExtensionBuilder.build());
147         networkBuilder.addAugmentation(NetworkProviderExtension.class, providerExtensionBuilder.build());
148     }
149
150     @Override
151     protected Network toMd(NeutronNetwork network) {
152         final NetworkBuilder networkBuilder = new NetworkBuilder();
153         toMdAdminAttributes(network, networkBuilder);
154         fillExtensions(networkBuilder, network);
155
156         if (network.getShared() != null) {
157             networkBuilder.setShared(network.getShared());
158         }
159
160         if (network.getVlanTransparent() != null) {
161             networkBuilder.setVlanTransparent(network.getVlanTransparent());
162         }
163
164         return networkBuilder.build();
165     }
166
167     @Override
168     protected boolean areAllDependenciesAvailable(ReadTransaction tx, NeutronNetwork network)
169             throws ReadFailedException {
170         return ifNonNull(network.getQosPolicyId(), qosPolicyId -> qosPolicyInterface.exists(qosPolicyId, tx));
171     }
172 }