muyu-apiserver/rpc/inventory/internal/svc/servicecontext.go

package svc

import (
	"muyu-apiserver/model"
	"muyu-apiserver/rpc/inventory/internal/config"

	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type ServiceContext struct {
	Config            config.Config
	ProductModel      model.InvProductModel
	YarnModel         model.InvYarnModel
	ProductBatchModel model.InvProductBatchModel
	PanModel          model.InvProductPanModel
	BoltModel         model.InvProductBoltModel
	SupplierModel     model.PurSupplierModel
	ImportLogModel    model.InvStockImportLogModel
	StockCheckModel   model.InvStockCheckModel
	CheckDetailModel  model.InvStockCheckDetailModel
	StockAdjustModel  model.InvStockAdjustModel
	AdjustDetailModel model.InvStockAdjustDetailModel
}

func NewServiceContext(c config.Config) *ServiceContext {
	conn := sqlx.NewMysql(c.DataSource)
	return &ServiceContext{
		Config:            c,
		ProductModel:      model.NewInvProductModel(conn, c.Cache),
		YarnModel:         model.NewInvYarnModel(conn),
		ProductBatchModel: model.NewInvProductBatchModel(conn),
		PanModel:          model.NewInvProductPanModel(conn),
		BoltModel:         model.NewInvProductBoltModel(conn),
		SupplierModel:     model.NewPurSupplierModel(conn, c.Cache),
		ImportLogModel:    model.NewInvStockImportLogModel(conn, c.Cache),
		StockCheckModel:   model.NewInvStockCheckModel(conn, c.Cache),
		CheckDetailModel:  model.NewInvStockCheckDetailModel(conn, c.Cache),
		StockAdjustModel:  model.NewInvStockAdjustModel(conn, c.Cache),
		AdjustDetailModel: model.NewInvStockAdjustDetailModel(conn, c.Cache),
	}
}