106 lines
3.0 KiB
Go
106 lines
3.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"gotest/internal/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// FoodRepository 外卖数据访问层
|
|
type FoodRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewFoodRepository 创建外卖仓库实例
|
|
func NewFoodRepository(db *gorm.DB) *FoodRepository {
|
|
return &FoodRepository{db: db}
|
|
}
|
|
|
|
// ListCategories 获取餐厅分类列表
|
|
func (r *FoodRepository) ListCategories() ([]model.RestaurantCategory, error) {
|
|
var categories []model.RestaurantCategory
|
|
if err := r.db.Order("sort ASC, id ASC").Find(&categories).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return categories, nil
|
|
}
|
|
|
|
// ListRestaurants 获取餐厅列表，支持分类过滤和关键词搜索
|
|
func (r *FoodRepository) ListRestaurants(categoryID uint, keyword string, sort string, page, pageSize int) ([]model.Restaurant, int64, error) {
|
|
var restaurants []model.Restaurant
|
|
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 ? OR tags LIKE ?", "%"+keyword+"%", "%"+keyword+"%")
|
|
}
|
|
|
|
if err := query.Model(&model.Restaurant{}).Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
switch sort {
|
|
case "rating":
|
|
query = query.Order("rating DESC")
|
|
case "sales":
|
|
query = query.Order("sales DESC")
|
|
case "delivery_time":
|
|
query = query.Order("delivery_time ASC")
|
|
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(&restaurants).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return restaurants, total, nil
|
|
}
|
|
|
|
// GetRestaurant 根据 ID 获取餐厅详情
|
|
func (r *FoodRepository) GetRestaurant(id uint) (*model.Restaurant, error) {
|
|
var restaurant model.Restaurant
|
|
if err := r.db.First(&restaurant, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &restaurant, nil
|
|
}
|
|
|
|
// ListFeaturedRestaurants 获取推荐餐厅（按销量）
|
|
func (r *FoodRepository) ListFeaturedRestaurants(limit int) ([]model.Restaurant, error) {
|
|
var restaurants []model.Restaurant
|
|
if err := r.db.Where("status = 1").Order("sales DESC").Limit(limit).Find(&restaurants).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return restaurants, nil
|
|
}
|
|
|
|
// ListMenuCategories 获取餐厅的菜品分类
|
|
func (r *FoodRepository) ListMenuCategories(restaurantID uint) ([]model.MenuCategory, error) {
|
|
var categories []model.MenuCategory
|
|
if err := r.db.Where("restaurant_id = ?", restaurantID).Order("sort ASC, id ASC").Find(&categories).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return categories, nil
|
|
}
|
|
|
|
// ListMenuItems 获取餐厅的菜品列表，可按菜品分类过滤
|
|
func (r *FoodRepository) ListMenuItems(restaurantID uint, categoryID uint) ([]model.MenuItem, error) {
|
|
var items []model.MenuItem
|
|
query := r.db.Where("restaurant_id = ? AND status = 1", restaurantID)
|
|
if categoryID > 0 {
|
|
query = query.Where("category_id = ?", categoryID)
|
|
}
|
|
if err := query.Order("sort ASC, id ASC").Find(&items).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return items, nil
|
|
}
|