/*
* Copyright (c) 2008-2017, 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.spi;
import com.hazelcast.internal.cluster.ClusterClock;
import com.hazelcast.internal.partition.InternalPartition;
import com.hazelcast.logging.ILogger;
import com.hazelcast.logging.Logger;
import com.hazelcast.nio.Address;
import com.hazelcast.nio.Connection;
import com.hazelcast.nio.ObjectDataInput;
import com.hazelcast.nio.ObjectDataOutput;
import com.hazelcast.nio.serialization.DataSerializable;
import com.hazelcast.spi.exception.RetryableException;
import com.hazelcast.spi.exception.SilentException;
import com.hazelcast.spi.properties.GroupProperty;
import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import java.io.IOException;
import java.util.concurrent.atomic.AtomicLongFieldUpdater;
import java.util.logging.Level;
import static com.hazelcast.spi.ExceptionAction.RETRY_INVOCATION;
import static com.hazelcast.spi.ExceptionAction.THROW_EXCEPTION;
import static com.hazelcast.util.EmptyStatement.ignore;
import static com.hazelcast.util.StringUtil.timeToString;
/**
* An operation could be compared to a {@link Runnable}. It contains logic that is going to be executed; this logic
* will be placed in the {@link Operation#run()} method.
*/
public abstract class Operation implements DataSerializable {
/** Marks an {@link Operation} as non-partition-specific. */
public static final int GENERIC_PARTITION_ID = -1;
static final int BITMASK_VALIDATE_TARGET = 1;
static final int BITMASK_CALLER_UUID_SET = 1 << 1;
static final int BITMASK_REPLICA_INDEX_SET = 1 << 2;
static final int BITMASK_WAIT_TIMEOUT_SET = 1 << 3;
static final int BITMASK_PARTITION_ID_32_BIT = 1 << 4;
static final int BITMASK_CALL_TIMEOUT_64_BIT = 1 << 5;
static final int BITMASK_SERVICE_NAME_SET = 1 << 6;
private static final AtomicLongFieldUpdater<Operation> CALL_ID =
AtomicLongFieldUpdater.newUpdater(Operation.class, "callId");
// serialized
private volatile long callId;
private String serviceName;
private int partitionId = GENERIC_PARTITION_ID;
private int replicaIndex;
private short flags;
private long invocationTime = -1;
private long callTimeout = Long.MAX_VALUE;
private long waitTimeout = -1;
private String callerUuid;
// injected
private transient NodeEngine nodeEngine;
private transient Object service;
private transient Address callerAddress;
private transient Connection connection;
private transient OperationResponseHandler responseHandler;
protected Operation() {
setFlag(true, BITMASK_VALIDATE_TARGET);
setFlag(true, BITMASK_CALL_TIMEOUT_64_BIT);
}
public boolean executedLocally() {
return nodeEngine.getThisAddress().equals(callerAddress);
}
public boolean isUrgent() {
return this instanceof UrgentSystemOperation;
}
// runs before wait-support
public void beforeRun() throws Exception {
}
// runs after wait-support, supposed to do actual operation
public abstract void run() throws Exception;
// runs after backups, before wait-notify
public void afterRun() throws Exception {
}
public boolean returnsResponse() {
return true;
}
public Object getResponse() {
return null;
}
// Gets the actual service name without looking at overriding methods. This method only exists for testing purposes.
String getRawServiceName() {
return serviceName;
}
public String getServiceName() {
return serviceName;
}
@SuppressFBWarnings("ES_COMPARING_PARAMETER_STRING_WITH_EQ")
public final Operation setServiceName(String serviceName) {
// If the name of the service is the same as the name already provided, the call is skipped.
// We can do a == instead of an equals because serviceName are typically constants, and it will
// prevent serialization of the service-name if it is already provided by the getServiceName.
if (serviceName == getServiceName()) {
return this;
}
this.serviceName = serviceName;
setFlag(serviceName != null, BITMASK_SERVICE_NAME_SET);
return this;
}
/**
* Returns the id of the partition that this Operation will be executed upon.
*
* If the partitionId is equal or larger than 0, it means that it is tied to a specific partition: for example,
* a map.get('foo'). If it is smaller than 0, than it means that it isn't bound to a particular partition.
*
* The partitionId should never be equal or larger than the total number of partitions. For example, if there are 271
* partitions, the maximum partitionId is 270.
*
* The partitionId is used by the OperationService to figure out which member owns a specific partition, and to send
* the operation to that member.
*
* @return the id of the partition.
* @see #setPartitionId(int)
*/
public final int getPartitionId() {
return partitionId;
}
/**
* Sets the partition id.
*
* @param partitionId the id of the partition.
* @return the updated Operation.
* @see #getPartitionId()
*/
public final Operation setPartitionId(int partitionId) {
this.partitionId = partitionId;
setFlag(partitionId > Short.MAX_VALUE, BITMASK_PARTITION_ID_32_BIT);
return this;
}
public final int getReplicaIndex() {
return replicaIndex;
}
public final Operation setReplicaIndex(int replicaIndex) {
if (replicaIndex < 0 || replicaIndex >= InternalPartition.MAX_REPLICA_COUNT) {
throw new IllegalArgumentException("Replica index is out of range [0-"
+ (InternalPartition.MAX_REPLICA_COUNT - 1) + "]");
}
setFlag(replicaIndex != 0, BITMASK_REPLICA_INDEX_SET);
this.replicaIndex = replicaIndex;
return this;
}
/**
* Gets the call ID of this operation. The call ID is used to associate the invocation of an operation
* on a remote system with the response from the execution of that operation.
* <ul>
* <li>Initially the call ID is zero.</li>
* <li>When an Invocation of the operation is created, call ID is assigned a positive value.</li>
* <li>When the invocation ends, the operation is {@link #deactivate()}d, but the call ID is preserved.</li>
* <li>The same operation may be involved in a further invocation (retrying); this will assign a
* new call ID and reactivate the operation.</li>
* </ul>
* @return the call ID
*/
public final long getCallId() {
return Math.abs(callId);
}
/**
* Tells whether this operation is involved in an ongoing invocation. Such an operation always has a
* positive call ID.
*
* @return {@code true} if the operation's invocation is active; {@code false} otherwise
*/
final boolean isActive() {
return callId > 0;
}
/**
* Marks this operation as "not involved in an ongoing invocation".
*
* @return {@code true} if this call deactivated the operation; {@code false} if it was already inactive
*/
final boolean deactivate() {
long c = callId;
if (c <= 0) {
return false;
}
if (CALL_ID.compareAndSet(this, c, -c)) {
return true;
}
if (callId > 0) {
throw new IllegalStateException("Operation concurrently re-activated while executing deactivate(). " + this);
}
return false;
}
/**
* Atomically ensures that the operation is not already involved in an invocation and sets the supplied call ID.
*
* @param newId the requested call ID, must be positive
* @throws IllegalArgumentException if the supplied call ID is non-positive
* @throws IllegalStateException if the operation already has an ongoing invocation
*/
// Accessed using OperationAccessor
final void setCallId(long newId) {
if (newId <= 0) {
throw new IllegalArgumentException(String.format("Attempted to set non-positive call ID %d on %s",
newId, this));
}
final long c = callId;
if (c > 0) {
throw new IllegalStateException(String.format(
"Attempt to overwrite the call ID of an active operation: current %d, requested %d. %s",
callId, newId, this));
}
if (!CALL_ID.compareAndSet(this, c, newId)) {
throw new IllegalStateException(String.format("Concurrent modification of call ID. Initially observed %d,"
+ " then attempted to set %d, then observed %d. %s", c, newId, callId, this));
}
onSetCallId(newId);
}
/**
* Called every time a new <tt>callId</tt> is set on the operation.
* A new <tt>callId</tt> is set before initial invocation and before every invocation retry.
* <p>
* By default this is a no-op method. Operation implementations which want to
* get notified on <tt>callId</tt> changes can override it.
* <p>
* For example an operation can distinguish the first invocation and invocation retries by keeping
* the initial <tt>callId</tt>.
*
* @param callId the new call ID that was set on the operation
*/
protected void onSetCallId(long callId) {
}
public boolean validatesTarget() {
return isFlagSet(BITMASK_VALIDATE_TARGET);
}
public final Operation setValidateTarget(boolean validateTarget) {
setFlag(validateTarget, BITMASK_VALIDATE_TARGET);
return this;
}
public final NodeEngine getNodeEngine() {
return nodeEngine;
}
public final Operation setNodeEngine(NodeEngine nodeEngine) {
this.nodeEngine = nodeEngine;
return this;
}
public final <T> T getService() {
if (service == null) {
// one might have overridden getServiceName() method...
final String name = serviceName != null ? serviceName : getServiceName();
service = nodeEngine.getService(name);
}
return (T) service;
}
public final Operation setService(Object service) {
this.service = service;
return this;
}
public final Address getCallerAddress() {
return callerAddress;
}
// Accessed using OperationAccessor
final Operation setCallerAddress(Address callerAddress) {
this.callerAddress = callerAddress;
return this;
}
public final Connection getConnection() {
return connection;
}
// Accessed using OperationAccessor
final Operation setConnection(Connection connection) {
this.connection = connection;
return this;
}
/**
* Gets the {@link OperationResponseHandler} tied to this Operation. The returned value can be null.
*
* @return the {@link OperationResponseHandler}
*/
public final OperationResponseHandler getOperationResponseHandler() {
return responseHandler;
}
/**
* Sets the {@link OperationResponseHandler}. Value is allowed to be null.
*
* @param responseHandler the {@link OperationResponseHandler} to set.
* @return this instance.
*/
public final Operation setOperationResponseHandler(OperationResponseHandler responseHandler) {
this.responseHandler = responseHandler;
return this;
}
public final void sendResponse(Object value) {
OperationResponseHandler responseHandler = getOperationResponseHandler();
responseHandler.sendResponse(this, value);
}
/**
* Gets the time in milliseconds since this invocation started.
*
* For more information, see {@link ClusterClock#getClusterTime()}.
*
* @return the time of the invocation start.
*/
public final long getInvocationTime() {
return invocationTime;
}
// Accessed using OperationAccessor
final Operation setInvocationTime(long invocationTime) {
this.invocationTime = invocationTime;
return this;
}
/**
* Gets the call timeout in milliseconds. For example, if a call should start execution within 60 seconds otherwise
* it should be aborted, then the call-timeout is 60000 milliseconds. Once an operation starts execution and runs for a
* long period (e.g. 5 minutes with an IExecutorService execute operation), then the call timeout isn't relevant any longer.
*
* For more information about the default value, see
* {@link GroupProperty#OPERATION_CALL_TIMEOUT_MILLIS}
*
* @return the call timeout in milliseconds.
* @see #setCallTimeout(long)
* @see com.hazelcast.spi.OperationAccessor#setCallTimeout(Operation, long)
*/
public final long getCallTimeout() {
return callTimeout;
}
/**
* Sets the call timeout.
*
* @param callTimeout the call timeout.
* @return the updated Operation.
* @see #getCallTimeout()
*/
// Accessed using OperationAccessor
final Operation setCallTimeout(long callTimeout) {
this.callTimeout = callTimeout;
setFlag(callTimeout > Integer.MAX_VALUE, BITMASK_CALL_TIMEOUT_64_BIT);
return this;
}
/**
* Returns the wait timeout in millis. -1 means infinite wait, and 0 means no waiting at all.
*
* The wait timeout is the amount of time a {@link BlockingOperation} is allowed to parked in the
* {@link com.hazelcast.spi.impl.operationparker.OperationParker}.
*
* Examples:
* <ol>
* <li>in case of ILock.tryLock(10, ms), the wait timeout is 10 ms</li>
* <li>in case of ILock.lock(), the wait timeout is -1</li>
* <li>in case of ILock.tryLock(), the wait timeout is 0.</li>
* </ol>
*
* The waitTimeout only has meaning for blocking operations. For non blocking operations the value is undefined.
*
* @return the wait timeout.
*/
public final long getWaitTimeout() {
return waitTimeout;
}
/**
* Sets the wait timeout in millis.
*
* @param timeout the wait timeout.
* @see #getWaitTimeout() for more detail.
*/
public final void setWaitTimeout(long timeout) {
this.waitTimeout = timeout;
setFlag(timeout != -1, BITMASK_WAIT_TIMEOUT_SET);
}
/**
* Called when an <tt>Exception</tt>/<tt>Error</tt> is thrown
* during an invocation. Invocation process will continue, retry
* or fail according to returned <tt>ExceptionAction</tt> result.
* <p/>
* This method is called on caller side of the invocation.
*
* @param throwable <tt>Exception</tt>/<tt>Error</tt> thrown during invocation
* @return <tt>ExceptionAction</tt>
*/
public ExceptionAction onInvocationException(Throwable throwable) {
return throwable instanceof RetryableException ? RETRY_INVOCATION : THROW_EXCEPTION;
}
public String getCallerUuid() {
return callerUuid;
}
public Operation setCallerUuid(String callerUuid) {
this.callerUuid = callerUuid;
setFlag(callerUuid != null, BITMASK_CALLER_UUID_SET);
return this;
}
protected final ILogger getLogger() {
final NodeEngine ne = nodeEngine;
return ne != null ? ne.getLogger(getClass()) : Logger.getLogger(getClass());
}
void setFlag(boolean value, int bitmask) {
if (value) {
flags |= bitmask;
} else {
flags &= ~bitmask;
}
}
boolean isFlagSet(int bitmask) {
return (flags & bitmask) != 0;
}
short getFlags() {
return flags;
}
/**
* Called when an <tt>Exception</tt>/<tt>Error</tt> is thrown during operation execution.
* <p/>
* By default this method does nothing.
* Operation implementations can override this behaviour due to their needs.
* <p/>
* This method is called on node & thread that's executing the operation.
*
* @param e Exception/Error thrown during operation execution
*/
public void onExecutionFailure(Throwable e) {
}
/**
* Logs <tt>Exception</tt>/<tt>Error</tt> thrown during operation execution.
* Operation implementations can override this behaviour due to their needs.
* <p/>
* This method is called on node & thread that's executing the operation.
*
* @param e Exception/Error thrown during operation execution
*/
public void logError(Throwable e) {
final ILogger logger = getLogger();
if (e instanceof SilentException) {
logger.finest(e.getMessage(), e);
} else if (e instanceof RetryableException) {
final Level level = returnsResponse() ? Level.FINEST : Level.WARNING;
if (logger.isLoggable(level)) {
logger.log(level, e.getClass().getName() + ": " + e.getMessage());
}
} else if (e instanceof OutOfMemoryError) {
try {
logger.severe(e.getMessage(), e);
} catch (Throwable ignored) {
ignore(ignored);
}
} else {
final Level level = nodeEngine != null && nodeEngine.isRunning() ? Level.SEVERE : Level.FINEST;
if (logger.isLoggable(level)) {
logger.log(level, e.getMessage(), e);
}
}
}
@Override
public final void writeData(ObjectDataOutput out) throws IOException {
// THIS HAS TO BE THE FIRST VALUE IN THE STREAM! DO NOT CHANGE!
// It is used to return deserialization exceptions to the caller.
out.writeLong(callId);
// write state next, so that it is first available on reading.
out.writeShort(flags);
if (isFlagSet(BITMASK_SERVICE_NAME_SET)) {
out.writeUTF(serviceName);
}
if (isFlagSet(BITMASK_PARTITION_ID_32_BIT)) {
out.writeInt(partitionId);
} else {
out.writeShort(partitionId);
}
if (isFlagSet(BITMASK_REPLICA_INDEX_SET)) {
out.writeByte(replicaIndex);
}
out.writeLong(invocationTime);
if (isFlagSet(BITMASK_CALL_TIMEOUT_64_BIT)) {
out.writeLong(callTimeout);
} else {
out.writeInt((int) callTimeout);
}
if (isFlagSet(BITMASK_WAIT_TIMEOUT_SET)) {
out.writeLong(waitTimeout);
}
if (isFlagSet(BITMASK_CALLER_UUID_SET)) {
out.writeUTF(callerUuid);
}
writeInternal(out);
}
@Override
public final void readData(ObjectDataInput in) throws IOException {
// THIS HAS TO BE THE FIRST VALUE IN THE STREAM! DO NOT CHANGE!
// It is used to return deserialization exceptions to the caller.
callId = in.readLong();
flags = in.readShort();
if (isFlagSet(BITMASK_SERVICE_NAME_SET)) {
serviceName = in.readUTF();
}
if (isFlagSet(BITMASK_PARTITION_ID_32_BIT)) {
partitionId = in.readInt();
} else {
partitionId = in.readShort();
}
if (isFlagSet(BITMASK_REPLICA_INDEX_SET)) {
replicaIndex = in.readByte();
}
invocationTime = in.readLong();
if (isFlagSet(BITMASK_CALL_TIMEOUT_64_BIT)) {
callTimeout = in.readLong();
} else {
callTimeout = in.readInt();
}
if (isFlagSet(BITMASK_WAIT_TIMEOUT_SET)) {
waitTimeout = in.readLong();
}
if (isFlagSet(BITMASK_CALLER_UUID_SET)) {
callerUuid = in.readUTF();
}
readInternal(in);
}
protected void writeInternal(ObjectDataOutput out) throws IOException {
}
protected void readInternal(ObjectDataInput in) throws IOException {
}
/**
* A template method allows for additional information to be passed into the {@link #toString()} method. So an Operation
* subclass can override this method and add additional debugging content. The default implementation does nothing so
* one is not forced to provide an empty implementation.
*
* It is a good practice always to call the super.toString(stringBuffer) when implementing this method to make sure
* that the super class can inject content if needed.
*
* @param sb the StringBuilder to add the debug info to.
*/
protected void toString(StringBuilder sb) {
}
@Override
public String toString() {
StringBuilder sb = new StringBuilder(getClass().getName()).append('{');
sb.append("serviceName='").append(getServiceName()).append('\'');
sb.append(", identityHash=").append(System.identityHashCode(this));
sb.append(", partitionId=").append(partitionId);
sb.append(", replicaIndex=").append(replicaIndex);
sb.append(", callId=").append(callId);
sb.append(", invocationTime=").append(invocationTime).append(" (").append(timeToString(invocationTime)).append(")");
sb.append(", waitTimeout=").append(waitTimeout);
sb.append(", callTimeout=").append(callTimeout);
toString(sb);
sb.append('}');
return sb.toString();
}
}