From 1c31c2d7068737af583d76ec0e7dc12125a5c34d Mon Sep 17 00:00:00 2001
From: Lennard Kittner <lennardkittner@icloud.com>
Date: Tue, 1 Apr 2025 12:07:07 +0200
Subject: Fix budget reset

---
 src/energy.rs | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src')

diff --git a/src/energy.rs b/src/energy.rs
index 439792d..35ead3b 100644
--- a/src/energy.rs
+++ b/src/energy.rs
@@ -145,7 +145,9 @@ impl EnergyService {
         match request {
             Request::NewTask(pid, task_info) => {
                 if let Some(info) = self.process_info.write().unwrap().get_mut(&pid) {
+                    let old_budget = task_info.read_budget();
                     info.task_info = task_info.clone();
+                    info.task_info.set_budget(old_budget);
                     return;
                 }
                 self.estimator.start_trace(
-- 
cgit v1.3.1