netconf/netconf-topology-singleton/src/main/java/org/opendaylight/netconf/topology/singleton/impl/ProxyDOMDataBroker.java

   1 /*
   2  * Copyright (c) 2017 Pantheon Technologies s.r.o. 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
   9 package org.opendaylight.netconf.topology.singleton.impl;
  10
  11 import akka.actor.ActorRef;
  12 import akka.actor.ActorSystem;
  13 import akka.pattern.Patterns;
  14 import akka.util.Timeout;
  15 import com.google.common.base.Verify;
  16 import java.util.Collections;
  17 import java.util.Map;
  18 import javax.annotation.Nonnull;
  19 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
  20 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
  21 import org.opendaylight.controller.md.sal.dom.api.DOMDataBrokerExtension;
  22 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadOnlyTransaction;
  23 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction;
  24 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
  25 import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
  26 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
  27 import org.opendaylight.netconf.topology.singleton.impl.tx.ProxyReadTransaction;
  28 import org.opendaylight.netconf.topology.singleton.impl.tx.ProxyReadWriteTransaction;
  29 import org.opendaylight.netconf.topology.singleton.impl.tx.ProxyWriteTransaction;
  30 import org.opendaylight.netconf.topology.singleton.messages.transactions.NewReadTransactionReply;
  31 import org.opendaylight.netconf.topology.singleton.messages.transactions.NewReadTransactionRequest;
  32 import org.opendaylight.netconf.topology.singleton.messages.transactions.NewReadWriteTransactionReply;
  33 import org.opendaylight.netconf.topology.singleton.messages.transactions.NewReadWriteTransactionRequest;
  34 import org.opendaylight.netconf.topology.singleton.messages.transactions.NewWriteTransactionReply;
  35 import org.opendaylight.netconf.topology.singleton.messages.transactions.NewWriteTransactionRequest;
  36 import scala.concurrent.Await;
  37 import scala.concurrent.Future;
  38
  39 public class ProxyDOMDataBroker implements DOMDataBroker {
  40
  41     private final Timeout askTimeout;
  42     private final RemoteDeviceId id;
  43     private final ActorRef masterNode;
  44     private final ActorSystem actorSystem;
  45
  46     /**
  47      * Constructor for {@code ProxyDOMDataBroker}.
  48      *
  49      * @param actorSystem system
  50      * @param id          id
  51      * @param masterNode  {@link org.opendaylight.netconf.topology.singleton.impl.actors.NetconfNodeActor} ref
  52      * @param askTimeout  ask timeout
  53      */
  54     public ProxyDOMDataBroker(final ActorSystem actorSystem, final RemoteDeviceId id,
  55                               final ActorRef masterNode, final Timeout askTimeout) {
  56         this.id = id;
  57         this.masterNode = masterNode;
  58         this.actorSystem = actorSystem;
  59         this.askTimeout = askTimeout;
  60     }
  61
  62     @SuppressWarnings("checkstyle:IllegalCatch")
  63     @Override
  64     public DOMDataReadOnlyTransaction newReadOnlyTransaction() {
  65         final Future<Object> txActorFuture = Patterns.ask(masterNode, new NewReadTransactionRequest(), askTimeout);
  66         final Object msg;
  67         try {
  68             msg = Await.result(txActorFuture, askTimeout.duration());
  69         } catch (Exception e) {
  70             throw new IllegalStateException("Can't create ProxyReadTransaction", e);
  71         }
  72
  73         if (msg instanceof Exception) {
  74             throw new IllegalStateException("Can't create ProxyReadTransaction", (Exception) msg);
  75         }
  76
  77         Verify.verify(msg instanceof NewReadTransactionReply);
  78         final NewReadTransactionReply reply = (NewReadTransactionReply) msg;
  79         return new ProxyReadTransaction(reply.getTxActor(), id, actorSystem, askTimeout);
  80     }
  81
  82     @SuppressWarnings("checkstyle:IllegalCatch")
  83     @Override
  84     public DOMDataReadWriteTransaction newReadWriteTransaction() {
  85         final Future<Object> txActorFuture = Patterns.ask(masterNode, new NewReadWriteTransactionRequest(), askTimeout);
  86         final Object msg;
  87         try {
  88             msg = Await.result(txActorFuture, askTimeout.duration());
  89         } catch (Exception e) {
  90             throw new IllegalStateException("Can't create ProxyReadWriteTransaction", e);
  91         }
  92
  93         if (msg instanceof Exception) {
  94             throw new IllegalStateException("Can't create ProxyReadWriteTransaction", (Exception) msg);
  95         }
  96
  97         Verify.verify(msg instanceof NewReadWriteTransactionReply);
  98         final NewReadWriteTransactionReply reply = (NewReadWriteTransactionReply) msg;
  99         return new ProxyReadWriteTransaction(reply.getTxActor(), id, actorSystem, askTimeout);
 100     }
 101
 102     @SuppressWarnings("checkstyle:IllegalCatch")
 103     @Override
 104     public DOMDataWriteTransaction newWriteOnlyTransaction() {
 105         final Future<Object> txActorFuture = Patterns.ask(masterNode, new NewWriteTransactionRequest(), askTimeout);
 106         final Object msg;
 107         try {
 108             msg = Await.result(txActorFuture, askTimeout.duration());
 109         } catch (Exception e) {
 110             throw new IllegalStateException("Can't create ProxyWriteTransaction", e);
 111         }
 112
 113         if (msg instanceof Exception) {
 114             throw new IllegalStateException("Can't create ProxyWriteTransaction", (Exception) msg);
 115         }
 116
 117         Verify.verify(msg instanceof NewWriteTransactionReply);
 118         final NewWriteTransactionReply reply = (NewWriteTransactionReply) msg;
 119         return new ProxyWriteTransaction(reply.getTxActor(), id, actorSystem, askTimeout);
 120     }
 121
 122     @Override
 123     public DOMTransactionChain createTransactionChain(final TransactionChainListener listener) {
 124         throw new UnsupportedOperationException(id + ": Transaction chains not supported for netconf mount point");
 125     }
 126
 127     @Nonnull
 128     @Override
 129     public Map<Class<? extends DOMDataBrokerExtension>, DOMDataBrokerExtension> getSupportedExtensions() {
 130         return Collections.emptyMap();
 131     }
 132 }