micro-document/internel/dao/press_releases.go

package dao

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

	"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)
	}
	err = session.Order("updated_at desc").Offset((int(req.Page-1) * int(req.PageSize))).Limit(int(req.PageSize)).Count(&total).Find(&data).Error
	return
}

func (pressReleasesDao) Display(req *pressreleases.DisplayReq) (data []model.PressReleases, err error) {
	session := db.DocDB.Model(&model.PressReleases{}).Where(&model.PressReleases{Display: 2, Status: 2})
	if req.Query != "" {
		session = session.Where("title like ?", "%"+req.Query+"%")
	}
	err = session.Offset(int(req.Page-1) * int(req.PageSize)).Limit(int(req.PageSize)).Order(clause.OrderBy{Columns: []clause.OrderByColumn{
		{Column: clause.Column{Name: "sort"}, Desc: false},
		{Column: clause.Column{Name: "updated_at"}, Desc: true},
	}}).Find(&data).Error
	return
}

func (pressReleasesDao) Edit(req *pressreleases.EditReq) (err error) {
	return db.DocDB.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),
		Model: gorm.Model{
			CreatedAt: time.UnixMilli(int64(req.CreatedAt)),
			UpdatedAt: time.Now(),
		},
	}).Error
}

func (pressReleasesDao) Create(req *pressreleases.CreateReq) (err error) {
	return db.DocDB.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),
		Model: gorm.Model{
			CreatedAt: time.UnixMilli(int64(req.CreatedAt)),
			UpdatedAt: time.Now(),
		},
	}).Error
}

func (pressReleasesDao) Delete(req *pressreleases.DeleteReq) (err error) {
	return db.DocDB.Delete(&model.PressReleases{
		Model: gorm.Model{
			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
}