youngee_b_api/app/dao/product_dao.go

package dao

import (
	"errors"
	"github.com/sirupsen/logrus"
	"gorm.io/gorm"
	"youngee_b_api/app/entity"
)

type ProductDAO struct{}

func (d ProductDAO) GetProductsByEnterpriseID(enterpriseId string, page int, pageSize int) ([]entity.Product, int64, error) {
	var products []entity.Product
	var total int64
	Db.Model(&entity.Product{}).Where("enterprise_id = ?", enterpriseId).Count(&total)
	// 计算偏移量
	offset := (page - 1) * pageSize
	// 分页查询
	err := Db.Where("enterprise_id = ?", enterpriseId).Order("created_at desc").Offset(offset).Limit(pageSize).Find(&products).Error
	if err != nil {
		return nil, 0, err
	}
	return products, total, nil
}

func (d ProductDAO) GetProductsBySubAccountId(subAccountId int64, page int, pageSize int) ([]entity.Product, int64, error) {
	var products []entity.Product
	var total int64
	Db.Model(&entity.Product{}).Where("sub_account_id = ?", subAccountId).Count(&total)
	// 计算偏移量
	offset := (page - 1) * pageSize
	// 分页查询
	err := Db.Where("sub_account_id = ?", subAccountId).Order("created_at desc").Offset(offset).Limit(pageSize).Find(&products).Error
	if err != nil {
		return nil, 0, err
	}
	return products, total, nil
}

func (d ProductDAO) CreateProduct(product entity.Product) (int64, error) {
	err := Db.Create(&product).Error
	if err != nil {
		return 0, err
	}
	return product.ProductID, nil
}

func (d ProductDAO) GetProductByID(productId int64) (*entity.Product, error) {
	var product entity.Product
	err := Db.Where("product_id = ?", productId).First(&product).Error
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, nil
		} else {
			return nil, err
		}
	}
	return &product, nil
}

func (d ProductDAO) GetProductBySelectionId(selectionId string) (*entity.Product, error) {
	productId := 0
	err := Db.Model(entity.SelectionInfo{}).Select("product_id").Where("selection_id = ?", selectionId).Find(&productId).Error
	if err != nil {
		logrus.Errorf("[GetProductInfo] error query mysql, err:%+v", err)
		return nil, err
	}
	productInfo := entity.Product{}
	err = Db.Model(entity.Product{}).Where("product_id = ?", productId).Find(&productInfo).Error
	if err != nil {
		logrus.Errorf("[GetProductInfo] error query mysql, err:%+v", err)
		return nil, err
	}
	return &productInfo, nil
}

func (d ProductDAO) GetProductByProjectId(projectId string) (*entity.Product, error) {
	productId := 0
	err := Db.Model(entity.Project{}).Select("product_id").Where("project_id = ?", projectId).Find(&productId).Error
	if err != nil {
		logrus.Errorf("[GetProductInfo] error query mysql, err:%+v", err)
		return nil, err
	}
	productInfo := entity.Product{}
	err = Db.Model(entity.Product{}).Where("product_id = ?", productId).Find(&productInfo).Error
	if err != nil {
		logrus.Errorf("[GetProductInfo] error query mysql, err:%+v", err)
		return nil, err
	}
	return &productInfo, nil
}