Convert ProxyDOMDataBroker tx creation to async
[netconf.git] / netconf / netconf-topology-singleton / src / main / java / org / opendaylight / netconf / topology / singleton / impl / actors / WriteAdapter.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.actors;
10
11 import akka.actor.ActorContext;
12 import akka.actor.ActorRef;
13 import akka.actor.Status.Failure;
14 import akka.actor.Status.Success;
15 import com.google.common.util.concurrent.CheckedFuture;
16 import com.google.common.util.concurrent.FutureCallback;
17 import com.google.common.util.concurrent.Futures;
18 import com.google.common.util.concurrent.MoreExecutors;
19 import javax.annotation.Nonnull;
20 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
21 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
22 import org.opendaylight.netconf.topology.singleton.messages.NormalizedNodeMessage;
23 import org.opendaylight.netconf.topology.singleton.messages.transactions.CancelRequest;
24 import org.opendaylight.netconf.topology.singleton.messages.transactions.DeleteRequest;
25 import org.opendaylight.netconf.topology.singleton.messages.transactions.MergeRequest;
26 import org.opendaylight.netconf.topology.singleton.messages.transactions.PutRequest;
27 import org.opendaylight.netconf.topology.singleton.messages.transactions.SubmitRequest;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
30
31 class WriteAdapter {
32
33     private static final Logger LOG = LoggerFactory.getLogger(WriteAdapter.class);
34
35     private final DOMDataWriteTransaction tx;
36
37     WriteAdapter(final DOMDataWriteTransaction tx) {
38         this.tx = tx;
39     }
40
41     private void cancel(final ActorContext context, final ActorRef sender, final ActorRef self) {
42         final boolean cancelled = tx.cancel();
43         sender.tell(cancelled, self);
44         context.stop(self);
45     }
46
47     private void submit(final ActorRef requester, final ActorRef self, final ActorContext context) {
48         final CheckedFuture<Void, TransactionCommitFailedException> submitFuture = tx.submit();
49         context.stop(self);
50         Futures.addCallback(submitFuture, new FutureCallback<Void>() {
51             @Override
52             public void onSuccess(final Void result) {
53                 requester.tell(new Success(null), self);
54             }
55
56             @Override
57             public void onFailure(@Nonnull final Throwable throwable) {
58                 requester.tell(new Failure(throwable), self);
59             }
60         }, MoreExecutors.directExecutor());
61     }
62
63     @SuppressWarnings("checkstyle:IllegalCatch")
64     public void handle(final Object message, final ActorRef sender, final ActorContext context, final ActorRef self) {
65         // we need to catch everything, since an unchecked exception can be thrown from the underlying parse.
66         // TODO Maybe we should store it and fail the submit immediately?.
67         try {
68             if (message instanceof MergeRequest) {
69                 final MergeRequest mergeRequest = (MergeRequest) message;
70                 final NormalizedNodeMessage data = mergeRequest.getNormalizedNodeMessage();
71                 tx.merge(mergeRequest.getStore(), data.getIdentifier(), data.getNode());
72             } else if (message instanceof PutRequest) {
73                 final PutRequest putRequest = (PutRequest) message;
74                 final NormalizedNodeMessage data = putRequest.getNormalizedNodeMessage();
75                 tx.put(putRequest.getStore(), data.getIdentifier(), data.getNode());
76             } else if (message instanceof DeleteRequest) {
77                 final DeleteRequest deleteRequest = (DeleteRequest) message;
78                 tx.delete(deleteRequest.getStore(), deleteRequest.getPath());
79             } else if (message instanceof CancelRequest) {
80                 cancel(context, sender, self);
81             } else if (message instanceof SubmitRequest) {
82                 submit(sender, self, context);
83             }
84
85         } catch (final RuntimeException exception) {
86             LOG.error("Write command has failed.", exception);
87         }
88     }
89 }