Wenjun Ruan
5 months ago
committed by
GitHub
14 changed files with 434 additions and 208 deletions
@ -1,85 +0,0 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.server.master.runner; |
||||
|
||||
import org.apache.dolphinscheduler.common.utils.DateUtils; |
||||
import org.apache.dolphinscheduler.dao.entity.ProcessInstance; |
||||
import org.apache.dolphinscheduler.dao.entity.TaskInstance; |
||||
import org.apache.dolphinscheduler.plugin.task.api.TaskExecutionContext; |
||||
|
||||
import java.util.concurrent.Delayed; |
||||
import java.util.concurrent.TimeUnit; |
||||
|
||||
public abstract class PriorityDelayTaskExecuteRunnable extends BaseTaskExecuteRunnable implements Delayed { |
||||
|
||||
public PriorityDelayTaskExecuteRunnable(ProcessInstance workflowInstance, |
||||
TaskInstance taskInstance, |
||||
TaskExecutionContext taskExecutionContext) { |
||||
super(workflowInstance, taskInstance, taskExecutionContext); |
||||
} |
||||
|
||||
@Override |
||||
public long getDelay(TimeUnit unit) { |
||||
return unit.convert( |
||||
DateUtils.getRemainTime(taskExecutionContext.getFirstSubmitTime(), |
||||
taskExecutionContext.getDelayTime() * 60L), |
||||
TimeUnit.SECONDS); |
||||
} |
||||
|
||||
@Override |
||||
public int compareTo(Delayed o) { |
||||
if (o == null) { |
||||
return 1; |
||||
} |
||||
int delayTimeCompareResult = |
||||
Long.compare(this.getDelay(TimeUnit.MILLISECONDS), o.getDelay(TimeUnit.MILLISECONDS)); |
||||
if (delayTimeCompareResult != 0) { |
||||
return delayTimeCompareResult; |
||||
} |
||||
PriorityDelayTaskExecuteRunnable other = (PriorityDelayTaskExecuteRunnable) o; |
||||
// the smaller dispatch fail times, the higher priority
|
||||
int dispatchFailTimesCompareResult = taskExecutionContext.getDispatchFailTimes() |
||||
- other.getTaskExecutionContext().getDispatchFailTimes(); |
||||
if (dispatchFailTimesCompareResult != 0) { |
||||
return dispatchFailTimesCompareResult; |
||||
} |
||||
int workflowInstancePriorityCompareResult = workflowInstance.getProcessInstancePriority().getCode() |
||||
- other.getWorkflowInstance().getProcessInstancePriority().getCode(); |
||||
if (workflowInstancePriorityCompareResult != 0) { |
||||
return workflowInstancePriorityCompareResult; |
||||
} |
||||
long workflowInstanceIdCompareResult = workflowInstance.getId().compareTo(other.getWorkflowInstance().getId()); |
||||
if (workflowInstanceIdCompareResult != 0) { |
||||
return workflowInstancePriorityCompareResult; |
||||
} |
||||
int taskInstancePriorityCompareResult = taskInstance.getTaskInstancePriority().getCode() |
||||
- other.getTaskInstance().getTaskInstancePriority().getCode(); |
||||
if (taskInstancePriorityCompareResult != 0) { |
||||
return taskInstancePriorityCompareResult; |
||||
} |
||||
// larger number, higher priority
|
||||
int taskGroupPriorityCompareResult = |
||||
taskInstance.getTaskGroupPriority() - other.getTaskInstance().getTaskGroupPriority(); |
||||
if (taskGroupPriorityCompareResult != 0) { |
||||
return -taskGroupPriorityCompareResult; |
||||
} |
||||
// The task instance shouldn't be equals
|
||||
return taskInstance.getId().compareTo(other.getTaskInstance().getId()); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,82 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.server.master.runner.queue; |
||||
|
||||
import static com.google.common.base.Preconditions.checkNotNull; |
||||
|
||||
import java.util.Objects; |
||||
import java.util.concurrent.Delayed; |
||||
import java.util.concurrent.TimeUnit; |
||||
|
||||
import lombok.Getter; |
||||
|
||||
import org.jetbrains.annotations.NotNull; |
||||
|
||||
public class DelayEntry<V extends Comparable<V>> implements Delayed { |
||||
|
||||
private final long delayTimeMills; |
||||
|
||||
private final long triggerTimeMills; |
||||
|
||||
@Getter |
||||
private final V data; |
||||
|
||||
public DelayEntry(long delayTimeMills, V data) { |
||||
this.delayTimeMills = delayTimeMills; |
||||
this.triggerTimeMills = System.currentTimeMillis() + delayTimeMills; |
||||
this.data = checkNotNull(data, "data is null"); |
||||
} |
||||
|
||||
@Override |
||||
public long getDelay(@NotNull TimeUnit unit) { |
||||
long remainTimeMills = triggerTimeMills - System.currentTimeMillis(); |
||||
if (TimeUnit.MILLISECONDS.equals(unit)) { |
||||
return remainTimeMills; |
||||
} |
||||
return unit.convert(remainTimeMills, TimeUnit.MILLISECONDS); |
||||
} |
||||
|
||||
@Override |
||||
public int compareTo(@NotNull Delayed o) { |
||||
DelayEntry<V> other = (DelayEntry<V>) o; |
||||
int delayTimeMillsCompareResult = Long.compare(delayTimeMills, other.delayTimeMills); |
||||
if (delayTimeMillsCompareResult != 0) { |
||||
return delayTimeMillsCompareResult; |
||||
} |
||||
|
||||
if (data == null || other.data == null) { |
||||
return 0; |
||||
} |
||||
return data.compareTo(other.data); |
||||
} |
||||
|
||||
@Override |
||||
public boolean equals(Object o) { |
||||
if (this == o) |
||||
return true; |
||||
if (o == null || getClass() != o.getClass()) |
||||
return false; |
||||
DelayEntry<?> that = (DelayEntry<?>) o; |
||||
return delayTimeMills == that.delayTimeMills && Objects.equals(data, that.data); |
||||
} |
||||
|
||||
@Override |
||||
public int hashCode() { |
||||
return Objects.hash(delayTimeMills, data); |
||||
} |
||||
} |
||||
@ -0,0 +1,41 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.server.master.runner.queue; |
||||
|
||||
import java.util.concurrent.DelayQueue; |
||||
|
||||
import lombok.SneakyThrows; |
||||
|
||||
public class PriorityDelayQueue<V extends DelayEntry> { |
||||
|
||||
private final DelayQueue<V> queue = new DelayQueue<>(); |
||||
|
||||
public void add(V v) { |
||||
queue.put(v); |
||||
} |
||||
|
||||
@SneakyThrows |
||||
public V take() { |
||||
return queue.take(); |
||||
} |
||||
|
||||
public int size() { |
||||
return queue.size(); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,184 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.server.master.runner; |
||||
|
||||
import static com.google.common.truth.Truth.assertThat; |
||||
import static org.awaitility.Awaitility.await; |
||||
import static org.junit.jupiter.api.Assertions.assertThrowsExactly; |
||||
|
||||
import org.apache.dolphinscheduler.common.enums.Priority; |
||||
import org.apache.dolphinscheduler.dao.entity.ProcessInstance; |
||||
import org.apache.dolphinscheduler.dao.entity.TaskInstance; |
||||
import org.apache.dolphinscheduler.plugin.task.api.TaskExecutionContext; |
||||
import org.apache.dolphinscheduler.server.master.runner.operator.TaskExecuteRunnableOperatorManager; |
||||
|
||||
import org.apache.commons.lang3.time.DateUtils; |
||||
|
||||
import java.time.Duration; |
||||
import java.util.Date; |
||||
import java.util.concurrent.CompletableFuture; |
||||
|
||||
import org.awaitility.Awaitility; |
||||
import org.awaitility.core.ConditionTimeoutException; |
||||
import org.junit.jupiter.api.Assertions; |
||||
import org.junit.jupiter.api.BeforeEach; |
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
class GlobalTaskDispatchWaitingQueueTest { |
||||
|
||||
private GlobalTaskDispatchWaitingQueue globalTaskDispatchWaitingQueue; |
||||
|
||||
@BeforeEach |
||||
public void setUp() { |
||||
globalTaskDispatchWaitingQueue = new GlobalTaskDispatchWaitingQueue(); |
||||
} |
||||
|
||||
@Test |
||||
void submitTaskExecuteRunnable() { |
||||
TaskExecuteRunnable taskExecuteRunnable = createTaskExecuteRunnable(); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable); |
||||
Awaitility.await() |
||||
.atMost(Duration.ofSeconds(1)) |
||||
.untilAsserted( |
||||
() -> Assertions.assertNotNull(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable())); |
||||
} |
||||
|
||||
@Test |
||||
void testSubmitTaskExecuteRunnableWithDelay() { |
||||
TaskExecuteRunnable taskExecuteRunnable = createTaskExecuteRunnable(); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnableWithDelay(taskExecuteRunnable, 3_000L); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable); |
||||
|
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable()).isNotNull(); |
||||
Awaitility.await() |
||||
.atLeast(Duration.ofSeconds(2)) |
||||
.atMost(Duration.ofSeconds(4)) |
||||
.untilAsserted( |
||||
() -> Assertions.assertNotNull(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable())); |
||||
} |
||||
|
||||
@Test |
||||
void takeTaskExecuteRunnable_NoElementShouldBlock() { |
||||
CompletableFuture<Void> completableFuture = |
||||
CompletableFuture.runAsync(() -> globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable()); |
||||
assertThrowsExactly(ConditionTimeoutException.class, |
||||
() -> await() |
||||
.atLeast(Duration.ofSeconds(2)) |
||||
.timeout(Duration.ofSeconds(3)) |
||||
.until(completableFuture::isDone)); |
||||
} |
||||
|
||||
@Test |
||||
void takeTaskExecuteRunnable_withDifferentTaskInstancePriority() { |
||||
TaskExecuteRunnable taskExecuteRunnable1 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable1.getTaskInstance().setId(1); |
||||
taskExecuteRunnable1.getTaskInstance().setTaskInstancePriority(Priority.MEDIUM); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable1); |
||||
|
||||
TaskExecuteRunnable taskExecuteRunnable2 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable2.getTaskInstance().setId(2); |
||||
taskExecuteRunnable2.getTaskInstance().setTaskInstancePriority(Priority.HIGH); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable2); |
||||
|
||||
TaskExecuteRunnable taskExecuteRunnable3 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable3.getTaskInstance().setId(3); |
||||
taskExecuteRunnable3.getTaskInstance().setTaskInstancePriority(Priority.LOW); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable3); |
||||
|
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(2); |
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(1); |
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(3); |
||||
} |
||||
|
||||
@Test |
||||
void takeTaskExecuteRunnable_withDifferentTaskGroupPriority() { |
||||
TaskExecuteRunnable taskExecuteRunnable1 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable1.getTaskInstance().setId(1); |
||||
taskExecuteRunnable1.getTaskInstance().setTaskGroupPriority(Priority.MEDIUM.getCode()); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable1); |
||||
|
||||
TaskExecuteRunnable taskExecuteRunnable2 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable2.getTaskInstance().setId(2); |
||||
taskExecuteRunnable2.getTaskInstance().setTaskGroupPriority(Priority.HIGH.getCode()); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable2); |
||||
|
||||
TaskExecuteRunnable taskExecuteRunnable3 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable3.getTaskInstance().setId(3); |
||||
taskExecuteRunnable3.getTaskInstance().setTaskGroupPriority(Priority.LOW.getCode()); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable3); |
||||
|
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(3); |
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(1); |
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(2); |
||||
} |
||||
|
||||
@Test |
||||
void takeTaskExecuteRunnable_withDifferentSubmitTime() { |
||||
Date now = new Date(); |
||||
|
||||
TaskExecuteRunnable taskExecuteRunnable1 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable1.getTaskInstance().setId(1); |
||||
taskExecuteRunnable1.getTaskInstance().setFirstSubmitTime(now); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable1); |
||||
|
||||
TaskExecuteRunnable taskExecuteRunnable2 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable2.getTaskInstance().setId(2); |
||||
taskExecuteRunnable2.getTaskInstance().setFirstSubmitTime(DateUtils.addMinutes(now, 1)); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable2); |
||||
|
||||
TaskExecuteRunnable taskExecuteRunnable3 = createTaskExecuteRunnable(); |
||||
taskExecuteRunnable3.getTaskInstance().setId(3); |
||||
taskExecuteRunnable3.getTaskInstance().setFirstSubmitTime(DateUtils.addMinutes(now, -1)); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable3); |
||||
|
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(3); |
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(1); |
||||
assertThat(globalTaskDispatchWaitingQueue.takeTaskExecuteRunnable().getTaskInstance().getId()) |
||||
.isEqualTo(2); |
||||
} |
||||
|
||||
@Test |
||||
void getWaitingDispatchTaskNumber() { |
||||
Assertions.assertEquals(0, globalTaskDispatchWaitingQueue.getWaitingDispatchTaskNumber()); |
||||
TaskExecuteRunnable taskExecuteRunnable = createTaskExecuteRunnable(); |
||||
globalTaskDispatchWaitingQueue.dispatchTaskExecuteRunnable(taskExecuteRunnable); |
||||
Assertions.assertEquals(1, globalTaskDispatchWaitingQueue.getWaitingDispatchTaskNumber()); |
||||
} |
||||
|
||||
private TaskExecuteRunnable createTaskExecuteRunnable() { |
||||
ProcessInstance processInstance = new ProcessInstance(); |
||||
processInstance.setProcessInstancePriority(Priority.MEDIUM); |
||||
|
||||
TaskInstance taskInstance = new TaskInstance(); |
||||
taskInstance.setTaskInstancePriority(Priority.MEDIUM); |
||||
taskInstance.setFirstSubmitTime(new Date()); |
||||
|
||||
TaskExecutionContext taskExecutionContext = new TaskExecutionContext(); |
||||
|
||||
return new DefaultTaskExecuteRunnable(processInstance, taskInstance, taskExecutionContext, |
||||
new TaskExecuteRunnableOperatorManager()); |
||||
} |
||||
} |
||||
@ -1,67 +0,0 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.server.master.runner.execute; |
||||
|
||||
import org.apache.dolphinscheduler.common.enums.Priority; |
||||
import org.apache.dolphinscheduler.dao.entity.ProcessInstance; |
||||
import org.apache.dolphinscheduler.dao.entity.TaskInstance; |
||||
import org.apache.dolphinscheduler.plugin.task.api.TaskExecutionContext; |
||||
import org.apache.dolphinscheduler.server.master.runner.DefaultTaskExecuteRunnable; |
||||
import org.apache.dolphinscheduler.server.master.runner.PriorityDelayTaskExecuteRunnable; |
||||
import org.apache.dolphinscheduler.server.master.runner.operator.TaskExecuteRunnableOperatorManager; |
||||
|
||||
import org.junit.jupiter.api.Assertions; |
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
public class PriorityDelayTaskExecuteRunnableTest { |
||||
|
||||
@Test |
||||
public void testCompareTo() { |
||||
TaskExecuteRunnableOperatorManager taskOperatorManager = new TaskExecuteRunnableOperatorManager(); |
||||
|
||||
ProcessInstance workflowInstance = new ProcessInstance(); |
||||
workflowInstance.setId(1); |
||||
workflowInstance.setProcessInstancePriority(Priority.HIGH); |
||||
|
||||
TaskInstance t1 = new TaskInstance(); |
||||
t1.setId(1); |
||||
t1.setTaskInstancePriority(Priority.HIGH); |
||||
|
||||
TaskInstance t2 = new TaskInstance(); |
||||
t2.setId(1); |
||||
t2.setTaskInstancePriority(Priority.HIGH); |
||||
|
||||
TaskExecutionContext context1 = new TaskExecutionContext(); |
||||
TaskExecutionContext context2 = new TaskExecutionContext(); |
||||
PriorityDelayTaskExecuteRunnable p1 = |
||||
new DefaultTaskExecuteRunnable(workflowInstance, t1, context1, taskOperatorManager); |
||||
PriorityDelayTaskExecuteRunnable p2 = |
||||
new DefaultTaskExecuteRunnable(workflowInstance, t2, context2, taskOperatorManager); |
||||
|
||||
Assertions.assertEquals(0, p1.compareTo(p2)); |
||||
|
||||
// the higher priority, the higher priority
|
||||
t2.setTaskInstancePriority(Priority.MEDIUM); |
||||
Assertions.assertTrue(p1.compareTo(p2) < 0); |
||||
|
||||
// the smaller dispatch fail times, the higher priority
|
||||
context1.setDispatchFailTimes(1); |
||||
Assertions.assertTrue(p1.compareTo(p2) > 0); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,35 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.server.master.runner.queue; |
||||
|
||||
import java.util.concurrent.TimeUnit; |
||||
|
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
import com.google.common.truth.Truth; |
||||
|
||||
class DelayEntryTest { |
||||
|
||||
@Test |
||||
void getDelay() { |
||||
DelayEntry<String> delayEntry = new DelayEntry<>(1_000L, "Item"); |
||||
Truth.assertThat(delayEntry.getDelay(TimeUnit.NANOSECONDS)) |
||||
.isWithin(100) |
||||
.of(TimeUnit.NANOSECONDS.convert(1_000L, TimeUnit.MILLISECONDS)); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue