commit e57204bd55d27f258c19e7334c10cc4802cf14c7
author Disha Talreja <dishatalreja@cloudera.com> Mon Sep 20 16:25:21 2021 -0400
committer Sarath Subramanian <sarath@apache.org> Mon Sep 20 21:12:48 2021 -0700
tree f5b3ceab6236c4b6b24f175323cbb73b667ad58c
parent 3ad3f30ee5b74c1b231e82c4dd414da07b6a36c1

ATLAS-4431: Add additional null check when retrieving tasks from vertex

Signed-off-by: Sarath Subramanian <sarath@apache.org>

repository/src/main/java/org/apache/atlas/tasks/TaskRegistry.java

diff --git a/repository/src/main/java/org/apache/atlas/tasks/TaskRegistry.java b/repository/src/main/java/org/apache/atlas/tasks/TaskRegistry.java
index fae8a4f..32e0ad9 100644
--- a/repository/src/main/java/org/apache/atlas/tasks/TaskRegistry.java
+++ b/repository/src/main/java/org/apache/atlas/tasks/TaskRegistry.java
@@ -185,8 +185,16 @@
         ret.setType(v.getProperty(Constants.TASK_TYPE, String.class));
         ret.setStatus(v.getProperty(Constants.TASK_STATUS, String.class));
         ret.setCreatedBy(v.getProperty(Constants.TASK_CREATED_BY, String.class));
-        ret.setCreatedTime(new Date(v.getProperty(Constants.TASK_CREATED_TIME, Long.class)));
-        ret.setUpdatedTime(new Date(v.getProperty(Constants.TASK_UPDATED_TIME, Long.class)));
+
+        Long createdTime = v.getProperty(Constants.TASK_CREATED_TIME, Long.class);
+        if (createdTime != null) {
+            ret.setCreatedTime(new Date(createdTime));
+        }
+
+        Long updatedTime = v.getProperty(Constants.TASK_UPDATED_TIME, Long.class);
+        if (updatedTime != null) {
+            ret.setUpdatedTime(new Date(updatedTime));
+        }
 
         Long startTime = v.getProperty(Constants.TASK_START_TIME, Long.class);
         if (startTime != null) {