From fb24f44e22aef1530d08f5d2c9408a8ece17ecc7 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Wed, 13 Dec 2023 19:45:26 +0800
Subject: [PATCH] :art: Improve callback functions

---
 pkg/administration/sites.go | 3 +--
 pkg/sign/pm.go              | 8 ++++++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/pkg/administration/sites.go b/pkg/administration/sites.go
index 2937d51..8291599 100644
--- a/pkg/administration/sites.go
+++ b/pkg/administration/sites.go
@@ -29,7 +29,6 @@ func doSyncSite(c *fiber.Ctx) error {
 		file.Write(raw)
 		defer file.Close()
 	}
-
 	if site, ok := lo.Find(sign.App.Sites, func(item *sign.SiteConfig) bool {
 		return item.ID == id
 	}); ok {
@@ -40,7 +39,7 @@ func doSyncSite(c *fiber.Ctx) error {
 
 	// Reload
 	sign.ReadInConfig(viper.GetString("paths.configs"))
-	sign.App.PreheatProcesses(func(total int, success int) {})
+	sign.App.PreheatProcesses()
 
 	return c.SendStatus(fiber.StatusOK)
 }
diff --git a/pkg/sign/pm.go b/pkg/sign/pm.go
index 325b6dc..f48295d 100644
--- a/pkg/sign/pm.go
+++ b/pkg/sign/pm.go
@@ -80,7 +80,7 @@ func (v *ProcessConfig) StopProcess() error {
 	return nil
 }
 
-func (v *RoadApp) PreheatProcesses(callback func(total int, success int)) {
+func (v *RoadApp) PreheatProcesses(callbacks ...func(total int, success int)) {
 	var processes []*ProcessConfig
 	for _, site := range v.Sites {
 		for _, process := range site.Processes {
@@ -97,5 +97,9 @@ func (v *RoadApp) PreheatProcesses(callback func(total int, success int)) {
 		}
 	}
 
-	callback(len(processes), success)
+	if len(callbacks) > 0 {
+		for _, callback := range callbacks {
+			callback(len(processes), success)
+		}
+	}
 }