3dshow-customer/3dshow/api/shopCart.go

package api

import (
	"3dshow/db/model"
	"3dshow/db/repo"
	"errors"
	"fmt"
	"time"

	"github.com/gin-gonic/gin"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

func ShopCart(r *GinRouter) {
	r.POSTJWT("/shopCart/deleteBatch", ShopCartDeleteBatch)
	r.GETJWT("/shopCart/groupList", shopCartList)

	CreateCRUD(r, "/shopCart", &CRUDOption{
		Collection: repo.CollectionShopCart,
		NewModel: func(c *gin.Context, apictx *ApiSession) (interface{}, error) {
			entity := &model.ShopCart{}
			err := c.ShouldBindJSON(entity)
			if err != nil {
				return nil, errors.New("参数错误")
			}
			if len(entity.SupplyId) < 12 {
				return nil, errors.New("供应链id错误")
			}
			if len(entity.ProductId) < 12 {
				return nil, errors.New("商品id错误")
			}
			entity.CreateTime = time.Now()
			_userId := apictx.User.ID
			userId, _ := primitive.ObjectIDFromHex(_userId)
			entity.UserId = userId

			return entity, nil
		},
		EmtyModel: func(c *gin.Context, apictx *ApiSession) interface{} {
			return &model.ShopCart{}
		},
		JWT: true,
		SearchFilter: func(_ *gin.Context, apictx *ApiSession, query map[string]interface{}) map[string]interface{} {
			_userId := apictx.User.ID
			userId, _ := primitive.ObjectIDFromHex(_userId)
			query["userId"] = userId
			return query
		},
		OnUpdate: func(_ *gin.Context, _ *ApiSession, entity interface{}) {
			ety := entity.(*model.ShopCart)
			ety.UpdateTime = time.Now()
		},
	})
}

type ShopCartDeleteBatchReq struct {
	Ids []string `json:"ids"`
}

func ShopCartDeleteBatch(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	var form ShopCartDeleteBatchReq
	err := c.ShouldBindJSON(&form)
	if err != nil {
		fmt.Println(form)
		return nil, errors.New("参数错误")
	}
	if len(form.Ids) > 0 {
		// 批量删除
		for _, v := range form.Ids {
			// id 错误
			_, err := primitive.ObjectIDFromHex(v)
			if err != nil {
				continue
			}
			// 删除
			repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionShopCart, v)

		}
	}

	return true, nil
}

func shopCartList(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	var shopCarts []*model.ShopCart
	userId, _ := primitive.ObjectIDFromHex(apictx.User.ID)
	option := &repo.PageSearchOptions{
		CollectName: repo.CollectionShopCart,
		Query:       repo.Map{"userId": userId},
	}

	repo.RepoDocsSearch(apictx.CreateRepoCtx(), option, &shopCarts)

	supplyArr := map[primitive.ObjectID]struct{}{}
	supplyList := make([]map[string]interface{}, 0)

	if len(shopCarts) > 0 {
		for _, v := range shopCarts {
			supplyArr[v.SupplyId] = struct{}{}
		}
		// supplyId分组列表
		for k, _ := range supplyArr {
			// 查询产品信息
			supply := model.Supply{}
			// 供应商信息
			repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
				CollectName: repo.CollectionSupply,
				Query:       repo.Map{"_id": k},
				Project:     []string{"name", "avatar", "contact"},
			}, &supply)
			products := make([]*model.ShopCart, 0)
			for _, v := range shopCarts {
				if k == v.SupplyId {
					products = append(products, v)

				}
			}
			supplyList = append(supplyList, map[string]interface{}{
				"supply":   supply,
				"products": products,
			})
		}
	}

	return supplyList, nil
}