bj_power/bj_power_wms/internal/handler/internalapi.go

package handler

import (
	"net/http"
	"time"

	"bj_power_wms/ent/inboundorder"
	"bj_power_wms/ent/inventory"
	"bj_power_wms/ent/ordermaterialledger"
	"bj_power_wms/ent/outboundorder"
	"bj_power_wms/ent/semifinished"
	"bj_power_wms/internal/svc"
)

// ledgerSyncRequest MES 下发工单 BOM 后同步台账
type ledgerSyncRequest struct {
	OrderNo     string `json:"orderNo"`
	ProductCode string `json:"productCode"`
	Operator    string `json:"operator"`
	Items       []struct {
		MaterialCode string `json:"materialCode"`
		MaterialName string `json:"materialName"`
		TotalQty     int    `json:"totalQty"`
	} `json:"items"`
}

// ledgerSyncHandler 同步/更新工单物料台账（幂等：按 orderNo+materialCode upsert）
func ledgerSyncHandler(ctx *svc.ServiceContext) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		var req ledgerSyncRequest
		if err := parseJSON(r, &req); err != nil {
			fail(w, http.StatusBadRequest, "参数错误: "+err.Error())
			return
		}
		if req.OrderNo == "" || len(req.Items) == 0 {
			fail(w, http.StatusBadRequest, "orderNo 和 items 必填")
			return
		}
		saved := 0
		for _, it := range req.Items {
			existing, err := ctx.EntClient.OrderMaterialLedger.Query().
				Where(
					ordermaterialledger.OrderNoEQ(req.OrderNo),
					ordermaterialledger.MaterialCodeEQ(it.MaterialCode),
				).Only(ctx0())
			if err == nil {
				ctx.EntClient.OrderMaterialLedger.UpdateOneID(existing.ID).
					SetTotalQty(it.TotalQty).
					SetStatus("进行中").
					SetUpdatedAt(time.Now().Unix()).
					ExecX(ctx0())
			} else {
				ctx.EntClient.OrderMaterialLedger.Create().
					SetOrderNo(req.OrderNo).
					SetMaterialCode(it.MaterialCode).
					SetNillableMaterialName(strPtr(it.MaterialName)).
					SetTotalQty(it.TotalQty).
					SetOutQty(0).
					SetStatus("进行中").
					SaveX(ctx0())
			}
			saved++
		}
		ok(w, map[string]any{"orderNo": req.OrderNo, "synced": saved})
	}
}

// internalDisplayHandler 库存总览（55寸大屏免登录用）
func internalDisplayHandler(ctx *svc.ServiceContext) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// 批次库存（结构件）总量（按质量状态）
		batches, _ := ctx.EntClient.Inventory.Query().Where(inventory.ManageModeEQ(1)).All(ctx0())
		totalQty := 0
		lockedQty := 0
		type zoneRow struct {
			Zone, Quality string
			Qty           int
		}
		zones := map[string]int{}
		batchCount := len(batches)
		for _, b := range batches {
			totalQty += b.Quantity
			lockedQty += b.LockedQty
			zc := b.ZoneCode
			if zc == "" {
				zc = "未分区"
			}
			zones[zc+"/"+b.QualityStatus] += b.Quantity
		}
		snInStock, _ := ctx.EntClient.Inventory.Query().Where(inventory.ManageModeEQ(2), inventory.StatusIn("在库", "锁定")).Count(ctx0())
		snOut, _ := ctx.EntClient.Inventory.Query().Where(inventory.ManageModeEQ(2), inventory.StatusEQ("出库")).Count(ctx0())
		semiInStock, _ := ctx.EntClient.SemiFinished.Query().Where(semifinished.StatusEQ("在库")).Count(ctx0())
		pkgCount, _ := ctx.EntClient.OutboundOrder.Query().Where(outboundorder.BoxNoNEQ("")).Count(ctx0())

		dayStart := time.Now().Truncate(24 * time.Hour).Unix()
		inToday, _ := ctx.EntClient.InboundOrder.Query().
			Where(inboundorder.CreatedAtGTE(dayStart)).Count(ctx0())
		outToday, _ := ctx.EntClient.OutboundOrder.Query().
			Where(outboundorder.CreatedAtGTE(dayStart)).Count(ctx0())

		ok(w, map[string]any{
			"batchCount":   batchCount,
			"totalQty":     totalQty,
			"lockedQty":    lockedQty,
			"snInStock":    snInStock,
			"snOut":        snOut,
			"semiInStock":  semiInStock,
			"packageCount": pkgCount,
			"inToday":      inToday,
			"outToday":     outToday,
			"zoneDetail":   zones,
			"time":         time.Now().Format("2006-01-02 15:04:05"),
		})
	}
}