diff --git a/dolphinscheduler-common/src/main/java/org/apache/dolphinscheduler/common/utils/DateUtils.java b/dolphinscheduler-common/src/main/java/org/apache/dolphinscheduler/common/utils/DateUtils.java index f883d5ca68b7..f6e38ac50322 100644 --- a/dolphinscheduler-common/src/main/java/org/apache/dolphinscheduler/common/utils/DateUtils.java +++ b/dolphinscheduler-common/src/main/java/org/apache/dolphinscheduler/common/utils/DateUtils.java @@ -646,4 +646,23 @@ public static long toDurationHours(long d) { } + /** + * transform timeStamp to local date + * + * @param timeStamp time stamp (milliseconds) + * @return local date + */ + public static @Nullable Date timeStampToDate(long timeStamp) { + return timeStamp <= 0L ? null : new Date(timeStamp); + } + + /** + * transform date to timeStamp + * @param date date + * @return time stamp (milliseconds) + */ + public static long dateToTimeStamp(Date date) { + return date == null ? 0L : date.getTime(); + } + } diff --git a/dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/utils/DateUtilsTest.java b/dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/utils/DateUtilsTest.java index 0233fcbd9113..0b0b7c84fdd9 100644 --- a/dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/utils/DateUtilsTest.java +++ b/dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/utils/DateUtilsTest.java @@ -260,4 +260,41 @@ public void testDateToString() { String utcNowStr = DateUtils.dateToString(asiaShNow, utc); Assert.assertEquals(asiaShNowStr, utcNowStr); } + + @Test + public void testDateToTimeStamp() throws ParseException { + // Beijing Date + String timeString = "2022-09-29 21:00:00"; + SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); + sdf.setTimeZone(TimeZone.getTimeZone("Asia/Shanghai")); + Date date = sdf.parse(timeString); + long timeStamp = DateUtils.dateToTimeStamp(date); + Assert.assertEquals(1664456400000L, timeStamp); + + // Tokyo Date + String tokyoTime = "2022-09-29 22:00:00"; + sdf.setTimeZone(TimeZone.getTimeZone("Asia/Tokyo")); + date = sdf.parse(tokyoTime); + timeStamp = DateUtils.dateToTimeStamp(date); + Assert.assertEquals(1664456400000L, timeStamp); + + date = null; + Assert.assertEquals(0L, DateUtils.dateToTimeStamp(date)); + } + + @Test + public void testTimeStampToDate() { + long timeStamp = 1664456400000L; + SimpleDateFormat sdf=new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); + sdf.setTimeZone(TimeZone.getTimeZone("Asia/Shanghai")); + String sd = sdf.format(new Date(timeStamp)); + Assert.assertEquals("2022-09-29 21:00:00", sd); + + sdf.setTimeZone(TimeZone.getTimeZone("Asia/Tokyo")); + sd = sdf.format(new Date(timeStamp)); + Assert.assertEquals("2022-09-29 22:00:00", sd); + + Date date = DateUtils.timeStampToDate(0L); + Assert.assertNull(date); + } } diff --git a/dolphinscheduler-master/src/main/java/org/apache/dolphinscheduler/server/master/builder/TaskExecutionContextBuilder.java b/dolphinscheduler-master/src/main/java/org/apache/dolphinscheduler/server/master/builder/TaskExecutionContextBuilder.java index 5dfd708f56f0..262a5acc62d3 100644 --- a/dolphinscheduler-master/src/main/java/org/apache/dolphinscheduler/server/master/builder/TaskExecutionContextBuilder.java +++ b/dolphinscheduler-master/src/main/java/org/apache/dolphinscheduler/server/master/builder/TaskExecutionContextBuilder.java @@ -20,6 +20,7 @@ import static org.apache.dolphinscheduler.common.Constants.SEC_2_MINUTES_TIME_UNIT; import org.apache.dolphinscheduler.common.enums.TimeoutFlag; +import org.apache.dolphinscheduler.common.utils.DateUtils; import org.apache.dolphinscheduler.dao.entity.ProcessDefinition; import org.apache.dolphinscheduler.dao.entity.ProcessInstance; import org.apache.dolphinscheduler.dao.entity.TaskDefinition; @@ -61,8 +62,8 @@ public static TaskExecutionContextBuilder get() { public TaskExecutionContextBuilder buildTaskInstanceRelatedInfo(TaskInstance taskInstance) { taskExecutionContext.setTaskInstanceId(taskInstance.getId()); taskExecutionContext.setTaskName(taskInstance.getName()); - taskExecutionContext.setFirstSubmitTime(taskInstance.getFirstSubmitTime()); - taskExecutionContext.setStartTime(taskInstance.getStartTime()); + taskExecutionContext.setFirstSubmitTime(DateUtils.dateToTimeStamp(taskInstance.getFirstSubmitTime())); + taskExecutionContext.setStartTime(DateUtils.dateToTimeStamp(taskInstance.getStartTime())); taskExecutionContext.setTaskType(taskInstance.getTaskType()); taskExecutionContext.setLogPath(taskInstance.getLogPath()); taskExecutionContext.setWorkerGroup(taskInstance.getWorkerGroup()); @@ -102,7 +103,7 @@ public TaskExecutionContextBuilder buildTaskDefinitionRelatedInfo(TaskDefinition */ public TaskExecutionContextBuilder buildProcessInstanceRelatedInfo(ProcessInstance processInstance) { taskExecutionContext.setProcessInstanceId(processInstance.getId()); - taskExecutionContext.setScheduleTime(processInstance.getScheduleTime()); + taskExecutionContext.setScheduleTime(DateUtils.dateToTimeStamp(processInstance.getScheduleTime())); taskExecutionContext.setGlobalParams(processInstance.getGlobalParams()); taskExecutionContext.setExecutorId(processInstance.getExecutorId()); taskExecutionContext.setCmdTypeIfComplement(processInstance.getCmdTypeIfComplement().getCode()); diff --git a/dolphinscheduler-master/src/main/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskEvent.java b/dolphinscheduler-master/src/main/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskEvent.java index dbbca6cab31f..72fc02665e40 100644 --- a/dolphinscheduler-master/src/main/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskEvent.java +++ b/dolphinscheduler-master/src/main/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskEvent.java @@ -18,6 +18,7 @@ package org.apache.dolphinscheduler.server.master.processor.queue; import org.apache.dolphinscheduler.common.enums.TaskEventType; +import org.apache.dolphinscheduler.common.utils.DateUtils; import org.apache.dolphinscheduler.plugin.task.api.enums.TaskExecutionStatus; import org.apache.dolphinscheduler.remote.command.TaskExecuteResultCommand; import org.apache.dolphinscheduler.remote.command.TaskExecuteRunningCommand; @@ -110,7 +111,7 @@ public static TaskEvent newRunningEvent(TaskExecuteRunningCommand command, Chann event.setProcessInstanceId(command.getProcessInstanceId()); event.setTaskInstanceId(command.getTaskInstanceId()); event.setState(command.getStatus()); - event.setStartTime(command.getStartTime()); + event.setStartTime(DateUtils.timeStampToDate(command.getStartTime())); event.setExecutePath(command.getExecutePath()); event.setLogPath(command.getLogPath()); event.setAppIds(command.getAppIds()); @@ -125,10 +126,10 @@ public static TaskEvent newResultEvent(TaskExecuteResultCommand command, Channel event.setProcessInstanceId(command.getProcessInstanceId()); event.setTaskInstanceId(command.getTaskInstanceId()); event.setState(TaskExecutionStatus.of(command.getStatus())); - event.setStartTime(command.getStartTime()); + event.setStartTime(DateUtils.timeStampToDate(command.getStartTime())); event.setExecutePath(command.getExecutePath()); event.setLogPath(command.getLogPath()); - event.setEndTime(command.getEndTime()); + event.setEndTime(DateUtils.timeStampToDate(command.getEndTime())); event.setProcessId(command.getProcessId()); event.setAppIds(command.getAppIds()); event.setVarPool(command.getVarPool()); diff --git a/dolphinscheduler-master/src/test/java/org/apache/dolphinscheduler/server/master/processor/TaskAckProcessorTest.java b/dolphinscheduler-master/src/test/java/org/apache/dolphinscheduler/server/master/processor/TaskAckProcessorTest.java index 6b4b8811d924..5919bcbdb055 100644 --- a/dolphinscheduler-master/src/test/java/org/apache/dolphinscheduler/server/master/processor/TaskAckProcessorTest.java +++ b/dolphinscheduler-master/src/test/java/org/apache/dolphinscheduler/server/master/processor/TaskAckProcessorTest.java @@ -69,7 +69,7 @@ public void before() { taskExecuteRunningMessage.setExecutePath("/dolphinscheduler/worker"); taskExecuteRunningMessage.setHost("localhost"); taskExecuteRunningMessage.setLogPath("/temp/worker.log"); - taskExecuteRunningMessage.setStartTime(new Date()); + taskExecuteRunningMessage.setStartTime(System.currentTimeMillis()); taskExecuteRunningMessage.setTaskInstanceId(1); taskExecuteRunningMessage.setProcessInstanceId(1); } diff --git a/dolphinscheduler-master/src/test/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskResponseServiceTest.java b/dolphinscheduler-master/src/test/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskResponseServiceTest.java index 37b147fe250f..119ea9702b35 100644 --- a/dolphinscheduler-master/src/test/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskResponseServiceTest.java +++ b/dolphinscheduler-master/src/test/java/org/apache/dolphinscheduler/server/master/processor/queue/TaskResponseServiceTest.java @@ -82,7 +82,7 @@ public void before() { taskExecuteRunningMessage.setExecutePath("path"); taskExecuteRunningMessage.setLogPath("logPath"); taskExecuteRunningMessage.setHost("127.*.*.*"); - taskExecuteRunningMessage.setStartTime(new Date()); + taskExecuteRunningMessage.setStartTime(System.currentTimeMillis()); ackEvent = TaskEvent.newRunningEvent(taskExecuteRunningMessage, channel, @@ -94,7 +94,7 @@ public void before() { taskExecuteResultMessage.setProcessInstanceId(1); taskExecuteResultMessage.setTaskInstanceId(22); taskExecuteResultMessage.setStatus(TaskExecutionStatus.SUCCESS.getCode()); - taskExecuteResultMessage.setEndTime(new Date()); + taskExecuteResultMessage.setEndTime(System.currentTimeMillis()); taskExecuteResultMessage.setVarPool("varPol"); taskExecuteResultMessage.setAppIds("ids"); taskExecuteResultMessage.setProcessId(1); diff --git a/dolphinscheduler-remote/src/main/java/org/apache/dolphinscheduler/remote/command/TaskExecuteResultCommand.java b/dolphinscheduler-remote/src/main/java/org/apache/dolphinscheduler/remote/command/TaskExecuteResultCommand.java index 12dfa56976f1..fb24438426a8 100644 --- a/dolphinscheduler-remote/src/main/java/org/apache/dolphinscheduler/remote/command/TaskExecuteResultCommand.java +++ b/dolphinscheduler-remote/src/main/java/org/apache/dolphinscheduler/remote/command/TaskExecuteResultCommand.java @@ -57,7 +57,7 @@ public TaskExecuteResultCommand(String messageSenderAddress, String messageRecei /** * startTime */ - private Date startTime; + private long startTime; /** * host @@ -78,7 +78,7 @@ public TaskExecuteResultCommand(String messageSenderAddress, String messageRecei /** * end time */ - private Date endTime; + private long endTime; /** diff --git a/dolphinscheduler-remote/src/main/java/org/apache/dolphinscheduler/remote/command/TaskExecuteRunningCommand.java b/dolphinscheduler-remote/src/main/java/org/apache/dolphinscheduler/remote/command/TaskExecuteRunningCommand.java index 3002d57ad482..ed0231b374a3 100644 --- a/dolphinscheduler-remote/src/main/java/org/apache/dolphinscheduler/remote/command/TaskExecuteRunningCommand.java +++ b/dolphinscheduler-remote/src/main/java/org/apache/dolphinscheduler/remote/command/TaskExecuteRunningCommand.java @@ -19,8 +19,6 @@ import org.apache.dolphinscheduler.common.utils.JSONUtils; -import java.util.Date; - import lombok.Data; import lombok.EqualsAndHashCode; import lombok.NoArgsConstructor; @@ -49,7 +47,7 @@ public class TaskExecuteRunningCommand extends BaseCommand { /** * startTime */ - private Date startTime; + private long startTime; /** * host diff --git a/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/utils/LogUtils.java b/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/utils/LogUtils.java index 116051f14521..8c4c5d5c9e17 100644 --- a/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/utils/LogUtils.java +++ b/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/utils/LogUtils.java @@ -74,7 +74,8 @@ public static String getTaskLogPath(Date firstSubmitTime, Long processDefineCode * get task log path by TaskExecutionContext */ public static String getTaskLogPath(TaskExecutionContext taskExecutionContext) { - return getTaskLogPath(taskExecutionContext.getFirstSubmitTime(), taskExecutionContext.getProcessDefineCode(), + return getTaskLogPath(DateUtils.timeStampToDate(taskExecutionContext.getFirstSubmitTime()), + taskExecutionContext.getProcessDefineCode(), taskExecutionContext.getProcessDefineVersion(), taskExecutionContext.getProcessInstanceId(), taskExecutionContext.getTaskInstanceId()); diff --git a/dolphinscheduler-service/src/test/java/org/apache/dolphinscheduler/service/utils/LogUtilsTest.java b/dolphinscheduler-service/src/test/java/org/apache/dolphinscheduler/service/utils/LogUtilsTest.java index 8c9d2affe833..c0a1c70a44aa 100644 --- a/dolphinscheduler-service/src/test/java/org/apache/dolphinscheduler/service/utils/LogUtilsTest.java +++ b/dolphinscheduler-service/src/test/java/org/apache/dolphinscheduler/service/utils/LogUtilsTest.java @@ -47,7 +47,7 @@ public void testGetTaskLogPath() { taskExecutionContext.setTaskInstanceId(1000); taskExecutionContext.setProcessDefineCode(1L); taskExecutionContext.setProcessDefineVersion(1); - taskExecutionContext.setFirstSubmitTime(firstSubmitTime); + taskExecutionContext.setFirstSubmitTime(firstSubmitTime.getTime()); Logger rootLogger = (Logger) LoggerFactory.getILoggerFactory().getLogger("ROOT"); Assert.assertNotNull(rootLogger); diff --git a/dolphinscheduler-spi/src/main/java/org/apache/dolphinscheduler/spi/utils/DateUtils.java b/dolphinscheduler-spi/src/main/java/org/apache/dolphinscheduler/spi/utils/DateUtils.java index 695e70a7ba76..3f187f92a6eb 100644 --- a/dolphinscheduler-spi/src/main/java/org/apache/dolphinscheduler/spi/utils/DateUtils.java +++ b/dolphinscheduler-spi/src/main/java/org/apache/dolphinscheduler/spi/utils/DateUtils.java @@ -26,6 +26,7 @@ import java.util.Objects; import java.util.TimeZone; +import javax.annotation.Nullable; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -437,4 +438,15 @@ public static TimeZone getTimezone(String timezoneId) { public static String getTimestampString() { return String.valueOf(System.currentTimeMillis()); } + + /** + * transform timeStamp to local date + * + * @param timeStamp time stamp (milliseconds) + * @return local date + */ + public static @Nullable + Date timeStampToDate(long timeStamp) { + return timeStamp <= 0L ? null : new Date(timeStamp); + } } diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/AbstractCommandExecutor.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/AbstractCommandExecutor.java index 49e224e95d2b..b773b798981b 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/AbstractCommandExecutor.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/AbstractCommandExecutor.java @@ -403,7 +403,7 @@ private String findVarPool(String line) { * @return remain time */ private long getRemainTime() { - long usedTime = (System.currentTimeMillis() - taskRequest.getStartTime().getTime()) / 1000; + long usedTime = (System.currentTimeMillis() - taskRequest.getStartTime()) / 1000; long remainTime = taskRequest.getTaskTimeout() - usedTime; if (remainTime < 0) { diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/TaskExecutionContext.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/TaskExecutionContext.java index ad6d6b50d12e..31a0ca0be084 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/TaskExecutionContext.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/TaskExecutionContext.java @@ -23,7 +23,6 @@ import org.apache.dolphinscheduler.plugin.task.api.parameters.resource.ResourceParametersHelper; import java.io.Serializable; -import java.util.Date; import java.util.Map; import lombok.AllArgsConstructor; @@ -55,12 +54,12 @@ public class TaskExecutionContext implements Serializable { /** * task first submit time. */ - private Date firstSubmitTime; + private long firstSubmitTime; /** * task start time */ - private Date startTime; + private long startTime; /** * task type @@ -115,7 +114,7 @@ public class TaskExecutionContext implements Serializable { /** * process instance schedule time */ - private Date scheduleTime; + private long scheduleTime; /** * process instance global parameters @@ -223,7 +222,7 @@ public class TaskExecutionContext implements Serializable { /** * endTime */ - private Date endTime; + private long endTime; /** * sql TaskExecutionContext diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-pigeon/src/test/java/org/apache/dolphinscheduler/plugin/task/pigeon/PigeonTaskTest.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-pigeon/src/test/java/org/apache/dolphinscheduler/plugin/task/pigeon/PigeonTaskTest.java index f748315b657d..679cca1eb7d1 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-pigeon/src/test/java/org/apache/dolphinscheduler/plugin/task/pigeon/PigeonTaskTest.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-pigeon/src/test/java/org/apache/dolphinscheduler/plugin/task/pigeon/PigeonTaskTest.java @@ -61,7 +61,7 @@ public void before() throws Exception { Mockito.when(taskExecutionContext.getExecutePath()).thenReturn("/tmp"); Mockito.when(taskExecutionContext.getTaskAppId()).thenReturn(UUID.randomUUID().toString()); Mockito.when(taskExecutionContext.getTenantCode()).thenReturn("root"); - Mockito.when(taskExecutionContext.getStartTime()).thenReturn(new Date()); + Mockito.when(taskExecutionContext.getStartTime()).thenReturn(System.currentTimeMillis()); Mockito.when(taskExecutionContext.getTaskTimeout()).thenReturn(10000); Mockito.when(taskExecutionContext.getLogPath()).thenReturn("/tmp/dx"); // Mockito.when(taskExecutionContext.getVarPool()) diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-sql/src/main/java/org/apache/dolphinscheduler/plugin/task/sql/SqlTask.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-sql/src/main/java/org/apache/dolphinscheduler/plugin/task/sql/SqlTask.java index 2f766cca7f30..2322fc27ff1f 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-sql/src/main/java/org/apache/dolphinscheduler/plugin/task/sql/SqlTask.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-sql/src/main/java/org/apache/dolphinscheduler/plugin/task/sql/SqlTask.java @@ -33,6 +33,7 @@ import org.apache.dolphinscheduler.plugin.task.api.parser.ParameterUtils; import org.apache.dolphinscheduler.spi.datasource.BaseConnectionParam; import org.apache.dolphinscheduler.spi.enums.DbType; +import org.apache.dolphinscheduler.spi.utils.DateUtils; import org.apache.dolphinscheduler.spi.utils.JSONUtils; import org.apache.dolphinscheduler.spi.utils.StringUtils; @@ -483,7 +484,7 @@ private SqlBinds getSqlAndSqlParamsMap(String sql) { //new //replace variable TIME with $[YYYYmmddd...] in sql when history run job and batch complement job - sql = ParameterUtils.replaceScheduleTime(sql, taskExecutionContext.getScheduleTime()); + sql = ParameterUtils.replaceScheduleTime(sql, DateUtils.timeStampToDate(taskExecutionContext.getScheduleTime())); // special characters need to be escaped, ${} needs to be escaped setSqlParamsMap(sql, rgex, sqlParamsMap, paramsMap,taskExecutionContext.getTaskInstanceId()); //Replace the original value in sql ！{...} ，Does not participate in precompilation diff --git a/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/processor/TaskDispatchProcessor.java b/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/processor/TaskDispatchProcessor.java index 85f703d25cac..d73873233477 100644 --- a/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/processor/TaskDispatchProcessor.java +++ b/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/processor/TaskDispatchProcessor.java @@ -116,7 +116,7 @@ public void process(Channel channel, Command command) { taskExecutionContext.setLogPath(LogUtils.getTaskLogPath(taskExecutionContext)); // delay task process - long remainTime = DateUtils.getRemainTime(taskExecutionContext.getFirstSubmitTime(), + long remainTime = DateUtils.getRemainTime(DateUtils.timeStampToDate(taskExecutionContext.getFirstSubmitTime()), taskExecutionContext.getDelayTime() * 60L); if (remainTime > 0) { logger.info("Current taskInstance is choose delay execution, delay time: {}s", remainTime); diff --git a/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/runner/WorkerDelayTaskExecuteRunnable.java b/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/runner/WorkerDelayTaskExecuteRunnable.java index 73e14c51327d..6e6e854ee4fe 100644 --- a/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/runner/WorkerDelayTaskExecuteRunnable.java +++ b/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/runner/WorkerDelayTaskExecuteRunnable.java @@ -47,7 +47,8 @@ public long getDelay(TimeUnit unit) { TaskExecutionContext taskExecutionContext = getTaskExecutionContext(); return unit.convert( DateUtils.getRemainTime( - taskExecutionContext.getFirstSubmitTime(), taskExecutionContext.getDelayTime() * 60L), TimeUnit.SECONDS); + DateUtils.timeStampToDate(taskExecutionContext.getFirstSubmitTime()), + taskExecutionContext.getDelayTime() * 60L), TimeUnit.SECONDS); } @Override diff --git a/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/runner/WorkerTaskExecuteRunnable.java b/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/runner/WorkerTaskExecuteRunnable.java index de2780af860c..d858ee38c50e 100644 --- a/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/runner/WorkerTaskExecuteRunnable.java +++ b/dolphinscheduler-worker/src/main/java/org/apache/dolphinscheduler/server/worker/runner/WorkerTaskExecuteRunnable.java @@ -23,6 +23,7 @@ import org.apache.dolphinscheduler.common.enums.WarningType; import org.apache.dolphinscheduler.common.storage.StorageOperate; import org.apache.dolphinscheduler.common.utils.CommonUtils; +import org.apache.dolphinscheduler.common.utils.DateUtils; import org.apache.dolphinscheduler.common.utils.JSONUtils; import org.apache.dolphinscheduler.common.utils.LoggerUtils; import org.apache.dolphinscheduler.plugin.task.api.AbstractTask; @@ -44,6 +45,7 @@ import org.apache.dolphinscheduler.service.task.TaskPluginManager; import org.apache.dolphinscheduler.service.utils.ProcessUtils; +import javax.annotation.Nullable; import java.io.File; import java.io.IOException; import java.nio.file.NoSuchFileException; @@ -90,7 +92,7 @@ protected WorkerTaskExecuteRunnable( this.alertClientService = alertClientService; this.taskPluginManager = taskPluginManager; this.storageOperate = storageOperate; - String taskLogName = LoggerUtils.buildTaskId(taskExecutionContext.getFirstSubmitTime(), + String taskLogName = LoggerUtils.buildTaskId(DateUtils.timeStampToDate(taskExecutionContext.getFirstSubmitTime()), taskExecutionContext.getProcessDefineCode(), taskExecutionContext.getProcessDefineVersion(), taskExecutionContext.getProcessInstanceId(), @@ -118,7 +120,7 @@ protected void afterThrowing(Throwable throwable) throws TaskException { cancelTask(); TaskExecutionContextCacheManager.removeByTaskInstanceId(taskExecutionContext.getTaskInstanceId()); taskExecutionContext.setCurrentExecutionStatus(TaskExecutionStatus.FAILURE); - taskExecutionContext.setEndTime(new Date()); + taskExecutionContext.setEndTime(System.currentTimeMillis()); workerMessageSender.sendMessageWithRetry(taskExecutionContext, masterAddress, CommandType.TASK_EXECUTE_RESULT); logger.info( "Get a exception when execute the task, will send the task execute result to master, the current task execute result is {}", @@ -157,7 +159,7 @@ public void run() { if (Constants.DRY_RUN_FLAG_YES == taskExecutionContext.getDryRun()) { taskExecutionContext.setCurrentExecutionStatus(TaskExecutionStatus.SUCCESS); - taskExecutionContext.setEndTime(new Date()); + taskExecutionContext.setEndTime(System.currentTimeMillis()); TaskExecutionContextCacheManager.removeByTaskInstanceId(taskExecutionContext.getTaskInstanceId()); workerMessageSender.sendMessageWithRetry(taskExecutionContext, masterAddress, CommandType.TASK_EXECUTE_RESULT); @@ -185,7 +187,7 @@ public void run() { protected void initializeTask() { logger.info("Begin to initialize task"); - Date taskStartTime = new Date(); + long taskStartTime = System.currentTimeMillis(); taskExecutionContext.setStartTime(taskStartTime); logger.info("Set task startTime: {}", taskStartTime); @@ -251,7 +253,7 @@ protected void sendAlertIfNeeded() { protected void sendTaskResult() { taskExecutionContext.setCurrentExecutionStatus(task.getExitStatus()); - taskExecutionContext.setEndTime(new Date()); + taskExecutionContext.setEndTime(System.currentTimeMillis()); taskExecutionContext.setProcessId(task.getProcessId()); taskExecutionContext.setAppIds(task.getAppIds()); taskExecutionContext.setVarPool(JSONUtils.toJsonString(task.getParameters().getVarPool())); diff --git a/dolphinscheduler-worker/src/test/java/org/apache/dolphinscheduler/server/worker/processor/TaskDispatchProcessorTest.java b/dolphinscheduler-worker/src/test/java/org/apache/dolphinscheduler/server/worker/processor/TaskDispatchProcessorTest.java index 0bf8a26c452b..d88a589a8595 100644 --- a/dolphinscheduler-worker/src/test/java/org/apache/dolphinscheduler/server/worker/processor/TaskDispatchProcessorTest.java +++ b/dolphinscheduler-worker/src/test/java/org/apache/dolphinscheduler/server/worker/processor/TaskDispatchProcessorTest.java @@ -97,7 +97,7 @@ public TaskExecutionContext getTaskExecutionContext() { taskExecutionContext.setProcessDefineCode(1L); taskExecutionContext.setProcessDefineVersion(1); taskExecutionContext.setTaskType("SQL"); - taskExecutionContext.setFirstSubmitTime(new Date()); + taskExecutionContext.setFirstSubmitTime(System.currentTimeMillis()); taskExecutionContext.setDelayTime(0); taskExecutionContext.setLogPath("/tmp/test.log"); taskExecutionContext.setHost("localhost"); diff --git a/dolphinscheduler-worker/src/test/java/org/apache/dolphinscheduler/server/worker/runner/DefaultWorkerDelayTaskExecuteRunnableTest.java b/dolphinscheduler-worker/src/test/java/org/apache/dolphinscheduler/server/worker/runner/DefaultWorkerDelayTaskExecuteRunnableTest.java index 5a5d63da6cf8..ca9af127f743 100644 --- a/dolphinscheduler-worker/src/test/java/org/apache/dolphinscheduler/server/worker/runner/DefaultWorkerDelayTaskExecuteRunnableTest.java +++ b/dolphinscheduler-worker/src/test/java/org/apache/dolphinscheduler/server/worker/runner/DefaultWorkerDelayTaskExecuteRunnableTest.java @@ -53,7 +53,7 @@ public void testDryRun() { .dryRun(Constants.DRY_RUN_FLAG_YES) .taskInstanceId(0) .processDefineId(0) - .firstSubmitTime(new Date()) + .firstSubmitTime(System.currentTimeMillis()) .taskLogName("TestLogName") .build(); WorkerTaskExecuteRunnable workerTaskExecuteRunnable = new DefaultWorkerDelayTaskExecuteRunnable( @@ -77,7 +77,7 @@ public void testErrorboundTestDataSource() { .testFlag(Constants.TEST_FLAG_YES) .taskInstanceId(0) .processDefineId(0) - .firstSubmitTime(new Date()) + .firstSubmitTime(System.currentTimeMillis()) .taskLogName("TestLogName") .taskType("SQL") .taskParams("{\"localParams\":[],\"resourceList\":[],\"type\":\"POSTGRESQL\",\"datasource\":null,\"sql\":\"select * from t_ds_user\",\"sqlType\":\"0\",\"preStatements\":[],\"postStatements\":[],\"segmentSeparator\":\"\",\"displayRows\":10,\"conditionResult\":\"null\",\"dependence\":\"null\",\"switchResult\":\"null\",\"waitStartTimeout\":null}")