bj_power/bj_power_wms/internal/svc/service_context.go

package svc

import (
	"bj_power_wms/ent"
	"bj_power_wms/internal/agv"
	"bj_power_wms/internal/config"
	"bj_power_wms/internal/db"
	"bj_power_wms/internal/eventlog"
	"bj_power_wms/internal/mesclient"
)

type ServiceContext struct {
	Config    config.Config
	EntClient *ent.Client
	EventLog  *eventlog.Writer
	Agv       agv.Client
	Mes       *mesclient.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,
		EventLog:  eventlog.NewWriter(entClient),
		Agv:       agv.New(c.Agv),
		Mes:       mesclient.New(c.Mes.BaseURL, c.Mes.Token),
	}
}