128 lines
4.0 KiB
Go
128 lines
4.0 KiB
Go
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
|
|
}
|