Examples with TaskAction org.apache.druid.indexing.common.actions.TaskAction used on opensource projects

Example 1 with TaskAction

use of org.apache.druid.indexing.common.actions.TaskAction in project druid by druid-io.

the class CountingLocalTaskActionClientForTest method submit.

@Override
public <RetType> RetType submit(TaskAction<RetType> taskAction) {
    final RetType result = delegate.submit(taskAction);
    final TaskAction actionKey;
    if (taskAction instanceof SurrogateAction) {
        actionKey = ((SurrogateAction) taskAction).getTaskAction();
    } else {
        actionKey = taskAction;
    }
    actionCountMap.computeIfAbsent(actionKey.getClass(), k -> new AtomicInteger()).incrementAndGet();
    return result;
}