87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"gotest/internal/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// ShopRepository 电商数据访问层
|
|
type ShopRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewShopRepository 创建电商仓库实例
|
|
func NewShopRepository(db *gorm.DB) *ShopRepository {
|
|
return &ShopRepository{db: db}
|
|
}
|
|
|
|
// ListCategories 获取所有分类
|
|
func (r *ShopRepository) ListCategories() ([]model.ProductCategory, error) {
|
|
var categories []model.ProductCategory
|
|
if err := r.db.Order("sort ASC, id ASC").Find(&categories).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return categories, nil
|
|
}
|
|
|
|
// ListProducts 获取商品列表，支持分类过滤和关键词搜索
|
|
func (r *ShopRepository) ListProducts(categoryID uint, keyword string, sort string, page, pageSize int) ([]model.Product, int64, error) {
|
|
var products []model.Product
|
|
var total int64
|
|
|
|
query := r.db.Where("status = 1")
|
|
if categoryID > 0 {
|
|
query = query.Where("category_id = ?", categoryID)
|
|
}
|
|
if keyword != "" {
|
|
query = query.Where("name LIKE ?", "%"+keyword+"%")
|
|
}
|
|
|
|
if err := query.Model(&model.Product{}).Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
// 排序
|
|
switch sort {
|
|
case "price_asc":
|
|
query = query.Order("price ASC")
|
|
case "price_desc":
|
|
query = query.Order("price DESC")
|
|
case "sales":
|
|
query = query.Order("sales DESC")
|
|
case "rating":
|
|
query = query.Order("rating DESC")
|
|
default:
|
|
query = query.Order("sort ASC, id DESC")
|
|
}
|
|
|
|
if page > 0 && pageSize > 0 {
|
|
offset := (page - 1) * pageSize
|
|
query = query.Offset(offset).Limit(pageSize)
|
|
}
|
|
|
|
if err := query.Find(&products).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return products, total, nil
|
|
}
|
|
|
|
// GetProduct 根据 ID 获取商品详情
|
|
func (r *ShopRepository) GetProduct(id uint) (*model.Product, error) {
|
|
var product model.Product
|
|
if err := r.db.First(&product, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &product, nil
|
|
}
|
|
|
|
// ListFeaturedProducts 获取推荐商品（按销量）
|
|
func (r *ShopRepository) ListFeaturedProducts(limit int) ([]model.Product, error) {
|
|
var products []model.Product
|
|
if err := r.db.Where("status = 1").Order("sales DESC").Limit(limit).Find(&products).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return products, nil
|
|
}
|