后端重构

internal/plugin/announcement/service/enter.go

@@ -0,0 +1,18 @@
+package service
+
+import "gorm.io/gorm"
+
+var Service *service
+
+type service struct {
+	Info *Info
+}
+
+// NewService 创建服务实例
+func NewService(db *gorm.DB) *service {
+	s := &service{
+		Info: NewInfoService(db),
+	}
+	Service = s
+	return s
+}

internal/plugin/announcement/service/info.go

@@ -0,0 +1,81 @@
+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
+}