cutego/modules/core/dao/user_post_dao.go

49 lines
1.0 KiB
Go

package dao
import (
"cutego/modules/core/dataobject"
"cutego/pkg/logging"
"cutego/refs"
)
type UserPostDao struct {
}
// BatchInsert 批量新增用户岗位信息
func (d UserPostDao) BatchInsert(posts []dataobject.SysUserPost) {
session := refs.SqlDB.NewSession()
session.Begin()
_, err := session.Table(dataobject.SysUserPost{}.TableName()).Insert(&posts)
if err != nil {
logging.ErrorLog(err)
session.Rollback()
return
}
session.Commit()
}
// Delete 删除用户和岗位关系
func (d UserPostDao) Delete(id int64) {
post := dataobject.SysUserPost{
UserId: id,
}
session := refs.SqlDB.NewSession()
session.Begin()
_, err := session.Delete(&post)
if err != nil {
logging.ErrorLog(err)
session.Rollback()
}
session.Commit()
}
// CountById 通过岗位ID查询岗位使用数量
func (d UserPostDao) CountById(id int64) int64 {
count, err := refs.SqlDB.NewSession().Table("sys_user_post").Cols("post_id").Where("post_id = ?", id).Count()
if err != nil {
logging.ErrorLog(err)
return 0
}
return count
}