Dockerfile

dao/asset/asset_sku_dao.go

@@ -0,0 +1,130 @@
+package dao
+
+import (
+	"assets/consts/public"
+	"assets/model/dto/asset"
+	"assets/model/entity/asset"
+	"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/frame/g"
+	"go.mongodb.org/mongo-driver/v2/bson"
+)
+
+var AssetSku = new(assetSku)
+
+type assetSku struct {
+}
+
+// Insert 插入SKU
+func (d *assetSku) Insert(ctx context.Context, req *dto.CreateAssetSkuReq) (ids []any, err error) {
+	var result *entity.AssetSku
+	if err = utils.Struct(req, &result); err != nil {
+		return
+	}
+	ids, err = mongo.DB().Insert(ctx, []interface{}{&result}, public.AssetSkuCollection)
+	return
+}
+
+// Update 更新SKU
+func (d *assetSku) Update(ctx context.Context, req *dto.UpdateAssetSkuReq) (err error) {
+	buildUpdateData, err := mongo.BuildUpdateData(ctx, req)
+	if err != nil {
+		return
+	}
+	filter := bson.M{"_id": req.Id}
+	update := bson.M{"$set": buildUpdateData}
+	if !g.IsEmpty(req.Stock) {
+		// 从$set中移除stock字段避免$set和$inc冲突
+		delete(buildUpdateData, "stock")
+		update = bson.M{
+			"$inc": bson.M{
+				"stock": req.Stock,
+			},
+		}
+		if len(buildUpdateData) > 0 {
+			update["$set"] = buildUpdateData
+		}
+	}
+	_, err = mongo.DB().Update(ctx, filter, update, public.AssetSkuCollection)
+	return
+}
+
+// DeleteFake 删除SKU-根据id进行假删
+func (d *assetSku) DeleteFake(ctx context.Context, req *dto.DeleteAssetSkuReq) (err error) {
+	filter := bson.M{"_id": req.Id}
+	_, err = mongo.DB().DeleteSoft(ctx, filter, public.AssetSkuCollection)
+	return
+}
+
+// GetOne 获取单个SKU
+func (d *assetSku) GetOne(ctx context.Context, req *dto.GetAssetSkuReq, noTenantId bool) (res *entity.AssetSku, err error) {
+	filter := bson.M{"_id": req.Id}
+	if noTenantId {
+		err = mongo.DB().NoTenantId().FindOne(ctx, filter, &res, public.AssetSkuCollection)
+	} else {
+		err = mongo.DB().FindOne(ctx, filter, &res, public.AssetSkuCollection)
+	}
+	return
+}
+
+// GetListByAssetIdExcludeCurrentSku 根据资产ID获取SKU列表并且排除当前SKU
+func (d *assetSku) GetListByAssetIdExcludeCurrentSku(ctx context.Context, assetId *bson.ObjectID, req *dto.ListAssetSkuReq) (res []entity.AssetSku, total int64, err error) {
+	filter := bson.M{"assetId": assetId, "_id": bson.M{"$ne": req.Id}}
+	total, err = mongo.DB().Find(ctx, filter, &res, public.AssetSkuCollection, req.Page, req.OrderBy)
+	return
+}
+
+// List 获取SKU列表
+func (d *assetSku) List(ctx context.Context, req *dto.ListAssetSkuReq, noTenantId bool) (res []entity.AssetSku, total int64, err error) {
+	// 构建查询过滤条件
+	filter, err := d.buildListFilter(ctx, req)
+	if err != nil {
+		return
+	}
+	// 排序处理
+	req.OrderBy = []beans.OrderBy{
+		{Field: "sort", Order: beans.Asc},
+		{Field: "createdAt", Order: beans.Desc},
+	}
+	if noTenantId {
+		total, err = mongo.DB().NoTenantId().Find(ctx, filter, &res, public.AssetSkuCollection, req.Page, req.OrderBy)
+	} else {
+		total, err = mongo.DB().Find(ctx, filter, &res, public.AssetSkuCollection, req.Page, req.OrderBy)
+	}
+	return
+}
+
+// buildListFilter 构建列表查询的过滤条件
+func (d *assetSku) buildListFilter(ctx context.Context, req *dto.ListAssetSkuReq) (filter bson.M, err error) {
+	_ = ctx
+	filter = bson.M{}
+	if !g.IsEmpty(req.AssetId) {
+		filter["assetId"] = req.AssetId
+	}
+	if !g.IsEmpty(req.Status) {
+		filter["status"] = req.Status
+	}
+	if !g.IsEmpty(req.CategoryPath) {
+		filter["categoryPath"] = bson.M{"$regex": "^" + req.CategoryPath, "$options": "i"}
+	}
+	if !g.IsEmpty(req.Keyword) {
+		orConditions := bson.A{
+			bson.M{"skuName": bson.M{"$regex": req.Keyword, "$options": "i"}},
+			bson.M{"assetName": bson.M{"$regex": req.Keyword, "$options": "i"}},
+		}
+		filter["$or"] = orConditions
+	}
+	if req.MinPrice > 0 {
+		filter["price"] = bson.M{"$gte": req.MinPrice}
+	}
+	if req.MaxPrice > 0 {
+		if filter["price"] == nil {
+			filter["price"] = bson.M{}
+		}
+		filter["price"].(bson.M)["$lte"] = req.MaxPrice
+	}
+	return
+}