sys_role.go 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301
  1. package system
  2. import (
  3. "eta_gn/eta_api/global"
  4. "eta_gn/eta_api/utils"
  5. "github.com/rdlucklib/rdluck_tools/paging"
  6. "time"
  7. )
  8. type SysRoleAddReq struct {
  9. RoleName string `description:"角色名称"`
  10. RoleType string `description:"角色类型"`
  11. }
  12. type SysRole struct {
  13. RoleId int `gorm:"primaryKey"`
  14. RoleName string `description:"角色名称"`
  15. RoleType string `description:"角色类型"`
  16. RoleTypeCode string `description:"角色类型编码"`
  17. CreateTime time.Time `description:"创建时间"`
  18. ModifyTime time.Time
  19. }
  20. type SysRoleItem struct {
  21. RoleId int `orm:"column(role_id);pk" description:"角色ID"`
  22. RoleName string `description:"角色名称"`
  23. RoleType string `description:"角色类型"`
  24. RoleTypeCode string `description:"角色类型编码"`
  25. CreateTime time.Time `description:"创建时间"`
  26. RoleLevel int `description:"角色等级:0-表示一级角色,每个账号只能绑定一个一级角色 ,1-表示二级角色,每个账号可以绑定多个二级 角色"`
  27. }
  28. func GetSysRoleCount(roleName string) (count int, err error) {
  29. sql := `SELECT COUNT(1) AS count FROM sys_role WHERE role_name=? `
  30. // o := orm.NewOrm()
  31. //err = o.Raw(sql, roleName).QueryRow(&count)
  32. err = global.DEFAULT_DmSQL.Raw(sql, roleName).First(&count).Error
  33. return
  34. }
  35. func AddSysRole(item *SysRole) (lastId int64, err error) {
  36. // o := orm.NewOrm()
  37. //lastId, err = o.Insert(item)
  38. err = global.DEFAULT_DmSQL.Create(item).Error
  39. return
  40. }
  41. type SysRoleEditReq struct {
  42. RoleId int `description:"角色ID"`
  43. RoleName string `description:"角色名称"`
  44. RoleType string `description:"角色类型"`
  45. }
  46. func GetSysRoleById(roleId int) (item *SysRoleItem, err error) {
  47. sql := `SELECT * FROM sys_role WHERE role_id=? `
  48. // o := orm.NewOrm()
  49. //err = o.Raw(sql, roleId).QueryRow(&item)
  50. err = global.DEFAULT_DmSQL.Raw(sql, roleId).First(&item).Error
  51. return
  52. }
  53. func GetSysRoleByName(roleName string) (item *SysRoleItem, err error) {
  54. sql := `SELECT * FROM sys_role WHERE role_name=? `
  55. // o := orm.NewOrm()
  56. //err = o.Raw(sql, roleName).QueryRow(&item)
  57. err = global.DEFAULT_DmSQL.Raw(sql, roleName).First(&item).Error
  58. return
  59. }
  60. // GetSysRoleByRoleTypeCode role_type_code
  61. func GetSysRoleByRoleTypeCode(roleTypeCode string) (item *SysRoleItem, err error) {
  62. sql := `SELECT * FROM sys_role WHERE role_type_code=? `
  63. // o := orm.NewOrm()
  64. //err = o.Raw(sql, roleTypeCode).QueryRow(&item)
  65. err = global.DEFAULT_DmSQL.Raw(sql, roleTypeCode).First(&item).Error
  66. return
  67. }
  68. func ModifySysRole(roleName, roleType, roleTypeCode string, groupId int) (err error) {
  69. sql := `UPDATE sys_role SET role_name=?,role_type=?,role_type_code=?,modify_time=NOW() WHERE role_id=? `
  70. // o := orm.NewOrm()
  71. //_, err = o.Raw(sql, roleName, roleType, roleTypeCode, groupId).Exec()
  72. err = global.DEFAULT_DmSQL.Exec(sql, roleName, roleType, roleTypeCode, groupId).Error
  73. return
  74. }
  75. type SysRoleDeleteReq struct {
  76. RoleId int `description:"角色ID"`
  77. }
  78. func DeleteSysRole(roleId int) (err error) {
  79. sql := `DELETE FROM sys_role WHERE role_id=? `
  80. // o := orm.NewOrm()
  81. //_, err = o.Raw(sql, roleId).Exec()
  82. err = global.DEFAULT_DmSQL.Exec(sql, roleId).Error
  83. return
  84. }
  85. func GetSysRoleListCount(condition string, pars []interface{}) (count int, err error) {
  86. // o := orm.NewOrm()
  87. sql := `SELECT COUNT(1) AS count FROM sys_role WHERE 1=1 `
  88. if condition != "" {
  89. sql += condition
  90. }
  91. //err = o.Raw(sql, pars).QueryRow(&count)
  92. err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&count).Error
  93. return
  94. }
  95. func GetSysRoleList(condition string, pars []interface{}, startSize, pageSize int) (items []*SysRoleItem, err error) {
  96. // o := orm.NewOrm()
  97. sql := `SELECT * FROM sys_role WHERE 1=1 `
  98. if condition != "" {
  99. sql += condition
  100. }
  101. sql += `ORDER BY modify_time DESC LIMIT ?,?`
  102. //_, err = o.Raw(sql, pars...).QueryRows(&items)
  103. pars = append(pars, startSize)
  104. pars = append(pars, pageSize)
  105. err = global.DEFAULT_DmSQL.Raw(sql, pars...).Find(&items).Error
  106. return
  107. }
  108. type SysRoleListResp struct {
  109. List []*SysRoleItem
  110. Paging *paging.PagingItem `description:"分页数据"`
  111. }
  112. type RoleMenu struct {
  113. MenuId int `description:"导航唯一标识"`
  114. Name string `description:"导航名称"`
  115. NameEn string `description:"导航名称(英文)"`
  116. }
  117. type RoleMenuList struct {
  118. MenuId int `description:"导航唯一标识"`
  119. Name string `description:"导航名称"`
  120. NameEn string `description:"导航名称(英文)"`
  121. Child []*RoleMenu `gorm:"-"`
  122. CheckList []int
  123. }
  124. type RoleMenuListResp struct {
  125. List []*RoleMenuList
  126. }
  127. func GetRoleMenuList() (items []*RoleMenuList, err error) {
  128. sql := `SELECT * FROM sys_menu WHERE parent_id=0 ORDER BY sort ASC `
  129. //_, err = orm.NewOrm().Raw(sql).QueryRows(&items)
  130. err = global.DEFAULT_DmSQL.Raw(sql).Find(&items).Error
  131. return
  132. }
  133. func GetRoleMenuByParentId(parentId int) (items []*RoleMenu, err error) {
  134. sql := `SELECT * FROM sys_menu WHERE parent_id=? ORDER BY sort ASC `
  135. //_, err = orm.NewOrm().Raw(sql, parentId).QueryRows(&items)
  136. err = global.DEFAULT_DmSQL.Raw(sql, parentId).Find(&items).Error
  137. return
  138. }
  139. func GetCheckListRoleMenu(roleId, parentId int) (items []*RoleMenuList, err error) {
  140. sql := ` SELECT b.menu_id FROM sys_role_menu AS a
  141. INNER JOIN sys_menu AS b ON a.menu_id=b.menu_id
  142. WHERE a.role_id=? AND a.type = 0
  143. AND b.parent_id=? `
  144. //_, err = orm.NewOrm().Raw(sql, roleId, parentId).QueryRows(&items)
  145. err = global.DEFAULT_DmSQL.Raw(sql, roleId, parentId).Find(&items).Error
  146. return
  147. }
  148. type RoleMenusAddReq struct {
  149. RoleId int `description:"角色ID"`
  150. MenuIdStr string `description:"MenuId,多个用英文逗号隔开"`
  151. MenuButtonIdStr string `description:"menu_button_id,多个用英文逗号隔开"`
  152. }
  153. func DeleteRoleMenuByRoleId(roleId int) (err error) {
  154. sql := `DELETE FROM sys_role_menu WHERE role_id=? `
  155. //_, err = orm.NewOrm().Raw(sql, roleId).Exec()
  156. err = global.DEFAULT_DmSQL.Exec(sql, roleId).Error
  157. return
  158. }
  159. type SysRoleMenu struct {
  160. RoleMenuId int `gorm:"primaryKey"`
  161. RoleId int
  162. MenuId int
  163. Type int `description:"类型:0-全选;1-半选(此字段仅用于前端的回显)"`
  164. }
  165. func AddSysRoleMenu(item *SysRoleMenu) (lastId int64, err error) {
  166. // o := orm.NewOrm()
  167. //lastId, err = o.Insert(item)
  168. err = global.DEFAULT_DmSQL.Create(item).Error
  169. return
  170. }
  171. // CreateMultiSysRoleMenu 删除并新增角色权限
  172. func CreateMultiSysRoleMenu(roleId int, items []*SysRoleMenu) (err error) {
  173. if roleId == 0 {
  174. return
  175. }
  176. //o := orm.NewOrm()
  177. //tx, err := o.Begin()
  178. //if err != nil {
  179. // return
  180. //}
  181. //defer func() {
  182. // if err != nil {
  183. // _ = tx.Rollback()
  184. // return
  185. // }
  186. // _ = tx.Commit()
  187. //}()
  188. //
  189. //sql := `DELETE FROM sys_role_menu WHERE role_id = ?`
  190. //_, err = tx.Raw(sql, roleId).Exec()
  191. //if err != nil {
  192. // return
  193. //}
  194. //
  195. //if len(items) > 0 {
  196. // _, err = tx.InsertMulti(len(items), items)
  197. //}
  198. tx := global.DEFAULT_DmSQL.Begin()
  199. defer func() {
  200. if err != nil {
  201. _ = tx.Rollback()
  202. return
  203. }
  204. _ = tx.Commit()
  205. }()
  206. sql := `DELETE FROM sys_role_menu WHERE role_id = ?`
  207. //_, err = tx.Raw(sql, roleId).Exec()
  208. err = tx.Exec(sql, roleId).Error
  209. if err != nil {
  210. return
  211. }
  212. if len(items) > 0 {
  213. //_, err = tx.InsertMulti(len(items), items)
  214. err = tx.CreateInBatches(items, utils.MultiAddNum).Error
  215. }
  216. return
  217. }
  218. func GetSysRoleAll(levelFlag bool) (items []*SysRoleItem, err error) {
  219. // o := orm.NewOrm()
  220. sql := `SELECT * FROM sys_role WHERE 1=1`
  221. if levelFlag {
  222. sql += ` and role_level = 0`
  223. }
  224. sql += ` ORDER BY create_time DESC `
  225. //_, err = o.Raw(sql).QueryRows(&items)
  226. err = global.DEFAULT_DmSQL.Raw(sql).Find(&items).Error
  227. return
  228. }
  229. type SysRoleAllResp struct {
  230. List []*SysRoleItem
  231. }
  232. // GetEnglishAuthRoleIds 获取英文权限角色IDs
  233. func GetEnglishAuthRoleIds(roleCodes []string) (roleIds []int, err error) {
  234. if len(roleCodes) == 0 {
  235. return
  236. }
  237. // o := orm.NewOrm()
  238. sql := `SELECT role_id FROM sys_role WHERE role_type_code IN (` + utils.GetOrmInReplace(len(roleCodes)) + `)`
  239. //_, err = o.Raw(sql, roleCodes).QueryRows(&roleIds)
  240. err = global.DEFAULT_DmSQL.Raw(sql, roleCodes).Scan(&roleIds).Error
  241. return
  242. }
  243. // GetSysRoleMenuByRoleId 获取角色关联菜单
  244. func GetSysRoleMenuByRoleId(roleId int) (items []*SysRoleMenu, err error) {
  245. sql := `SELECT
  246. *
  247. FROM
  248. sys_role_menu
  249. WHERE
  250. role_id = ?`
  251. //_, err = orm.NewOrm().Raw(sql, roleId).QueryRows(&items)
  252. err = global.DEFAULT_DmSQL.Raw(sql, roleId).Find(&items).Error
  253. return
  254. }
  255. // GetEnglishAuthRoleItems 获取英文权限角色
  256. func GetEnglishAuthRoleItems(roleCodes []string) (items []*SysRole, err error) {
  257. if len(roleCodes) == 0 {
  258. return
  259. }
  260. // o := orm.NewOrm()
  261. sql := `SELECT * FROM sys_role WHERE role_type_code IN (` + utils.GetOrmInReplace(len(roleCodes)) + `)`
  262. //_, err = o.Raw(sql, roleCodes).QueryRows(&items)
  263. err = global.DEFAULT_DmSQL.Raw(sql, roleCodes).Find(&items).Error
  264. return
  265. }