第 5 阶段：纠纷、通知与后台-1

backend/internal/modules/notification/repository.go

@@ -0,0 +1,82 @@
+package notification
+
+import (
+	"time"
+
+	"hfb_sys/backend/internal/model"
+
+	"gorm.io/gorm"
+)
+
+type Repository struct {
+	db *gorm.DB
+}
+
+type Entry struct {
+	UserID  uint64
+	Type    string
+	Title   string
+	Content string
+	BizType string
+	BizID   *uint64
+}
+
+func NewRepository(db *gorm.DB) *Repository {
+	return &Repository{db: db}
+}
+
+func (r *Repository) List(userID uint64) ([]NotificationDTO, error) {
+	var rows []model.Notification
+	if err := r.db.Where("user_id = ?", userID).Order("id DESC").Limit(100).Find(&rows).Error; err != nil {
+		return nil, err
+	}
+	items := make([]NotificationDTO, 0, len(rows))
+	for _, row := range rows {
+		items = append(items, toDTO(row))
+	}
+	return items, nil
+}
+
+func (r *Repository) MarkRead(userID uint64, id uint64) error {
+	now := time.Now()
+	return r.db.Model(&model.Notification{}).
+		Where("id = ? AND user_id = ?", id, userID).
+		Update("read_at", now).Error
+}
+
+func Append(tx *gorm.DB, entries ...Entry) error {
+	for _, entry := range entries {
+		if entry.UserID == 0 || entry.Title == "" {
+			continue
+		}
+		row := model.Notification{
+			UserID:  entry.UserID,
+			Type:    entry.Type,
+			Title:   entry.Title,
+			Content: entry.Content,
+			BizType: entry.BizType,
+			BizID:   entry.BizID,
+		}
+		if row.Type == "" {
+			row.Type = "system"
+		}
+		if err := tx.Create(&row).Error; err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func toDTO(row model.Notification) NotificationDTO {
+	return NotificationDTO{
+		ID:        row.ID,
+		UserID:    row.UserID,
+		Type:      row.Type,
+		Title:     row.Title,
+		Content:   row.Content,
+		BizType:   row.BizType,
+		BizID:     row.BizID,
+		ReadAt:    row.ReadAt,
+		CreatedAt: row.CreatedAt,
+	}
+}