迁移公告，邮件

internal/data/announcement.go

@@ -0,0 +1,127 @@
+package data
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"time"
+
+	"kra/internal/biz"
+
+	"gorm.io/gorm"
+)
+
+type announcementPO struct {
+	ID          uint `gorm:"primaryKey"`
+	CreatedAt   time.Time
+	UpdatedAt   time.Time
+	DeletedAt   gorm.DeletedAt `gorm:"index"`
+	Title       string
+	Content     string `gorm:"type:text"`
+	UserID      *uint  `gorm:"column:user_id"`
+	Attachments []byte `gorm:"type:json"`
+}
+
+func (announcementPO) TableName() string { return "gva_announcements_info" }
+
+type announcementRepo struct{ data *Data }
+
+func NewAnnouncementRepo(data *Data) biz.AnnouncementRepo { return &announcementRepo{data: data} }
+
+func newAnnouncement(item *biz.Announcement) announcementPO {
+	attachments := []byte(item.Attachments)
+	if len(attachments) == 0 || !json.Valid(attachments) {
+		attachments = []byte("[]")
+	}
+	return announcementPO{ID: item.ID, Title: item.Title, Content: item.Content, UserID: item.UserID, Attachments: attachments}
+}
+
+func announcementToBiz(item announcementPO) *biz.Announcement {
+	attachments := json.RawMessage(item.Attachments)
+	if len(attachments) == 0 || !json.Valid(attachments) {
+		attachments = json.RawMessage("[]")
+	}
+	return &biz.Announcement{ID: item.ID, CreatedAt: item.CreatedAt, UpdatedAt: item.UpdatedAt, Title: item.Title, Content: item.Content, UserID: item.UserID, Attachments: attachments}
+}
+
+func (r *announcementRepo) Create(ctx context.Context, item *biz.Announcement) error {
+	po := newAnnouncement(item)
+	if err := r.data.gormDB.WithContext(ctx).Create(&po).Error; err != nil {
+		return err
+	}
+	item.ID, item.CreatedAt, item.UpdatedAt = po.ID, po.CreatedAt, po.UpdatedAt
+	return nil
+}
+
+func (r *announcementRepo) Delete(ctx context.Context, id uint) error {
+	return r.data.gormDB.WithContext(ctx).Delete(&announcementPO{}, id).Error
+}
+
+func (r *announcementRepo) DeleteByIDs(ctx context.Context, ids []uint) error {
+	return r.data.gormDB.WithContext(ctx).Where("id IN ?", ids).Delete(&announcementPO{}).Error
+}
+
+func (r *announcementRepo) Update(ctx context.Context, item *biz.Announcement) error {
+	po := newAnnouncement(item)
+	result := r.data.gormDB.WithContext(ctx).Model(&announcementPO{}).Where("id = ?", item.ID).Updates(map[string]any{
+		"title": po.Title, "content": po.Content, "user_id": po.UserID, "attachments": po.Attachments,
+	})
+	if result.Error != nil {
+		return result.Error
+	}
+	if result.RowsAffected == 0 {
+		return gorm.ErrRecordNotFound
+	}
+	return nil
+}
+
+func (r *announcementRepo) Find(ctx context.Context, id uint) (*biz.Announcement, error) {
+	var po announcementPO
+	if err := r.data.gormDB.WithContext(ctx).First(&po, id).Error; err != nil {
+		return nil, err
+	}
+	return announcementToBiz(po), nil
+}
+
+func (r *announcementRepo) List(ctx context.Context, filter biz.AnnouncementFilter) ([]*biz.Announcement, int64, error) {
+	db := r.data.gormDB.WithContext(ctx).Model(&announcementPO{})
+	if filter.StartCreatedAt != nil && filter.EndCreatedAt != nil {
+		db = db.Where("created_at BETWEEN ? AND ?", filter.StartCreatedAt, filter.EndCreatedAt)
+	}
+	var total int64
+	if err := db.Count(&total).Error; err != nil {
+		return nil, 0, err
+	}
+	if filter.PageSize > 0 {
+		page := filter.Page
+		if page < 1 {
+			page = 1
+		}
+		db = db.Offset((page - 1) * filter.PageSize).Limit(filter.PageSize)
+	}
+	var pos []announcementPO
+	if err := db.Find(&pos).Error; err != nil {
+		return nil, 0, err
+	}
+	items := make([]*biz.Announcement, 0, len(pos))
+	for _, po := range pos {
+		items = append(items, announcementToBiz(po))
+	}
+	return items, total, nil
+}
+
+func (r *announcementRepo) UserOptions(ctx context.Context) ([]biz.UserOption, error) {
+	var rows []struct {
+		Label string
+		Value uint
+	}
+	err := r.data.gormDB.WithContext(ctx).Table("sys_users").Select("nick_name AS label, id AS value").Where("deleted_at IS NULL").Scan(&rows).Error
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, err
+	}
+	items := make([]biz.UserOption, 0, len(rows))
+	for _, row := range rows {
+		items = append(items, biz.UserOption{Label: row.Label, Value: row.Value})
+	}
+	return items, nil
+}