[ISSUE-53] Fix NPE in DiffData job on printing final counts

abinaya21 · abinaya21 · commit 7ee4ade4d140 · 2022-12-20T14:00:18.000+05:30
diff --git a/pom.xml b/pom.xml
@@ -3,7 +3,7 @@
 
   <groupId>datastax.astra.migrate</groupId>
   <artifactId>cassandra-data-migrator</artifactId>
-  <version>2.10.1</version>
+  <version>2.10.2</version>
   <packaging>jar</packaging>
 
   <properties>
diff --git a/src/main/java/datastax/astra/migrate/AbstractJobSession.java b/src/main/java/datastax/astra/migrate/AbstractJobSession.java
@@ -24,6 +24,9 @@ protected AbstractJobSession(CqlSession sourceSession, CqlSession astraSession,
     }
 
     protected AbstractJobSession(CqlSession sourceSession, CqlSession astraSession, SparkConf sc, boolean isJobMigrateRowsFromFile) {
+        if (sourceSession == null) {
+            return;
+        }
         this.sourceSession = sourceSession;
         this.astraSession = astraSession;
 

Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,9 @@ protected AbstractJobSession(CqlSession sourceSession, CqlSession astraSession,`
`24`	`24`	`}`
`25`	`25`
`26`	`26`	`protected AbstractJobSession(CqlSession sourceSession, CqlSession astraSession, SparkConf sc, boolean isJobMigrateRowsFromFile) {`
	`27`	`+ if (sourceSession == null) {`
	`28`	`+ return;`
	`29`	`+ }`
`27`	`30`	`this.sourceSession = sourceSession;`
`28`	`31`	`this.astraSession = astraSession;`
`29`	`32`