opendaylight/md-sal/sal-binding-broker/src/main/java/org/opendaylight/controller/md/sal/binding/impl/ForwardedBackwardsCompatibleDataBroker.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.controller.md.sal.binding.impl;
   9
  10 import java.util.ArrayList;
  11 import java.util.Collections;
  12 import java.util.HashMap;
  13 import java.util.HashSet;
  14 import java.util.List;
  15 import java.util.Map;
  16 import java.util.Set;
  17 import java.util.concurrent.Callable;
  18 import java.util.concurrent.ConcurrentHashMap;
  19 import java.util.concurrent.ExecutionException;
  20
  21 import org.opendaylight.controller.md.sal.binding.api.BindingDataChangeListener;
  22 import org.opendaylight.controller.md.sal.common.api.RegistrationListener;
  23 import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
  24 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  25 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  26 import org.opendaylight.controller.md.sal.common.api.data.DataChangeEvent;
  27 import org.opendaylight.controller.md.sal.common.api.data.DataCommitHandler;
  28 import org.opendaylight.controller.md.sal.common.api.data.DataCommitHandler.DataCommitTransaction;
  29 import org.opendaylight.controller.md.sal.common.api.data.DataCommitHandlerRegistration;
  30 import org.opendaylight.controller.md.sal.common.api.data.DataReader;
  31 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  32 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
  33 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction;
  34 import org.opendaylight.controller.sal.binding.api.data.DataChangeListener;
  35 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
  36 import org.opendaylight.controller.sal.binding.api.data.DataProviderService;
  37 import org.opendaylight.controller.sal.common.util.Rpcs;
  38 import org.opendaylight.yangtools.concepts.AbstractObjectRegistration;
  39 import org.opendaylight.yangtools.concepts.Delegator;
  40 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  41 import org.opendaylight.yangtools.concepts.Registration;
  42 import org.opendaylight.yangtools.concepts.util.ListenerRegistry;
  43 import org.opendaylight.yangtools.yang.binding.DataObject;
  44 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  45 import org.opendaylight.yangtools.yang.common.RpcError;
  46 import org.opendaylight.yangtools.yang.common.RpcResult;
  47 import org.opendaylight.yangtools.yang.data.impl.codec.BindingIndependentMappingService;
  48 import org.slf4j.Logger;
  49 import org.slf4j.LoggerFactory;
  50
  51 import com.google.common.base.Function;
  52 import com.google.common.util.concurrent.AsyncFunction;
  53 import com.google.common.util.concurrent.FutureCallback;
  54 import com.google.common.util.concurrent.Futures;
  55 import com.google.common.util.concurrent.ListenableFuture;
  56 import com.google.common.util.concurrent.ListeningExecutorService;
  57
  58 public class ForwardedBackwardsCompatibleDataBroker extends AbstractForwardedDataBroker implements DataProviderService, AutoCloseable {
  59
  60     private static final Logger LOG = LoggerFactory.getLogger(ForwardedBackwardsCompatibleDataBroker.class);
  61
  62     private final ConcurrentHashMap<InstanceIdentifier<?>, CommitHandlerRegistrationImpl> commitHandlers = new ConcurrentHashMap<>();
  63     private final ListenerRegistry<DataChangeListener> fakeRegistry = ListenerRegistry.create();
  64     private final ListeningExecutorService executorService;
  65
  66     public ForwardedBackwardsCompatibleDataBroker(final DOMDataBroker domDataBroker,
  67             final BindingIndependentMappingService mappingService, final ListeningExecutorService executor) {
  68         super(domDataBroker, mappingService);
  69         executorService = executor;
  70         LOG.info("ForwardedBackwardsCompatibleBroker started.");
  71     }
  72
  73     @Override
  74     public DataModificationTransaction beginTransaction() {
  75         return new ForwardedBackwardsCompatibleTransacion(getDelegate().newReadWriteTransaction(), getCodec());
  76     }
  77
  78     @Override
  79     public DataObject readConfigurationData(final InstanceIdentifier<? extends DataObject> path) {
  80         DataModificationTransaction tx = beginTransaction();
  81         return tx.readConfigurationData(path);
  82     }
  83
  84     @Override
  85     public DataObject readOperationalData(final InstanceIdentifier<? extends DataObject> path) {
  86         DataModificationTransaction tx = beginTransaction();
  87         return tx.readOperationalData(path);
  88     }
  89
  90     @Override
  91     public Registration<DataCommitHandler<InstanceIdentifier<? extends DataObject>, DataObject>> registerCommitHandler(
  92             final InstanceIdentifier<? extends DataObject> path,
  93             final DataCommitHandler<InstanceIdentifier<? extends DataObject>, DataObject> commitHandler) {
  94
  95
  96         //transformingCommitHandler = new TransformingDataChangeListener
  97         //fakeCommitHandler =  registerDataChangeListener(LogicalDatastoreType.CONFIGURATION, path, listener, DataChangeScope.SUBTREE);
  98
  99         CommitHandlerRegistrationImpl reg = new CommitHandlerRegistrationImpl(path, commitHandler);
 100         commitHandlers.put(path, reg);
 101         return reg;
 102     }
 103
 104     @Override
 105     @Deprecated
 106     public ListenerRegistration<RegistrationListener<DataCommitHandlerRegistration<InstanceIdentifier<? extends DataObject>, DataObject>>> registerCommitHandlerListener(
 107             final RegistrationListener<DataCommitHandlerRegistration<InstanceIdentifier<? extends DataObject>, DataObject>> commitHandlerListener) {
 108         throw new UnsupportedOperationException("Not supported contract.");
 109     }
 110
 111     @Override
 112     public ListenerRegistration<DataChangeListener> registerDataChangeListener(
 113             final InstanceIdentifier<? extends DataObject> path, final DataChangeListener listener) {
 114
 115
 116         BindingDataChangeListener asyncOperListener = new BackwardsCompatibleOperationalDataChangeInvoker(listener);
 117         BindingDataChangeListener asyncCfgListener = new BackwardsCompatibleConfigurationDataChangeInvoker(listener);
 118
 119         ListenerRegistration<BindingDataChangeListener> cfgReg = registerDataChangeListener(LogicalDatastoreType.CONFIGURATION, path, asyncCfgListener, DataChangeScope.SUBTREE);
 120         ListenerRegistration<BindingDataChangeListener> operReg = registerDataChangeListener(LogicalDatastoreType.OPERATIONAL, path, asyncOperListener, DataChangeScope.SUBTREE);
 121
 122         return new LegacyListenerRegistration(listener,cfgReg,operReg);
 123     }
 124
 125     @Override
 126     public Registration<DataReader<InstanceIdentifier<? extends DataObject>, DataObject>> registerDataReader(
 127             final InstanceIdentifier<? extends DataObject> path,
 128             final DataReader<InstanceIdentifier<? extends DataObject>, DataObject> reader) {
 129         throw new UnsupportedOperationException("Data reader contract is not supported.");
 130     }
 131
 132     @Override
 133     public void close() throws Exception {
 134         // TODO Auto-generated method stub
 135
 136     }
 137
 138     public ListenableFuture<RpcResult<TransactionStatus>> commit(final ForwardedBackwardsCompatibleTransacion tx) {
 139
 140         final List<DataCommitTransaction<InstanceIdentifier<? extends DataObject>, DataObject>> subTrans = new ArrayList<>();
 141         LOG.debug("Tx: {} Submitted.",tx.getIdentifier());
 142         ListenableFuture<Boolean> requestCommit = executorService.submit(new Callable<Boolean>() {
 143
 144             @Override
 145             public Boolean call() throws Exception {
 146                 try {
 147                     for (CommitHandlerRegistrationImpl handler : commitHandlers.values()) {
 148
 149                         DataCommitTransaction<InstanceIdentifier<? extends DataObject>, DataObject> subTx = handler
 150                                 .getInstance().requestCommit(tx);
 151                         subTrans.add(subTx);
 152                     }
 153                 } catch (Exception e) {
 154                     LOG.error("Tx: {} Rollback.",tx.getIdentifier(),e);
 155                     for (DataCommitTransaction<InstanceIdentifier<? extends DataObject>, DataObject> subTx : subTrans) {
 156                         subTx.rollback();
 157                     }
 158                     return false;
 159                 }
 160                 LOG.debug("Tx: {} Can Commit True.",tx.getIdentifier());
 161                 return true;
 162             }
 163
 164         });
 165
 166         ListenableFuture<RpcResult<TransactionStatus>> dataStoreCommit = Futures.transform(requestCommit, new AsyncFunction<Boolean, RpcResult<TransactionStatus>>() {
 167
 168             @Override
 169             public ListenableFuture<RpcResult<TransactionStatus>> apply(final Boolean requestCommitSuccess) throws Exception {
 170                 if(requestCommitSuccess) {
 171                     return tx.getDelegate().commit();
 172                 }
 173                 return Futures.immediateFuture(Rpcs.getRpcResult(false, TransactionStatus.FAILED, Collections.<RpcError>emptySet()));
 174             }
 175         });
 176
 177         return Futures.transform(dataStoreCommit, new Function<RpcResult<TransactionStatus>,RpcResult<TransactionStatus>>() {
 178             @Override
 179             public RpcResult<TransactionStatus> apply(final RpcResult<TransactionStatus> input) {
 180                 if(input.isSuccessful()) {
 181                     for(DataCommitTransaction<InstanceIdentifier<? extends DataObject>, DataObject> subTx : subTrans ) {
 182                         subTx.finish();
 183                     }
 184                 } else {
 185                     LOG.error("Tx: {} Rollback - Datastore commit failed.",tx.getIdentifier());
 186                     for(DataCommitTransaction<InstanceIdentifier<? extends DataObject>, DataObject> subTx : subTrans ) {
 187                         subTx.rollback();
 188                     }
 189                 }
 190                 return input;
 191             }
 192         });
 193     }
 194
 195     private class ForwardedBackwardsCompatibleTransacion extends
 196             AbstractForwardedTransaction<DOMDataReadWriteTransaction> implements DataModificationTransaction {
 197
 198         private final Map<InstanceIdentifier<? extends DataObject>, DataObject> updated = new HashMap<>();
 199         private final Map<InstanceIdentifier<? extends DataObject>, DataObject> created = new HashMap<>();
 200         private final Set<InstanceIdentifier<? extends DataObject>> removed = new HashSet<>();
 201         private final Map<InstanceIdentifier<? extends DataObject>, DataObject> original = new HashMap<>();
 202         private TransactionStatus status = TransactionStatus.NEW;
 203
 204         private final Set<InstanceIdentifier<? extends DataObject>> posponedRemovedOperational = new HashSet<>();
 205         private final Set<InstanceIdentifier<? extends DataObject>> posponedRemovedConfiguration = new HashSet<>();
 206
 207
 208         @Override
 209         public final TransactionStatus getStatus() {
 210             return status;
 211         }
 212
 213         protected ForwardedBackwardsCompatibleTransacion(final DOMDataReadWriteTransaction delegate,
 214                 final BindingToNormalizedNodeCodec codec) {
 215             super(delegate, codec);
 216             LOG.debug("Tx {} allocated.",getIdentifier());
 217         }
 218
 219         @Override
 220         public void putOperationalData(final InstanceIdentifier<? extends DataObject> path, final DataObject data) {
 221             posponedRemovedOperational.remove(path);
 222             doPutWithEnsureParents(getDelegate(), LogicalDatastoreType.OPERATIONAL, path, data);
 223         }
 224
 225         @Override
 226         public void putConfigurationData(final InstanceIdentifier<? extends DataObject> path, final DataObject data) {
 227             posponedRemovedConfiguration.remove(path);
 228             DataObject originalObj = readConfigurationData(path);
 229             if (originalObj != null) {
 230                 original.put(path, originalObj);
 231
 232             } else {
 233                 created.put(path, data);
 234             }
 235             updated.put(path, data);
 236             doPutWithEnsureParents(getDelegate(), LogicalDatastoreType.CONFIGURATION, path, data);
 237         }
 238
 239         @Override
 240         public void removeOperationalData(final InstanceIdentifier<? extends DataObject> path) {
 241             posponedRemovedOperational.add(path);
 242         }
 243
 244         @Override
 245         public void removeConfigurationData(final InstanceIdentifier<? extends DataObject> path) {
 246             posponedRemovedConfiguration.add(path);
 247         }
 248
 249         @Override
 250         public Map<InstanceIdentifier<? extends DataObject>, DataObject> getCreatedOperationalData() {
 251             return Collections.emptyMap();
 252         }
 253
 254         @Override
 255         public Map<InstanceIdentifier<? extends DataObject>, DataObject> getCreatedConfigurationData() {
 256             return created;
 257         }
 258
 259         @Override
 260         public Map<InstanceIdentifier<? extends DataObject>, DataObject> getUpdatedOperationalData() {
 261             return Collections.emptyMap();
 262         }
 263
 264         @Override
 265         public Map<InstanceIdentifier<? extends DataObject>, DataObject> getUpdatedConfigurationData() {
 266             return updated;
 267         }
 268
 269         @Override
 270         public Set<InstanceIdentifier<? extends DataObject>> getRemovedConfigurationData() {
 271             return removed;
 272         }
 273
 274         @Override
 275         public Set<InstanceIdentifier<? extends DataObject>> getRemovedOperationalData() {
 276             return Collections.emptySet();
 277         }
 278
 279         @Override
 280         public Map<InstanceIdentifier<? extends DataObject>, DataObject> getOriginalConfigurationData() {
 281             return original;
 282         }
 283
 284         @Override
 285         public Map<InstanceIdentifier<? extends DataObject>, DataObject> getOriginalOperationalData() {
 286             return Collections.emptyMap();
 287         }
 288
 289         @Override
 290         public DataObject readOperationalData(final InstanceIdentifier<? extends DataObject> path) {
 291             try {
 292                 return doRead(getDelegate(), LogicalDatastoreType.OPERATIONAL, path).get().orNull();
 293             } catch (InterruptedException | ExecutionException e) {
 294                 LOG.error("Read of {} failed.", path,e);
 295                 return null;
 296             }
 297         }
 298
 299         @Override
 300         public DataObject readConfigurationData(final InstanceIdentifier<? extends DataObject> path) {
 301             try {
 302                 return doRead(getDelegate(), LogicalDatastoreType.CONFIGURATION, path).get().orNull();
 303             } catch (InterruptedException | ExecutionException e) {
 304                 LOG.error("Read of {} failed.", path,e);
 305                 return null;
 306             }
 307         }
 308
 309         @Override
 310         public Object getIdentifier() {
 311             return getDelegate().getIdentifier();
 312         }
 313
 314         private void changeStatus(final TransactionStatus status) {
 315             LOG.trace("Transaction {} changed status to {}", getIdentifier(), status);
 316             this.status = status;
 317         }
 318
 319         @Override
 320         public ListenableFuture<RpcResult<TransactionStatus>> commit() {
 321
 322             for(InstanceIdentifier<? extends DataObject> path : posponedRemovedConfiguration) {
 323                 doDelete(getDelegate(), LogicalDatastoreType.CONFIGURATION, path);
 324             }
 325
 326             for(InstanceIdentifier<? extends DataObject> path : posponedRemovedOperational) {
 327                 doDelete(getDelegate(), LogicalDatastoreType.OPERATIONAL, path);
 328             }
 329
 330             final ListenableFuture<RpcResult<TransactionStatus>> f = ForwardedBackwardsCompatibleDataBroker.this.commit(this);
 331
 332             changeStatus(TransactionStatus.SUBMITED);
 333
 334             Futures.addCallback(f, new FutureCallback<RpcResult<TransactionStatus>>() {
 335                 @Override
 336                 public void onSuccess(final RpcResult<TransactionStatus> result) {
 337                     changeStatus(result.getResult());
 338                 }
 339
 340                 @Override
 341                 public void onFailure(final Throwable t) {
 342                     LOG.error("Transaction {} failed to complete", getIdentifier(), t);
 343                     changeStatus(TransactionStatus.FAILED);
 344                 }
 345             });
 346
 347             return f;
 348         }
 349
 350         @Override
 351         public ListenerRegistration<DataTransactionListener> registerListener(final DataTransactionListener listener) {
 352             throw new UnsupportedOperationException();
 353         }
 354
 355     }
 356
 357     private class CommitHandlerRegistrationImpl extends
 358             AbstractObjectRegistration<DataCommitHandler<InstanceIdentifier<? extends DataObject>, DataObject>> {
 359
 360         private final InstanceIdentifier<? extends DataObject> path;
 361
 362         public CommitHandlerRegistrationImpl(final InstanceIdentifier<? extends DataObject> path,
 363                 final DataCommitHandler<InstanceIdentifier<? extends DataObject>, DataObject> commitHandler) {
 364             super(commitHandler);
 365             this.path = path;
 366         }
 367
 368         @Override
 369         protected void removeRegistration() {
 370             commitHandlers.remove(path, this);
 371         }
 372
 373     }
 374
 375
 376     private static final class LegacyListenerRegistration implements ListenerRegistration<DataChangeListener> {
 377
 378         private final DataChangeListener instance;
 379         private final ListenerRegistration<BindingDataChangeListener> cfgReg;
 380         private final ListenerRegistration<BindingDataChangeListener> operReg;
 381
 382         public LegacyListenerRegistration(final DataChangeListener listener,
 383                 final ListenerRegistration<BindingDataChangeListener> cfgReg,
 384                 final ListenerRegistration<BindingDataChangeListener> operReg) {
 385             this.instance = listener;
 386             this.cfgReg = cfgReg;
 387             this.operReg = operReg;
 388         }
 389
 390         @Override
 391         public DataChangeListener getInstance() {
 392             return instance;
 393         }
 394
 395         @Override
 396         public void close() {
 397             cfgReg.close();
 398             operReg.close();
 399         }
 400
 401     }
 402
 403     private static class BackwardsCompatibleOperationalDataChangeInvoker implements BindingDataChangeListener, Delegator<DataChangeListener> {
 404
 405         private final org.opendaylight.controller.md.sal.common.api.data.DataChangeListener<?,?> delegate;
 406
 407
 408         public BackwardsCompatibleOperationalDataChangeInvoker(final DataChangeListener listener) {
 409             this.delegate = listener;
 410         }
 411
 412         @SuppressWarnings({ "unchecked", "rawtypes" })
 413         @Override
 414         public void onDataChanged(final AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
 415
 416             DataChangeEvent legacyChange = LegacyDataChangeEvent.createOperational(change);
 417             delegate.onDataChanged(legacyChange);
 418
 419         }
 420
 421         @Override
 422         public DataChangeListener getDelegate() {
 423             return (DataChangeListener) delegate;
 424         }
 425
 426     }
 427
 428     private static class BackwardsCompatibleConfigurationDataChangeInvoker implements BindingDataChangeListener, Delegator<DataChangeListener> {
 429
 430
 431         @SuppressWarnings("rawtypes")
 432         private final org.opendaylight.controller.md.sal.common.api.data.DataChangeListener<?,?> delegate;
 433
 434         public BackwardsCompatibleConfigurationDataChangeInvoker(final DataChangeListener listener) {
 435             this.delegate = listener;
 436         }
 437
 438         @SuppressWarnings({ "unchecked", "rawtypes" })
 439         @Override
 440         public void onDataChanged(final AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
 441
 442             DataChangeEvent legacyChange = LegacyDataChangeEvent.createConfiguration(change);
 443
 444             delegate.onDataChanged(legacyChange);
 445
 446         }
 447
 448         @Override
 449         public DataChangeListener getDelegate() {
 450             return (DataChangeListener) delegate;
 451         }
 452
 453     }
 454 }