role_service.go 33KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971
  1. package service
  2. import (
  3. "database/sql"
  4. "fmt"
  5. "strconv"
  6. "strings"
  7. "time"
  8. "encoding/json"
  9. "XT_New/models"
  10. "XT_New/utils"
  11. "github.com/jinzhu/gorm"
  12. )
  13. func GetRoles(orgID int64, appID int64, page int, count int) ([]*models.Role, int, error) {
  14. var roles []*models.Role
  15. var totalCount int
  16. err := readUserDb.Model(&models.Role{}).
  17. Where("org_id = ? AND app_id = ?", orgID, appID).
  18. Count(&totalCount).
  19. Order("ctime asc").Limit(count).Offset((page - 1) * count).
  20. Find(&roles).
  21. Error
  22. if err != nil {
  23. if err == gorm.ErrRecordNotFound {
  24. return make([]*models.Role, 0), 0, nil
  25. } else {
  26. return nil, 0, err
  27. }
  28. }
  29. return roles, totalCount, nil
  30. }
  31. func GetAppRole(orgID int64, appID int64, adminUserID int64) (*models.App_Role, error) {
  32. var appRole models.App_Role
  33. err := readUserDb.Model(models.App_Role{}).Preload("AdminUser", "status = 1").Where("org_id = ? and app_id = ? and admin_user_id = ?", orgID, appID, adminUserID).First(&appRole).Error
  34. if err != nil {
  35. if err == gorm.ErrRecordNotFound {
  36. return nil, nil
  37. } else {
  38. return nil, err
  39. }
  40. }
  41. if len(appRole.RoleIds) > 0 {
  42. role_ids := strings.Split(appRole.RoleIds, ",")
  43. for _, item := range role_ids {
  44. id, _ := strconv.ParseInt(item, 10, 64)
  45. if id != 0 {
  46. role, _ := GetRoleByRoleID(id)
  47. if role != nil {
  48. if role.IsSystem == 1 && role.RoleName == "子管理员" {
  49. appRole.IsSubSuperAdmin = true
  50. }
  51. }
  52. }
  53. }
  54. }
  55. return &appRole, nil
  56. }
  57. func CreateRole(adminUserID int64, orgID int64, appID int64, name string, intro string) (*models.Role, error) {
  58. role := models.Role{
  59. RoleName: name,
  60. RoleIntro: intro,
  61. Creator: adminUserID,
  62. OrgId: orgID,
  63. AppId: appID,
  64. IsSuperAdmin: false,
  65. Status: 1,
  66. CreateTime: time.Now().Unix(),
  67. ModifyTime: time.Now().Unix(),
  68. }
  69. tx := writeUserDb.Begin()
  70. if err := tx.Create(&role).Error; err != nil {
  71. tx.Rollback()
  72. return nil, err
  73. }
  74. tx.Commit()
  75. return &role, nil
  76. }
  77. func GetRoleByRoleID(roleID int64) (*models.Role, error) {
  78. var role models.Role
  79. err := readUserDb.Model(&models.Role{}).Where("id = ?", roleID).First(&role).Error
  80. if err != nil {
  81. if err == gorm.ErrRecordNotFound {
  82. return nil, nil
  83. }
  84. return nil, err
  85. } else {
  86. return &role, nil
  87. }
  88. }
  89. func ModifyRole(role *models.Role) error {
  90. tx := writeUserDb.Begin()
  91. if err := tx.Save(role).Error; err != nil {
  92. tx.Rollback()
  93. return err
  94. }
  95. return tx.Commit().Error
  96. }
  97. // 拥有xx角色的管理员的数量
  98. func RoleAdminUserCount(orgID int64, appID int64, roleID int64) (int, error) {
  99. var count int
  100. err := readUserDb.Model(models.App_Role{}).
  101. Where("org_id = ? AND app_id = ? AND role_id = ? AND status = 1", orgID, appID, roleID).
  102. Count(&count).
  103. Error
  104. if err != nil {
  105. return 0, err
  106. } else {
  107. return count, nil
  108. }
  109. }
  110. type AdminUserManageViewModel struct {
  111. AdminUserId int `gorm:"admin_user_id" json:"user_id"`
  112. UserName string `gorm:"user_name" json:"user_name"`
  113. RoleName string `gorm:"role_name" json:"role_name"`
  114. UserTitle int `gorm:"user_title" json:"user_title"`
  115. Ip string `gorm:"ip" json:"ip"`
  116. Ctime int64 `gorm:"ctime" json:"last_login_time"`
  117. Status int `gorm:"status" json:"status"`
  118. Avatar string `gorm:"avatar" json:"avatar"`
  119. RoleIds string `gorm:"role_ids" json:"role_ids"`
  120. // LastLoginTimeStr string `gorm:"-" json:"last_login_time_formatted"`
  121. TitleName string `gorm:"-" json:"title_name"`
  122. IsSubSuperAdmin bool `gorm:"-" json:"is_sub_super_admin"`
  123. //mobile string `gorm:"-" json:"mobile"`
  124. Sort int `gorm:"sort" json:"sort"`
  125. IsMark int `gorm:"is_mark" json:"is_mark"`
  126. UserType int `gorm:"user_type" json:"user_type"`
  127. }
  128. func GetAdminUsersAndLoginInfo(orgID int64, appID int64, page int, count int) ([]*AdminUserManageViewModel, int, error) {
  129. if count <= 0 {
  130. return []*AdminUserManageViewModel{}, 0, nil
  131. }
  132. if page < 1 {
  133. page = 1
  134. }
  135. var viewModels []*AdminUserManageViewModel = make([]*AdminUserManageViewModel, 0)
  136. rows, err := readUserDb.Raw("SELECT u_a_r.admin_user_id, u_a_r.user_name, u_a_r.user_title, u_l.ip, u_l.ctime, u_a_r.status,u_a_r.avatar,u_a_r.role_ids,u_a_r.sort,u_a_r.is_mark,u_a_r.user_type FROM sgj_user_admin_role AS u_a_r LEFT JOIN (SELECT * FROM (SELECT admin_user_id, org_id, app_id, ip, ctime FROM sgj_user_admin_login_log WHERE org_id = ? AND app_id = ? ORDER BY ctime DESC) AS t GROUP BY admin_user_id) AS u_l ON u_a_r.org_id = u_l.org_id AND u_a_r.app_id = u_l.app_id AND u_a_r.admin_user_id = u_l.admin_user_id WHERE u_a_r.org_id = ? AND u_a_r.app_id = ? GROUP BY u_a_r.admin_user_id LIMIT ? OFFSET ?;", orgID, appID, orgID, appID, count, (page-1)*count).Rows()
  137. defer rows.Close()
  138. if err != nil {
  139. if err == gorm.ErrRecordNotFound {
  140. return viewModels, 0, nil
  141. } else {
  142. return nil, 0, err
  143. }
  144. }
  145. for rows.Next() {
  146. var viewModel AdminUserManageViewModel
  147. readUserDb.ScanRows(rows, &viewModel)
  148. title, _ := models.UserTitle[viewModel.UserTitle]
  149. viewModel.TitleName = title
  150. // if viewModel.Ctime == 0 {
  151. // viewModel.LastLoginTimeStr = ""
  152. // } else {
  153. // loginTime := time.Unix(viewModel.Ctime, 0)
  154. // viewModel.LastLoginTimeStr = loginTime.Format("2006-01-02 15:04")
  155. // }
  156. viewModels = append(viewModels, &viewModel)
  157. }
  158. for _, item := range viewModels {
  159. if len(item.RoleIds) > 0 {
  160. ids := strings.Split(item.RoleIds, ",")
  161. for _, id := range ids {
  162. id, _ := strconv.ParseInt(id, 10, 64)
  163. if id != 0 {
  164. role, _ := GetRoleByRoleID(id)
  165. if role != nil {
  166. if role.IsSystem == 1 && role.RoleName == "子管理员" {
  167. item.IsSubSuperAdmin = true
  168. }
  169. if len(item.RoleName) == 0 {
  170. item.RoleName = role.RoleName
  171. } else {
  172. item.RoleName = item.RoleName + "," + role.RoleName
  173. }
  174. }
  175. }
  176. }
  177. }
  178. //admin, _ := GetAdminUserByUserID(int64(item.AdminUserId))
  179. //item.mobile = admin.Mobile
  180. }
  181. total := 0
  182. //readUserDb.Table("sgj_user_admin_role as u_a_r").Joins("join sgj_user_role as u_r on u_r.org_id = u_a_r.org_id AND u_r.app_id = u_a_r.app_id AND u_r.id = u_a_r.role_id").Where("u_a_r.org_id = ? AND u_a_r.app_id = ?", orgID, appID).Count(&total)
  183. return viewModels, total, nil
  184. }
  185. func GetValidRoleCount(orgID int64, appID int64, superAdminUserID int64) (int, error) {
  186. var count int
  187. err := readUserDb.Model(models.Role{}).
  188. Where("org_id = ? AND app_id = ? AND creator = ? AND is_super_admin = 0 AND status = 1", orgID, appID, superAdminUserID).
  189. Count(&count).
  190. Error
  191. if err != nil {
  192. return 0, err
  193. } else {
  194. return count, nil
  195. }
  196. }
  197. func GetAllValidRoles(orgID int64, appID int64) ([]*models.Role, error) {
  198. var roles []*models.Role
  199. err := readUserDb.Model(models.Role{}).
  200. Where("org_id = ? AND app_id = ? AND status = 1", orgID, appID).
  201. Order("number desc,ctime").
  202. Find(&roles).
  203. Error
  204. if err != nil {
  205. if err == gorm.ErrRecordNotFound {
  206. return make([]*models.Role, 0), nil
  207. } else {
  208. return nil, err
  209. }
  210. }
  211. return roles, nil
  212. }
  213. func IsRoleExist(orgID int64, appID int64, roleID int64) (bool, error) {
  214. var count int
  215. err := readUserDb.Model(models.Role{}).Where("org_id = ? AND app_id = ? AND id = ? AND status = 1", orgID, appID, roleID).Count(&count).Error
  216. if err != nil {
  217. return false, err
  218. } else {
  219. return count > 0, nil
  220. }
  221. }
  222. type AdminUserEditViewModel struct {
  223. AdminUserId int64 `gorm:"admin_user_id" json:"user_id"`
  224. UserName string `gorm:"user_name" json:"user_name"`
  225. Mobile string `gorm:"mobile" json:"mobile"`
  226. UserType int `gorm:"user_type" json:"user_type"`
  227. UserTitle int `gorm:"user_title" json:"user_title"`
  228. Intro string `gorm:"intro" json:"intro"`
  229. RoleId int64 `gorm:"role_id" json:"role_id"`
  230. }
  231. func GetGeneralAdminUser(orgID int64, appID int64, user_id int64) (*AdminUserEditViewModel, error) {
  232. rows, err := readUserDb.Raw("SELECT u_a.mobile, u_a_r.admin_user_id, u_a_r.user_name, u_a_r.user_type, u_a_r.user_title, u_a_r.intro, u_a_r.role_id FROM sgj_user_admin AS u_a, sgj_user_admin_role AS u_a_r WHERE u_a.id = u_a_r.admin_user_id AND u_a.id = ? AND u_a_r.status = 1 AND u_a_r.org_id = ? AND u_a_r.app_id = ?;", user_id, orgID, appID).Rows()
  233. defer rows.Close()
  234. if err != nil {
  235. if err == gorm.ErrRecordNotFound {
  236. return nil, nil
  237. } else {
  238. return nil, err
  239. }
  240. }
  241. if rows.Next() {
  242. var viewModel AdminUserEditViewModel
  243. err := readUserDb.ScanRows(rows, &viewModel)
  244. if err != nil {
  245. return nil, err
  246. } else {
  247. return &viewModel, nil
  248. }
  249. }
  250. return nil, nil
  251. }
  252. func SaveAppRole(appRole *models.App_Role) error {
  253. tx := writeUserDb.Begin()
  254. if err := tx.Model(&models.App_Role{}).Save(appRole).Error; err != nil {
  255. tx.Rollback()
  256. return err
  257. }
  258. return tx.Commit().Error
  259. }
  260. func IsMobileDidUsedAtApp(orgID int64, appID int64, mobile string) (bool, error) {
  261. var count int
  262. rows, err := readUserDb.Raw("select count(u_a.id) as count from sgj_user_admin as u_a, sgj_user_admin_role as u_a_r where u_a_r.org_id = ? and u_a_r.app_id = ? and u_a.mobile = ? and u_a.id = u_a_r.admin_user_id;", orgID, appID, mobile).Rows()
  263. defer rows.Close()
  264. if err != nil {
  265. if err == gorm.ErrRecordNotFound {
  266. return false, nil
  267. } else {
  268. return true, err
  269. }
  270. }
  271. if rows.Next() {
  272. rows.Scan(&count)
  273. return count > 0, nil
  274. }
  275. return true, nil
  276. }
  277. func IsUserSuperAdminWithMobile(mobile string) (bool, error) {
  278. var user models.AdminUser
  279. err := readUserDb.Where("mobile = ?", mobile).First(&user).Error
  280. if err != nil {
  281. if err == gorm.ErrRecordNotFound {
  282. return false, nil
  283. } else {
  284. return false, err
  285. }
  286. }
  287. return user.IsSuperAdmin, nil
  288. }
  289. func CreateGeneralAdminUserOne(orgID int64, appID int64, mobile string, name string, userTitle string, roleIds string, user_type int, user_title int, department_id int64, deapartment_name string, sex int64, age int64, nation string, card_type int64, idCard string, education int64, studyMajorName string, workMajorName string, roletype int64, medicalCode string, doctorcode string, licensing int64, jobnumber string, prescriptionQualificationIdentification int64, identificationOutpatients int64, startime int64, medicalRangeCode int64, medicalLevel int64, medicalTypeJob int64, pharmacistRegistrationNumber string, doctorRangeCode int64, doctorLevel int64, doctorTypeJob int64, doctorNumber string, outpatientIllnessCategory string, isActive int64, activeStatus int64) (*models.AdminUser, string, error) {
  290. now := time.Now().Unix()
  291. tx := writeUserDb.Begin()
  292. var adminUser models.AdminUser
  293. err := readUserDb.Where("mobile = ? AND status = 1", mobile).First(&adminUser).Error
  294. password := ""
  295. if err != nil {
  296. if err != gorm.ErrRecordNotFound {
  297. return nil, "", err
  298. } else {
  299. password = utils.RandomNumberString(6)
  300. adminUser.Mobile = mobile
  301. adminUser.Password = utils.String2md5(password)
  302. adminUser.IsSuperAdmin = false
  303. adminUser.Status = 1
  304. adminUser.CreateTime = now
  305. adminUser.ModifyTime = now
  306. if createErr := tx.Create(&adminUser).Error; createErr != nil {
  307. tx.Rollback()
  308. return nil, "", createErr
  309. }
  310. }
  311. }
  312. app_role := models.App_Role{
  313. AdminUserId: adminUser.Id,
  314. OrgId: orgID,
  315. AppId: appID,
  316. Avatar: "",
  317. UserName: name,
  318. UserTitleName: userTitle,
  319. UserTitle: int8(user_title),
  320. UserType: int8(user_type),
  321. Status: 1,
  322. CreateTime: now,
  323. ModifyTime: now,
  324. RoleIds: roleIds,
  325. IsSort: 1,
  326. Department: deapartment_name,
  327. DepartmentId: department_id,
  328. Sex: sex,
  329. Age: age,
  330. Nation: nation,
  331. CardType: card_type,
  332. IdCard: idCard,
  333. Education: education,
  334. StudyMajorName: studyMajorName,
  335. WorkMajorName: workMajorName,
  336. RoleType: roletype,
  337. MedicalCode: medicalCode,
  338. DoctorCode: doctorcode,
  339. Licensing: licensing,
  340. JobNumber: jobnumber,
  341. PharmacistRegistrationNumber: pharmacistRegistrationNumber,
  342. IdentificationOutpatients: identificationOutpatients,
  343. StartTime: startime,
  344. MedicalRangeCode: medicalRangeCode,
  345. MedicalLevel: medicalLevel,
  346. MedicalTypeJob: medicalTypeJob,
  347. DoctorNumber: doctorNumber,
  348. OutpatientIllnesscategory: outpatientIllnessCategory,
  349. IsActive: isActive,
  350. ActiveStatus: activeStatus,
  351. PrescriptionQualificationIdentification: prescriptionQualificationIdentification,
  352. DoctorRangeCode: doctorRangeCode,
  353. DoctorTypeJob: doctorTypeJob,
  354. DoctorLevel: doctorLevel,
  355. }
  356. if createApp_RoleErr := tx.Create(&app_role).Error; createApp_RoleErr != nil {
  357. tx.Rollback()
  358. return nil, "", createApp_RoleErr
  359. }
  360. tx.Commit()
  361. return &adminUser, password, nil
  362. }
  363. func CreateGeneralAdminUser(orgID int64, appID int64, mobile string, name string, userTitle string, roleIds string, user_type int, user_title int, department_id int64, deapartment_name string) (*models.AdminUser, string, error) {
  364. now := time.Now().Unix()
  365. tx := writeUserDb.Begin()
  366. var adminUser models.AdminUser
  367. err := readUserDb.Where("mobile = ? AND status = 1", mobile).First(&adminUser).Error
  368. password := ""
  369. if err != nil {
  370. if err != gorm.ErrRecordNotFound {
  371. return nil, "", err
  372. } else {
  373. password = utils.RandomNumberString(6)
  374. adminUser.Mobile = mobile
  375. adminUser.Password = utils.String2md5(password)
  376. adminUser.IsSuperAdmin = false
  377. adminUser.Status = 1
  378. adminUser.CreateTime = now
  379. adminUser.ModifyTime = now
  380. if createErr := tx.Create(&adminUser).Error; createErr != nil {
  381. tx.Rollback()
  382. return nil, "", createErr
  383. }
  384. }
  385. }
  386. app_role := models.App_Role{
  387. AdminUserId: adminUser.Id,
  388. OrgId: orgID,
  389. AppId: appID,
  390. Avatar: "",
  391. UserName: name,
  392. UserTitleName: userTitle,
  393. UserTitle: int8(user_title),
  394. UserType: int8(user_type),
  395. Status: 1,
  396. CreateTime: now,
  397. ModifyTime: now,
  398. RoleIds: roleIds,
  399. IsSort: 1,
  400. Department: deapartment_name,
  401. DepartmentId: department_id,
  402. }
  403. if createApp_RoleErr := tx.Create(&app_role).Error; createApp_RoleErr != nil {
  404. tx.Rollback()
  405. return nil, "", createApp_RoleErr
  406. }
  407. tx.Commit()
  408. return &adminUser, password, nil
  409. }
  410. type PurviewTreeViewModel struct {
  411. ID int64 `json:"id"`
  412. PID int64 `json:"pid"`
  413. Name string `json:"name"`
  414. Childs []*PurviewTreeViewModel `json:"childs"`
  415. Number int64 `json:"number"`
  416. Url string `json:"url"`
  417. ParentUrl string `json:"parent_url"`
  418. }
  419. func GetAllGeneralPurviewVMsProcessed(module int) ([]*PurviewTreeViewModel, error) {
  420. var originPurviews []*models.Purview
  421. getPurviewErr := readUserDb.Model(models.Purview{}).Where("module = ? AND status = 1", module).Order("listorder asc").Order("id asc").Find(&originPurviews).Error
  422. if getPurviewErr != nil {
  423. return nil, getPurviewErr
  424. }
  425. // 加工这些规则:树形化
  426. purviewVMs := make([]*PurviewTreeViewModel, 0)
  427. pid_childs := make(map[int64][]*PurviewTreeViewModel)
  428. for _, purview := range originPurviews {
  429. // warning:下面这个算法只适用最多两层树形结构的菜单,对于两层以上的会丢失掉第三层及其以下的节点
  430. // 因为取出 originPurviews 的时候已经排过序了,所以顶级节点肯定最先处理,不需要担心子节点比父节点先处理
  431. pvm := &PurviewTreeViewModel{
  432. ID: purview.Id,
  433. PID: purview.Parentid,
  434. Name: purview.Name,
  435. Number: 2,
  436. }
  437. if purview.Parentid == 0 {
  438. purviewVMs = append(purviewVMs, pvm)
  439. } else {
  440. childs := pid_childs[purview.Parentid]
  441. if childs == nil {
  442. childs = make([]*PurviewTreeViewModel, 0)
  443. }
  444. childs = append(childs, pvm)
  445. pid_childs[purview.Parentid] = childs
  446. }
  447. }
  448. for _, vm := range purviewVMs {
  449. vm.Childs = pid_childs[vm.ID]
  450. }
  451. return purviewVMs, nil
  452. }
  453. func GetAllGeneralFuncPurviewVMsProcessed() ([]*PurviewTreeViewModel, error) {
  454. var originPurviews []*models.SgjUserOperatePurview
  455. getPurviewErr := readUserDb.Model(models.SgjUserOperatePurview{}).Where(" status = 1").Order("number asc").Find(&originPurviews).Error
  456. if getPurviewErr != nil {
  457. return nil, getPurviewErr
  458. }
  459. // 加工这些规则:树形化
  460. purviewVMs := make([]*PurviewTreeViewModel, 0)
  461. for _, purview := range originPurviews {
  462. // warning:下面这个算法只适用最多两层树形结构的菜单,对于两层以上的会丢失掉第三层及其以下的节点
  463. // 因为取出 originPurviews 的时候已经排过序了,所以顶级节点肯定最先处理,不需要担心子节点比父节点先处理
  464. pvm := &PurviewTreeViewModel{
  465. ID: purview.ID,
  466. PID: purview.Parentid,
  467. Name: purview.Name,
  468. }
  469. purviewVMs = append(purviewVMs, pvm)
  470. }
  471. return purviewVMs, nil
  472. }
  473. func GetRolePurviewIds(roleID int64) (string, error) {
  474. var rolePurview models.RolePurview
  475. err := readUserDb.Where("role_id = ?", roleID).First(&rolePurview).Error
  476. if err != nil {
  477. if err == gorm.ErrRecordNotFound {
  478. return "", nil
  479. } else {
  480. return "", err
  481. }
  482. }
  483. return rolePurview.PurviewIds, nil
  484. }
  485. func SaveRolePurviewIds(orgID int64, appID int64, roleID int64, purviewIds string) error {
  486. var rolePurview models.RolePurview
  487. getRPErr := readUserDb.Where("org_id = ? AND app_id = ? AND role_id = ?", orgID, appID, roleID).First(&rolePurview).Error
  488. if getRPErr != nil {
  489. if getRPErr == gorm.ErrRecordNotFound {
  490. rolePurview = models.RolePurview{
  491. RoleId: roleID,
  492. OrgId: orgID,
  493. AppId: appID,
  494. Status: 1,
  495. CreateTime: time.Now().Unix(),
  496. }
  497. } else {
  498. return getRPErr
  499. }
  500. }
  501. rolePurview.PurviewIds = purviewIds
  502. rolePurview.ModifyTime = time.Now().Unix()
  503. tx := writeUserDb.Begin()
  504. if err := tx.Save(&rolePurview).Error; err != nil {
  505. tx.Rollback()
  506. return err
  507. }
  508. return tx.Commit().Error
  509. }
  510. func SaveFuncRolePurviewIds(orgID int64, appID int64, roleID int64, funcPurviewIds string) error {
  511. var rolePurview models.SgjUserRoleFuncPurview
  512. getRPErr := readUserDb.Where("org_id = ? AND app_id = ? AND role_id = ?", orgID, appID, roleID).First(&rolePurview).Error
  513. if getRPErr != nil {
  514. if getRPErr == gorm.ErrRecordNotFound {
  515. rolePurview = models.SgjUserRoleFuncPurview{
  516. RoleId: roleID,
  517. OrgId: orgID,
  518. AppId: appID,
  519. Status: 1,
  520. Ctime: time.Now().Unix(),
  521. }
  522. } else {
  523. return getRPErr
  524. }
  525. }
  526. rolePurview.PurviewIds = funcPurviewIds
  527. rolePurview.Mtime = time.Now().Unix()
  528. tx := writeUserDb.Begin()
  529. if err := tx.Save(&rolePurview).Error; err != nil {
  530. tx.Rollback()
  531. return err
  532. }
  533. return tx.Commit().Error
  534. }
  535. func ModifyAdminUserInfo(adminUserID int64, orgID int64, appID int64, name string, avatar string, newPassword string) error {
  536. tx := writeUserDb.Begin()
  537. editInfoErr := tx.Exec("update sgj_user_admin_role set user_name = ?, avatar = ?, mtime = ? where admin_user_id = ? and org_id = ?", name, avatar, time.Now().Unix(), adminUserID, orgID).Error
  538. if editInfoErr != nil {
  539. tx.Rollback()
  540. return editInfoErr
  541. }
  542. if len(newPassword) > 0 {
  543. editPwdErr := tx.Exec("update sgj_user_admin set password = ?, mtime = ? where id = ?", newPassword, time.Now().Unix(), adminUserID).Error
  544. if editPwdErr != nil {
  545. tx.Rollback()
  546. return editPwdErr
  547. }
  548. }
  549. tx.Commit()
  550. return nil
  551. }
  552. func GetOrgApp(orgID int64, app_type int) (*models.OrgApp, error) {
  553. var apps models.OrgApp
  554. err := readUserDb.Where("app_type = ? AND org_id = ? AND status = 1", app_type, orgID).First(&apps).Error
  555. if err != nil {
  556. return nil, err
  557. }
  558. return &apps, nil
  559. }
  560. func GetAllOrgValidRoles(orgID int64, isSubSuperAdmin bool) ([]*models.Role, error) {
  561. var roles []*models.Role
  562. db := readUserDb.Model(models.Org{})
  563. if isSubSuperAdmin {
  564. db = db.Where("role_name != '子管理员' AND is_system != 1")
  565. }
  566. err := db.Where("org_id = ? AND status = 1", orgID).
  567. Order("number desc,ctime").
  568. Find(&roles).
  569. Error
  570. if err != nil {
  571. if err == gorm.ErrRecordNotFound {
  572. return make([]*models.Role, 0), nil
  573. } else {
  574. return nil, err
  575. }
  576. }
  577. for _, item := range roles {
  578. var total int64
  579. readUserDb.Model(&models.App_Role{}).Where("org_id = ? AND find_in_set(?, role_ids)", orgID, item.Id).Count(&total)
  580. item.StaffNumber = total
  581. }
  582. return roles, nil
  583. }
  584. func CreateUserRole(userRole *models.App_Role) (err error) {
  585. err = writeUserDb.Create(&userRole).Error
  586. return
  587. }
  588. func CreateOrgRole(role *models.Role) (err error) {
  589. err = writeUserDb.Create(&role).Error
  590. return
  591. }
  592. func GetRolePurview(roleID int64) (models.RolePurview, error) {
  593. var rolePurviews models.RolePurview
  594. var err error
  595. err = readUserDb.Where("role_id = ?", roleID).First(&rolePurviews).Error
  596. return rolePurviews, err
  597. }
  598. func GetFuncRolePurview(roleID int64) (models.SgjUserRoleFuncPurview, error) {
  599. var rolePurviews models.SgjUserRoleFuncPurview
  600. var err error
  601. err = readUserDb.Where("role_id = ?", roleID).First(&rolePurviews).Error
  602. return rolePurviews, err
  603. }
  604. func SaveRolePurview(purview *models.RolePurview) (err error) {
  605. err = writeUserDb.Save(&purview).Error
  606. return
  607. }
  608. func CreateRolePurview(purview *models.RolePurview) (err error) {
  609. err = writeUserDb.Create(&purview).Error
  610. return
  611. }
  612. func CreateFuncRolePurview(purview *models.SgjUserRoleFuncPurview) (err error) {
  613. err = writeUserDb.Create(&purview).Error
  614. return
  615. }
  616. func GetAllPurview(module int64, pid int64) ([]*models.Purview, error) {
  617. var originPurviews []*models.Purview
  618. getPurviewErr := readUserDb.Model(models.Purview{}).Preload("Purview", "status = 1").Where("module = ? AND status = 1 AND parentid = ? AND super_admin_exclusive = 0", module, pid).Order("listorder asc").Order("id asc").Find(&originPurviews).Error
  619. return originPurviews, getPurviewErr
  620. }
  621. type NewAdminUserModel struct {
  622. ID int64 `gorm:"id" json:"id"`
  623. AdminUserId int `gorm:"admin_user_id" json:"user_id"`
  624. UserName string `gorm:"user_name" json:"user_name"`
  625. UserTitle int `gorm:"user_title_name" json:"user_title_name"`
  626. Ip string `gorm:"ip" json:"ip"`
  627. Ctime int64 `gorm:"ctime" json:"last_login_time"`
  628. Status int `gorm:"status" json:"status"`
  629. Avatar string `gorm:"avatar" json:"avatar"`
  630. RoleIds string `gorm:"role_ids" json:"role_ids"`
  631. IsSubAdmin bool `gorm:"-" json:"is_sub_admin"`
  632. }
  633. func GetAllAdminUsersAndRole(orgID int64, appID int64, page int, count int) ([]*NewAdminUserModel, int, error) {
  634. var viewModels []*NewAdminUserModel = make([]*NewAdminUserModel, 0)
  635. var rows *sql.Rows
  636. var err error
  637. rows, err = readUserDb.Raw("SELECT u_a_r.id, u_a_r.admin_user_id, u_a_r.user_name, u_a_r.user_title, u_a_r.status,u_a_r.avatar,u_a_r.role_ids FROM sgj_user_admin_role AS u_a_r WHERE org_id = ? AND app_id = ? ORDER BY ctime asc", orgID, appID).Rows()
  638. defer rows.Close()
  639. if err != nil {
  640. if err == gorm.ErrRecordNotFound {
  641. return viewModels, 0, nil
  642. } else {
  643. return nil, 0, err
  644. }
  645. }
  646. for rows.Next() {
  647. var viewModel NewAdminUserModel
  648. readUserDb.ScanRows(rows, &viewModel)
  649. viewModels = append(viewModels, &viewModel)
  650. }
  651. for _, items := range viewModels {
  652. if len(items.RoleIds) > 0 {
  653. ids := strings.Split(items.RoleIds, ",")
  654. for _, ids := range ids {
  655. id, _ := strconv.ParseInt(ids, 10, 64)
  656. if id != 0 {
  657. role, _ := GetRoleByRoleID(id)
  658. if role != nil {
  659. if role.IsSystem == 1 && role.RoleName == "子管理员" {
  660. items.IsSubAdmin = true
  661. }
  662. }
  663. }
  664. }
  665. }
  666. }
  667. total := 0
  668. return viewModels, total, nil
  669. }
  670. func FindAdminUserByID(admin_user_id int64, org_id int64) (total int64, err error) {
  671. err = readUserDb.Model(&models.App_Role{}).Where("admin_user_id = ? AND org_id = ?", admin_user_id, org_id).Count(&total).Error
  672. return
  673. }
  674. func GetRoleAndPurviewById(role_id int64, org_id int64, app_id int64) (rolePurview models.RolePurview, err error) {
  675. err = readUserDb.Model(&models.RolePurview{}).Preload("Role", "status = 1").Where("role_id = ? AND status = 1 AND org_id = ? AND app_id = ?", role_id, org_id, app_id).First(&rolePurview).Error
  676. return
  677. }
  678. func SaveRole(role *models.Role) (err error) {
  679. err = writeUserDb.Save(&role).Error
  680. return
  681. }
  682. // 拥有xx角色的管理员的数量
  683. func RoleAdminUserCountTwo(orgID int64, appID int64, roleID int64) (int, error) {
  684. var count int
  685. err := readUserDb.Model(models.App_Role{}).
  686. Where("org_id = ? AND app_id = ? AND find_in_set(?, role_ids)", orgID, appID, roleID).
  687. Count(&count).
  688. Error
  689. if err != nil {
  690. return 0, err
  691. } else {
  692. return count, nil
  693. }
  694. }
  695. func FindAdminUserID(id int64) (role models.App_Role, err error) {
  696. err = readUserDb.Model(&models.App_Role{}).Where("id = ?", id).First(&role).Error
  697. return
  698. }
  699. func SaveAdminUser(role *models.App_Role) (err error) {
  700. err = writeUserDb.Save(&role).Error
  701. return
  702. }
  703. func GetFunctionPurview(pid int64) (purview []*models.SgjUserOperatePurview, err error) {
  704. err = readUserDb.Model(&models.SgjUserOperatePurview{}).Where("status = 1 AND parentid = ?", pid).Order("number asc").Find(&purview).Error
  705. return
  706. }
  707. func GetRoleFuncPurviewIds(roleID int64) (string, error) {
  708. var rolePurview models.SgjUserRoleFuncPurview
  709. err := readUserDb.Where("role_id = ?", roleID).First(&rolePurview).Error
  710. if err != nil {
  711. if err == gorm.ErrRecordNotFound {
  712. return "", nil
  713. } else {
  714. return "", err
  715. }
  716. }
  717. return rolePurview.PurviewIds, nil
  718. }
  719. func GetRoleFuncPurview(roleID int64) (rolePurviews models.SgjUserRoleFuncPurview, err error) {
  720. err = readUserDb.Where("role_id = ?", roleID).First(&rolePurviews).Error
  721. return
  722. }
  723. func SaveRoleFuncPurview(purview *models.SgjUserRoleFuncPurview) (err error) {
  724. err = writeUserDb.Save(purview).Error
  725. return
  726. }
  727. func GetUserAllRole(org_id int64, admin_user_id int64) (role models.App_Role, err error) {
  728. redis := RedisClient()
  729. defer redis.Close()
  730. // cur_date := time.Now().Format("2006-01-02")
  731. key := "user_all_role:" + strconv.FormatInt(org_id, 10) + ":"+ strconv.FormatInt(admin_user_id, 10)
  732. role_str, _ := redis.Get(key).Result()
  733. if len(role_str) == 0 { //没有到缓存数据,从数据库中获取数据,进行缓存到redis
  734. err = readUserDb.Model(&models.App_Role{}).Where("status = 1 AND org_id = ? AND admin_user_id = ?", org_id, admin_user_id).First(&role).Error
  735. if err != nil {
  736. if err == gorm.ErrRecordNotFound {
  737. return role, nil
  738. } else {
  739. return role, err
  740. }
  741. } else {
  742. if role.Id > 0 {
  743. //缓存数据
  744. role_json, err := json.Marshal(role)
  745. if err == nil {
  746. redis.Set(key, role_json, time.Second*60*60*18)
  747. }
  748. }
  749. return role, nil
  750. }
  751. } else { //缓存数据了数据,将redis缓存的json字符串转为map
  752. if err := json.Unmarshal([]byte(role_str), &role); err == nil {
  753. } else {
  754. }
  755. return role, nil
  756. }
  757. }
  758. func GetAllFunctionPurview() (purview []models.SgjUserOperatePurview, err error) {
  759. err = readUserDb.Model(&models.SgjUserOperatePurview{}).Where("status = 1 ").Order("number asc").Find(&purview).Error
  760. return
  761. }
  762. func FindErrorMsgByStr(str string) (string, error) {
  763. var purview models.SgjUserOperatePurview
  764. var err error
  765. err = readUserDb.Model(&models.SgjUserOperatePurview{}).Where("find_in_set(?, urlfor)", str).First(&purview).Error
  766. return purview.ErrorMsg, err
  767. }
  768. func GetNewAllOrgValidRoles(orgID int64) ([]*models.Role, error) {
  769. var roles []*models.Role
  770. db := readUserDb.Model(models.Org{})
  771. err := db.Where("org_id = ? AND status = 1", orgID).
  772. Order("number desc,ctime").
  773. Find(&roles).
  774. Error
  775. if err != nil {
  776. if err == gorm.ErrRecordNotFound {
  777. return make([]*models.Role, 0), nil
  778. } else {
  779. return nil, err
  780. }
  781. }
  782. for _, item := range roles {
  783. var total int64
  784. readUserDb.Model(&models.App_Role{}).Where("org_id = ? AND find_in_set(?, role_ids)", orgID, item.Id).Count(&total)
  785. item.StaffNumber = total
  786. }
  787. return roles, nil
  788. }
  789. func FindRoleRecordByRoleName(name string, org_id int64) (total int64) {
  790. readUserDb.Model(&models.Role{}).Where("status = 1 AND role_name = ? AND org_id = ?", name, org_id).Count(&total)
  791. return
  792. }
  793. func FindAllFuncPurview(ids []string) (purview []*models.SgjUserOperatePurview, err error) {
  794. err = readUserDb.Model(&models.SgjUserOperatePurview{}).Where("status = 1 AND id in (?)", ids).Order("number asc").Find(&purview).Error
  795. return
  796. }
  797. func GetSystemPurview() (purview []*models.Purview, err error) {
  798. err = readUserDb.Model(&models.Purview{}).Where("status = 1 AND module = 3 AND super_admin_exclusive = 1").Find(&purview).Error
  799. return
  800. }
  801. func GetAllSystemPurview() (purview []*models.Purview, err error) {
  802. err = readUserDb.Model(&models.Purview{}).Where("status = 1 AND module = 3 ").Find(&purview).Error
  803. return
  804. }
  805. type Purview struct {
  806. Id int64 `gorm:"PRIMARY_KEY;AUTO_INCREMENT" json:"id"`
  807. Parentid int64 `json:"pid"` // 父级ID
  808. Module int8 `json:"module"` // 所属应用类型 1.病友经营;2.科普号;3.血透经营;4.慢病管理;
  809. Name string `json:"name"` // 规则名称(也是菜单名)
  810. Urlfor string `json:"urlfor"` // 规则唯一英文标识,(Controller.Method)
  811. MenuIconClass string `gorm:"menu_icon_class" json:"menu_icon_class"` // 菜单图标的 css 样式
  812. SuperAdminExclusive int8 `gorm:"super_admin_exclusive" json:"super_admin_exclusive"` // 是否为超级管理员专属 0否1是
  813. Listorder int `json:"-"` // 排序ID
  814. Status int8 `json:"status"` // 状态 0.无效 1.有效 2.禁用
  815. CreateTime int64 `gorm:"column:ctime" json:"-"` // 创建时间
  816. ModifyTime int64 `gorm:"column:mtime" json:"-"` // 修改时间
  817. Purview []*Purview `gorm:"ForeignKey:Parentid;AssociationForeignKey:ID" json:"sub_purview"`
  818. IconUrl string `gorm:"column:icon_url" json:"icon_url" form:"icon_url"`
  819. Desc string `gorm:"column:desc" json:"desc" form:"desc"`
  820. ParentUrl string `gorm:"column:parent_url" json:"parent_url" form:"parent_url"`
  821. Link string `gorm:"-" json:"link"` // urlfor 解析后的链接
  822. Childs []*Purview `gorm:"-" json:"childs"` // 子节点
  823. }
  824. func (Purview) TableName() string {
  825. return "sgj_user_purview"
  826. }
  827. func GetNewAllPurview(url string) ([]*Purview, error) {
  828. var originPurviews []*Purview
  829. getPurviewErr := readUserDb.Model(Purview{}).Preload("Purview", "status = 1").Where("parent_url = ? AND status = 1", url).Order("listorder asc").Order("id asc").Find(&originPurviews).Error
  830. return originPurviews, getPurviewErr
  831. }
  832. func GetOtherNewAllPurview(url string) ([]*Purview, error) {
  833. var originPurviews []*Purview
  834. getPurviewErr := readUserDb.Model(Purview{}).Preload("Purview", func(db *gorm.DB) *gorm.DB {
  835. return db.Preload("Purview", "status = 1").Where("status = 1")
  836. }).Where("parent_url = ? AND status = 1", url).Order("listorder asc").Order("id asc").Find(&originPurviews).Error
  837. return originPurviews, getPurviewErr
  838. }
  839. func GetAllFuncPurview() ([]*models.SgjUserOperatePurview, error) {
  840. var originPurviews []*models.SgjUserOperatePurview
  841. getPurviewErr := readUserDb.Model(&models.SgjUserOperatePurview{}).Where(" status = 1").Error
  842. return originPurviews, getPurviewErr
  843. }
  844. func GetOtherAllGeneralPurviewVMsProcessed(url string) ([]*PurviewTreeViewModel, error, []*models.Purview) {
  845. var originPurviews []*models.Purview
  846. getPurviewErr := readUserDb.Model(Purview{}).Where("status = 1 AND parent_url = ?", url).Order("listorder asc").Order("id asc").Find(&originPurviews).Error
  847. if getPurviewErr != nil {
  848. return nil, getPurviewErr, nil
  849. }
  850. // 加工这些规则:树形化
  851. purviewVMs := make([]*PurviewTreeViewModel, 0)
  852. pid_childs := make(map[int64][]*PurviewTreeViewModel)
  853. fmt.Println(originPurviews)
  854. for _, purview := range originPurviews {
  855. // warning:下面这个算法只适用最多两层树形结构的菜单,对于两层以上的会丢失掉第三层及其以下的节点
  856. // 因为取出 originPurviews 的时候已经排过序了,所以顶级节点肯定最先处理,不需要担心子节点比父节点先处理
  857. pvm := &PurviewTreeViewModel{
  858. ID: purview.Id,
  859. PID: purview.Parentid,
  860. Name: purview.Name,
  861. Number: 2,
  862. ParentUrl: purview.ParentUrl,
  863. Url: purview.Urlfor,
  864. }
  865. if purview.Parentid == 0 {
  866. purviewVMs = append(purviewVMs, pvm)
  867. } else {
  868. childs := pid_childs[purview.Parentid]
  869. if childs == nil {
  870. childs = make([]*PurviewTreeViewModel, 0)
  871. }
  872. childs = append(childs, pvm)
  873. pid_childs[purview.Parentid] = childs
  874. }
  875. }
  876. for _, vm := range purviewVMs {
  877. vm.Childs = pid_childs[vm.ID]
  878. }
  879. return purviewVMs, nil, originPurviews
  880. }