Refresh neurons after spawning or disbursing (#5160)

open-chat-labs · Jan 11, 2024 · 9ce72c9 · 9ce72c9
1 parent 6691d1b
commit 9ce72c9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/backend/canisters/neuron_controller/CHANGELOG.md b/backend/canisters/neuron_controller/CHANGELOG.md
@@ -8,6 +8,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 ### Changed
 
 - Separate spawning neurons from active neurons ([#5159](https://github.com/open-chat-labs/open-chat/pull/5159))
+- Refresh neurons after spawning or disbursing ([#5160](https://github.com/open-chat-labs/open-chat/pull/5160))
 
 ## [[2.0.1002](https://github.com/open-chat-labs/open-chat/releases/tag/v2.0.1002-neuron_controller)] - 2024-01-10
 

diff --git a/backend/canisters/neuron_controller/impl/src/jobs/process_neurons.rs b/backend/canisters/neuron_controller/impl/src/jobs/process_neurons.rs
@@ -75,12 +75,20 @@ async fn run_async() {
             }
         });
 
+        let mut neurons_updated = false;
         if !neurons_to_spawn.is_empty() {
             spawn_neurons(neurons_to_spawn).await;
+            neurons_updated = true;
         }
 
         if !neurons_to_disburse.is_empty() {
             disburse_neurons(neurons_to_disburse).await;
+            neurons_updated = true;
+        }
+
+        if neurons_updated {
+            // Refresh the neurons again given that they've been updated
+            ic_cdk_timers::set_timer(Duration::ZERO, || ic_cdk::spawn(run_async()));
         }
     }
 }