add connector

datarts-tech · Jan 8, 2024 · bf07eaf · bf07eaf
1 parent 74be535
commit bf07eaf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tap_cassandra/streams.py b/tap_cassandra/streams.py
@@ -36,8 +36,8 @@ def get_records(self, context):
         cql = f"select {selected_column_string} from {self.name.split('-')[1]}"
 
         if self.config.get('skip_hot_partitions'):
-            for row in self.execute_with_skip(cql, self.catalog_entry['key_properties'][0]):
+            for row in self.connector.execute_with_skip(cql, self.catalog_entry['key_properties'][0]):
                 yield row
         else:
-            for row in self.execute(cql):
+            for row in self.connector.execute(cql):
                 yield row