Migrate to UntypedAbstractActor
[netconf.git] / netconf / netconf-topology-singleton / src / main / java / org / opendaylight / netconf / topology / singleton / impl / actors / ReadWriteTransactionActor.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.Props;
12 import akka.actor.ReceiveTimeout;
13 import akka.actor.UntypedAbstractActor;
14 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction;
15 import org.opendaylight.netconf.topology.singleton.messages.transactions.ReadActorMessage;
16 import org.opendaylight.netconf.topology.singleton.messages.transactions.WriteActorMessage;
17 import org.slf4j.Logger;
18 import org.slf4j.LoggerFactory;
19 import scala.concurrent.duration.Duration;
20
21 public final class ReadWriteTransactionActor extends UntypedAbstractActor {
22
23     private static final Logger LOG = LoggerFactory.getLogger(ReadWriteTransactionActor.class);
24
25     private final DOMDataReadWriteTransaction tx;
26     private final long idleTimeout;
27     private final ReadAdapter readAdapter;
28     private final WriteAdapter writeAdapter;
29
30     private ReadWriteTransactionActor(final DOMDataReadWriteTransaction tx, final Duration idleTimeout) {
31         this.tx = tx;
32         this.idleTimeout = idleTimeout.toSeconds();
33         if (this.idleTimeout > 0) {
34             context().setReceiveTimeout(idleTimeout);
35         }
36         readAdapter = new ReadAdapter(tx);
37         writeAdapter = new WriteAdapter(tx);
38     }
39
40     /**
41      * Creates new actor Props.
42      *
43      * @param tx          delegate device read write transaction
44      * @param idleTimeout idle time in seconds, after which transaction is closed automatically
45      * @return props
46      */
47     static Props props(final DOMDataReadWriteTransaction tx, final Duration idleTimeout) {
48         return Props.create(ReadWriteTransactionActor.class, () -> new ReadWriteTransactionActor(tx, idleTimeout));
49     }
50
51     @Override
52     public void onReceive(final Object message) throws Throwable {
53         if (message instanceof ReadActorMessage) {
54             readAdapter.handle(message, sender(), self());
55         } else if (message instanceof WriteActorMessage) {
56             writeAdapter.handle(message, sender(), context(), self());
57         } else if (message instanceof ReceiveTimeout) {
58             LOG.warn("Haven't received any message for {} seconds, cancelling transaction and stopping actor",
59                     idleTimeout);
60             tx.cancel();
61             context().stop(self());
62         } else {
63             unhandled(message);
64         }
65     }
66
67 }