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"
"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() {