18278715334@163.com

2025-12-06 13:56:00 +08:00
parent 05b6623e75
commit 89367fb2ee
22 changed files with 2186 additions and 74 deletions
--- a/internal/infrastructure/database/repositories/article/gorm_announcement_repository.go
+++ b/internal/infrastructure/database/repositories/article/gorm_announcement_repository.go
@@ -0,0 +1,328 @@
+package repositories
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+	"tyapi-server/internal/domains/article/entities"
+	"tyapi-server/internal/domains/article/repositories"
+	repoQueries "tyapi-server/internal/domains/article/repositories/queries"
+	"tyapi-server/internal/shared/interfaces"
+
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+)
+
+// GormAnnouncementRepository GORM公告仓储实现
+type GormAnnouncementRepository struct {
+	db     *gorm.DB
+	logger *zap.Logger
+}
+
+// 编译时检查接口实现
+var _ repositories.AnnouncementRepository = (*GormAnnouncementRepository)(nil)
+
+// NewGormAnnouncementRepository 创建GORM公告仓储
+func NewGormAnnouncementRepository(db *gorm.DB, logger *zap.Logger) *GormAnnouncementRepository {
+	return &GormAnnouncementRepository{
+		db:     db,
+		logger: logger,
+	}
+}
+
+// Create 创建公告
+func (r *GormAnnouncementRepository) Create(ctx context.Context, entity entities.Announcement) (entities.Announcement, error) {
+	r.logger.Info("创建公告", zap.String("id", entity.ID), zap.String("title", entity.Title))
+
+	err := r.db.WithContext(ctx).Create(&entity).Error
+
+	if err != nil {
+		r.logger.Error("创建公告失败", zap.Error(err))
+		return entity, err
+	}
+
+	return entity, nil
+}
+
+// GetByID 根据ID获取公告
+func (r *GormAnnouncementRepository) GetByID(ctx context.Context, id string) (entities.Announcement, error) {
+	var entity entities.Announcement
+
+	err := r.db.WithContext(ctx).
+		Where("id = ?", id).
+		First(&entity).Error
+
+	if err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return entity, fmt.Errorf("公告不存在")
+		}
+		r.logger.Error("获取公告失败", zap.String("id", id), zap.Error(err))
+		return entity, err
+	}
+
+	return entity, nil
+}
+
+// Update 更新公告
+func (r *GormAnnouncementRepository) Update(ctx context.Context, entity entities.Announcement) error {
+	r.logger.Info("更新公告", zap.String("id", entity.ID))
+
+	err := r.db.WithContext(ctx).Save(&entity).Error
+	if err != nil {
+		r.logger.Error("更新公告失败", zap.String("id", entity.ID), zap.Error(err))
+		return err
+	}
+
+	return nil
+}
+
+// Delete 删除公告
+func (r *GormAnnouncementRepository) Delete(ctx context.Context, id string) error {
+	r.logger.Info("删除公告", zap.String("id", id))
+
+	err := r.db.WithContext(ctx).Delete(&entities.Announcement{}, "id = ?", id).Error
+	if err != nil {
+		r.logger.Error("删除公告失败", zap.String("id", id), zap.Error(err))
+		return err
+	}
+
+	return nil
+}
+
+// FindByStatus 根据状态查找公告
+func (r *GormAnnouncementRepository) FindByStatus(ctx context.Context, status entities.AnnouncementStatus) ([]*entities.Announcement, error) {
+	var announcements []entities.Announcement
+
+	err := r.db.WithContext(ctx).
+		Where("status = ?", status).
+		Order("created_at DESC").
+		Find(&announcements).Error
+
+	if err != nil {
+		r.logger.Error("根据状态查找公告失败", zap.String("status", string(status)), zap.Error(err))
+		return nil, err
+	}
+
+	// 转换为指针切片
+	result := make([]*entities.Announcement, len(announcements))
+	for i := range announcements {
+		result[i] = &announcements[i]
+	}
+
+	return result, nil
+}
+
+// FindScheduled 查找定时发布的公告
+func (r *GormAnnouncementRepository) FindScheduled(ctx context.Context) ([]*entities.Announcement, error) {
+	var announcements []entities.Announcement
+	now := time.Now()
+
+	err := r.db.WithContext(ctx).
+		Where("status = ? AND scheduled_at IS NOT NULL AND scheduled_at <= ?", entities.AnnouncementStatusDraft, now).
+		Order("scheduled_at ASC").
+		Find(&announcements).Error
+
+	if err != nil {
+		r.logger.Error("查找定时发布公告失败", zap.Error(err))
+		return nil, err
+	}
+
+	// 转换为指针切片
+	result := make([]*entities.Announcement, len(announcements))
+	for i := range announcements {
+		result[i] = &announcements[i]
+	}
+
+	return result, nil
+}
+
+// ListAnnouncements 获取公告列表
+func (r *GormAnnouncementRepository) ListAnnouncements(ctx context.Context, query *repoQueries.ListAnnouncementQuery) ([]*entities.Announcement, int64, error) {
+	var announcements []entities.Announcement
+	var total int64
+
+	dbQuery := r.db.WithContext(ctx).Model(&entities.Announcement{})
+
+	// 应用筛选条件
+	if query.Status != "" {
+		dbQuery = dbQuery.Where("status = ?", query.Status)
+	}
+
+	if query.Title != "" {
+		dbQuery = dbQuery.Where("title ILIKE ?", "%"+query.Title+"%")
+	}
+
+	// 获取总数
+	if err := dbQuery.Count(&total).Error; err != nil {
+		r.logger.Error("获取公告列表总数失败", zap.Error(err))
+		return nil, 0, err
+	}
+
+	// 应用排序
+	if query.OrderBy != "" {
+		orderDir := "DESC"
+		if query.OrderDir != "" {
+			orderDir = strings.ToUpper(query.OrderDir)
+		}
+		dbQuery = dbQuery.Order(fmt.Sprintf("%s %s", query.OrderBy, orderDir))
+	} else {
+		dbQuery = dbQuery.Order("created_at DESC")
+	}
+
+	// 应用分页
+	if query.Page > 0 && query.PageSize > 0 {
+		offset := (query.Page - 1) * query.PageSize
+		dbQuery = dbQuery.Offset(offset).Limit(query.PageSize)
+	}
+
+	// 获取数据
+	if err := dbQuery.Find(&announcements).Error; err != nil {
+		r.logger.Error("获取公告列表失败", zap.Error(err))
+		return nil, 0, err
+	}
+
+	// 转换为指针切片
+	result := make([]*entities.Announcement, len(announcements))
+	for i := range announcements {
+		result[i] = &announcements[i]
+	}
+
+	return result, total, nil
+}
+
+// CountByStatus 根据状态统计公告数量
+func (r *GormAnnouncementRepository) CountByStatus(ctx context.Context, status entities.AnnouncementStatus) (int64, error) {
+	var count int64
+
+	err := r.db.WithContext(ctx).Model(&entities.Announcement{}).
+		Where("status = ?", status).
+		Count(&count).Error
+
+	if err != nil {
+		r.logger.Error("统计公告数量失败", zap.String("status", string(status)), zap.Error(err))
+		return 0, err
+	}
+
+	return count, nil
+}
+
+// UpdateStatistics 更新统计信息
+// 注意：公告实体目前没有统计字段，此方法预留扩展
+func (r *GormAnnouncementRepository) UpdateStatistics(ctx context.Context, announcementID string) error {
+	r.logger.Info("更新公告统计信息", zap.String("announcement_id", announcementID))
+	// TODO: 如果将来需要统计字段（如阅读量等），可以在这里实现
+	return nil
+}
+
+// ================ 实现 BaseRepository 接口的其他方法 ================
+
+// Count 统计数量
+func (r *GormAnnouncementRepository) Count(ctx context.Context, options interfaces.CountOptions) (int64, error) {
+	dbQuery := r.db.WithContext(ctx).Model(&entities.Announcement{})
+
+	// 应用筛选条件
+	if options.Filters != nil {
+		for key, value := range options.Filters {
+			dbQuery = dbQuery.Where(key+" = ?", value)
+		}
+	}
+
+	if options.Search != "" {
+		search := "%" + options.Search + "%"
+		dbQuery = dbQuery.Where("title LIKE ? OR content LIKE ?", search, search)
+	}
+
+	var count int64
+	err := dbQuery.Count(&count).Error
+	return count, err
+}
+
+// Exists 检查是否存在
+func (r *GormAnnouncementRepository) Exists(ctx context.Context, id string) (bool, error) {
+	var count int64
+	err := r.db.WithContext(ctx).Model(&entities.Announcement{}).
+		Where("id = ?", id).
+		Count(&count).Error
+	return count > 0, err
+}
+
+// SoftDelete 软删除
+func (r *GormAnnouncementRepository) SoftDelete(ctx context.Context, id string) error {
+	return r.db.WithContext(ctx).Delete(&entities.Announcement{}, "id = ?", id).Error
+}
+
+// Restore 恢复软删除
+func (r *GormAnnouncementRepository) Restore(ctx context.Context, id string) error {
+	return r.db.WithContext(ctx).Unscoped().Model(&entities.Announcement{}).
+		Where("id = ?", id).
+		Update("deleted_at", nil).Error
+}
+
+// CreateBatch 批量创建
+func (r *GormAnnouncementRepository) CreateBatch(ctx context.Context, entities []entities.Announcement) error {
+	return r.db.WithContext(ctx).Create(&entities).Error
+}
+
+// GetByIDs 根据ID列表获取
+func (r *GormAnnouncementRepository) GetByIDs(ctx context.Context, ids []string) ([]entities.Announcement, error) {
+	var announcements []entities.Announcement
+	err := r.db.WithContext(ctx).Where("id IN ?", ids).Find(&announcements).Error
+	return announcements, err
+}
+
+// UpdateBatch 批量更新
+func (r *GormAnnouncementRepository) UpdateBatch(ctx context.Context, entities []entities.Announcement) error {
+	return r.db.WithContext(ctx).Save(&entities).Error
+}
+
+// DeleteBatch 批量删除
+func (r *GormAnnouncementRepository) DeleteBatch(ctx context.Context, ids []string) error {
+	return r.db.WithContext(ctx).Delete(&entities.Announcement{}, "id IN ?", ids).Error
+}
+
+// List 列表查询
+func (r *GormAnnouncementRepository) List(ctx context.Context, options interfaces.ListOptions) ([]entities.Announcement, error) {
+	var announcements []entities.Announcement
+
+	dbQuery := r.db.WithContext(ctx).Model(&entities.Announcement{})
+
+	// 应用筛选条件
+	if options.Filters != nil {
+		for key, value := range options.Filters {
+			dbQuery = dbQuery.Where(key+" = ?", value)
+		}
+	}
+
+	if options.Search != "" {
+		search := "%" + options.Search + "%"
+		dbQuery = dbQuery.Where("title LIKE ? OR content LIKE ?", search, search)
+	}
+
+	// 应用排序
+	if options.Sort != "" {
+		order := "DESC"
+		if options.Order != "" {
+			order = strings.ToUpper(options.Order)
+		}
+		dbQuery = dbQuery.Order(fmt.Sprintf("%s %s", options.Sort, order))
+	} else {
+		dbQuery = dbQuery.Order("created_at DESC")
+	}
+
+	// 应用分页
+	if options.Page > 0 && options.PageSize > 0 {
+		offset := (options.Page - 1) * options.PageSize
+		dbQuery = dbQuery.Offset(offset).Limit(options.PageSize)
+	}
+
+	// 预加载关联数据
+	if len(options.Include) > 0 {
+		for _, include := range options.Include {
+			dbQuery = dbQuery.Preload(include)
+		}
+	}
+
+	err := dbQuery.Find(&announcements).Error
+	return announcements, err
+}