Merge "Fix so that operational store correctly removes protocol entries"
[ovsdb.git] / southbound / southbound-impl / src / main / java / org / opendaylight / ovsdb / southbound / transactions / md / OvsdbBridgeUpdateCommand.java
1 package org.opendaylight.ovsdb.southbound.transactions.md;
2
3 import java.util.ArrayList;
4 import java.util.List;
5 import java.util.Map;
6 import java.util.Map.Entry;
7 import java.util.Set;
8
9 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
10 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
11 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
12 import org.opendaylight.ovsdb.lib.message.TableUpdates;
13 import org.opendaylight.ovsdb.lib.notation.UUID;
14 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
15 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
16 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
17 import org.opendaylight.ovsdb.southbound.OvsdbClientKey;
18 import org.opendaylight.ovsdb.southbound.SouthboundConstants;
19 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.DatapathId;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentationBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeName;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeProtocolBase;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeRef;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentationBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeRef;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.BridgeExternalIds;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.BridgeExternalIdsBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.BridgeOtherConfigs;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.BridgeOtherConfigsBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ProtocolEntry;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ProtocolEntryKey;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntry;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntryBuilder;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
41 import org.opendaylight.yangtools.yang.binding.DataObject;
42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
43 import org.slf4j.Logger;
44 import org.slf4j.LoggerFactory;
45
46 import com.google.common.base.Optional;
47 import com.google.common.base.Preconditions;
48
49 public class OvsdbBridgeUpdateCommand extends AbstractTransactionCommand {
50     private static final Logger LOG = LoggerFactory.getLogger(OvsdbBridgeUpdateCommand.class);
51     private Map<UUID,Bridge> updatedBridgeRows;
52     private Map<UUID, Bridge> oldBridgeRows;
53
54     public OvsdbBridgeUpdateCommand(OvsdbClientKey key, TableUpdates updates,
55             DatabaseSchema dbSchema) {
56         super(key,updates,dbSchema);
57         updatedBridgeRows = TyperUtils.extractRowsUpdated(Bridge.class, getUpdates(), getDbSchema());
58         oldBridgeRows = TyperUtils.extractRowsOld(Bridge.class, getUpdates(), getDbSchema());
59     }
60
61     @Override
62     public void execute(ReadWriteTransaction transaction) {
63         for (Entry<UUID, Bridge> entry : updatedBridgeRows.entrySet()) {
64             updateBridge(transaction, entry.getValue());
65         }
66     }
67
68     private void updateBridge(ReadWriteTransaction transaction,
69             Bridge bridge) {
70         final InstanceIdentifier<Node> connectionIId = getKey().toInstanceIndentifier();
71         Optional<Node> connection = readNode(transaction, getKey().toInstanceIndentifier());
72         if (connection.isPresent()) {
73             LOG.debug("Connection {} is present",connection);
74
75             // Update the connection node to let it know it manages this bridge
76             Node connectionNode = buildConnectionNode(bridge);
77             transaction.merge(LogicalDatastoreType.OPERATIONAL, connectionIId, connectionNode);
78
79             // Update the bridge node with whatever data we are getting
80             InstanceIdentifier<Node> bridgeIid = SouthboundMapper.createInstanceIdentifier(getKey(),bridge);
81             Node bridgeNode = buildBridgeNode(bridge);
82             transaction.merge(LogicalDatastoreType.OPERATIONAL, bridgeIid, bridgeNode);
83             deleteEntries(transaction, protocolEntriesToRemove(bridgeIid,bridge));
84         }
85     }
86
87     private <T extends DataObject> void deleteEntries(ReadWriteTransaction transaction,
88             List<InstanceIdentifier<T>> entryIids) {
89         for (InstanceIdentifier<T> entryIid: entryIids) {
90             transaction.delete(LogicalDatastoreType.OPERATIONAL, entryIid);
91         }
92     }
93
94
95     private List<InstanceIdentifier<ProtocolEntry>> protocolEntriesToRemove(
96             InstanceIdentifier<Node> bridgeIid, Bridge bridge) {
97         Preconditions.checkNotNull(bridgeIid);
98         Preconditions.checkNotNull(bridge);
99         List<InstanceIdentifier<ProtocolEntry>> result =
100                 new ArrayList<InstanceIdentifier<ProtocolEntry>>();
101         Bridge oldBridge = oldBridgeRows.get(bridge.getUuid());
102
103         if (oldBridge != null && oldBridge.getProtocolsColumn() != null) {
104             for (String protocol: oldBridge.getProtocolsColumn().getData()) {
105                 if (bridge.getProtocolsColumn() == null
106                         || !bridge.getProtocolsColumn().getData().contains(protocol)) {
107                     Class<? extends OvsdbBridgeProtocolBase> proto =
108                             SouthboundConstants.OVSDB_PROTOCOL_MAP.inverse().get(protocol);
109                     InstanceIdentifier<ProtocolEntry> iid = bridgeIid
110                             .augmentation(OvsdbBridgeAugmentation.class)
111                             .child(ProtocolEntry.class,
112                                     new ProtocolEntryKey(proto));
113                     result.add(iid);
114                 }
115             }
116         }
117         return result;
118     }
119
120     private Optional<Node> readNode(ReadWriteTransaction transaction,
121             final InstanceIdentifier<Node> connectionIid) {
122         Optional<Node> node = Optional.absent();
123         try {
124             node = transaction.read(LogicalDatastoreType.OPERATIONAL, connectionIid).checkedGet();
125         } catch (final ReadFailedException e) {
126             LOG.debug("Read Operational/DS for Node fail! {}", connectionIid, e);
127         }
128         return node;
129     }
130
131     private Node buildConnectionNode(
132             Bridge bridge) {
133         //Update node with managed node reference
134         NodeBuilder connectionNode = new NodeBuilder();
135         connectionNode.setNodeId(SouthboundMapper.createNodeId(getKey().getIp(),getKey().getPort()));
136
137         OvsdbNodeAugmentationBuilder ovsdbConnectionAugmentationBuilder = new OvsdbNodeAugmentationBuilder();
138         List<ManagedNodeEntry> managedBridges = new ArrayList<ManagedNodeEntry>();
139         InstanceIdentifier<Node> bridgeIid = SouthboundMapper.createInstanceIdentifier(getKey(),bridge);
140         ManagedNodeEntry managedBridge = new ManagedNodeEntryBuilder().setBridgeRef(
141                 new OvsdbBridgeRef(bridgeIid)).build();
142         managedBridges.add(managedBridge);
143         ovsdbConnectionAugmentationBuilder.setManagedNodeEntry(managedBridges);
144
145         connectionNode.addAugmentation(OvsdbNodeAugmentation.class, ovsdbConnectionAugmentationBuilder.build());
146
147         LOG.debug("Update node with bridge node ref {}",ovsdbConnectionAugmentationBuilder.toString());
148         return connectionNode.build();
149     }
150
151     private Node buildBridgeNode(Bridge bridge) {
152         NodeBuilder bridgeNodeBuilder = new NodeBuilder();
153         InstanceIdentifier<Node> bridgeIid = SouthboundMapper.createInstanceIdentifier(getKey(),bridge);
154         NodeId bridgeNodeId = SouthboundMapper.createManagedNodeId(bridgeIid);
155         bridgeNodeBuilder.setNodeId(bridgeNodeId);
156         OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentationBuilder = new OvsdbBridgeAugmentationBuilder();
157         ovsdbBridgeAugmentationBuilder.setBridgeName(new OvsdbBridgeName(bridge.getName()));
158         ovsdbBridgeAugmentationBuilder.setBridgeUuid(new Uuid(bridge.getUuid().toString()));
159         setDataPath(ovsdbBridgeAugmentationBuilder, bridge);
160         setDataPathType(ovsdbBridgeAugmentationBuilder, bridge);
161         setProtocol(ovsdbBridgeAugmentationBuilder, bridge);
162         setExternalIds(ovsdbBridgeAugmentationBuilder, bridge);
163         setOtherConfig(ovsdbBridgeAugmentationBuilder, bridge);
164         setFailMode(ovsdbBridgeAugmentationBuilder, bridge);
165         setManagedBy(ovsdbBridgeAugmentationBuilder);
166         bridgeNodeBuilder.addAugmentation(OvsdbBridgeAugmentation.class, ovsdbBridgeAugmentationBuilder.build());
167
168         LOG.debug("Built with the intent to store bridge data {}",
169                 ovsdbBridgeAugmentationBuilder.toString());
170         return bridgeNodeBuilder.build();
171     }
172
173     private void setManagedBy(OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentationBuilder) {
174         InstanceIdentifier<Node> connectionNodePath = getKey().toInstanceIndentifier();
175         ovsdbBridgeAugmentationBuilder.setManagedBy(new OvsdbNodeRef(connectionNodePath));
176     }
177
178     private void setDataPathType(OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentationBuilder,
179             Bridge bridge) {
180         ovsdbBridgeAugmentationBuilder.setDatapathType(
181                 SouthboundMapper.createDatapathType(bridge.getDatapathTypeColumn().getData()));
182     }
183
184     private void setFailMode(OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentationBuilder,
185             Bridge bridge) {
186         if (bridge.getFailModeColumn() != null
187                 && bridge.getFailModeColumn().getData() != null
188                 && !bridge.getFailModeColumn().getData().isEmpty()) {
189             String[] failmodeArray = new String[bridge.getFailModeColumn().getData().size()];
190             bridge.getFailModeColumn().getData().toArray(failmodeArray);
191             ovsdbBridgeAugmentationBuilder.setFailMode(
192                     SouthboundConstants.OVSDB_FAIL_MODE_MAP.inverse().get(failmodeArray[0]));
193         }
194     }
195
196     private void setOtherConfig(OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentationBuilder,
197             Bridge bridge) {
198         Map<String, String> otherConfigs = bridge
199                 .getOtherConfigColumn().getData();
200         if (otherConfigs != null && !otherConfigs.isEmpty()) {
201             Set<String> otherConfigKeys = otherConfigs.keySet();
202             List<BridgeOtherConfigs> otherConfigList = new ArrayList<BridgeOtherConfigs>();
203             String otherConfigValue;
204             for (String otherConfigKey : otherConfigKeys) {
205                 otherConfigValue = otherConfigs.get(otherConfigKey);
206                 if (otherConfigKey != null && otherConfigValue != null) {
207                     otherConfigList.add(new BridgeOtherConfigsBuilder()
208                             .setBridgeOtherConfigKey(otherConfigKey)
209                             .setBridgeOtherConfigValue(otherConfigValue)
210                             .build());
211                 }
212             }
213             ovsdbBridgeAugmentationBuilder.setBridgeOtherConfigs(otherConfigList);
214         }
215     }
216
217     private void setExternalIds(OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentationBuilder,
218             Bridge bridge) {
219         Map<String, String> externalIds = bridge.getExternalIdsColumn()
220                 .getData();
221         if (externalIds != null && !externalIds.isEmpty()) {
222             Set<String> externalIdKeys = externalIds.keySet();
223             List<BridgeExternalIds> externalIdsList = new ArrayList<BridgeExternalIds>();
224             String externalIdValue;
225             for (String externalIdKey : externalIdKeys) {
226                 externalIdValue = externalIds.get(externalIdKey);
227                 if (externalIdKey != null && externalIdValue != null) {
228                     externalIdsList.add(new BridgeExternalIdsBuilder()
229                             .setBridgeExternalIdKey(externalIdKey)
230                             .setBridgeExternalIdValue(externalIdValue)
231                             .build());
232                 }
233             }
234             ovsdbBridgeAugmentationBuilder.setBridgeExternalIds(externalIdsList);
235         }
236     }
237
238     private void setProtocol(OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentationBuilder,
239             Bridge bridge) {
240         if (SouthboundMapper.createMdsalProtocols(bridge) != null
241                 && SouthboundMapper.createMdsalProtocols(bridge).size() > 0) {
242             ovsdbBridgeAugmentationBuilder.setProtocolEntry(SouthboundMapper.createMdsalProtocols(bridge));
243         }
244     }
245
246     private void setDataPath(OvsdbBridgeAugmentationBuilder ovsdbBridgeAugmentationBuilder,
247             Bridge bridge) {
248         DatapathId dpid = SouthboundMapper.createDatapathId(bridge);
249         if (dpid != null) {
250             ovsdbBridgeAugmentationBuilder.setDatapathId(dpid);
251         }
252     }
253 }