From ab22309c8d679a83d1558ca9c046f12f5380f269 Mon Sep 17 00:00:00 2001 From: generall Date: Sun, 16 Nov 2025 20:26:07 +0100 Subject: [PATCH 1/2] make .initial_state usable --- src/builders/create_shard_key_builder.rs | 5 +++-- tests/snippet_tests/test_create_shard_key.rs | 4 +++- tests/snippets/create_shard_key.rs | 4 +++- 3 files changed, 9 insertions(+), 4 deletions(-) diff --git a/src/builders/create_shard_key_builder.rs b/src/builders/create_shard_key_builder.rs index fc09c9ba..b799efe2 100644 --- a/src/builders/create_shard_key_builder.rs +++ b/src/builders/create_shard_key_builder.rs @@ -51,11 +51,12 @@ impl CreateShardKeyBuilder { /// /// Use with caution! Setting arbirray replica states here may break your Qdrant cluster. #[allow(unused_mut)] - pub fn initial_state(self, value: i32) -> Self { + pub fn initial_state(self, value: ReplicaState) -> Self { let mut new = self; - new.initial_state = Option::Some(Option::Some(value)); + new.initial_state = Option::Some(Option::Some(value as i32)); new } + fn build_inner(self) -> Result { Ok(CreateShardKey { shard_key: self.shard_key.unwrap_or_default(), diff --git a/tests/snippet_tests/test_create_shard_key.rs b/tests/snippet_tests/test_create_shard_key.rs index 52fcc842..14f9a8e4 100644 --- a/tests/snippet_tests/test_create_shard_key.rs +++ b/tests/snippet_tests/test_create_shard_key.rs @@ -6,6 +6,7 @@ async fn test_create_shard_key() { // Please, modify the snippet in the `../snippets/create_shard_key.rs` file use qdrant_client::qdrant::shard_key::Key; use qdrant_client::qdrant::{CreateShardKeyBuilder, CreateShardKeyRequestBuilder}; + use qdrant_client::qdrant::ReplicaState; use qdrant_client::Qdrant; let client = Qdrant::from_url("http://localhost:6334").build()?; @@ -14,7 +15,8 @@ async fn test_create_shard_key() { .create_shard_key( CreateShardKeyRequestBuilder::new("{collection_name}").request( CreateShardKeyBuilder::default() - .shard_key(Key::Keyword("{shard_key}".to_string())), + .shard_key(Key::Keyword("{shard_key}".to_string())) + .initial_state(ReplicaState::Active), ), ) .await?; diff --git a/tests/snippets/create_shard_key.rs b/tests/snippets/create_shard_key.rs index 2b4ce631..6559eb74 100644 --- a/tests/snippets/create_shard_key.rs +++ b/tests/snippets/create_shard_key.rs @@ -1,5 +1,6 @@ use qdrant_client::qdrant::shard_key::Key; use qdrant_client::qdrant::{CreateShardKeyBuilder, CreateShardKeyRequestBuilder}; +use qdrant_client::qdrant::ReplicaState; use qdrant_client::Qdrant; let client = Qdrant::from_url("http://localhost:6334").build()?; @@ -8,7 +9,8 @@ client .create_shard_key( CreateShardKeyRequestBuilder::new("{collection_name}").request( CreateShardKeyBuilder::default() - .shard_key(Key::Keyword("{shard_key}".to_string())), + .shard_key(Key::Keyword("{shard_key}".to_string())) + .initial_state(ReplicaState::Active), ), ) .await?; From d644401a342da099d22d9966d7ded6080278a70f Mon Sep 17 00:00:00 2001 From: generall Date: Sun, 16 Nov 2025 20:51:13 +0100 Subject: [PATCH 2/2] fmt --- src/builders/create_shard_key_builder.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/builders/create_shard_key_builder.rs b/src/builders/create_shard_key_builder.rs index b799efe2..ccf2f7e4 100644 --- a/src/builders/create_shard_key_builder.rs +++ b/src/builders/create_shard_key_builder.rs @@ -56,7 +56,7 @@ impl CreateShardKeyBuilder { new.initial_state = Option::Some(Option::Some(value as i32)); new } - + fn build_inner(self) -> Result { Ok(CreateShardKey { shard_key: self.shard_key.unwrap_or_default(),