Fix pool management

src-tauri/grpc/src/manager.rs

@@ -150,34 +150,41 @@ impl GrpcConnection {
 }
 
 pub struct GrpcHandle {
-    connections: HashMap<String, GrpcConnection>,
     pools: HashMap<String, DescriptorPool>,
 }
 
 impl Default for GrpcHandle {
     fn default() -> Self {
-        let connections = HashMap::new();
         let pools = HashMap::new();
-        Self { connections, pools }
+        Self { pools }
     }
 }
 
 impl GrpcHandle {
     pub async fn services_from_files(
-        &self,
+        &mut self,
+        id: &str,
+        uri: &Uri,
         paths: Vec<PathBuf>,
     ) -> Result<Vec<ServiceDefinition>, String> {
         let pool = fill_pool_from_files(paths).await?;
+        self.pools.insert(self.get_pool_key(id, uri), pool.clone());
         Ok(self.services_from_pool(&pool))
     }
     pub async fn services_from_reflection(
         &mut self,
+        id: &str,
         uri: &Uri,
     ) -> Result<Vec<ServiceDefinition>, String> {
         let pool = fill_pool(uri).await?;
+        self.pools.insert(self.get_pool_key(id, uri), pool.clone());
         Ok(self.services_from_pool(&pool))
     }
 
+    fn get_pool_key(&self, id: &str, uri: &Uri) -> String {
+        format!("{}-{}", id, uri)
+    }
+
     fn services_from_pool(&self, pool: &DescriptorPool) -> Vec<ServiceDefinition> {
         pool.services()
             .map(|s| {
@@ -268,7 +275,6 @@ impl GrpcHandle {
 
         let conn = get_transport();
         let connection = GrpcConnection { pool, conn, uri };
-        self.connections.insert(id.to_string(), connection.clone());
         Ok(connection)
     }
 }

src-tauri/grpc/src/proto.rs

@@ -24,6 +24,7 @@ use tonic_reflection::pb::server_reflection_response::MessageResponse;
 use tonic_reflection::pb::ServerReflectionRequest;
 
 pub async fn fill_pool_from_files(paths: Vec<PathBuf>) -> Result<DescriptorPool, String> {
+    println!("FILL POOL FROM FILES");
     let mut pool = DescriptorPool::new();
     let random_file_name = format!("{}.desc", uuid::Uuid::new_v4());
     let desc_path = temp_dir().join(random_file_name);
@@ -88,6 +89,7 @@ pub fn get_transport() -> Client<HttpsConnector<HttpConnector>, BoxBody> {
 }
 
 pub async fn fill_pool(uri: &Uri) -> Result<DescriptorPool, String> {
+    println!("FILL POOL FROM URI");
     let mut pool = DescriptorPool::new();
     let mut client = ServerReflectionClient::with_origin(get_transport(), uri.clone());