269 lines
8.2 KiB
Go
269 lines
8.2 KiB
Go
// 实物库存批次DAO层
|
||
// 职责：CRUD、IncrementAvailableQty原子操作(防并发超卖)、SumAvailableQtyByLocation聚合汇总
|
||
// 紧密耦合：service.PrivateStock、service.Capacity(容量计算入口)
|
||
// 注意：IncrementAvailableQty使用$inc+$gte条件，防止并发导致库存变负
|
||
package dao
|
||
|
||
import (
|
||
"assets/consts/public"
|
||
"assets/consts/stock"
|
||
dto "assets/model/dto/stock"
|
||
entity "assets/model/entity/stock"
|
||
"context"
|
||
|
||
"gitea.com/red-future/common/beans"
|
||
"gitea.com/red-future/common/db/mongo"
|
||
"gitea.com/red-future/common/utils"
|
||
"github.com/gogf/gf/v2/errors/gerror"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"go.mongodb.org/mongo-driver/v2/bson"
|
||
)
|
||
|
||
var PrivateStock = new(privateStock)
|
||
|
||
type privateStock struct {
|
||
}
|
||
|
||
// Insert 插入私域库存
|
||
func (d *privateStock) Insert(ctx context.Context, req *dto.CreatePrivateStockReq) (ids []interface{}, err error) {
|
||
var result *entity.PrivateStock
|
||
if err = utils.Struct(req, &result); err != nil {
|
||
return
|
||
}
|
||
|
||
// 设置StockType（如果未指定，默认为PrivateStock类型）
|
||
if !g.IsEmpty(req.StockType) {
|
||
result.StockType = req.StockType
|
||
} else {
|
||
result.StockType = stock.StockLocationTypePrivateStock
|
||
}
|
||
|
||
// 获取仓库信息（非必填）
|
||
if req.WarehouseId != nil && !req.WarehouseId.IsZero() {
|
||
warehouse, err := Warehouse.GetOne(ctx, &dto.GetWarehouseReq{Id: req.WarehouseId})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
result.WarehouseID = req.WarehouseId
|
||
result.WarehouseCode = warehouse.WarehouseCode
|
||
result.WarehouseName = warehouse.WarehouseName
|
||
}
|
||
|
||
// 如果有库区信息
|
||
if req.ZoneId != nil && !req.ZoneId.IsZero() {
|
||
zone, err := Zone.GetOne(ctx, &dto.GetZoneReq{Id: req.ZoneId})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
result.ZoneID = req.ZoneId
|
||
result.ZoneCode = zone.ZoneCode
|
||
result.ZoneName = zone.ZoneName
|
||
result.ZoneType = zone.ZoneType
|
||
}
|
||
|
||
// 如果有库位信息
|
||
if req.LocationId != nil && !req.LocationId.IsZero() {
|
||
location, err := Location.GetOne(ctx, &dto.GetLocationReq{Id: req.LocationId})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
result.LocationID = req.LocationId
|
||
result.LocationCode = location.LocationCode
|
||
result.LocationName = location.LocationName
|
||
result.LocationType = location.LocationType
|
||
}
|
||
|
||
ids, err = mongo.DB().Insert(ctx, []interface{}{&result}, public.PrivateStockCollection)
|
||
return
|
||
}
|
||
|
||
// Update 更新私域库存
|
||
func (d *privateStock) Update(ctx context.Context, req *dto.UpdatePrivateStockReq) (err error) {
|
||
filter := bson.M{"_id": req.Id}
|
||
update := bson.M{"$set": bson.M{}}
|
||
|
||
if req.WarehouseId != nil && !req.WarehouseId.IsZero() {
|
||
warehouse, err := Warehouse.GetOne(ctx, &dto.GetWarehouseReq{Id: req.WarehouseId})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
update["$set"].(bson.M)["warehouseId"] = req.WarehouseId
|
||
update["$set"].(bson.M)["warehouseCode"] = warehouse.WarehouseCode
|
||
update["$set"].(bson.M)["warehouseName"] = warehouse.WarehouseName
|
||
}
|
||
|
||
if req.ZoneId != nil && !req.ZoneId.IsZero() {
|
||
zone, err := Zone.GetOne(ctx, &dto.GetZoneReq{Id: req.ZoneId})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
update["$set"].(bson.M)["zoneId"] = req.ZoneId
|
||
update["$set"].(bson.M)["zoneCode"] = zone.ZoneCode
|
||
update["$set"].(bson.M)["zoneName"] = zone.ZoneName
|
||
update["$set"].(bson.M)["zoneType"] = zone.ZoneType
|
||
}
|
||
|
||
if req.LocationId != nil && !req.LocationId.IsZero() {
|
||
location, err := Location.GetOne(ctx, &dto.GetLocationReq{Id: req.LocationId})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
update["$set"].(bson.M)["locationId"] = req.LocationId
|
||
update["$set"].(bson.M)["locationCode"] = location.LocationCode
|
||
update["$set"].(bson.M)["locationName"] = location.LocationName
|
||
update["$set"].(bson.M)["locationType"] = location.LocationType
|
||
}
|
||
|
||
if req.PrivateSkuID != nil {
|
||
update["$set"].(bson.M)["privateSkuId"] = req.PrivateSkuID
|
||
}
|
||
if !g.IsEmpty(req.BatchNo) {
|
||
update["$set"].(bson.M)["batchNo"] = req.BatchNo
|
||
}
|
||
if req.BatchQty > 0 {
|
||
update["$set"].(bson.M)["batchQty"] = req.BatchQty
|
||
}
|
||
// AvailableQty已移除全量覆盖，请使用IncrementAvailableQty方法进行增量更新
|
||
if req.BatchStatus != nil {
|
||
update["$set"].(bson.M)["batchStatus"] = req.BatchStatus
|
||
}
|
||
if req.StockStatus != nil {
|
||
update["$set"].(bson.M)["stockStatus"] = req.StockStatus
|
||
}
|
||
if req.SupplierID != nil {
|
||
update["$set"].(bson.M)["supplierId"] = req.SupplierID
|
||
}
|
||
if req.SupportsRecycle != nil {
|
||
update["$set"].(bson.M)["supportsRecycle"] = *req.SupportsRecycle
|
||
}
|
||
if req.ProductionDate != nil {
|
||
update["$set"].(bson.M)["productionDate"] = req.ProductionDate
|
||
}
|
||
if req.ExpiryDate != nil {
|
||
update["$set"].(bson.M)["expiryDate"] = req.ExpiryDate
|
||
}
|
||
if req.ExpiryWarningDate != nil {
|
||
update["$set"].(bson.M)["expiryWarningDate"] = req.ExpiryWarningDate
|
||
}
|
||
if !g.IsEmpty(req.PrivateCategoryPath) {
|
||
update["$set"].(bson.M)["privateCategoryPath"] = req.PrivateCategoryPath
|
||
}
|
||
if !g.IsEmpty(req.StockType) {
|
||
update["$set"].(bson.M)["stockType"] = req.StockType
|
||
}
|
||
|
||
_, err = mongo.DB().Update(ctx, filter, update, public.PrivateStockCollection)
|
||
return
|
||
}
|
||
|
||
// DeleteFake 软删除私域库存
|
||
func (d *privateStock) DeleteFake(ctx context.Context, req *dto.DeletePrivateStockReq) error {
|
||
filter := bson.M{"_id": req.Id}
|
||
_, err := mongo.DB().DeleteSoft(ctx, filter, public.PrivateStockCollection)
|
||
return err
|
||
}
|
||
|
||
// GetOne 根据ID查询私域库存
|
||
func (d *privateStock) GetOne(ctx context.Context, req *dto.GetPrivateStockReq) (res *entity.PrivateStock, err error) {
|
||
filter := bson.M{"_id": req.Id}
|
||
err = mongo.DB().FindOne(ctx, filter, &res, public.PrivateStockCollection)
|
||
return
|
||
}
|
||
|
||
// IncrementAvailableQty 增量更新可用数量（并发安全）
|
||
// deltaQty：正数表示增加（入库），负数表示减少（出库）
|
||
// 减少时自动添加 availableQty >= |deltaQty| 条件，防止并发导致库存变负
|
||
func (d *privateStock) IncrementAvailableQty(ctx context.Context, id *bson.ObjectID, deltaQty int) (err error) {
|
||
filter := bson.M{"_id": id}
|
||
if deltaQty < 0 {
|
||
filter["availableQty"] = bson.M{"$gte": -deltaQty}
|
||
}
|
||
update := bson.M{
|
||
"$inc": bson.M{
|
||
"availableQty": deltaQty,
|
||
},
|
||
}
|
||
modifiedCount, err := mongo.DB().Update(ctx, filter, update, public.PrivateStockCollection)
|
||
if err != nil {
|
||
return
|
||
}
|
||
if deltaQty < 0 && modifiedCount == 0 {
|
||
err = gerror.Newf("库存不足，无法减少%d", -deltaQty)
|
||
}
|
||
return
|
||
}
|
||
|
||
// List 查询私域库存列表
|
||
func (d *privateStock) List(ctx context.Context, req *dto.ListPrivateStockReq) (res []entity.PrivateStock, total int64, err error) {
|
||
filter := bson.M{}
|
||
if req.WarehouseId != nil && !req.WarehouseId.IsZero() {
|
||
filter["warehouseId"] = req.WarehouseId
|
||
}
|
||
if req.ZoneId != nil && !req.ZoneId.IsZero() {
|
||
filter["zoneId"] = req.ZoneId
|
||
}
|
||
if req.LocationId != nil && !req.LocationId.IsZero() {
|
||
filter["locationId"] = req.LocationId
|
||
}
|
||
if req.PrivateSkuID != nil && !req.PrivateSkuID.IsZero() {
|
||
filter["privateSkuId"] = req.PrivateSkuID
|
||
}
|
||
if req.BatchStatus != nil {
|
||
filter["batchStatus"] = req.BatchStatus
|
||
}
|
||
if req.StockStatus != nil {
|
||
filter["stockStatus"] = req.StockStatus
|
||
}
|
||
if req.SupplierID != nil && !req.SupplierID.IsZero() {
|
||
filter["supplierId"] = req.SupplierID
|
||
}
|
||
if !g.IsEmpty(req.PrivateCategoryPath) {
|
||
filter["privateCategoryPath"] = req.PrivateCategoryPath
|
||
}
|
||
if !g.IsEmpty(req.StockType) {
|
||
filter["stockType"] = req.StockType
|
||
}
|
||
|
||
// 默认排序
|
||
if len(req.OrderBy) == 0 {
|
||
req.OrderBy = []beans.OrderBy{
|
||
{Field: "createdAt", Order: beans.Desc},
|
||
}
|
||
}
|
||
|
||
total, err = mongo.DB().Find(ctx, filter, &res, public.PrivateStockCollection, req.Page, req.OrderBy)
|
||
return
|
||
}
|
||
|
||
// SumAvailableQtyByLocation 按库位ID汇总所有库存的可用数量（使用MongoDB聚合管道）
|
||
func (d *privateStock) SumAvailableQtyByLocation(ctx context.Context, locationId *bson.ObjectID) (totalQty int, err error) {
|
||
pipeline := bson.A{
|
||
bson.M{"$match": bson.M{
|
||
"locationId": locationId,
|
||
"isDeleted": false,
|
||
}},
|
||
bson.M{"$group": bson.M{
|
||
"_id": nil,
|
||
"totalQty": bson.M{"$sum": "$availableQty"},
|
||
}},
|
||
}
|
||
|
||
coll := mongo.GetDB().Collection(public.PrivateStockCollection)
|
||
cursor, err := coll.Aggregate(ctx, pipeline)
|
||
if err != nil {
|
||
return
|
||
}
|
||
defer cursor.Close(ctx)
|
||
|
||
var results []struct {
|
||
TotalQty int `bson:"totalQty"`
|
||
}
|
||
if err = cursor.All(ctx, &results); err != nil {
|
||
return
|
||
}
|
||
if len(results) > 0 {
|
||
totalQty = results[0].TotalQty
|
||
}
|
||
return
|
||
}
|