fix(install): wait until db inited

This commit is contained in:
Jacky 2025-05-05 01:16:33 +00:00
parent c4b8b8f088
commit 910a2efbd9
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D

View file

@ -7,6 +7,7 @@ import (
"mime" "mime"
"path" "path"
"runtime" "runtime"
"sync"
"github.com/0xJacky/Nginx-UI/internal/analytic" "github.com/0xJacky/Nginx-UI/internal/analytic"
"github.com/0xJacky/Nginx-UI/internal/cache" "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() { func PostInstall() {
installChan <- struct{}{} installChan <- struct{}{}
dbInited.Wait()
} }
func InitDatabase(ctx context.Context) { func InitDatabase(ctx context.Context) {
@ -102,6 +111,7 @@ func InitDatabase(ctx context.Context) {
query.Init(db) query.Init(db)
InitAfterDatabase(ctx) InitAfterDatabase(ctx)
dbInited.Done()
} }
func InitNodeSecret() { func InitNodeSecret() {