fix: increment the cursor position

Once a row is delivered, we should increment the offset.
getdozer · Sep 15, 2023 · cea84e6 · cea84e6
1 parent c64a676
commit cea84e6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dozer-ingestion/src/connectors/snowflake/connection/client.rs b/dozer-ingestion/src/connectors/snowflake/connection/client.rs
@@ -487,7 +487,7 @@ fn exec_iter(pool: Pool, query: String) -> ExecIter {
     let schema_ref = schema.clone();
 
     let mut generator: Gen<Vec<Field>, (), _> = gen!({
-        let cursor_position = 0u64;
+        let mut cursor_position = 0u64;
         'retry: loop {
             let conn = pool.get_conn().map_err(QueryError)?;
             {
@@ -515,6 +515,7 @@ fn exec_iter(pool: Pool, query: String) -> ExecIter {
                     let fields =
                         get_fields_from_cursor(cursor, cols, schema.borrow().as_deref().unwrap())?;
                     yield_!(fields);
+                    cursor_position += 1;
                 }
             }
             conn.return_();