eta_mini_crm/models/sys_role.go

package models

import (
	"context"
	"time"

	"github.com/beego/beego/v2/client/orm"
)

type SysRole struct {
	SysRoleId   int       `orm:"pk" description:"角色id"`
	SysRoleName string    `description:"姓名"`
	CreateTime  time.Time `description:"创建时间"`
	ModifyTime  time.Time `description:"更新时间"`
}

func (s *SysRole) Add() (err error) {
	o := orm.NewOrmUsingDB("master")
	_, err = o.Insert(s)
	return
}

func DeleteSysRoleById(roleId int) (err error) {
	sql := `DELETE FROM sys_role WHERE sys_role_id=? `
	o := orm.NewOrmUsingDB("master")
	_, err = o.Raw(sql, roleId).Exec()
	return
}

func GetSysRoleCountByRoleName(roleName string) (count int, err error) {
	sql := `SELECT COUNT(1) AS count FROM sys_role WHERE sys_role_name=? `
	o := orm.NewOrmUsingDB("master")
	err = o.Raw(sql, roleName).QueryRow(&count)
	return
}

func UpdateSysUserRoleByRoleId(roleId int, roleName string) (err error) {
	o := orm.NewOrmUsingDB("master")
	err = o.DoTx(func(ctx context.Context, txOrm orm.TxOrmer) error {
		role := &SysRole{}
		role.SysRoleId = roleId
		role.SysRoleName = roleName
		role.ModifyTime = time.Now()
		_, e := txOrm.Update(role, "sys_role_id", "sys_role_name", "modify_time")
		if e != nil {
			return e
		}
		sql := `UPDATE sys_user SET sys_role_name=? WHERE sys_role_id=?`
		_, e = txOrm.Raw(sql, roleName, roleId).Exec()
		return e
	})
	return
}

func GetSysRoleById(id int) (item *SysRole, err error) {
	o := orm.NewOrmUsingDB("master")
	sql := `SELECT * FROM sys_role WHERE sys_role_id=?`
	err = o.Raw(sql, id).QueryRow(&item)
	return
}

func GetSysRoleListCount() (count int, err error) {
	o := orm.NewOrmUsingDB("master")
	sql := `SELECT COUNT(1) AS count FROM sys_role `
	err = o.Raw(sql).QueryRow(&count)
	return
}

func GetSysRoleList(startSize, pageSize int) (items []*SysRole, err error) {
	o := orm.NewOrmUsingDB("master")
	sql := `SELECT * FROM sys_role ORDER BY modify_time DESC LIMIT ?,?`
	_, err = o.Raw(sql, startSize, pageSize).QueryRows(&items)
	return
}

type SysMenu struct {
}

type SysRoleMenuMapping struct {
}