platform/web/services/product.go

package services

import (
	"platform/web/core"
	m "platform/web/models"
	q "platform/web/queries"
	"time"

	"github.com/shopspring/decimal"
	"gorm.io/gen/field"
)

var Product = &productService{}

type productService struct{}

// 获取所有产品
func (s *productService) AllProducts() ([]*m.Product, error) {
	return q.Product.
		Order(q.Product.Sort.Asc(), q.Product.CreatedAt.Desc()).
		Find()
}

func (s *productService) AllProductSaleInfos() ([]any, error) {
	products, err := q.Product.
		Select(
			q.Product.ID,
			q.Product.Code,
			q.Product.Name,
			q.Product.Description,
			q.Product.Sort,
		).
		Where(
			q.Product.Status.Eq(int(m.ProductStatusEnabled)),
		).
		Order(q.Product.Sort).
		Find()
	if err != nil {
		return nil, err
	}

	pids := make([]int32, len(products))
	for i, p := range products {
		pids[i] = p.ID
	}

	type SkuInfo struct {
		ID        int32           `json:"id"`
		ProductID int32           `json:"product_id"`
		Name      string          `json:"name"`
		Code      string          `json:"code"`
		Price     decimal.Decimal `json:"price"`
		CountMin  decimal.Decimal `json:"count_min"`
		Discount  float64         `json:"discount"`
	}
	var skus []*SkuInfo
	err = q.ProductSku.
		Select(
			q.ProductSku.ID,
			q.ProductSku.ProductID,
			q.ProductSku.Name,
			q.ProductSku.Code,
			q.ProductSku.Price,
			q.ProductSku.CountMin,
			q.ProductDiscount.Discount,
		).
		Where(
			q.ProductSku.ProductID.In(pids...),
			q.ProductSku.Status.Eq(int32(m.SkuStatusEnabled)),
		).
		LeftJoin(q.ProductDiscount, q.ProductDiscount.ID.EqCol(q.ProductSku.DiscountId)).
		Order(q.ProductSku.Sort).
		Scan(&skus)
	if err != nil {
		return nil, err
	}

	type ProductInfo struct {
		m.Product
		Skus []*SkuInfo `json:"skus,omitempty"`
	}
	pmap := make(map[int32]*ProductInfo, len(products))
	for _, p := range products {
		pmap[p.ID] = &ProductInfo{Product: *p, Skus: make([]*SkuInfo, 0)}
	}
	for _, s := range skus {
		if p, ok := pmap[s.ProductID]; ok {
			p.Skus = append(p.Skus, &SkuInfo{
				ID:        s.ID,
				ProductID: s.ProductID,
				Name:      s.Name,
				Code:      s.Code,
				Price:     s.Price,
				CountMin:  s.CountMin,
				Discount:  s.Discount,
			})
		}
	}

	plist := make([]any, 0, len(pmap))
	for _, p := range pmap {
		plist = append(plist, p)
	}

	return plist, nil
}

// 新增产品
func (s *productService) CreateProduct(create *CreateProductData) error {
	return q.Product.Create(&m.Product{
		Code:        create.Code,
		Name:        create.Name,
		Description: create.Description,
		Sort:        create.Sort,
		Status:      m.ProductStatus(create.Status),
	})
}

type CreateProductData struct {
	Code        string  `json:"code"`
	Name        string  `json:"name"`
	Description *string `json:"description"`
	Sort        int32   `json:"sort"`
	Status      int     `json:"status"`
}

// 更新产品
func (s *productService) UpdateProduct(update *UpdateProductData) error {
	if update == nil {
		return core.NewBizErr("更新数据不存在")
	}

	do := make([]field.AssignExpr, 0, 5)
	if update.Code != nil {
		do = append(do, q.Product.Code.Value(*update.Code))
	}
	if update.Name != nil {
		do = append(do, q.Product.Name.Value(*update.Name))
	}
	if update.Description != nil {
		do = append(do, q.Product.Description.Value(*update.Description))
	}
	if update.Sort != nil {
		do = append(do, q.Product.Sort.Value(*update.Sort))
	}
	if update.Status != nil {
		do = append(do, q.Product.Status.Value(*update.Status))
	}
	r, err := q.Product.Where(q.Product.ID.Eq(update.Id)).UpdateSimple(do...)
	if err != nil {
		return err
	}
	if r.RowsAffected == 0 {
		return core.NewBizErr("产品状态已过期")
	}
	return nil
}

type UpdateProductData struct {
	Id          int32   `json:"id"`
	Code        *string `json:"code"`
	Name        *string `json:"name"`
	Description *string `json:"description"`
	Sort        *int32  `json:"sort"`
	Status      *int    `json:"status"`
}

// 删除产品
func (s *productService) DeleteProduct(id int32) error {
	r, err := q.Product.Where(q.Product.ID.Eq(id)).UpdateColumn(q.Product.DeletedAt, time.Now())
	if err != nil {
		return err
	}
	if r.RowsAffected == 0 {
		return core.NewBizErr("产品状态已过期")
	}
	return nil
}