From c79a0574d2eb9761cd5e0d8e0c14d66a9a2a769c Mon Sep 17 00:00:00 2001
From: Ali Naqvi <syed.alinaqvi@gmail.com>
Date: Tue, 26 Nov 2024 12:07:34 +0800
Subject: [PATCH] chore: minor refactoring

---
 src/placeos-rest-api/controllers/drivers.cr | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/placeos-rest-api/controllers/drivers.cr b/src/placeos-rest-api/controllers/drivers.cr
index 260ece44..206960f8 100644
--- a/src/placeos-rest-api/controllers/drivers.cr
+++ b/src/placeos-rest-api/controllers/drivers.cr
@@ -132,10 +132,10 @@ module PlaceOS::Api
 
     def self.driver_reload(driver : ::PlaceOS::Model::Driver, request_id : String) : Tuple(Int32, String)
       cores = RemoteDriver.default_discovery.node_hash
-      wg = WaitGroup.new(cores.size)
+      wg = WaitGroup.new
       channel = Channel(Tuple(Int32, String)).new(cores.size)
       cores.each do |core_id, uri|
-        spawn do
+        wg.spawn do
           client = PlaceOS::Core::Client.new(uri: uri, request_id: request_id)
           resp = client.driver_reload(driver.id.as(String))
           channel.send(resp)
@@ -148,8 +148,6 @@ module PlaceOS::Api
             request_id: request_id,
           } }
           channel.send({500, "failed to request a driver reload on core #{uri}: error: #{error.message}"})
-        ensure
-          wg.done
         end
       end