82 lines
2.0 KiB
Go
82 lines
2.0 KiB
Go
package service
|
|
|
|
import (
|
|
"kra/internal/plugin/announcement/model"
|
|
"kra/internal/plugin/announcement/model/request"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Info 公告服务
|
|
type Info struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewInfoService 创建公告服务
|
|
func NewInfoService(db *gorm.DB) *Info {
|
|
return &Info{db: db}
|
|
}
|
|
|
|
// CreateInfo 创建公告记录
|
|
func (s *Info) CreateInfo(info *model.Info) (err error) {
|
|
err = s.db.Create(info).Error
|
|
return err
|
|
}
|
|
|
|
// DeleteInfo 删除公告记录
|
|
func (s *Info) DeleteInfo(ID string) (err error) {
|
|
err = s.db.Delete(&model.Info{}, "id = ?", ID).Error
|
|
return err
|
|
}
|
|
|
|
// DeleteInfoByIds 批量删除公告记录
|
|
func (s *Info) DeleteInfoByIds(IDs []string) (err error) {
|
|
err = s.db.Delete(&[]model.Info{}, "id in ?", IDs).Error
|
|
return err
|
|
}
|
|
|
|
// UpdateInfo 更新公告记录
|
|
func (s *Info) UpdateInfo(info model.Info) (err error) {
|
|
err = s.db.Model(&model.Info{}).Where("id = ?", info.ID).Updates(&info).Error
|
|
return err
|
|
}
|
|
|
|
// GetInfo 根据ID获取公告记录
|
|
func (s *Info) GetInfo(ID string) (info model.Info, err error) {
|
|
err = s.db.Where("id = ?", ID).First(&info).Error
|
|
return
|
|
}
|
|
|
|
// GetInfoInfoList 分页获取公告记录
|
|
func (s *Info) GetInfoInfoList(info request.InfoSearch) (list []model.Info, total int64, err error) {
|
|
limit := info.PageSize
|
|
offset := info.PageSize * (info.Page - 1)
|
|
// 创建db
|
|
db := s.db.Model(&model.Info{})
|
|
var infos []model.Info
|
|
// 如果有条件搜索 下方会自动创建搜索语句
|
|
if info.StartCreatedAt != nil && info.EndCreatedAt != nil {
|
|
db = db.Where("created_at BETWEEN ? AND ?", info.StartCreatedAt, info.EndCreatedAt)
|
|
}
|
|
err = db.Count(&total).Error
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if limit != 0 {
|
|
db = db.Limit(limit).Offset(offset)
|
|
}
|
|
err = db.Find(&infos).Error
|
|
return infos, total, err
|
|
}
|
|
|
|
// GetInfoDataSource 获取公告数据源
|
|
func (s *Info) GetInfoDataSource() (res map[string][]map[string]any, err error) {
|
|
res = make(map[string][]map[string]any)
|
|
|
|
userID := make([]map[string]any, 0)
|
|
s.db.Table("sys_users").Select("nick_name as label,id as value").Scan(&userID)
|
|
res["userID"] = userID
|
|
return
|
|
}
|