Fix bug with driver instantiation.

homeaway · Jul 5, 2019 · 94c07a6 · 94c07a6
1 parent 70caff6
commit 94c07a6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/com/homeaway/datatools/photon/driver/CassandraPhotonDriver.java b/src/main/java/com/homeaway/datatools/photon/driver/CassandraPhotonDriver.java
@@ -115,7 +115,7 @@ public BeamDao getBeamDao() {
     public BeamDataDao getBeamDataDao() {
         return Optional.ofNullable(beamDataDao)
                 .orElseGet(() -> {
-                    beamDataDao = new CassandraBeamDataDao(getSession(properties), partitionHelper, beamDataManifestDao);
+                    beamDataDao = new CassandraBeamDataDao(getSession(properties), getPartitionHelper(), getBeamDataManifestDao());
                     return beamDataDao;
                 });
     }
@@ -124,7 +124,7 @@ public BeamDataDao getBeamDataDao() {
     public BeamDataManifestDao getBeamDataManifestDao() {
         return Optional.ofNullable(beamDataManifestDao)
                 .orElseGet(() -> {
-                    beamDataManifestDao = new CassandraBeamDataManifestDao(getSession(properties), partitionHelper);
+                    beamDataManifestDao = new CassandraBeamDataManifestDao(getSession(properties), getPartitionHelper());
                     return beamDataManifestDao;
                 });
     }
@@ -133,7 +133,7 @@ public BeamDataManifestDao getBeamDataManifestDao() {
     public BeamProcessedDao getBeamProcessedDao() {
         return Optional.ofNullable(beamProcessedDao)
                 .orElseGet(() -> {
-                    beamProcessedDao = new CassandraBeamProcessedDao(getMultiRegionSession(properties), partitionHelper);
+                    beamProcessedDao = new CassandraBeamProcessedDao(getMultiRegionSession(properties), getPartitionHelper());
                     return beamProcessedDao;
                 });
     }