Use DataTreeChangeListener instead of DataChangeListener
[ovsdb.git] / southbound / southbound-impl / src / main / java / org / opendaylight / ovsdb / southbound / ovsdb / transact / ControllerRemovedCommand.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.Collection;
13 import java.util.Map;
14 import java.util.Set;
15
16 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
17 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
18 import org.opendaylight.ovsdb.lib.notation.Mutator;
19 import org.opendaylight.ovsdb.lib.notation.UUID;
20 import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
21 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
22 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
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.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
26 import org.opendaylight.yangtools.yang.binding.DataObject;
27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
28
29 import com.google.common.base.Optional;
30 import com.google.common.collect.Sets;
31
32 public class ControllerRemovedCommand implements TransactCommand {
33
34     @Override
35     public void execute(TransactionBuilder transaction, BridgeOperationalState state,
36                         AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> events) {
37         execute(transaction, state, TransactUtils.extractRemoved(events, ControllerEntry.class),
38                 TransactUtils.extractCreatedOrUpdatedOrRemoved(events, OvsdbBridgeAugmentation.class));
39     }
40
41     @Override
42     public void execute(TransactionBuilder transaction, BridgeOperationalState state,
43                         Collection<DataTreeModification<Node>> modifications) {
44         execute(transaction, state, TransactUtils.extractRemoved(modifications, ControllerEntry.class),
45                 TransactUtils.extractCreatedOrUpdatedOrRemoved(modifications, OvsdbBridgeAugmentation.class));
46     }
47
48     private void execute(TransactionBuilder transaction, BridgeOperationalState state,
49                          Set<InstanceIdentifier<ControllerEntry>> removedControllers,
50                          Map<InstanceIdentifier<OvsdbBridgeAugmentation>, OvsdbBridgeAugmentation>
51                                  modifiedBridges) {
52         for (InstanceIdentifier<ControllerEntry> controllerIid : removedControllers) {
53             InstanceIdentifier<OvsdbBridgeAugmentation> bridgeIid =
54                     controllerIid.firstIdentifierOf(OvsdbBridgeAugmentation.class);
55             OvsdbBridgeAugmentation ovsdbBridge = modifiedBridges.get(bridgeIid);
56             Optional<ControllerEntry> controllerEntryOptional = state.getControllerEntry(controllerIid);
57             if (ovsdbBridge != null && controllerEntryOptional.isPresent()) {
58                 ControllerEntry controllerEntry = controllerEntryOptional.get();
59                 Bridge bridge = TyperUtils.getTypedRowWrapper(transaction.getDatabaseSchema(), Bridge.class);
60                 bridge.setController(Sets.newHashSet(new UUID(controllerEntry.getControllerUuid().getValue())));
61                 transaction.add(op.mutate(bridge).addMutation(bridge.getControllerColumn().getSchema(),
62                         Mutator.DELETE, bridge.getControllerColumn().getData()));
63             }
64         }
65
66
67     }
68
69 }