Search in sources :

Example 1 with RecordActivityTaskHeartbeatResponse

use of com.uber.cadence.RecordActivityTaskHeartbeatResponse in project cadence-client by uber-java.

the class ActivityExecutionContextImpl method recordActivityHeartbeat.

/**
 * @throws CancellationException
 * @see ActivityExecutionContext#recordActivityHeartbeat(Object)
 */
@Override
public void recordActivityHeartbeat(Object details) throws ActivityCompletionException {
    // TODO: call service with the specified minimal interval (through @ActivityExecutionOptions)
    // allowing more frequent calls of this method.
    RecordActivityTaskHeartbeatRequest r = new RecordActivityTaskHeartbeatRequest();
    r.setTaskToken(task.getTaskToken());
    byte[] serialized = dataConverter.toData(details);
    r.setDetails(serialized);
    RecordActivityTaskHeartbeatResponse status;
    try {
        status = service.RecordActivityTaskHeartbeat(r);
        if (status.isCancelRequested()) {
            throw new ActivityCancelledException(task);
        }
    } catch (EntityNotExistsError e) {
        throw new ActivityNotExistsException(task, e);
    } catch (BadRequestError e) {
        throw new ActivityCompletionFailureException(task, e);
    } catch (TException e) {
        log.warn("Failure heartbeating on activityID=" + task.getActivityId() + " of Workflow=" + task.getWorkflowExecution(), e);
    // Not rethrowing to not fail activity implementation on intermittent connection or Cadence errors.
    }
}
Also used : ActivityNotExistsException(com.uber.cadence.client.ActivityNotExistsException) TException(org.apache.thrift.TException) ActivityCancelledException(com.uber.cadence.client.ActivityCancelledException) RecordActivityTaskHeartbeatResponse(com.uber.cadence.RecordActivityTaskHeartbeatResponse) ActivityCompletionFailureException(com.uber.cadence.client.ActivityCompletionFailureException) RecordActivityTaskHeartbeatRequest(com.uber.cadence.RecordActivityTaskHeartbeatRequest) EntityNotExistsError(com.uber.cadence.EntityNotExistsError) BadRequestError(com.uber.cadence.BadRequestError)

Example 2 with RecordActivityTaskHeartbeatResponse

use of com.uber.cadence.RecordActivityTaskHeartbeatResponse in project cadence-client by uber-java.

the class TestWorkflowMutableStateImpl method heartbeatActivityTask.

@Override
public RecordActivityTaskHeartbeatResponse heartbeatActivityTask(String activityId, RecordActivityTaskHeartbeatRequest request) throws InternalServiceError, EntityNotExistsError, BadRequestError {
    RecordActivityTaskHeartbeatResponse result = new RecordActivityTaskHeartbeatResponse();
    try {
        update(ctx -> {
            StateMachine<ActivityTaskData> activity = getActivity(activityId);
            activity.action(StateMachines.Action.UPDATE, ctx, request, 0);
            if (activity.getState() == StateMachines.State.CANCELLATION_REQUESTED) {
                result.setCancelRequested(true);
            }
            ActivityTaskData data = activity.getData();
            data.lastHeartbeatTime = clock.getAsLong();
            int startToCloseTimeout = data.scheduledEvent.getStartToCloseTimeoutSeconds();
            int heartbeatTimeout = data.scheduledEvent.getHeartbeatTimeoutSeconds();
            updateHeartbeatTimer(ctx, activityId, activity, startToCloseTimeout, heartbeatTimeout);
        });
    } catch (InternalServiceError | EntityNotExistsError e) {
        throw e;
    } catch (Exception e) {
        throw new InternalServiceError(Throwables.getStackTraceAsString(e));
    }
    return result;
}
Also used : RecordActivityTaskHeartbeatResponse(com.uber.cadence.RecordActivityTaskHeartbeatResponse) InternalServiceError(com.uber.cadence.InternalServiceError) EntityNotExistsError(com.uber.cadence.EntityNotExistsError) ActivityTaskData(com.uber.cadence.internal.testservice.StateMachines.ActivityTaskData) TException(org.apache.thrift.TException) IOException(java.io.IOException) ExecutionException(java.util.concurrent.ExecutionException)

Example 3 with RecordActivityTaskHeartbeatResponse

use of com.uber.cadence.RecordActivityTaskHeartbeatResponse in project cadence-client by uber-java.

the class ManualActivityCompletionClientImpl method recordHeartbeat.

@Override
public void recordHeartbeat(Object details) throws CancellationException {
    if (taskToken != null) {
        RecordActivityTaskHeartbeatRequest request = new RecordActivityTaskHeartbeatRequest();
        request.setDetails(dataConverter.toData(details));
        request.setTaskToken(taskToken);
        RecordActivityTaskHeartbeatResponse status = null;
        try {
            status = service.RecordActivityTaskHeartbeat(request);
            if (status.isCancelRequested()) {
                throw new ActivityCancelledException();
            }
        } catch (EntityNotExistsError e) {
            throw new ActivityNotExistsException(e);
        } catch (TException e) {
            throw new ActivityCompletionFailureException(e);
        }
    } else {
        throw new UnsupportedOperationException("Heartbeating by id is not implemented by Cadence service yet.");
    }
}
Also used : ActivityNotExistsException(com.uber.cadence.client.ActivityNotExistsException) TException(org.apache.thrift.TException) ActivityCancelledException(com.uber.cadence.client.ActivityCancelledException) RecordActivityTaskHeartbeatResponse(com.uber.cadence.RecordActivityTaskHeartbeatResponse) ActivityCompletionFailureException(com.uber.cadence.client.ActivityCompletionFailureException) RecordActivityTaskHeartbeatRequest(com.uber.cadence.RecordActivityTaskHeartbeatRequest) EntityNotExistsError(com.uber.cadence.EntityNotExistsError)

Aggregations

EntityNotExistsError (com.uber.cadence.EntityNotExistsError)3 RecordActivityTaskHeartbeatResponse (com.uber.cadence.RecordActivityTaskHeartbeatResponse)3 TException (org.apache.thrift.TException)3 RecordActivityTaskHeartbeatRequest (com.uber.cadence.RecordActivityTaskHeartbeatRequest)2 ActivityCancelledException (com.uber.cadence.client.ActivityCancelledException)2 ActivityCompletionFailureException (com.uber.cadence.client.ActivityCompletionFailureException)2 ActivityNotExistsException (com.uber.cadence.client.ActivityNotExistsException)2 BadRequestError (com.uber.cadence.BadRequestError)1 InternalServiceError (com.uber.cadence.InternalServiceError)1 ActivityTaskData (com.uber.cadence.internal.testservice.StateMachines.ActivityTaskData)1 IOException (java.io.IOException)1 ExecutionException (java.util.concurrent.ExecutionException)1