Updated example and docs for GetArrowBatches (databricks#171)

Updated the example and documentation for GetArrowBatches to include releasing the arrow batch. Signed-off-by: Raymond Cypher <[email protected]>
rcypher-databricks · Oct 2, 2023 · 951c01b · 951c01b
2 parents e95dd4a + f1676f2
commit 951c01b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/doc.go b/doc.go
@@ -322,6 +322,7 @@ Example usage:
 			log.Printf("batch %v: nRecords=%v\n", iBatch, b.NumRows())
 			iBatch += 1
 			nRows += int(b.NumRows())
+			b.Release()
 		}
 		log.Printf("NRows: %v\n", nRows)
 	}

diff --git a/examples/arrrowbatches/main.go b/examples/arrrowbatches/main.go
@@ -89,6 +89,7 @@ func loopWithHasNext(db *sql.DB) {
 		log.Printf("batch %v: nRecords=%v\n", iBatch, b.NumRows())
 		iBatch += 1
 		nRows += int(b.NumRows())
+		b.Release()
 	}
 	log.Printf("NRows: %v\n", nRows)
 }
@@ -128,6 +129,7 @@ func loopWithNext(db *sql.DB) {
 		log.Printf("batch %v: nRecords=%v\n", iBatch, b.NumRows())
 		iBatch += 1
 		nRows += int(b.NumRows())
+		b.Release()
 	}
 
 	log.Printf("NRows: %v\n", nRows)