2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.controller.cluster.databroker.actors.dds;
10 import com.google.common.base.Optional;
11 import com.google.common.util.concurrent.CheckedFuture;
12 import com.google.common.util.concurrent.Futures;
13 import com.google.common.util.concurrent.ListenableFuture;
14 import com.google.common.util.concurrent.SettableFuture;
15 import java.util.function.Consumer;
16 import javax.annotation.Nullable;
17 import org.opendaylight.controller.cluster.access.commands.AbortLocalTransactionRequest;
18 import org.opendaylight.controller.cluster.access.commands.AbstractLocalTransactionRequest;
19 import org.opendaylight.controller.cluster.access.commands.AbstractReadTransactionRequest;
20 import org.opendaylight.controller.cluster.access.commands.CommitLocalTransactionRequest;
21 import org.opendaylight.controller.cluster.access.commands.ExistsTransactionRequest;
22 import org.opendaylight.controller.cluster.access.commands.ExistsTransactionSuccess;
23 import org.opendaylight.controller.cluster.access.commands.IncrementTransactionSequenceRequest;
24 import org.opendaylight.controller.cluster.access.commands.ModifyTransactionRequest;
25 import org.opendaylight.controller.cluster.access.commands.ModifyTransactionRequestBuilder;
26 import org.opendaylight.controller.cluster.access.commands.PersistenceProtocol;
27 import org.opendaylight.controller.cluster.access.commands.ReadTransactionRequest;
28 import org.opendaylight.controller.cluster.access.commands.ReadTransactionSuccess;
29 import org.opendaylight.controller.cluster.access.commands.TransactionAbortRequest;
30 import org.opendaylight.controller.cluster.access.commands.TransactionDelete;
31 import org.opendaylight.controller.cluster.access.commands.TransactionDoCommitRequest;
32 import org.opendaylight.controller.cluster.access.commands.TransactionMerge;
33 import org.opendaylight.controller.cluster.access.commands.TransactionModification;
34 import org.opendaylight.controller.cluster.access.commands.TransactionPreCommitRequest;
35 import org.opendaylight.controller.cluster.access.commands.TransactionPurgeRequest;
36 import org.opendaylight.controller.cluster.access.commands.TransactionRequest;
37 import org.opendaylight.controller.cluster.access.commands.TransactionSuccess;
38 import org.opendaylight.controller.cluster.access.commands.TransactionWrite;
39 import org.opendaylight.controller.cluster.access.concepts.RequestFailure;
40 import org.opendaylight.controller.cluster.access.concepts.Response;
41 import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
42 import org.opendaylight.controller.cluster.datastore.util.AbstractDataTreeModificationCursor;
43 import org.opendaylight.mdsal.common.api.ReadFailedException;
44 import org.opendaylight.yangtools.util.concurrent.MappingCheckedFuture;
45 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
46 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
47 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
48 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
49 import org.slf4j.Logger;
50 import org.slf4j.LoggerFactory;
53 * An {@link AbstractProxyTransaction} for dispatching a transaction towards a shard leader whose location is currently
54 * not known or is known to be not co-located with the client.
57 * It packages operations and sends them via the client actor queue to the shard leader. That queue is responsible for
58 * maintaining any submitted operations until the leader is discovered.
61 * This class is not safe to access from multiple application threads, as is usual for transactions. Its internal state
62 * transitions based on backend responses are thread-safe.
64 * @author Robert Varga
66 final class RemoteProxyTransaction extends AbstractProxyTransaction {
67 private static final Logger LOG = LoggerFactory.getLogger(RemoteProxyTransaction.class);
69 // FIXME: make this tuneable
70 private static final int REQUEST_MAX_MODIFICATIONS = 1000;
72 private final ModifyTransactionRequestBuilder builder;
73 private final boolean sendReadyOnSeal;
74 private final boolean snapshotOnly;
76 private boolean builderBusy;
78 private volatile Exception operationFailure;
80 RemoteProxyTransaction(final ProxyHistory parent, final TransactionIdentifier identifier,
81 final boolean snapshotOnly, final boolean sendReadyOnSeal, final boolean isDone) {
82 super(parent, isDone);
83 this.snapshotOnly = snapshotOnly;
84 this.sendReadyOnSeal = sendReadyOnSeal;
85 builder = new ModifyTransactionRequestBuilder(identifier, localActor());
89 boolean isSnapshotOnly() {
94 public TransactionIdentifier getIdentifier() {
95 return builder.getIdentifier();
99 void doDelete(final YangInstanceIdentifier path) {
100 appendModification(new TransactionDelete(path), Optional.absent());
104 void doMerge(final YangInstanceIdentifier path, final NormalizedNode<?, ?> data) {
105 appendModification(new TransactionMerge(path, data), Optional.absent());
109 void doWrite(final YangInstanceIdentifier path, final NormalizedNode<?, ?> data) {
110 appendModification(new TransactionWrite(path, data), Optional.absent());
113 private <T> CheckedFuture<T, ReadFailedException> sendReadRequest(final AbstractReadTransactionRequest<?> request,
114 final Consumer<Response<?, ?>> completer, final ListenableFuture<T> future) {
115 // Check if a previous operation failed. If it has, do not bother sending anything and report a failure
116 final Exception local = operationFailure;
118 return Futures.immediateFailedCheckedFuture(new ReadFailedException("Previous operation failed", local));
121 // Make sure we send any modifications before issuing a read
122 ensureFlushedBuider();
123 sendRequest(request, completer);
124 return MappingCheckedFuture.create(future, ReadFailedException.MAPPER);
128 CheckedFuture<Boolean, ReadFailedException> doExists(final YangInstanceIdentifier path) {
129 final SettableFuture<Boolean> future = SettableFuture.create();
130 return sendReadRequest(new ExistsTransactionRequest(getIdentifier(), nextSequence(), localActor(), path,
131 isSnapshotOnly()), t -> completeExists(future, t), future);
135 CheckedFuture<Optional<NormalizedNode<?, ?>>, ReadFailedException> doRead(final YangInstanceIdentifier path) {
136 final SettableFuture<Optional<NormalizedNode<?, ?>>> future = SettableFuture.create();
137 return sendReadRequest(new ReadTransactionRequest(getIdentifier(), nextSequence(), localActor(), path,
138 isSnapshotOnly()), t -> completeRead(future, t), future);
141 private void ensureInitializedBuilder() {
143 builder.setSequence(nextSequence());
148 private void ensureFlushedBuider() {
149 ensureFlushedBuider(Optional.absent());
152 private void ensureFlushedBuider(final Optional<Long> enqueuedTicks) {
154 flushBuilder(enqueuedTicks);
158 private void flushBuilder(final Optional<Long> enqueuedTicks) {
159 final ModifyTransactionRequest request = builder.build();
162 sendModification(request, enqueuedTicks);
165 private void sendModification(final TransactionRequest<?> request, final Optional<Long> enqueuedTicks) {
166 if (enqueuedTicks.isPresent()) {
167 enqueueRequest(request, response -> completeModify(request, response), enqueuedTicks.get().longValue());
169 sendRequest(request, response -> completeModify(request, response));
173 private void appendModification(final TransactionModification modification) {
174 appendModification(modification, Optional.absent());
177 private void appendModification(final TransactionModification modification, final Optional<Long> enqueuedTicks) {
178 if (operationFailure == null) {
179 ensureInitializedBuilder();
181 builder.addModification(modification);
182 if (builder.size() >= REQUEST_MAX_MODIFICATIONS) {
183 flushBuilder(enqueuedTicks);
186 LOG.debug("Transaction {} failed, not attempting further transactions", getIdentifier());
190 private void completeModify(final TransactionRequest<?> request, final Response<?, ?> response) {
191 LOG.debug("Modification request {} completed with {}", request, response);
193 if (response instanceof TransactionSuccess) {
195 recordSuccessfulRequest(request);
197 recordFailedResponse(response);
201 private Exception recordFailedResponse(final Response<?, ?> response) {
202 final Exception failure;
203 if (response instanceof RequestFailure) {
204 failure = ((RequestFailure<?, ?>) response).getCause();
206 LOG.warn("Unhandled response {}", response);
207 failure = new IllegalArgumentException("Unhandled response " + response.getClass());
210 if (operationFailure == null) {
211 LOG.debug("Transaction {} failed", getIdentifier(), failure);
212 operationFailure = failure;
217 private void failFuture(final SettableFuture<?> future, final Response<?, ?> response) {
218 future.setException(recordFailedResponse(response));
221 private void completeExists(final SettableFuture<Boolean> future, final Response<?, ?> response) {
222 LOG.debug("Exists request completed with {}", response);
224 if (response instanceof ExistsTransactionSuccess) {
225 future.set(((ExistsTransactionSuccess) response).getExists());
227 failFuture(future, response);
230 recordFinishedRequest(response);
233 private void completeRead(final SettableFuture<Optional<NormalizedNode<?, ?>>> future,
234 final Response<?, ?> response) {
235 LOG.debug("Read request completed with {}", response);
237 if (response instanceof ReadTransactionSuccess) {
238 future.set(((ReadTransactionSuccess) response).getData());
240 failFuture(future, response);
243 recordFinishedRequest(response);
247 ModifyTransactionRequest abortRequest() {
248 ensureInitializedBuilder();
251 return builder.build();
255 ModifyTransactionRequest commitRequest(final boolean coordinated) {
256 ensureInitializedBuilder();
257 builder.setCommit(coordinated);
259 return builder.build();
262 private ModifyTransactionRequest readyRequest() {
263 ensureInitializedBuilder();
266 return builder.build();
270 boolean sealAndSend(final Optional<Long> enqueuedTicks) {
271 if (sendReadyOnSeal) {
272 ensureInitializedBuilder();
274 flushBuilder(enqueuedTicks);
276 return super.sealAndSend(enqueuedTicks);
280 void flushState(final AbstractProxyTransaction successor) {
282 final ModifyTransactionRequest request = builder.build();
284 forwardToSuccessor(successor, request, null);
289 void forwardToRemote(final RemoteProxyTransaction successor, final TransactionRequest<?> request,
290 final Consumer<Response<?, ?>> callback) {
291 successor.handleForwardedRequest(request, callback);
294 private void handleForwardedRequest(final TransactionRequest<?> request, final Consumer<Response<?, ?>> callback) {
295 if (request instanceof ModifyTransactionRequest) {
296 final ModifyTransactionRequest req = (ModifyTransactionRequest) request;
298 req.getModifications().forEach(this::appendModification);
300 final java.util.Optional<PersistenceProtocol> maybeProto = req.getPersistenceProtocol();
301 if (maybeProto.isPresent()) {
302 // Persistence protocol implies we are sealed, propagate the marker, but hold off doing other actions
303 // until we know what we are going to do.
308 final TransactionRequest<?> tmp;
309 switch (maybeProto.get()) {
311 tmp = abortRequest();
312 sendRequest(tmp, resp -> {
313 completeModify(tmp, resp);
314 callback.accept(resp);
318 tmp = commitRequest(false);
319 sendRequest(tmp, resp -> {
320 completeModify(tmp, resp);
321 callback.accept(resp);
325 tmp = commitRequest(true);
326 sendRequest(tmp, resp -> {
327 recordSuccessfulRequest(tmp);
328 callback.accept(resp);
332 tmp = readyRequest();
333 sendRequest(tmp, resp -> {
334 recordSuccessfulRequest(tmp);
335 callback.accept(resp);
339 throw new IllegalArgumentException("Unhandled protocol " + maybeProto.get());
342 } else if (request instanceof ReadTransactionRequest) {
343 ensureFlushedBuider();
344 sendRequest(new ReadTransactionRequest(getIdentifier(), nextSequence(), localActor(),
345 ((ReadTransactionRequest) request).getPath(), isSnapshotOnly()), resp -> {
346 recordFinishedRequest(resp);
347 callback.accept(resp);
349 } else if (request instanceof ExistsTransactionRequest) {
350 ensureFlushedBuider();
351 sendRequest(new ExistsTransactionRequest(getIdentifier(), nextSequence(), localActor(),
352 ((ExistsTransactionRequest) request).getPath(), isSnapshotOnly()), resp -> {
353 recordFinishedRequest(resp);
354 callback.accept(resp);
356 } else if (request instanceof TransactionPreCommitRequest) {
357 ensureFlushedBuider();
358 final TransactionRequest<?> tmp = new TransactionPreCommitRequest(getIdentifier(), nextSequence(),
360 sendRequest(tmp, resp -> {
361 recordSuccessfulRequest(tmp);
362 callback.accept(resp);
364 } else if (request instanceof TransactionDoCommitRequest) {
365 ensureFlushedBuider();
366 sendRequest(new TransactionDoCommitRequest(getIdentifier(), nextSequence(), localActor()), callback);
367 } else if (request instanceof TransactionAbortRequest) {
368 ensureFlushedBuider();
369 sendDoAbort(callback);
370 } else if (request instanceof TransactionPurgeRequest) {
371 enqueuePurge(callback);
373 throw new IllegalArgumentException("Unhandled request {}" + request);
378 void forwardToLocal(final LocalProxyTransaction successor, final TransactionRequest<?> request,
379 final Consumer<Response<?, ?>> callback) {
380 successor.handleForwardedRemoteRequest(request, callback);
384 void handleReplayedLocalRequest(final AbstractLocalTransactionRequest<?> request,
385 final Consumer<Response<?, ?>> callback, final long enqueuedTicks) {
386 if (request instanceof CommitLocalTransactionRequest) {
387 replayLocalCommitRequest((CommitLocalTransactionRequest) request, callback, enqueuedTicks);
388 } else if (request instanceof AbortLocalTransactionRequest) {
389 enqueueRequest(abortRequest(), callback, enqueuedTicks);
391 throw new IllegalStateException("Unhandled request " + request);
395 private void replayLocalCommitRequest(final CommitLocalTransactionRequest request,
396 final Consumer<Response<?, ?>> callback, final long enqueuedTicks) {
397 final DataTreeModification mod = request.getModification();
398 final Optional<Long> optTicks = Optional.of(Long.valueOf(enqueuedTicks));
400 mod.applyToCursor(new AbstractDataTreeModificationCursor() {
402 public void write(final PathArgument child, final NormalizedNode<?, ?> data) {
403 appendModification(new TransactionWrite(current().node(child), data), optTicks);
407 public void merge(final PathArgument child, final NormalizedNode<?, ?> data) {
408 appendModification(new TransactionMerge(current().node(child), data), optTicks);
412 public void delete(final PathArgument child) {
413 appendModification(new TransactionDelete(current().node(child)), optTicks);
417 enqueueRequest(commitRequest(request.isCoordinated()), callback, enqueuedTicks);
421 void handleReplayedRemoteRequest(final TransactionRequest<?> request,
422 final @Nullable Consumer<Response<?, ?>> callback, final long enqueuedTicks) {
423 final Consumer<Response<?, ?>> cb = callback != null ? callback : resp -> { };
424 final Optional<Long> optTicks = Optional.of(Long.valueOf(enqueuedTicks));
426 if (request instanceof ModifyTransactionRequest) {
427 final ModifyTransactionRequest req = (ModifyTransactionRequest) request;
428 for (TransactionModification mod : req.getModifications()) {
429 appendModification(mod, optTicks);
432 final java.util.Optional<PersistenceProtocol> maybeProto = req.getPersistenceProtocol();
433 if (maybeProto.isPresent()) {
434 // Persistence protocol implies we are sealed, propagate the marker, but hold off doing other actions
435 // until we know what we are going to do.
440 final TransactionRequest<?> tmp;
441 switch (maybeProto.get()) {
443 tmp = abortRequest();
444 enqueueRequest(tmp, resp -> {
445 completeModify(tmp, resp);
450 tmp = commitRequest(false);
451 enqueueRequest(tmp, resp -> {
452 completeModify(tmp, resp);
457 tmp = commitRequest(true);
458 enqueueRequest(tmp, resp -> {
459 recordSuccessfulRequest(tmp);
464 tmp = readyRequest();
465 enqueueRequest(tmp, resp -> {
466 recordSuccessfulRequest(tmp);
471 throw new IllegalArgumentException("Unhandled protocol " + maybeProto.get());
474 } else if (request instanceof ReadTransactionRequest) {
475 ensureFlushedBuider(optTicks);
476 enqueueRequest(new ReadTransactionRequest(getIdentifier(), nextSequence(), localActor(),
477 ((ReadTransactionRequest) request).getPath(), isSnapshotOnly()), resp -> {
478 recordFinishedRequest(resp);
481 } else if (request instanceof ExistsTransactionRequest) {
482 ensureFlushedBuider(optTicks);
483 enqueueRequest(new ExistsTransactionRequest(getIdentifier(), nextSequence(), localActor(),
484 ((ExistsTransactionRequest) request).getPath(), isSnapshotOnly()), resp -> {
485 recordFinishedRequest(resp);
488 } else if (request instanceof TransactionPreCommitRequest) {
489 ensureFlushedBuider(optTicks);
490 final TransactionRequest<?> tmp = new TransactionPreCommitRequest(getIdentifier(), nextSequence(),
492 enqueueRequest(tmp, resp -> {
493 recordSuccessfulRequest(tmp);
496 } else if (request instanceof TransactionDoCommitRequest) {
497 ensureFlushedBuider(optTicks);
498 enqueueRequest(new TransactionDoCommitRequest(getIdentifier(), nextSequence(), localActor()), callback,
500 } else if (request instanceof TransactionAbortRequest) {
501 ensureFlushedBuider(optTicks);
502 enqueueDoAbort(callback, enqueuedTicks);
503 } else if (request instanceof TransactionPurgeRequest) {
504 enqueuePurge(callback, enqueuedTicks);
505 } else if (request instanceof IncrementTransactionSequenceRequest) {
506 final IncrementTransactionSequenceRequest req = (IncrementTransactionSequenceRequest) request;
507 ensureFlushedBuider(optTicks);
508 enqueueRequest(new IncrementTransactionSequenceRequest(getIdentifier(), nextSequence(), localActor(),
509 snapshotOnly, req.getIncrement()), callback, enqueuedTicks);
510 incrementSequence(req.getIncrement());
512 throw new IllegalArgumentException("Unhandled request {}" + request);