fix: init db调整位置

main
youys 2 weeks ago
parent ddebaa3c41
commit 0d52bd0a34

@ -9,6 +9,7 @@ import (
"github.com/go-kratos/kratos/v2/transport/http" "github.com/go-kratos/kratos/v2/transport/http"
"os" "os"
"vgpu/internal/conf" "vgpu/internal/conf"
"vgpu/internal/database"
_ "go.uber.org/automaxprocs" _ "go.uber.org/automaxprocs"
) )
@ -32,6 +33,12 @@ func init() {
func main() { func main() {
flag.Parse() flag.Parse()
var ctx = context.Background() var ctx = context.Background()
if err := initDatabase(); err != nil {
log.Errorf("数据库初始化失败: %v", err)
os.Exit(1)
}
app, cleanup, err := initApp(flagconf, ctx) app, cleanup, err := initApp(flagconf, ctx)
if err != nil { if err != nil {
panic(err) panic(err)
@ -62,3 +69,15 @@ func newApp(ctx context.Context, logger log.Logger, gs *grpc.Server, hs *http.Se
func getNodeSelectors(c *conf.Bootstrap) map[string]string { func getNodeSelectors(c *conf.Bootstrap) map[string]string {
return c.NodeSelectors return c.NodeSelectors
} }
func initDatabase() error {
config, err := database.LoadConfig(flagconf)
log.Infof("config: %+v", config)
if err != nil {
log.Errorf("Failed to load config: %v", err)
return err
}
database.InitDB(&config.Database)
log.Infof("初始化%s成功", config.Database.Driver)
return nil
}

Loading…
Cancel
Save