Search in sources :

Example 1 with ServiceBusyError

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

the class ActivityPollTask method pollTask.

@Override
protected PollForActivityTaskResponse pollTask() throws TException {
    options.getMetricsScope().counter(MetricsType.ACTIVITY_POLL_COUNTER).inc(1);
    Stopwatch sw = options.getMetricsScope().timer(MetricsType.ACTIVITY_POLL_LATENCY).start();
    PollForActivityTaskRequest pollRequest = new PollForActivityTaskRequest();
    pollRequest.setDomain(domain);
    pollRequest.setIdentity(options.getIdentity());
    pollRequest.setTaskList(new TaskList().setName(taskList));
    if (options.getTaskListActivitiesPerSecond() > 0) {
        TaskListMetadata metadata = new TaskListMetadata();
        metadata.setMaxTasksPerSecond(options.getTaskListActivitiesPerSecond());
        pollRequest.setTaskListMetadata(metadata);
    }
    if (log.isDebugEnabled()) {
        log.debug("poll request begin: " + pollRequest);
    }
    PollForActivityTaskResponse result;
    try {
        result = service.PollForActivityTask(pollRequest);
    } catch (InternalServiceError | ServiceBusyError e) {
        options.getMetricsScope().counter(MetricsType.ACTIVITY_POLL_TRANSIENT_FAILED_COUNTER).inc(1);
        throw e;
    } catch (TException e) {
        options.getMetricsScope().counter(MetricsType.ACTIVITY_POLL_FAILED_COUNTER).inc(1);
        throw e;
    }
    if (result == null || result.getTaskToken() == null) {
        if (log.isDebugEnabled()) {
            log.debug("poll request returned no task");
        }
        options.getMetricsScope().counter(MetricsType.ACTIVITY_POLL_NO_TASK_COUNTER).inc(1);
        return null;
    }
    if (log.isTraceEnabled()) {
        log.trace("poll request returned " + result);
    }
    sw.stop();
    return result;
}
Also used : PollForActivityTaskRequest(com.uber.cadence.PollForActivityTaskRequest) TException(org.apache.thrift.TException) TaskListMetadata(com.uber.cadence.TaskListMetadata) TaskList(com.uber.cadence.TaskList) Stopwatch(com.uber.m3.tally.Stopwatch) PollForActivityTaskResponse(com.uber.cadence.PollForActivityTaskResponse) ServiceBusyError(com.uber.cadence.ServiceBusyError) InternalServiceError(com.uber.cadence.InternalServiceError)

Example 2 with ServiceBusyError

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

the class WorkflowPollTask method poll.

@Override
public PollForDecisionTaskResponse poll() throws TException {
    metricScope.counter(MetricsType.DECISION_POLL_COUNTER).inc(1);
    Stopwatch sw = metricScope.timer(MetricsType.DECISION_POLL_LATENCY).start();
    PollForDecisionTaskRequest pollRequest = new PollForDecisionTaskRequest();
    pollRequest.setDomain(domain);
    pollRequest.setIdentity(identity);
    pollRequest.setBinaryChecksum(BinaryChecksum.getBinaryChecksum());
    TaskList tl = new TaskList();
    tl.setName(taskList);
    pollRequest.setTaskList(tl);
    if (log.isDebugEnabled()) {
        log.debug("poll request begin: " + pollRequest);
    }
    PollForDecisionTaskResponse result;
    try {
        result = service.PollForDecisionTask(pollRequest);
    } catch (InternalServiceError | ServiceBusyError e) {
        metricScope.counter(MetricsType.DECISION_POLL_TRANSIENT_FAILED_COUNTER).inc(1);
        throw e;
    } catch (TException e) {
        metricScope.counter(MetricsType.DECISION_POLL_FAILED_COUNTER).inc(1);
        throw e;
    }
    if (log.isDebugEnabled()) {
        log.debug("poll request returned decision task: workflowType=" + result.getWorkflowType() + ", workflowExecution=" + result.getWorkflowExecution() + ", startedEventId=" + result.getStartedEventId() + ", previousStartedEventId=" + result.getPreviousStartedEventId() + (result.getQuery() != null ? ", queryType=" + result.getQuery().getQueryType() : ""));
    }
    if (result == null || result.getTaskToken() == null) {
        metricScope.counter(MetricsType.DECISION_POLL_NO_TASK_COUNTER).inc(1);
        return null;
    }
    Scope metricsScope = metricScope.tagged(ImmutableMap.of(MetricsTag.WORKFLOW_TYPE, result.getWorkflowType().getName()));
    metricsScope.counter(MetricsType.DECISION_POLL_SUCCEED_COUNTER).inc(1);
    metricsScope.timer(MetricsType.DECISION_SCHEDULED_TO_START_LATENCY).record(Duration.ofNanos(result.getStartedTimestamp() - result.getScheduledTimestamp()));
    sw.stop();
    return result;
}
Also used : TException(org.apache.thrift.TException) Scope(com.uber.m3.tally.Scope) TaskList(com.uber.cadence.TaskList) Stopwatch(com.uber.m3.tally.Stopwatch) ServiceBusyError(com.uber.cadence.ServiceBusyError) PollForDecisionTaskResponse(com.uber.cadence.PollForDecisionTaskResponse) InternalServiceError(com.uber.cadence.InternalServiceError) PollForDecisionTaskRequest(com.uber.cadence.PollForDecisionTaskRequest)

Aggregations

InternalServiceError (com.uber.cadence.InternalServiceError)2 ServiceBusyError (com.uber.cadence.ServiceBusyError)2 TaskList (com.uber.cadence.TaskList)2 Stopwatch (com.uber.m3.tally.Stopwatch)2 TException (org.apache.thrift.TException)2 PollForActivityTaskRequest (com.uber.cadence.PollForActivityTaskRequest)1 PollForActivityTaskResponse (com.uber.cadence.PollForActivityTaskResponse)1 PollForDecisionTaskRequest (com.uber.cadence.PollForDecisionTaskRequest)1 PollForDecisionTaskResponse (com.uber.cadence.PollForDecisionTaskResponse)1 TaskListMetadata (com.uber.cadence.TaskListMetadata)1 Scope (com.uber.m3.tally.Scope)1