bj_power/bj_power_wms/internal/svc/service_context.go

package svc

import (
	"bj_power_wms/ent"
	"bj_power_wms/internal/config"
	"bj_power_wms/internal/db"
)

type ServiceContext struct {
	Config    config.Config
	EntClient *ent.Client
}

func NewServiceContext(c config.Config) *ServiceContext {
	conf := db.DatabaseConf{
		Host:     c.Postgres.Host,
		Port:     c.Postgres.Port,
		User:     c.Postgres.User,
		Password: c.Postgres.Password,
		Dbname:   c.Postgres.DBName,
	}

	// 目标数据库不存在则自动创建
	if err := db.EnsureDB(conf); err != nil {
		panic(err)
	}

	entClient := db.MustNewDB(conf)

	// 启动时自动建表
	if err := db.AutoMigrate(entClient); err != nil {
		panic(err)
	}

	// 首次启动预置基础数据
	if err := db.SeedIfEmpty(entClient); err != nil {
		panic(err)
	}

	return &ServiceContext{
		Config:    c,
		EntClient: entClient,
	}
}