handle exit logic

internal/routes/routes.go

@@ -1,6 +1,8 @@
 package routes
 
 import (
+	"sync"
+
 	"github.com/usual2970/certimate/internal/notify"
 	"github.com/usual2970/certimate/internal/repository"
 	"github.com/usual2970/certimate/internal/rest"
@@ -11,14 +13,26 @@ import (
 	"github.com/pocketbase/pocketbase/apis"
 )
 
+var (
+	workflowSvc     rest.WorkflowService
+	workflowSvcOnce sync.Once
+)
+
+func getWorkflowService() rest.WorkflowService {
+	workflowSvcOnce.Do(func() {
+		workflowRepo := repository.NewWorkflowRepository()
+		workflowSvc = workflow.NewWorkflowService(workflowRepo)
+	})
+	return workflowSvc
+}
+
 func Register(e *echo.Echo) {
 	group := e.Group("/api", apis.RequireAdminAuth())
 
 	notifyRepo := repository.NewSettingsRepository()
 	notifySvc := notify.NewNotifyService(notifyRepo)
 
-	workflowRepo := repository.NewWorkflowRepository()
-	workflowSvc := workflow.NewWorkflowService(workflowRepo)
+	workflowSvc := getWorkflowService()
 
 	statisticsRepo := repository.NewStatisticsRepository()
 	statisticsSvc := statistics.NewStatisticsService(statisticsRepo)
@@ -29,3 +43,7 @@ func Register(e *echo.Echo) {
 
 	rest.NewStatisticsHandler(group, statisticsSvc)
 }
+
+func Unregister() {
+	getWorkflowService().Stop()
+}