From 6d9989d34812e456d8e0956d48c2f524d1333184 Mon Sep 17 00:00:00 2001
From: ata <17314593+asdxzaaaaa@user.noreply.gitee.com>
Date: Fri, 10 Jul 2026 10:06:12 +0000
Subject: [PATCH] Add internal/repository/shop.go

---
 internal/repository/shop.go | 86 +++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 internal/repository/shop.go

diff --git a/internal/repository/shop.go b/internal/repository/shop.go
new file mode 100644
index 0000000..052c56a
--- /dev/null
+++ b/internal/repository/shop.go
@@ -0,0 +1,86 @@
+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
+}