WIP

Author: tellet-q

src/qdrant_client/config.rs

@@ -34,6 +34,10 @@ pub struct QdrantConfig {
 
     /// Optional compression schema to use for API requests
     pub compression: Option<CompressionEncoding>,
+
+    /// Whether to check compatibility between the client and server versions
+    pub check_compatibility: bool,
+
 }
 
 impl QdrantConfig {
@@ -183,6 +187,7 @@ impl Default for QdrantConfig {
             keep_alive_while_idle: true,
             api_key: None,
             compression: None,
+            check_compatibility: true,
         }
     }
 }

src/qdrant_client/mod.rs

@@ -10,9 +10,10 @@ mod query;
 mod search;
 mod sharding_keys;
 mod snapshot;
+// mod version_check;
 
 use std::future::Future;
-
+// use tokio::runtime::Runtime;
 use tonic::codegen::InterceptedService;
 use tonic::transport::{Channel, Uri};
 use tonic::Status;
@@ -21,6 +22,7 @@ use crate::auth::TokenInterceptor;
 use crate::channel_pool::ChannelPool;
 use crate::qdrant::{qdrant_client, HealthCheckReply, HealthCheckRequest};
 use crate::qdrant_client::config::QdrantConfig;
+// use crate::qdrant_client::version_check::health_check_sync;
 use crate::QdrantError;
 
 /// [`Qdrant`] client result
@@ -95,6 +97,7 @@ impl Qdrant {
     ///
     /// Constructs the client and connects based on the given [`QdrantConfig`](config::QdrantConfig).
     pub fn new(config: QdrantConfig) -> QdrantResult<Self> {
+        // let uri = config.uri.clone();
         let channel = ChannelPool::new(
             config.uri.parse::<Uri>()?,
             config.timeout,
@@ -104,6 +107,15 @@ impl Qdrant {
 
         let client = Self { channel, config };
 
+        if client.config.check_compatibility {
+            let client_version = env!("CARGO_PKG_VERSION").to_string();
+            // let rt = Runtime::new()?;
+            // let server_version = rt.block_on(client.health_check())?;
+            let server_version = tokio::runtime::Handle::current().block_on(client.health_check())?;
+            println!("Connected to Qdrant version: {}", server_version.version);
+            println!("Qdrant client version: {}", client_version);
+        }
+
         Ok(client)
     }