36 lines
1.1 KiB
Go
36 lines
1.1 KiB
Go
|
package svc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"muyu-apiserver/model"
|
||
|
|
"muyu-apiserver/rpc/production/internal/config"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ServiceContext struct {
|
||
|
|
Config config.Config
|
||
|
|
Conn sqlx.SqlConn
|
||
|
|
PlanModel model.ProProductionPlanModel
|
||
|
|
YarnRatioModel model.ProPlanYarnRatioModel
|
||
|
|
ProcessStepModel model.ProPlanProcessStepModel
|
||
|
|
ShareLinkModel model.ProShareLinkModel
|
||
|
|
PlanFactoryModel model.ProPlanFactoryModel
|
||
|
|
InboundModel model.ProInboundRecordModel
|
||
|
|
InventoryLogModel model.InvInventoryLogModel
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewServiceContext(c config.Config) *ServiceContext {
|
||
|
|
conn := sqlx.NewMysql(c.DataSource)
|
||
|
|
return &ServiceContext{
|
||
|
|
Config: c,
|
||
|
|
Conn: conn,
|
||
|
|
PlanModel: model.NewProProductionPlanModel(conn, c.Cache),
|
||
|
|
YarnRatioModel: model.NewProPlanYarnRatioModel(conn),
|
||
|
|
ProcessStepModel: model.NewProPlanProcessStepModel(conn),
|
||
|
|
ShareLinkModel: model.NewProShareLinkModel(conn),
|
||
|
|
PlanFactoryModel: model.NewProPlanFactoryModel(conn),
|
||
|
|
InboundModel: model.NewProInboundRecordModel(conn),
|
||
|
|
InventoryLogModel: model.NewInvInventoryLogModel(conn),
|
||
|
|
}
|
||
|
|
}
|