role_service.go 34KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991
  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. //用来处理树形结构
  455. func Totree(map1 map[int64]int64, map2 map[int64]*PurviewTreeViewModel, tmp int64) []*PurviewTreeViewModel {
  456. var j int //用来计数
  457. var tmp_purview []*PurviewTreeViewModel
  458. for k, v := range map1 {
  459. if v == tmp {
  460. tmp_purview = append(tmp_purview, map2[k])
  461. delete(map1, k)
  462. for k1, v1 := range tmp_purview {
  463. if k1 == j {
  464. v1.Childs = Totree(map1, map2, k)
  465. }
  466. }
  467. j++
  468. }
  469. }
  470. return tmp_purview
  471. }
  472. func GetAllGeneralFuncPurviewVMsProcessed() ([]*PurviewTreeViewModel, error) {
  473. var originPurviews []*models.SgjUserOperatePurview
  474. getPurviewErr := readUserDb.Model(models.SgjUserOperatePurview{}).Where(" status = 1").Order("number asc").Find(&originPurviews).Error
  475. if getPurviewErr != nil {
  476. return nil, getPurviewErr
  477. }
  478. // 加工这些规则:树形化
  479. purviewVMs := make([]*PurviewTreeViewModel, 0)
  480. for _, purview := range originPurviews {
  481. // warning:下面这个算法只适用最多两层树形结构的菜单,对于两层以上的会丢失掉第三层及其以下的节点
  482. // 因为取出 originPurviews 的时候已经排过序了,所以顶级节点肯定最先处理,不需要担心子节点比父节点先处理
  483. pvm := &PurviewTreeViewModel{
  484. ID: purview.ID,
  485. PID: purview.Parentid,
  486. Name: purview.Name,
  487. }
  488. purviewVMs = append(purviewVMs, pvm)
  489. }
  490. return purviewVMs, nil
  491. }
  492. func GetRolePurviewIds(roleID int64) (string, error) {
  493. var rolePurview models.RolePurview
  494. err := readUserDb.Where("role_id = ?", roleID).First(&rolePurview).Error
  495. if err != nil {
  496. if err == gorm.ErrRecordNotFound {
  497. return "", nil
  498. } else {
  499. return "", err
  500. }
  501. }
  502. return rolePurview.PurviewIds, nil
  503. }
  504. func SaveRolePurviewIds(orgID int64, appID int64, roleID int64, purviewIds string) error {
  505. var rolePurview models.RolePurview
  506. getRPErr := readUserDb.Where("org_id = ? AND app_id = ? AND role_id = ?", orgID, appID, roleID).First(&rolePurview).Error
  507. if getRPErr != nil {
  508. if getRPErr == gorm.ErrRecordNotFound {
  509. rolePurview = models.RolePurview{
  510. RoleId: roleID,
  511. OrgId: orgID,
  512. AppId: appID,
  513. Status: 1,
  514. CreateTime: time.Now().Unix(),
  515. }
  516. } else {
  517. return getRPErr
  518. }
  519. }
  520. rolePurview.PurviewIds = purviewIds
  521. rolePurview.ModifyTime = time.Now().Unix()
  522. tx := writeUserDb.Begin()
  523. if err := tx.Save(&rolePurview).Error; err != nil {
  524. tx.Rollback()
  525. return err
  526. }
  527. return tx.Commit().Error
  528. }
  529. func SaveFuncRolePurviewIds(orgID int64, appID int64, roleID int64, funcPurviewIds string) error {
  530. var rolePurview models.SgjUserRoleFuncPurview
  531. getRPErr := readUserDb.Where("org_id = ? AND app_id = ? AND role_id = ?", orgID, appID, roleID).First(&rolePurview).Error
  532. if getRPErr != nil {
  533. if getRPErr == gorm.ErrRecordNotFound {
  534. rolePurview = models.SgjUserRoleFuncPurview{
  535. RoleId: roleID,
  536. OrgId: orgID,
  537. AppId: appID,
  538. Status: 1,
  539. Ctime: time.Now().Unix(),
  540. }
  541. } else {
  542. return getRPErr
  543. }
  544. }
  545. rolePurview.PurviewIds = funcPurviewIds
  546. rolePurview.Mtime = time.Now().Unix()
  547. tx := writeUserDb.Begin()
  548. if err := tx.Save(&rolePurview).Error; err != nil {
  549. tx.Rollback()
  550. return err
  551. }
  552. return tx.Commit().Error
  553. }
  554. func ModifyAdminUserInfo(adminUserID int64, orgID int64, appID int64, name string, avatar string, newPassword string) error {
  555. tx := writeUserDb.Begin()
  556. 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
  557. if editInfoErr != nil {
  558. tx.Rollback()
  559. return editInfoErr
  560. }
  561. if len(newPassword) > 0 {
  562. editPwdErr := tx.Exec("update sgj_user_admin set password = ?, mtime = ? where id = ?", newPassword, time.Now().Unix(), adminUserID).Error
  563. if editPwdErr != nil {
  564. tx.Rollback()
  565. return editPwdErr
  566. }
  567. }
  568. tx.Commit()
  569. return nil
  570. }
  571. func GetOrgApp(orgID int64, app_type int) (*models.OrgApp, error) {
  572. var apps models.OrgApp
  573. err := readUserDb.Where("app_type = ? AND org_id = ? AND status = 1", app_type, orgID).First(&apps).Error
  574. if err != nil {
  575. return nil, err
  576. }
  577. return &apps, nil
  578. }
  579. func GetAllOrgValidRoles(orgID int64, isSubSuperAdmin bool) ([]*models.Role, error) {
  580. var roles []*models.Role
  581. db := readUserDb.Model(models.Org{})
  582. if isSubSuperAdmin {
  583. db = db.Where("role_name != '子管理员' AND is_system != 1")
  584. }
  585. err := db.Where("org_id = ? AND status = 1", orgID).
  586. Order("number desc,ctime").
  587. Find(&roles).
  588. Error
  589. if err != nil {
  590. if err == gorm.ErrRecordNotFound {
  591. return make([]*models.Role, 0), nil
  592. } else {
  593. return nil, err
  594. }
  595. }
  596. for _, item := range roles {
  597. var total int64
  598. readUserDb.Model(&models.App_Role{}).Where("org_id = ? AND find_in_set(?, role_ids)", orgID, item.Id).Count(&total)
  599. item.StaffNumber = total
  600. }
  601. return roles, nil
  602. }
  603. func CreateUserRole(userRole *models.App_Role) (err error) {
  604. err = writeUserDb.Create(&userRole).Error
  605. return
  606. }
  607. func CreateOrgRole(role *models.Role) (err error) {
  608. err = writeUserDb.Create(&role).Error
  609. return
  610. }
  611. func GetRolePurview(roleID int64) (models.RolePurview, error) {
  612. var rolePurviews models.RolePurview
  613. var err error
  614. err = readUserDb.Where("role_id = ?", roleID).First(&rolePurviews).Error
  615. return rolePurviews, err
  616. }
  617. func GetFuncRolePurview(roleID int64) (models.SgjUserRoleFuncPurview, error) {
  618. var rolePurviews models.SgjUserRoleFuncPurview
  619. var err error
  620. err = readUserDb.Where("role_id = ?", roleID).First(&rolePurviews).Error
  621. return rolePurviews, err
  622. }
  623. func SaveRolePurview(purview *models.RolePurview) (err error) {
  624. err = writeUserDb.Save(&purview).Error
  625. return
  626. }
  627. func CreateRolePurview(purview *models.RolePurview) (err error) {
  628. err = writeUserDb.Create(&purview).Error
  629. return
  630. }
  631. func CreateFuncRolePurview(purview *models.SgjUserRoleFuncPurview) (err error) {
  632. err = writeUserDb.Create(&purview).Error
  633. return
  634. }
  635. func GetAllPurview(module int64, pid int64) ([]*models.Purview, error) {
  636. var originPurviews []*models.Purview
  637. 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
  638. return originPurviews, getPurviewErr
  639. }
  640. type NewAdminUserModel struct {
  641. ID int64 `gorm:"id" json:"id"`
  642. AdminUserId int `gorm:"admin_user_id" json:"user_id"`
  643. UserName string `gorm:"user_name" json:"user_name"`
  644. UserTitle int `gorm:"user_title_name" json:"user_title_name"`
  645. Ip string `gorm:"ip" json:"ip"`
  646. Ctime int64 `gorm:"ctime" json:"last_login_time"`
  647. Status int `gorm:"status" json:"status"`
  648. Avatar string `gorm:"avatar" json:"avatar"`
  649. RoleIds string `gorm:"role_ids" json:"role_ids"`
  650. IsSubAdmin bool `gorm:"-" json:"is_sub_admin"`
  651. }
  652. func GetAllAdminUsersAndRole(orgID int64, appID int64, page int, count int) ([]*NewAdminUserModel, int, error) {
  653. var viewModels []*NewAdminUserModel = make([]*NewAdminUserModel, 0)
  654. var rows *sql.Rows
  655. var err error
  656. 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()
  657. defer rows.Close()
  658. if err != nil {
  659. if err == gorm.ErrRecordNotFound {
  660. return viewModels, 0, nil
  661. } else {
  662. return nil, 0, err
  663. }
  664. }
  665. for rows.Next() {
  666. var viewModel NewAdminUserModel
  667. readUserDb.ScanRows(rows, &viewModel)
  668. viewModels = append(viewModels, &viewModel)
  669. }
  670. for _, items := range viewModels {
  671. if len(items.RoleIds) > 0 {
  672. ids := strings.Split(items.RoleIds, ",")
  673. for _, ids := range ids {
  674. id, _ := strconv.ParseInt(ids, 10, 64)
  675. if id != 0 {
  676. role, _ := GetRoleByRoleID(id)
  677. if role != nil {
  678. if role.IsSystem == 1 && role.RoleName == "子管理员" {
  679. items.IsSubAdmin = true
  680. }
  681. }
  682. }
  683. }
  684. }
  685. }
  686. total := 0
  687. return viewModels, total, nil
  688. }
  689. func FindAdminUserByID(admin_user_id int64, org_id int64) (total int64, err error) {
  690. err = readUserDb.Model(&models.App_Role{}).Where("admin_user_id = ? AND org_id = ?", admin_user_id, org_id).Count(&total).Error
  691. return
  692. }
  693. func GetRoleAndPurviewById(role_id int64, org_id int64, app_id int64) (rolePurview models.RolePurview, err error) {
  694. 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
  695. return
  696. }
  697. func SaveRole(role *models.Role) (err error) {
  698. err = writeUserDb.Save(&role).Error
  699. return
  700. }
  701. // 拥有xx角色的管理员的数量
  702. func RoleAdminUserCountTwo(orgID int64, appID int64, roleID int64) (int, error) {
  703. var count int
  704. err := readUserDb.Model(models.App_Role{}).
  705. Where("org_id = ? AND app_id = ? AND find_in_set(?, role_ids)", orgID, appID, roleID).
  706. Count(&count).
  707. Error
  708. if err != nil {
  709. return 0, err
  710. } else {
  711. return count, nil
  712. }
  713. }
  714. func FindAdminUserID(id int64) (role models.App_Role, err error) {
  715. err = readUserDb.Model(&models.App_Role{}).Where("id = ?", id).First(&role).Error
  716. return
  717. }
  718. func SaveAdminUser(role *models.App_Role) (err error) {
  719. err = writeUserDb.Save(&role).Error
  720. return
  721. }
  722. func GetFunctionPurview(pid int64) (purview []*models.SgjUserOperatePurview, err error) {
  723. err = readUserDb.Model(&models.SgjUserOperatePurview{}).Where("status = 1 AND parentid = ?", pid).Order("number asc").Find(&purview).Error
  724. return
  725. }
  726. func GetRoleFuncPurviewIds(roleID int64) (string, error) {
  727. var rolePurview models.SgjUserRoleFuncPurview
  728. err := readUserDb.Where("role_id = ?", roleID).First(&rolePurview).Error
  729. if err != nil {
  730. if err == gorm.ErrRecordNotFound {
  731. return "", nil
  732. } else {
  733. return "", err
  734. }
  735. }
  736. return rolePurview.PurviewIds, nil
  737. }
  738. func GetRoleFuncPurview(roleID int64) (rolePurviews models.SgjUserRoleFuncPurview, err error) {
  739. err = readUserDb.Where("role_id = ?", roleID).First(&rolePurviews).Error
  740. return
  741. }
  742. func SaveRoleFuncPurview(purview *models.SgjUserRoleFuncPurview) (err error) {
  743. err = writeUserDb.Save(purview).Error
  744. return
  745. }
  746. func GetUserAllRole(org_id int64, admin_user_id int64) (role models.App_Role, err error) {
  747. redis := RedisClient()
  748. defer redis.Close()
  749. // cur_date := time.Now().Format("2006-01-02")
  750. key := "user_all_role:" + strconv.FormatInt(org_id, 10) + ":" + strconv.FormatInt(admin_user_id, 10)
  751. role_str, _ := redis.Get(key).Result()
  752. if len(role_str) == 0 { //没有到缓存数据,从数据库中获取数据,进行缓存到redis
  753. err = readUserDb.Model(&models.App_Role{}).Where("status = 1 AND org_id = ? AND admin_user_id = ?", org_id, admin_user_id).First(&role).Error
  754. if err != nil {
  755. if err == gorm.ErrRecordNotFound {
  756. return role, nil
  757. } else {
  758. return role, err
  759. }
  760. } else {
  761. if role.Id > 0 {
  762. //缓存数据
  763. role_json, err := json.Marshal(role)
  764. if err == nil {
  765. redis.Set(key, role_json, time.Second*60*60*18)
  766. }
  767. }
  768. return role, nil
  769. }
  770. } else { //缓存数据了数据,将redis缓存的json字符串转为map
  771. if err := json.Unmarshal([]byte(role_str), &role); err == nil {
  772. } else {
  773. }
  774. return role, nil
  775. }
  776. }
  777. func GetAllFunctionPurview() (purview []models.SgjUserOperatePurview, err error) {
  778. err = readUserDb.Model(&models.SgjUserOperatePurview{}).Where("status = 1 ").Order("number asc").Find(&purview).Error
  779. return
  780. }
  781. func FindErrorMsgByStr(str string) (string, error) {
  782. var purview models.SgjUserOperatePurview
  783. var err error
  784. err = readUserDb.Model(&models.SgjUserOperatePurview{}).Where("find_in_set(?, urlfor)", str).First(&purview).Error
  785. return purview.ErrorMsg, err
  786. }
  787. func GetNewAllOrgValidRoles(orgID int64) ([]*models.Role, error) {
  788. var roles []*models.Role
  789. db := readUserDb.Model(models.Org{})
  790. err := db.Where("org_id = ? AND status = 1", orgID).
  791. Order("number desc,ctime").
  792. Find(&roles).
  793. Error
  794. if err != nil {
  795. if err == gorm.ErrRecordNotFound {
  796. return make([]*models.Role, 0), nil
  797. } else {
  798. return nil, err
  799. }
  800. }
  801. for _, item := range roles {
  802. var total int64
  803. readUserDb.Model(&models.App_Role{}).Where("org_id = ? AND find_in_set(?, role_ids)", orgID, item.Id).Count(&total)
  804. item.StaffNumber = total
  805. }
  806. return roles, nil
  807. }
  808. func FindRoleRecordByRoleName(name string, org_id int64) (total int64) {
  809. readUserDb.Model(&models.Role{}).Where("status = 1 AND role_name = ? AND org_id = ?", name, org_id).Count(&total)
  810. return
  811. }
  812. func FindAllFuncPurview(ids []string) (purview []*models.SgjUserOperatePurview, err error) {
  813. err = readUserDb.Model(&models.SgjUserOperatePurview{}).Where("status = 1 AND id in (?)", ids).Order("number asc").Find(&purview).Error
  814. return
  815. }
  816. func GetSystemPurview() (purview []*models.Purview, err error) {
  817. err = readUserDb.Model(&models.Purview{}).Where("status = 1 AND module = 3 AND super_admin_exclusive = 1").Find(&purview).Error
  818. return
  819. }
  820. func GetAllSystemPurview() (purview []*models.Purview, err error) {
  821. err = readUserDb.Model(&models.Purview{}).Where("status = 1 AND module = 3 ").Find(&purview).Error
  822. return
  823. }
  824. type Purview struct {
  825. Id int64 `gorm:"PRIMARY_KEY;AUTO_INCREMENT" json:"id"`
  826. Parentid int64 `json:"pid"` // 父级ID
  827. Module int8 `json:"module"` // 所属应用类型 1.病友经营;2.科普号;3.血透经营;4.慢病管理;
  828. Name string `json:"name"` // 规则名称(也是菜单名)
  829. Urlfor string `json:"urlfor"` // 规则唯一英文标识,(Controller.Method)
  830. MenuIconClass string `gorm:"menu_icon_class" json:"menu_icon_class"` // 菜单图标的 css 样式
  831. SuperAdminExclusive int8 `gorm:"super_admin_exclusive" json:"super_admin_exclusive"` // 是否为超级管理员专属 0否1是
  832. Listorder int `json:"-"` // 排序ID
  833. Status int8 `json:"status"` // 状态 0.无效 1.有效 2.禁用
  834. CreateTime int64 `gorm:"column:ctime" json:"-"` // 创建时间
  835. ModifyTime int64 `gorm:"column:mtime" json:"-"` // 修改时间
  836. Purview []*Purview `gorm:"ForeignKey:Parentid;AssociationForeignKey:ID" json:"sub_purview"`
  837. IconUrl string `gorm:"column:icon_url" json:"icon_url" form:"icon_url"`
  838. Desc string `gorm:"column:desc" json:"desc" form:"desc"`
  839. ParentUrl string `gorm:"column:parent_url" json:"parent_url" form:"parent_url"`
  840. Link string `gorm:"-" json:"link"` // urlfor 解析后的链接
  841. Childs []*Purview `gorm:"-" json:"childs"` // 子节点
  842. }
  843. func (Purview) TableName() string {
  844. return "sgj_user_purview"
  845. }
  846. func GetNewAllPurview(url string) ([]*Purview, error) {
  847. var originPurviews []*Purview
  848. getPurviewErr := readUserDb.Model(Purview{}).Preload("Purview", "status = 1").Where("parent_url = ? AND status = 1", url).Order("listorder asc").Order("id asc").Find(&originPurviews).Error
  849. return originPurviews, getPurviewErr
  850. }
  851. func GetOtherNewAllPurview(url string) ([]*Purview, error) {
  852. var originPurviews []*Purview
  853. getPurviewErr := readUserDb.Model(Purview{}).Preload("Purview", func(db *gorm.DB) *gorm.DB {
  854. return db.Preload("Purview", "status = 1").Where("status = 1")
  855. }).Where("parent_url = ? AND status = 1", url).Order("listorder asc").Order("id asc").Find(&originPurviews).Error
  856. return originPurviews, getPurviewErr
  857. }
  858. func GetAllFuncPurview() ([]*models.SgjUserOperatePurview, error) {
  859. var originPurviews []*models.SgjUserOperatePurview
  860. getPurviewErr := readUserDb.Model(&models.SgjUserOperatePurview{}).Where(" status = 1").Error
  861. return originPurviews, getPurviewErr
  862. }
  863. func GetOtherAllGeneralPurviewVMsProcessed(url string) ([]*PurviewTreeViewModel, error, []*models.Purview) {
  864. var originPurviews []*models.Purview
  865. getPurviewErr := readUserDb.Model(Purview{}).Where("status = 1 AND parent_url = ?", url).Order("listorder asc").Order("id asc").Find(&originPurviews).Error
  866. if getPurviewErr != nil {
  867. return nil, getPurviewErr, nil
  868. }
  869. // 加工这些规则:树形化
  870. purviewVMs := make([]*PurviewTreeViewModel, 0)
  871. pid_childs := make(map[int64][]*PurviewTreeViewModel)
  872. fmt.Println(originPurviews)
  873. for _, purview := range originPurviews {
  874. // warning:下面这个算法只适用最多两层树形结构的菜单,对于两层以上的会丢失掉第三层及其以下的节点
  875. // 因为取出 originPurviews 的时候已经排过序了,所以顶级节点肯定最先处理,不需要担心子节点比父节点先处理
  876. pvm := &PurviewTreeViewModel{
  877. ID: purview.Id,
  878. PID: purview.Parentid,
  879. Name: purview.Name,
  880. Number: 2,
  881. ParentUrl: purview.ParentUrl,
  882. Url: purview.Urlfor,
  883. }
  884. if purview.Parentid == 0 {
  885. purviewVMs = append(purviewVMs, pvm)
  886. } else {
  887. childs := pid_childs[purview.Parentid]
  888. if childs == nil {
  889. childs = make([]*PurviewTreeViewModel, 0)
  890. }
  891. childs = append(childs, pvm)
  892. pid_childs[purview.Parentid] = childs
  893. }
  894. }
  895. for _, vm := range purviewVMs {
  896. vm.Childs = pid_childs[vm.ID]
  897. }
  898. return purviewVMs, nil, originPurviews
  899. }