pekko/fun-test-java/src/test/java/se/scalablesolutions/akka/api/PersistentStateful.java

package se.scalablesolutions.akka.api;

import se.scalablesolutions.akka.kernel.state.*;
import se.scalablesolutions.akka.annotation.transactional;
import se.scalablesolutions.akka.annotation.state;

public class PersistentStateful {
  private TransactionalState factory = new TransactionalState();
  private TransactionalMap mapState =       factory.newPersistentMap(new CassandraStorageConfig());
  private TransactionalVector vectorState = factory.newPersistentVector(new CassandraStorageConfig());;
  private TransactionalRef refState =       factory.newPersistentRef(new CassandraStorageConfig());

  @transactional
  public String getMapState(String key) {
    return (String) mapState.get(key).get();
  }

  @transactional
  public String getVectorState(int index) {
    return (String) vectorState.get(index);
  }

  @transactional
  public String getRefState() {
    if (refState.isDefined()) {
      return (String) refState.get().get();
    } else throw new IllegalStateException("No such element");
  }

  @transactional
  public void setMapState(String key, String msg) {
    mapState.put(key, msg);
  }

  @transactional
  public void setVectorState(String msg) {
    vectorState.add(msg);
  }

  @transactional
  public void setRefState(String msg) {
    refState.swap(msg);
  }

  @transactional
  public void success(String key, String msg) {
    mapState.put(key, msg);
    vectorState.add(msg);
    refState.swap(msg);
  }

  @transactional
  public String failure(String key, String msg, PersistentFailer failer) {
    mapState.put(key, msg);
    vectorState.add(msg);
    refState.swap(msg);
    failer.fail();
    return msg;
  }

  public void success(String key, String msg, PersistentStatefulNested nested) {
    mapState.put(key, msg);
    vectorState.add(msg);
    refState.swap(msg);
    nested.success(key, msg);
  }

  @transactional
  public String failure(String key, String msg, PersistentStatefulNested nested, PersistentFailer failer) {
    mapState.put(key, msg);
    vectorState.add(msg);
    refState.swap(msg);
    nested.failure(key, msg, failer);
    return msg;
  }


  @transactional
  public void thisMethodHangs(String key, String msg, PersistentFailer failer) {
    setMapState(key, msg);
  }
}