role_service.go 33KB

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