From 6a85acda9eb944ab5e651f37e95328523ec7ac6a Mon Sep 17 00:00:00 2001 From: ghe0 Date: Tue, 20 May 2025 01:45:31 +0300 Subject: [PATCH] when node reconnects, send deleted VMs previously, active VMs were sent --- Cargo.lock | 2 +- src/grpc/vm.rs | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index b945966..3826a4e 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -1000,7 +1000,7 @@ dependencies = [ [[package]] name = "detee-shared" version = "0.1.0" -source = "git+ssh://git@gitea.detee.cloud/testnet/proto?branch=surreal_brain#7ef70689328191fe4951dcf80dae272afdfc8ff2" +source = "git+ssh://git@gitea.detee.cloud/testnet/proto?branch=surreal_brain#d6ca058d2de78b5257517034bca2b2c7d5929db8" dependencies = [ "bincode 2.0.1", "prost", diff --git a/src/grpc/vm.rs b/src/grpc/vm.rs index 7922615..11c6ad0 100644 --- a/src/grpc/vm.rs +++ b/src/grpc/vm.rs @@ -30,7 +30,7 @@ impl VmDaemonServer { #[tonic::async_trait] impl BrainVmDaemon for VmDaemonServer { type BrainMessagesStream = Pin> + Send>>; - type RegisterVmNodeStream = Pin> + Send>>; + type RegisterVmNodeStream = Pin> + Send>>; async fn register_vm_node( &self, @@ -59,12 +59,12 @@ impl BrainVmDaemon for VmDaemonServer { .register(&self.db) .await?; - info!("Sending existing contracts to {}", req.node_pubkey); - let contracts = db::ActiveVmWithNode::list_by_node(&self.db, &req.node_pubkey).await?; + info!("Sending deleted contracts to {}", req.node_pubkey); + let deleted_vms = db::DeletedVm::list_by_node(&self.db, &req.node_pubkey).await?; let (tx, rx) = mpsc::channel(6); tokio::spawn(async move { - for contract in contracts { - let _ = tx.send(Ok(contract.into())).await; + for deleted_vm in deleted_vms { + let _ = tx.send(Ok(deleted_vm.into())).await; } }); let output_stream = ReceiverStream::new(rx);