micro-document/internel/dao/press_releases.go

package dao

import (
	"errors"
	"micro-document/api/pressreleases"
	"micro-document/internel/model"
	"micro-document/pkg/db"
	"time"

	"dubbo.apache.org/dubbo-go/v3/common/logger"
	"github.com/duke-git/lancet/v2/datetime"
	"github.com/go-sql-driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

type pressReleasesDao struct{}

var PressReleaseDao = new(pressReleasesDao)

func (pressReleasesDao) List(req *pressreleases.ListReq) (data []model.PressReleases, total int64, err error) {
	session := db.DocDB.Model(&model.PressReleases{})
	if req.Title != "" {
		session = session.Where("title like ?", "%"+req.Title+"%")
	}
	if req.StartTime != 0 {
		session = session.Where("created_at >= ?", time.UnixMilli(int64(req.StartTime)))
	}
	if req.EndTime != 0 {
		session = session.Where("created_at <= ?", time.UnixMilli(int64(req.EndTime)))
	}
	if req.Display != 0 {
		session = session.Where("display = ?", req.Display)
	}
	if req.Status != 0 {
		session = session.Where("status = ?", req.Status)
	}
	session = session.Order("updated_at desc")
	if err = session.Count(&total).Error; err != nil {
		return
	}
	err = session.Offset((int(req.Page-1) * int(req.PageSize))).Limit(int(req.PageSize)).Find(&data).Error
	return
}

func (pressReleasesDao) Display(req *pressreleases.DisplayReq) (data []model.PressReleases, total int64, err error) {
	session := db.DocDB.Model(&model.PressReleases{}).Where(&model.PressReleases{Status: 2})
	if req.Display != 0 {
		session = session.Where("display = ?", req.Display)
	}
	if req.Query != "" {
		session = session.Where("title like ?", "%"+req.Query+"%")
	}
	session = session.Order(clause.OrderBy{Columns: []clause.OrderByColumn{
		{Column: clause.Column{Name: "sort"}, Desc: false},
		{Column: clause.Column{Name: "updated_at"}, Desc: true},
	}})
	if req.Display != 0 {
		session = session.Where("display = ?", req.Display)
	}
	if req.TimeStart != 0 {
		session = session.Where("created_at > ?", time.UnixMilli(req.TimeStart))
	}
	if req.TimeEnd != 0 {
		session = session.Where("created_at < ?", time.UnixMilli(req.TimeEnd))
	} else {
		if req.TimeStart != 0 {
			session = session.Where("created_at < ?", datetime.EndOfYear(time.UnixMilli(req.TimeStart)))
		}
	}
	err = session.Count(&total).Error
	if err != nil {
		return
	}
	if req.Page != 0 && req.PageSize != 0 {
		session = session.Offset(int(req.Page-1) * int(req.PageSize)).Limit(int(req.PageSize))
	}
	err = session.Find(&data).Error
	return
}

func (pressReleasesDao) Edit(req *pressreleases.EditReq) (err error) {
	return db.DocDB.Transaction(func(tx *gorm.DB) error {
		err := tx.Model(&model.PressReleases{}).Where("id = ?", req.Id).UpdateColumns(&model.PressReleases{
			Sort:           uint(req.Sort),
			Status:         int(req.Status),
			Title:          req.Title,
			Attachment:     req.Attachment,
			AttachmentName: req.AttachmentName,
			Operator:       req.Operator,
			Summary:        req.Summary,
			OperatorID:     uint(req.OperatorId),
			Content:        req.Content,
			Display:        int(req.Display),
			CreatedAt:      time.UnixMilli(int64(req.CreatedAt)),
			UpdatedAt:      time.Now(),
		}).Error
		if err != nil {
			var mysqlErr *mysql.MySQLError
			if errors.As(err, &mysqlErr) && mysqlErr.Number == 1062 {
				return errors.New("已存在标题相同的数据")
			}
			return errors.New("操作失败")
		}
		var n int64
		if err := tx.Model(&model.PressReleases{}).Where("display = ?", 2).Count(&n).Error; err != nil {
			logger.Error(err)
			return errors.New("操作失败")
		}
		if n > 5 {
			return errors.New("展示数量不能超过5个")
		}
		return nil
	})
}

func (pressReleasesDao) Create(req *pressreleases.CreateReq) (err error) {
	return db.DocDB.Transaction(func(tx *gorm.DB) error {
		err := tx.Model(&model.PressReleases{}).Create(&model.PressReleases{
			Sort:           uint(req.Sort),
			Status:         int(req.Status),
			Title:          req.Title,
			Attachment:     req.Attachment,
			AttachmentName: req.AttachmentName,
			Operator:       req.Operator,
			OperatorID:     uint(req.OperatorId),
			Content:        req.Content,
			Summary:        req.Summary,
			Display:        int(req.Display),
			CreatedAt:      time.UnixMilli(int64(req.CreatedAt)),
			UpdatedAt:      time.Now(),
		}).Error
		if err != nil {
			var mysqlErr *mysql.MySQLError
			if errors.As(err, &mysqlErr) && mysqlErr.Number == 1062 {
				return errors.New("已存在标题相同的数据")
			}
			return errors.New("操作失败")
		}
		var n int64
		if err := tx.Model(&model.PressReleases{}).Where("display = ?", 2).Count(&n).Error; err != nil {
			return err
		}
		if n > 5 {
			return errors.New("展示数量不能超过5个")
		}
		return nil

	})

}

func (pressReleasesDao) Delete(req *pressreleases.DeleteReq) (err error) {
	return db.DocDB.Delete(&model.PressReleases{
		ID: uint(req.Id),
	}).Error
}

func (pressReleasesDao) GetById(id int) (data model.PressReleases, err error) {
	err = db.DocDB.Model(&model.PressReleases{}).Where("id = ?", id).First(&data).Error
	return
}