import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.Futures;
+import java.util.Collections;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ExecutionException;
import org.opendaylight.bgpcep.topology.TopologyReference;
+import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
import org.opendaylight.controller.md.sal.binding.api.DataBroker;
import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
+import org.opendaylight.controller.md.sal.common.api.data.AsyncTransaction;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.controller.md.sal.common.api.data.TransactionChain;
+import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
import org.opendaylight.protocol.bgp.rib.RibReference;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.Route;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
+import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
+import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.TopologyTypes;
import org.opendaylight.yangtools.yang.binding.DataObject;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-public abstract class AbstractTopologyBuilder<T extends Route> implements AutoCloseable, DataChangeListener, LocRIBListener, TopologyReference {
+public abstract class AbstractTopologyBuilder<T extends Route> implements AutoCloseable, DataChangeListener, LocRIBListener, TopologyReference, TransactionChainListener {
private static final Logger LOG = LoggerFactory.getLogger(AbstractTopologyBuilder.class);
- private final RibReference locRibReference;
private final InstanceIdentifier<Topology> topology;
- private final DataBroker dataProvider;
+ private final BindingTransactionChain chain;
+ private final RibReference locRibReference;
private final Class<T> idClass;
protected AbstractTopologyBuilder(final DataBroker dataProvider, final RibReference locRibReference,
final TopologyId topologyId, final TopologyTypes types, final Class<T> idClass) {
- this.dataProvider = Preconditions.checkNotNull(dataProvider);
this.locRibReference = Preconditions.checkNotNull(locRibReference);
this.idClass = Preconditions.checkNotNull(idClass);
+ this.chain = dataProvider.createTransactionChain(this);
final TopologyKey tk = new TopologyKey(Preconditions.checkNotNull(topologyId));
this.topology = InstanceIdentifier.builder(NetworkTopology.class).child(Topology.class, tk).toInstance();
LOG.debug("Initiating topology builder from {} at {}", locRibReference, this.topology);
- final ReadWriteTransaction t = dataProvider.newReadWriteTransaction();
+ final ReadWriteTransaction t = this.chain.newReadWriteTransaction();
final Optional<Topology> o;
try {
o = t.read(LogicalDatastoreType.OPERATIONAL, this.topology).get();
Preconditions.checkState(!o.isPresent(), "Data provider conflict detected on object {}", this.topology);
t.put(LogicalDatastoreType.OPERATIONAL, this.topology,
- new TopologyBuilder().setKey(tk).setServerProvided(Boolean.TRUE).setTopologyTypes(types).build());
+ new TopologyBuilder().setKey(tk).setServerProvided(Boolean.TRUE).setTopologyTypes(types)
+ .setLink(Collections.<Link>emptyList()).setNode(Collections.<Node>emptyList()).build(), true);
Futures.addCallback(t.submit(), new FutureCallback<Void>() {
@Override
public void onSuccess(final Void result) {
protected abstract void removeObject(ReadWriteTransaction trans, InstanceIdentifier<T> id, T value);
- public final DataBroker getDataProvider() {
- return this.dataProvider;
- }
-
@Override
public final InstanceIdentifier<Topology> getInstanceIdentifier() {
return this.topology;
}
final Map<InstanceIdentifier<?>, ? extends DataObject> o = event.getOriginalData();
- final Map<InstanceIdentifier<?>, DataObject> n = event.getUpdatedData();
+ final Map<InstanceIdentifier<?>, DataObject> u = event.getUpdatedData();
+ final Map<InstanceIdentifier<?>, DataObject> c = event.getCreatedData();
for (final InstanceIdentifier<T> i : ids) {
final T oldValue = this.idClass.cast(o.get(i));
- final T newValue = this.idClass.cast(n.get(i));
+ T newValue = this.idClass.cast(u.get(i));
+ if (newValue == null) {
+ newValue = this.idClass.cast(c.get(i));
+ }
LOG.debug("Updating object {} value {} -> {}", i, oldValue, newValue);
if (oldValue != null) {
@Override
public final void close() throws TransactionCommitFailedException {
LOG.info("Shutting down builder for {}", getInstanceIdentifier());
- final WriteTransaction trans = this.dataProvider.newWriteOnlyTransaction();
+ final WriteTransaction trans = this.chain.newWriteOnlyTransaction();
trans.delete(LogicalDatastoreType.OPERATIONAL, getInstanceIdentifier());
trans.submit().checkedGet();
+ this.chain.close();
}
@Override
public final void onDataChanged(final AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
- final ReadWriteTransaction trans = this.dataProvider.newReadWriteTransaction();
+ final ReadWriteTransaction trans = this.chain.newReadWriteTransaction();
try {
onLocRIBChange(trans, change);
return;
}
}
+
+ @Override
+ public final void onTransactionChainFailed(final TransactionChain<?, ?> chain, final AsyncTransaction<?, ?> transaction, final Throwable cause) {
+ // TODO: restart?
+ LOG.error("Topology builder for {} failed in transaction {}", getInstanceIdentifier(), transaction.getIdentifier(), cause);
+ }
+
+ @Override
+ public final void onTransactionChainSuccessful(final TransactionChain<?, ?> chain) {
+ LOG.info("Topology builder for {} shut down", getInstanceIdentifier());
+ }
}