assets/dao/stock/stock_batch_dao.go

// 批次库存DAO层(逻辑库存)
// 职责：批次CRUD、使用$inc原子操作更新数量
// 紧密耦合：service.StockBatch、service.StockManage(入库出库)
// 注意：Update使用$inc原子操作，GetOne使用NoCache()跳过缓存
package dao

import (
	"assets/consts/public"
	dto "assets/model/dto/stock"
	entity "assets/model/entity/stock"
	"context"

	"gitea.redpowerfuture.com/red-future/common/db/gfdb"
	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
)

var StockBatch = new(stockBatch)

type stockBatch struct {
}

// Insert 插入
func (d *stockBatch) Insert(ctx context.Context, req *dto.CreateSockBatchReq) (id int64, err error) {
	var res *entity.StockBatch
	if err = gconv.Struct(req, &res); err != nil {
		return
	}
	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameStockBatch).Data(&res).Insert()
	if err != nil {
		return
	}
	return r.LastInsertId()
}

func (d *stockBatch) Update(ctx context.Context, req *dto.UpdateSockBatchReq) (rows int64, err error) {
	model := gfdb.DB(ctx).Model(ctx, public.TableNameStockBatch).Where(entity.StockBatchCol.Id, req.Id)
	model.Data(entity.StockBatchCol.BatchQty, &gdb.Counter{
		Field: entity.StockBatchCol.BatchQty,
		Value: gconv.Float64(req.BatchQty),
	})
	model.Data(entity.StockBatchCol.AvailableQty, &gdb.Counter{
		Field: entity.StockBatchCol.AvailableQty,
		Value: gconv.Float64(req.AvailableQty),
	})
	r, err := model.Update()
	if err != nil {
		return
	}
	return r.RowsAffected()
}

// One 根据批次号查询（使用NoCache跳过缓存，确保获取最新数据）
func (d *stockBatch) One(ctx context.Context, req *dto.GetSockBatchReq, fields ...string) (res *entity.StockBatch, err error) {
	model := gfdb.DB(ctx).Model(ctx, public.TableNameStockBatch)
	if !g.IsEmpty(req.Id) {
		model.Where(entity.StockBatchCol.Id, req.Id)
	}
	if !g.IsEmpty(req.BatchNo) {
		model.Where(entity.StockBatchCol.BatchNo, req.BatchNo)
	}
	r, err := model.Fields(fields).One()
	if err != nil {
		return
	}
	err = r.Struct(&res)
	return
}