Merge commit 'd803bfe2b1fe7f5e219e50ac20d6801a0a58ac75' as 'vendor/ruvector'

vendor/ruvector/crates/ruvector-server/src/state.rs

@@ -0,0 +1,60 @@
+//! Shared application state
+
+use dashmap::DashMap;
+use ruvector_core::VectorDB;
+use std::sync::Arc;
+
+/// Shared application state
+#[derive(Clone)]
+pub struct AppState {
+    /// Map of collection name to VectorDB
+    pub collections: Arc<DashMap<String, Arc<VectorDB>>>,
+}
+
+impl AppState {
+    /// Create a new application state
+    pub fn new() -> Self {
+        Self {
+            collections: Arc::new(DashMap::new()),
+        }
+    }
+
+    /// Get a collection by name
+    pub fn get_collection(&self, name: &str) -> Option<Arc<VectorDB>> {
+        self.collections.get(name).map(|c| c.clone())
+    }
+
+    /// Insert a collection
+    pub fn insert_collection(&self, name: String, db: Arc<VectorDB>) {
+        self.collections.insert(name, db);
+    }
+
+    /// Remove a collection
+    pub fn remove_collection(&self, name: &str) -> Option<Arc<VectorDB>> {
+        self.collections.remove(name).map(|(_, c)| c)
+    }
+
+    /// Check if a collection exists
+    pub fn contains_collection(&self, name: &str) -> bool {
+        self.collections.contains_key(name)
+    }
+
+    /// Get all collection names
+    pub fn collection_names(&self) -> Vec<String> {
+        self.collections
+            .iter()
+            .map(|entry| entry.key().clone())
+            .collect()
+    }
+
+    /// Get the number of collections
+    pub fn collection_count(&self) -> usize {
+        self.collections.len()
+    }
+}
+
+impl Default for AppState {
+    fn default() -> Self {
+        Self::new()
+    }
+}