From 910a2efbd92052e520da7cad4a2729fdc17011d1 Mon Sep 17 00:00:00 2001
From: Jacky <me@jackyu.cn>
Date: Mon, 5 May 2025 01:16:33 +0000
Subject: [PATCH] fix(install): wait until db inited

---
 internal/kernel/boot.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/internal/kernel/boot.go b/internal/kernel/boot.go
index 97fe8f9e..0f3c212a 100644
--- a/internal/kernel/boot.go
+++ b/internal/kernel/boot.go
@@ -7,6 +7,7 @@ import (
 	"mime"
 	"path"
 	"runtime"
+	"sync"
 
 	"github.com/0xJacky/Nginx-UI/internal/analytic"
 	"github.com/0xJacky/Nginx-UI/internal/cache"
@@ -80,10 +81,18 @@ func recovery() {
 	}
 }
 
-var installChan = make(chan struct{})
+var (
+	installChan = make(chan struct{})
+	dbInited    = sync.WaitGroup{}
+)
+
+func init() {
+	dbInited.Add(1)
+}
 
 func PostInstall() {
 	installChan <- struct{}{}
+	dbInited.Wait()
 }
 
 func InitDatabase(ctx context.Context) {
@@ -102,6 +111,7 @@ func InitDatabase(ctx context.Context) {
 	query.Init(db)
 
 	InitAfterDatabase(ctx)
+	dbInited.Done()
 }
 
 func InitNodeSecret() {