cutego/core/dao/notice_dao.go

package dao

import (
	"cutego/core/api/v1/request"
	"cutego/core/entity"
	"cutego/pkg/common"
	"cutego/pkg/page"
	"github.com/druidcaesa/gotool"
)

type NoticeDao struct {
}

// SelectPage 查询集合
func (d NoticeDao) SelectPage(query request.NoticeQuery) (*[]entity.SysNotice, int64) {
	notices := make([]entity.SysNotice, 0)
	session := SqlDB.NewSession().Table(entity.SysNotice{}.TableName())
	if gotool.StrUtils.HasNotEmpty(query.NoticeTitle) {
		session.And("notice_title like concat('%', ?, '%')", query.NoticeTitle)
	}
	if gotool.StrUtils.HasNotEmpty(query.NoticeType) {
		session.And("notice_type = ?", query.NoticeType)
	}
	if gotool.StrUtils.HasNotEmpty(query.CreateBy) {
		session.And("create_by like concat('%', ?, '%')", query.CreateBy)
	}
	total, _ := page.GetTotal(session.Clone())
	err := session.Limit(query.PageSize, page.StartSize(query.PageNum, query.PageSize)).Find(&notices)
	if err != nil {
		common.ErrorLog(err)
		return nil, 0
	}
	return &notices, total
}

// Insert 添加数据
func (d NoticeDao) Insert(notice entity.SysNotice) int64 {
	session := SqlDB.NewSession()
	session.Begin()
	insert, err := session.Insert(&notice)
	if err != nil {
		session.Rollback()
		common.ErrorLog(err)
		return 0
	}
	session.Commit()
	return insert
}

// Delete 批量删除
func (d NoticeDao) Delete(list []int64) int64 {
	session := SqlDB.NewSession()
	session.Begin()
	i, err := session.In("notice_id", list).Delete(&entity.SysNotice{})
	if err != nil {
		session.Rollback()
		common.ErrorLog(err)
		return 0
	}
	session.Commit()
	return i
}

// SelectById 查询
func (d NoticeDao) SelectById(id int64) *entity.SysNotice {
	notice := entity.SysNotice{}
	_, err := SqlDB.NewSession().Where("notice_id = ?", id).Get(&notice)
	if err != nil {
		common.ErrorLog(err)
		return nil
	}
	return &notice
}

// Update 修改数据
func (d NoticeDao) Update(notice entity.SysNotice) int64 {
	session := SqlDB.NewSession()
	session.Begin()
	update, err := session.Where("notice_id = ?", notice.NoticeId).Update(&notice)
	if err != nil {
		session.Rollback()
		common.ErrorLog(err)
		return 0
	}
	session.Commit()
	return update
}