/*
* Copyright (c) 2008-2013, Hazelcast, Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hazelcast.executor;
import com.hazelcast.core.ICompletableFuture;
import com.hazelcast.nio.Address;
import com.hazelcast.spi.InvocationBuilder;
import com.hazelcast.spi.NodeEngine;
import com.hazelcast.spi.OperationService;
import com.hazelcast.util.executor.DelegatingFuture;
import java.util.concurrent.CancellationException;
import java.util.concurrent.Future;
import static com.hazelcast.util.ExceptionUtil.rethrow;
final class CancellableDelegatingFuture<V> extends DelegatingFuture<V> {
public static final int CANCEL_TRY_COUNT = 50;
public static final int CANCEL_TRY_PAUSE_MILLIS = 250;
private final NodeEngine nodeEngine;
private final String uuid;
private final int partitionId;
private final Address target;
private volatile boolean cancelled;
CancellableDelegatingFuture(ICompletableFuture future, NodeEngine nodeEngine, String uuid, int partitionId) {
super(future, nodeEngine.getSerializationService());
this.nodeEngine = nodeEngine;
this.uuid = uuid;
this.partitionId = partitionId;
this.target = null;
}
CancellableDelegatingFuture(ICompletableFuture future, NodeEngine nodeEngine, String uuid, Address target) {
super(future, nodeEngine.getSerializationService());
this.nodeEngine = nodeEngine;
this.uuid = uuid;
this.target = target;
this.partitionId = -1;
}
CancellableDelegatingFuture(ICompletableFuture future, V defaultValue, NodeEngine nodeEngine,
String uuid, int partitionId) {
super(future, nodeEngine.getSerializationService(), defaultValue);
this.nodeEngine = nodeEngine;
this.uuid = uuid;
this.partitionId = partitionId;
this.target = null;
}
@Override
public boolean cancel(boolean mayInterruptIfRunning) {
if (isDone() || cancelled) {
return false;
}
Future f = invokeCancelOperation(mayInterruptIfRunning);
try {
Boolean b = (Boolean) f.get();
if (b != null && b) {
setError(new CancellationException());
cancelled = true;
return true;
}
return false;
} catch (Exception e) {
throw rethrow(e);
} finally {
setDone();
}
}
private Future invokeCancelOperation(boolean mayInterruptIfRunning) {
CancellationOperation op = new CancellationOperation(uuid, mayInterruptIfRunning);
OperationService opService = nodeEngine.getOperationService();
InvocationBuilder builder;
if (partitionId > -1) {
builder = opService.createInvocationBuilder(DistributedExecutorService.SERVICE_NAME, op, partitionId);
} else {
builder = opService.createInvocationBuilder(DistributedExecutorService.SERVICE_NAME, op, target);
}
builder.setTryCount(CANCEL_TRY_COUNT).setTryPauseMillis(CANCEL_TRY_PAUSE_MILLIS);
return builder.invoke();
}
@Override
public boolean isCancelled() {
return cancelled;
}
}