[GOBBLIN-2041] Fix NPE in CopySource::getCopyEntityClass (#3920)

apache · Apr 12, 2024 · 97fc864 · 97fc864
1 parent 19d1620
commit 97fc864
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/...lin-data-management/src/main/java/org/apache/gobblin/data/management/copy/CopySource.java b/...lin-data-management/src/main/java/org/apache/gobblin/data/management/copy/CopySource.java
@@ -552,10 +552,13 @@ public static void serializeCopyEntity(State state, CopyEntity copyEntity) {
     state.setProp(COPY_ENTITY_CLASS, copyEntity.getClass().getName());
   }
 
-  public static Class<?> getCopyEntityClass(State state)
-      throws IOException {
+  public static Class<?> getCopyEntityClass(State state) throws IOException {
+    Optional<String> optClassName = Optional.fromNullable(state.getProp(COPY_ENTITY_CLASS));
+    if (!optClassName.isPresent()) {
+      throw new IOException("property '" + COPY_ENTITY_CLASS + "' not found in state: " + state);
+    }
     try {
-      return Class.forName(state.getProp(COPY_ENTITY_CLASS));
+      return Class.forName(optClassName.get());
     } catch (ClassNotFoundException cnfe) {
       throw new IOException(cnfe);
     }