- private void abortOperations(final Throwable cause) {
- // This acts as a barrier, application threads check this after they have added an entry in the maps,
- // and if they observe aborted being non-null, they will perform their cleanup and not return the handle.
- aborted = cause;
-
- for (ClientLocalHistory h : histories.values()) {
- h.localAbort(cause);
- }
- histories.clear();
- }
-
- private DistributedDataStoreClientBehavior shutdown(final ClientActorBehavior currentBehavior) {
- abortOperations(new IllegalStateException("Client " + getIdentifier() + " has been shut down"));
- return null;
- }
-
- @Override
- protected DistributedDataStoreClientBehavior onCommand(final Object command) {
- if (command instanceof GetClientRequest) {
- ((GetClientRequest) command).getReplyTo().tell(new Status.Success(this), ActorRef.noSender());
- } else {
- LOG.warn("{}: ignoring unhandled command {}", persistenceId(), command);
- }
-
- return this;
- }
-
- //
- //
- // Methods below are invoked from application threads
- //
- //
-
- @SuppressWarnings("checkstyle:IllegalCatch")
- private static <K, V extends LocalAbortable> V returnIfOperational(final Map<K , V> map, final K key, final V value,
- final Throwable aborted) {
- Verify.verify(map.put(key, value) == null);
-
- if (aborted != null) {
- try {
- value.localAbort(aborted);
- } catch (Exception e) {
- LOG.debug("Close of {} failed", value, e);
- }
- map.remove(key, value);
- throw Throwables.propagate(aborted);
- }
-
- return value;
- }
-
- @Override
- public ClientLocalHistory createLocalHistory() {
- final LocalHistoryIdentifier historyId = new LocalHistoryIdentifier(getIdentifier(),
- nextHistoryId.getAndIncrement());
- final ClientLocalHistory history = new ClientLocalHistory(this, historyId);
- LOG.debug("{}: creating a new local history {}", persistenceId(), history);
-
- return returnIfOperational(histories, historyId, history, aborted);