Break out ControllerUpdate and Remove into seperate commands
[ovsdb.git] / southbound / southbound-impl / src / main / java / org / opendaylight / ovsdb / southbound / ovsdb / transact / ControllerUpdateCommand.java
1 /*
2  * Copyright (c) 2014 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.ovsdb.southbound.ovsdb.transact;
9
10 import static org.opendaylight.ovsdb.lib.operations.Operations.op;
11
12 import java.util.Map;
13 import java.util.Map.Entry;
14
15 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
16 import org.opendaylight.ovsdb.lib.notation.Mutator;
17 import org.opendaylight.ovsdb.lib.notation.UUID;
18 import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
19 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
20 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
21 import org.opendaylight.ovsdb.schema.openvswitch.Controller;
22 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ControllerEntry;
25 import org.opendaylight.yangtools.yang.binding.DataObject;
26 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
27
28 import com.google.common.base.Optional;
29 import com.google.common.collect.Sets;
30
31 public class ControllerUpdateCommand extends AbstractTransactCommand {
32
33
34     public ControllerUpdateCommand(BridgeOperationalState state,
35             AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> changes) {
36         super(state, changes);
37     }
38
39     @Override
40     public void execute(TransactionBuilder transaction) {
41         Map<InstanceIdentifier<ControllerEntry>, ControllerEntry> controllers =
42                 TransactUtils.extractCreatedOrUpdated(getChanges(), ControllerEntry.class);
43         Map<InstanceIdentifier<OvsdbBridgeAugmentation>, OvsdbBridgeAugmentation> bridges =
44                 TransactUtils.extractCreatedOrUpdated(getChanges(), OvsdbBridgeAugmentation.class);
45         for (Entry<InstanceIdentifier<ControllerEntry>, ControllerEntry> entry: controllers.entrySet()) {
46             Optional<ControllerEntry> operationalControllerEntryOptional =
47                     getOperationalState().getControllerEntry(entry.getKey());
48             if (!operationalControllerEntryOptional.isPresent()) {
49                 InstanceIdentifier<OvsdbBridgeAugmentation> bridgeIid =
50                         entry.getKey().firstIdentifierOf(OvsdbBridgeAugmentation.class);
51                 Optional<OvsdbBridgeAugmentation> bridgeOptional =
52                         getOperationalState().getOvsdbBridgeAugmentation(bridgeIid);
53                 OvsdbBridgeAugmentation ovsdbBridge = null;
54                 if (bridgeOptional.isPresent()) {
55                     ovsdbBridge = bridgeOptional.get();
56                 } else {
57                     ovsdbBridge = bridges.get(bridgeIid);
58                 }
59                 if (ovsdbBridge != null
60                         && ovsdbBridge.getBridgeName() != null
61                         && entry.getValue() != null
62                         && entry.getValue().getTarget() != null) {
63                     Controller controller =
64                             TyperUtils.getTypedRowWrapper(transaction.getDatabaseSchema(), Controller.class);
65                     controller.setTarget(entry.getValue().getTarget().getValue());
66                     String controllerNamedUuidString = SouthboundMapper.getRandomUUID();
67                     UUID controllerNamedUuid = new UUID(controllerNamedUuidString);
68                     transaction.add(op.insert(controller).withId(controllerNamedUuidString));
69
70                     Bridge bridge = TyperUtils.getTypedRowWrapper(transaction.getDatabaseSchema(), Bridge.class);
71                     bridge.setName(ovsdbBridge.getBridgeName().getValue());
72                     bridge.setController(Sets.newHashSet(controllerNamedUuid));
73                     transaction.add(op.mutate(bridge)
74                             .addMutation(bridge.getControllerColumn().getSchema(), Mutator.INSERT,
75                                     bridge.getControllerColumn().getData())
76                             .where(bridge.getNameColumn().getSchema().opEqual(bridge.getNameColumn().getData()))
77                             .build());
78                 }
79             }
80         }
81
82     }
83
84 }