protected static List<Tag<?>> tagsForTask(TaskState taskState) { List<Tag<?>> tags = Lists.newArrayList(); tags.add(new Tag<>(TaskEvent.METADATA_TASK_ID, taskState.getTaskId())); tags.add(new Tag<>(TaskEvent.METADATA_TASK_ATTEMPT_ID, taskState.getTaskAttemptId().or(""))); tags.add(new Tag<>(ConfigurationKeys.DATASET_URN_KEY, taskState.getProp(ConfigurationKeys.DATASET_URN_KEY, ConfigurationKeys.DEFAULT_DATASET_URN))); tags.addAll(getCustomTagsFromState(taskState)); return tags; }
protected void submitTaskCommittedEvent() { MetricContext taskMetricContext = TaskMetrics.get(this.taskState).getMetricContext(); EventSubmitter eventSubmitter = new EventSubmitter.Builder(taskMetricContext, "gobblin.runtime.task").build(); eventSubmitter.submit(TaskEvent.TASK_COMMITTED_EVENT_NAME, ImmutableMap .of(TaskEvent.METADATA_TASK_ID, this.taskId, TaskEvent.METADATA_TASK_ATTEMPT_ID, this.taskState.getTaskAttemptId().or(""))); }
Mockito.when(taskState.getWorkunit()).thenReturn(workUnit); Mockito.when(taskState.getJobId()).thenReturn("123"); Mockito.when(taskState.getTaskAttemptId()).thenReturn(Optional.of("555")); Mockito.when(taskState.getTaskId()).thenReturn("888"); Mockito.when(limiter.acquirePermits(1)).thenReturn(null);
Mockito.when(taskState.getWorkunit()).thenReturn(workUnit); Mockito.when(taskState.getJobId()).thenReturn("123"); Mockito.when(taskState.getTaskAttemptId()).thenReturn(Optional.of("555")); Mockito.when(taskState.getTaskId()).thenReturn("888"); Mockito.when(limiter.acquirePermits(1)).thenReturn(null);
Mockito.when(taskState.getWorkunit()).thenReturn(workUnit); Mockito.when(taskState.getJobId()).thenReturn("123"); Mockito.when(taskState.getTaskAttemptId()).thenReturn(Optional.of("555")); Mockito.when(taskState.getTaskId()).thenReturn("888"); Mockito.when(limiter.acquirePermits(1)).thenReturn(null);
public TaskState(TaskState taskState) { super(taskState.getWorkunit(), taskState.getJobState(), taskState.getTaskBrokerNullable()); addAll(taskState); this.jobId = taskState.getProp(ConfigurationKeys.JOB_ID_KEY); this.taskId = taskState.getProp(ConfigurationKeys.TASK_ID_KEY); this.taskAttemptId = taskState.getTaskAttemptId(); this.setId(this.taskId); }
this.taskAttemptId = this.taskState.getTaskAttemptId();
protected static List<Tag<?>> tagsForTask(TaskState taskState) { List<Tag<?>> tags = Lists.newArrayList(); tags.add(new Tag<>(TaskEvent.METADATA_TASK_ID, taskState.getTaskId())); tags.add(new Tag<>(TaskEvent.METADATA_TASK_ATTEMPT_ID, taskState.getTaskAttemptId().or(""))); tags.add(new Tag<>(ConfigurationKeys.DATASET_URN_KEY, taskState.getProp(ConfigurationKeys.DATASET_URN_KEY, ConfigurationKeys.DEFAULT_DATASET_URN))); tags.addAll(getCustomTagsFromState(taskState)); return tags; }
protected void submitTaskCommittedEvent() { MetricContext taskMetricContext = TaskMetrics.get(this.taskState).getMetricContext(); EventSubmitter eventSubmitter = new EventSubmitter.Builder(taskMetricContext, "gobblin.runtime.task").build(); eventSubmitter.submit(TaskEvent.TASK_COMMITTED_EVENT_NAME, ImmutableMap .of(TaskEvent.METADATA_TASK_ID, this.taskId, TaskEvent.METADATA_TASK_ATTEMPT_ID, this.taskState.getTaskAttemptId().or(""))); }
public TaskState(TaskState taskState) { super(taskState.getWorkunit(), taskState.getJobState(), taskState.getTaskBrokerNullable()); addAll(taskState); this.jobId = taskState.getProp(ConfigurationKeys.JOB_ID_KEY); this.taskId = taskState.getProp(ConfigurationKeys.TASK_ID_KEY); this.taskAttemptId = taskState.getTaskAttemptId(); this.setId(this.taskId); }
this.taskAttemptId = this.taskState.getTaskAttemptId();