leekeiabstraction commented on code in PR #216:
URL: https://github.com/apache/fluss-rust/pull/216#discussion_r2751300045
##########
crates/fluss/tests/integration/kv_table.rs:
##########
@@ -851,4 +851,126 @@ mod kv_table_test {
.await
.expect("Failed to drop table");
}
+
+ /// Integration test for concurrent batched lookups.
+ #[tokio::test]
+ async fn batched_concurrent_lookups() {
+ use futures::stream::{FuturesUnordered, StreamExt};
+
+ let cluster = get_fluss_cluster();
+ let connection = cluster.get_fluss_connection().await;
+
+ let admin = connection.get_admin().await.expect("Failed to get admin");
+
+ let table_path = TablePath::new("fluss".to_string(),
"test_batched_lookups".to_string());
+
+ let table_descriptor = TableDescriptor::builder()
+ .schema(
+ Schema::builder()
+ .column("id", DataTypes::int())
+ .column("name", DataTypes::string())
+ .column("value", DataTypes::bigint())
+ .primary_key(vec!["id".to_string()])
+ .build()
+ .expect("Failed to build schema"),
+ )
+ .build()
+ .expect("Failed to build table");
+
+ create_table(&admin, &table_path, &table_descriptor).await;
+
+ let table = connection
+ .get_table(&table_path)
+ .await
+ .expect("Failed to get table");
+
+ let table_upsert = table.new_upsert().expect("Failed to create
upsert");
+ let mut upsert_writer = table_upsert
+ .create_writer()
+ .expect("Failed to create writer");
+
+ // Insert only even-numbered records (0, 2, 4, ..., 98)
Review Comment:
This part took over 20 seconds to run on my machine. Maybe we can
parallelise upsert as well.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
