southbound/southbound-impl/src/main/java/org/opendaylight/ovsdb/southbound/ovsdb/transact/ProtocolUpdateCommand.java

   1 package org.opendaylight.ovsdb.southbound.ovsdb.transact;
   2
   3 import static org.opendaylight.ovsdb.lib.operations.Operations.op;
   4
   5 import java.util.Map;
   6 import java.util.Map.Entry;
   7
   8 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
   9 import org.opendaylight.ovsdb.lib.notation.Mutator;
  10 import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
  11 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
  12 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
  13 import org.opendaylight.ovsdb.southbound.SouthboundConstants;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ProtocolEntry;
  16 import org.opendaylight.yangtools.yang.binding.DataObject;
  17 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  18
  19 import com.google.common.base.Optional;
  20 import com.google.common.collect.Sets;
  21
  22 public class ProtocolUpdateCommand extends AbstractTransactCommand {
  23
  24     private Map<InstanceIdentifier<ProtocolEntry>, ProtocolEntry> protocols;
  25     private Map<InstanceIdentifier<OvsdbBridgeAugmentation>, OvsdbBridgeAugmentation> bridges;
  26
  27     public ProtocolUpdateCommand(BridgeOperationalState state,
  28             AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> changes) {
  29         super(state, changes);
  30         protocols = TransactUtils.extractCreatedOrUpdated(getChanges(), ProtocolEntry.class);
  31         bridges = TransactUtils.extractCreatedOrUpdated(getChanges(), OvsdbBridgeAugmentation.class);
  32     }
  33
  34     @Override
  35     public void execute(TransactionBuilder transaction) {
  36         for (Entry<InstanceIdentifier<ProtocolEntry>, ProtocolEntry> entry: protocols.entrySet()) {
  37             Optional<ProtocolEntry> operationalProtocolEntryOptional =
  38                     getOperationalState().getProtocolEntry(entry.getKey());
  39             if (!operationalProtocolEntryOptional.isPresent()) {
  40                 InstanceIdentifier<OvsdbBridgeAugmentation> bridgeIid =
  41                         entry.getKey().firstIdentifierOf(OvsdbBridgeAugmentation.class);
  42                 Optional<OvsdbBridgeAugmentation> bridgeOptional =
  43                         getOperationalState().getOvsdbBridgeAugmentation(bridgeIid);
  44                 OvsdbBridgeAugmentation ovsdbBridge = null;
  45                 if (bridgeOptional.isPresent()) {
  46                     ovsdbBridge = bridgeOptional.get();
  47                 } else {
  48                     ovsdbBridge = bridges.get(bridgeIid);
  49                 }
  50                 if (ovsdbBridge != null
  51                         && ovsdbBridge.getBridgeName() != null
  52                         && entry.getValue() != null
  53                         && entry.getValue().getProtocol() != null) {
  54                     String protocolString = SouthboundConstants.OVSDB_PROTOCOL_MAP.get(entry.getValue().getProtocol());
  55                     if (protocolString != null) {
  56                         Bridge bridge = TyperUtils.getTypedRowWrapper(transaction.getDatabaseSchema(), Bridge.class);
  57                         bridge.setName(ovsdbBridge.getBridgeName().getValue());
  58                         bridge.setProtocols(Sets.newHashSet(protocolString));
  59                         transaction.add(op.mutate(bridge)
  60                                 .addMutation(bridge.getProtocolsColumn().getSchema(),
  61                                         Mutator.INSERT,bridge.getProtocolsColumn().getData())
  62                                 .where(bridge.getNameColumn().getSchema().opEqual(bridge.getNameColumn().getData()))
  63                                 .build());
  64                     }
  65                 }
  66             }
  67         }
  68     }
  69
  70 }