rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-10-12 14:52:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

docs: pass a RecordBatch in the merge_insert example

Browse Source 
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
Will Jones
2026-10-10 17:12:05 +00:00
co-authored by Claude Opus 5.5
parent 8bdc515f1e
commit 300d494d6e
1 changed files with 14 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rust/lancedb/src/table.rs
+14 -20
View File
@@ -1605,8 +1605,7 @@ impl Table {
///
/// ```no_run
/// # use std::sync::Arc;
/// # use arrow_array::{FixedSizeListArray, types::Float32Type, RecordBatch,
/// # RecordBatchIterator, Int32Array};
/// # use arrow_array::{FixedSizeListArray, types::Float32Type, RecordBatch, Int32Array};
/// # use arrow_schema::{Schema, Field, DataType};
/// # tokio::runtime::Runtime::new().unwrap().block_on(async {
/// let tmpdir = tempfile::tempdir().unwrap();
@@ -1620,30 +1619,25 @@ impl Table {
/// # Field::new("vector", DataType::FixedSizeList(
/// # Arc::new(Field::new("item", DataType::Float32, true)), 128), true),
/// # ]));
/// let new_data = RecordBatchIterator::new(
/// vec![RecordBatch::try_new(
/// schema.clone(),
/// vec![
/// Arc::new(Int32Array::from_iter_values(0..10)),
/// Arc::new(
/// FixedSizeListArray::from_iter_primitive::<Float32Type, _, _>(
/// (0..10).map(|_| Some(vec![Some(1.0); 128])),
/// 128,
/// ),
/// ),
/// ],
/// )
/// .unwrap()]
/// .into_iter()
/// .map(Ok),
/// let new_data = RecordBatch::try_new(
/// schema.clone(),
/// );
/// vec![
/// Arc::new(Int32Array::from_iter_values(0..10)),
/// Arc::new(
/// FixedSizeListArray::from_iter_primitive::<Float32Type, _, _>(
/// (0..10).map(|_| Some(vec![Some(1.0); 128])),
/// 128,
/// ),
/// ),
/// ],
/// )
/// .unwrap();
/// // Perform an upsert operation
/// let mut merge_insert = tbl.merge_insert(&["id"]);
/// merge_insert
/// .when_matched_update_all(None)
/// .when_not_matched_insert_all();
/// merge_insert.execute(Box::new(new_data) as Box<dyn RecordBatchReader + Send>).await.unwrap();
/// merge_insert.execute(new_data).await.unwrap();
/// # });
/// ```
pub fn merge_insert(&self, on: &[&str]) -> MergeInsertBuilder {