Dockerfile

dao/procurement/purchase_order_dao.go

@@ -0,0 +1,113 @@
+package dao
+
+import (
+	"assets/consts/public"
+	dto "assets/model/dto/procurement"
+	entity "assets/model/entity/procurement"
+	"context"
+
+	"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 PurchaseOrder = new(purchaseOrder)
+
+type purchaseOrder struct{}
+
+// Insert 插入采购订单
+func (d *purchaseOrder) Insert(ctx context.Context, req *dto.CreatePurchaseOrderReq) (ids []any, err error) {
+	var result *entity.PurchaseOrder
+	if err = utils.Struct(req, &result); err != nil {
+		return
+	}
+	ids, err = mongo.DB().Insert(ctx, []interface{}{&result}, public.PurchaseOrderCollection)
+	return
+}
+
+// BatchInsert 批量插入采购订单
+func (d *purchaseOrder) BatchInsert(ctx context.Context, req *dto.BatchCreatePurchaseOrdersReq) (ids []any, err error) {
+	items := make([]*entity.PurchaseOrder, 0, len(req.Orders))
+	for _, item := range req.Orders {
+		var result *entity.PurchaseOrder
+		if err = utils.Struct(item, &result); err != nil {
+			return
+		}
+		items = append(items, result)
+	}
+	// 转换为 interface{} 切片
+	interfaces := make([]interface{}, len(items))
+	for i, item := range items {
+		interfaces[i] = item
+	}
+	ids, err = mongo.DB().Insert(ctx, interfaces, public.PurchaseOrderCollection)
+	return
+}
+
+// GetOne 获取单个采购订单
+func (d *purchaseOrder) GetOne(ctx context.Context, id *bson.ObjectID) (order *entity.PurchaseOrder, err error) {
+	filter := bson.M{"_id": id}
+	err = mongo.DB().FindOne(ctx, filter, &order, public.PurchaseOrderCollection)
+	return
+}
+
+// Update 更新采购订单
+func (d *purchaseOrder) Update(ctx context.Context, req *dto.UpdatePurchaseOrderReq) (err error) {
+	buildUpdateData, err := mongo.BuildUpdateData(ctx, req)
+	if err != nil {
+		return
+	}
+	filter := bson.M{"_id": req.ID}
+	update := bson.M{"$set": buildUpdateData}
+	_, err = mongo.DB().Update(ctx, filter, update, public.PurchaseOrderCollection)
+	return
+}
+
+// DeleteFake 删除采购订单-根据id进行假删
+func (d *purchaseOrder) DeleteFake(ctx context.Context, id *bson.ObjectID) (err error) {
+	filter := bson.M{"_id": id}
+	_, err = mongo.DB().DeleteSoft(ctx, filter, public.PurchaseOrderCollection)
+	return
+}
+
+// List 获取采购订单列表
+func (d *purchaseOrder) List(ctx context.Context, req *dto.ListPurchaseOrdersReq) (res []*entity.PurchaseOrder, total int64, err error) {
+	filter, err := d.buildListFilter(ctx, req)
+	if err != nil {
+		return
+	}
+
+	total, err = mongo.DB().Find(ctx, filter, &res, public.PurchaseOrderCollection, nil, nil)
+	return
+}
+
+// ListByBuyerId 根据采购方ID获取采购订单列表
+func (d *purchaseOrder) ListByBuyerId(ctx context.Context, buyerId *bson.ObjectID) (res []*entity.PurchaseOrder, err error) {
+	filter := bson.M{"buyerId": buyerId}
+	_, err = mongo.DB().Find(ctx, filter, &res, public.PurchaseOrderCollection, nil, nil)
+	return
+}
+
+// buildListFilter 构建列表查询的过滤条件
+func (d *purchaseOrder) buildListFilter(ctx context.Context, req *dto.ListPurchaseOrdersReq) (filter bson.M, err error) {
+	_ = ctx
+	filter = bson.M{}
+
+	if !g.IsEmpty(req.OrderNo) {
+		filter["orderNo"] = req.OrderNo
+	}
+	if !g.IsEmpty(req.Title) {
+		filter["title"] = bson.M{"$regex": req.Title, "$options": "i"}
+	}
+	if req.BuyerId != nil {
+		filter["buyerId"] = req.BuyerId
+	}
+	if req.OrderType != "" {
+		filter["orderType"] = req.OrderType
+	}
+	if req.Status != nil {
+		filter["status"] = *req.Status
+	}
+	return
+}