初始化2

bj_power_mes/internal/logic/material.go

@@ -0,0 +1,61 @@
+package logic
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	"bj_power_mes/ent"
+	"bj_power_mes/ent/materialrequest"
+)
+
+// GenerateMaterialRequest 按日排产自动算料生成备料单
+// 数量 = 该日所有工单计划数量 × BOM单台用量×(1+损耗率)，材料来源=BOM
+func (s *Service) GenerateMaterialRequest(ctx context.Context, planDate string, operator string) (int, error) {
+	plans, err := s.ListDailyPlans(ctx, "", planDate)
+	if err != nil {
+		return 0, err
+	}
+	if len(plans) == 0 {
+		return 0, errors.New("该日期没有排产计划")
+	}
+	created := 0
+	for _, plan := range plans {
+		bom, err := s.ListBom(ctx, plan.OrderNo)
+		if err != nil {
+			continue
+		}
+		for _, item := range bom {
+			reqQty := float64(plan.PlanQty) * item.UnitQty * (1 + item.LossRate/100)
+			if reqQty <= 0 {
+				continue
+			}
+			reqNo := fmt.Sprintf("MR%s%03d", time.Now().Format("20060102150405"), len(bom))
+			_ = s.ctx.EntClient.MaterialRequest.Create().
+				SetRequestNo(reqNo).SetOrderNo(plan.OrderNo).SetPlanDate(plan.PlanDate).
+				SetMaterialCode(item.MaterialCode).SetMaterialName(item.MaterialName).
+				SetUnit(item.Unit).SetManageMode(item.ManageMode).SetReqQty(reqQty).
+				SetStatus("PENDING").SetOperator(operator).
+				SetTargetDock("").Exec(ctx)
+			created++
+		}
+	}
+	s.ctx.EventLog.Write(ctx, "material.request.generate", "", operator, "material_request", planDate, "按日排产自动生成备料单", map[string]any{"planDate": planDate, "count": created})
+	return created, nil
+}
+
+// ListMaterialRequests 查询备料单
+func (s *Service) ListMaterialRequests(ctx context.Context, orderNo, planDate, status string) ([]*ent.MaterialRequest, error) {
+	q := s.ctx.EntClient.MaterialRequest.Query()
+	if orderNo != "" {
+		q = q.Where(materialrequest.OrderNo(orderNo))
+	}
+	if planDate != "" {
+		q = q.Where(materialrequest.PlanDate(planDate))
+	}
+	if status != "" {
+		q = q.Where(materialrequest.Status(status))
+	}
+	return q.Order(materialrequest.ByID()).All(ctx)
+}