Merge pull request #40 from qdrant/tonic-0.9

update to tonic 0.9

Author: agourlay

Cargo.toml

@@ -13,9 +13,9 @@ readme = "README.md"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-tonic = { version = "0.8.3", features = ["tls", "tls-roots"] }
-prost = "0.11.8"
-prost-types = "0.11.8"
+tonic = { version = "0.9.2", features = ["tls", "tls-roots"] }
+prost = "0.11.9"
+prost-types = "0.11.9"
 anyhow = "1"
 
 # If you don't want to have serde as a dependency, please consider contributing a feature flag
@@ -25,7 +25,7 @@ reqwest = { version = "0.11.16", optional = true, features = ["stream"] }
 futures-util = { version = "0.3.28", optional = true }
 
 [dev-dependencies]
-tonic-build = { version = "0.8.4", features = ["prost"] }
+tonic-build = { version = "0.9.2", features = ["prost"] }
 tokio = { version = "1.27.0", features = ["rt-multi-thread"] }
 
 [features]

src/client.rs

@@ -228,17 +228,23 @@ pub struct QdrantClient {
 }
 
 impl QdrantClient {
+    /// Wraps a channel with a token interceptor
+    fn with_api_key(&self, channel: Channel) -> InterceptedService<Channel, TokenInterceptor> {
+        let interceptor = TokenInterceptor::new(self.cfg.api_key.clone());
+        InterceptedService::new(channel, interceptor)
+    }
+
     pub async fn with_snapshot_client<T, O: Future<Output = Result<T, Status>>>(
         &self,
         f: impl Fn(SnapshotsClient<InterceptedService<Channel, TokenInterceptor>>) -> O,
     ) -> Result<T, Status> {
         self.channel
             .with_channel(
                 |channel| {
-                    f(SnapshotsClient::with_interceptor(
-                        channel,
-                        TokenInterceptor::new(self.cfg.api_key.clone()),
-                    ))
+                    let service = self.with_api_key(channel);
+                    let client = SnapshotsClient::new(service);
+                    let client = client.max_decoding_message_size(usize::MAX);
+                    f(client)
                 },
                 false,
             )
@@ -253,10 +259,10 @@ impl QdrantClient {
         self.channel
             .with_channel(
                 |channel| {
-                    f(CollectionsClient::with_interceptor(
-                        channel,
-                        TokenInterceptor::new(self.cfg.api_key.clone()),
-                    ))
+                    let service = self.with_api_key(channel);
+                    let client = CollectionsClient::new(service);
+                    let client = client.max_decoding_message_size(usize::MAX);
+                    f(client)
                 },
                 false,
             )
@@ -271,10 +277,10 @@ impl QdrantClient {
         self.channel
             .with_channel(
                 |channel| {
-                    f(PointsClient::with_interceptor(
-                        channel,
-                        TokenInterceptor::new(self.cfg.api_key.clone()),
-                    ))
+                    let service = self.with_api_key(channel);
+                    let client = PointsClient::new(service);
+                    let client = client.max_decoding_message_size(usize::MAX);
+                    f(client)
                 },
                 true,
             )
@@ -289,10 +295,10 @@ impl QdrantClient {
         self.channel
             .with_channel(
                 |channel| {
-                    f(qdrant_client::QdrantClient::with_interceptor(
-                        channel,
-                        TokenInterceptor::new(self.cfg.api_key.clone()),
-                    ))
+                    let service = self.with_api_key(channel);
+                    let client = qdrant_client::QdrantClient::new(service);
+                    let client = client.max_decoding_message_size(usize::MAX);
+                    f(client)
                 },
                 true,
             )