user.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "eta/eta_mini_api/models"
  5. "eta/eta_mini_api/models/request"
  6. "eta/eta_mini_api/models/response"
  7. "eta/eta_mini_api/services"
  8. "eta/eta_mini_api/services/alarm_msg"
  9. "eta/eta_mini_api/services/go_redis"
  10. "eta/eta_mini_api/utils"
  11. "fmt"
  12. "strings"
  13. "time"
  14. )
  15. type UserController struct {
  16. BaseCommonController
  17. }
  18. type UserAuthController struct {
  19. BaseAuthController
  20. }
  21. // @Title 用户登录接口
  22. // @Description 用户登录
  23. // @Param request body models.LoginReq true "type json string"
  24. // @Success 200 {object} models.LoginResp
  25. // @router /login [post]
  26. func (this *UserAuthController) Login() {
  27. br := new(models.BaseResponse).Init()
  28. defer func() {
  29. this.Data["json"] = br
  30. this.ServeJSON()
  31. }()
  32. session := this.Session
  33. var req request.LoginReq
  34. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  35. if err != nil {
  36. br.Msg = "参数解析失败"
  37. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  38. return
  39. }
  40. switch req.LoginType {
  41. case 1:
  42. if req.Phone == "" {
  43. br.Msg = "请输入手机号"
  44. br.ErrMsg = "请输入手机号"
  45. return
  46. }
  47. if req.SmsCode == "" {
  48. br.Msg = "请输入验证码"
  49. br.ErrMsg = "请输入验证码"
  50. return
  51. }
  52. phone := req.AreaCode + req.Phone
  53. item, err := models.GetMsgCode(phone, req.SmsCode)
  54. if err != nil {
  55. if err.Error() == utils.ErrNoRow() {
  56. br.Msg = "验证码错误,请重新输入"
  57. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  58. return
  59. } else {
  60. br.Msg = "验证码错误,请重新输入"
  61. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  62. return
  63. }
  64. }
  65. if item == nil {
  66. br.Msg = "验证码错误,请重新输入"
  67. return
  68. }
  69. case 2:
  70. if req.Email == "" {
  71. br.Msg = "请输入邮箱"
  72. br.ErrMsg = "请输入邮箱"
  73. return
  74. }
  75. if req.SmsCode == "" {
  76. br.Msg = "请输入验证码"
  77. br.ErrMsg = "请输入验证码"
  78. return
  79. }
  80. item, err := models.GetMsgCode(req.Email, req.SmsCode)
  81. if err != nil {
  82. if err.Error() == utils.ErrNoRow() {
  83. br.Msg = "验证码错误,请重新输入"
  84. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  85. return
  86. } else {
  87. br.Msg = "验证码错误,请重新输入"
  88. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  89. return
  90. }
  91. }
  92. if item == nil {
  93. br.Msg = "验证码错误,请重新输入"
  94. return
  95. }
  96. }
  97. _, errMsg, err := services.BindUser(session.UnionId, session.OpenId, req.Phone, req.Email, req.AreaCode)
  98. if err != nil && err.Error() != utils.ErrNoRow() {
  99. br.Msg = "登录失败,系统处理中,请稍后重试"
  100. if errMsg != "" {
  101. br.Msg = errMsg
  102. }
  103. br.ErrMsg = "登录失败:" + err.Error()
  104. return
  105. }
  106. br.Msg = "登录成功"
  107. br.Success = true
  108. br.Ret = 200
  109. }
  110. // @Title 获取短信/邮箱验证码
  111. // @Description 用户登录
  112. // @Param request body models.LoginReq true "type json string"
  113. // @Success 200 {object} models.LoginResp
  114. // @router /getVerifyCode [post]
  115. func (this *UserController) GetVerifyCode() {
  116. br := new(models.BaseResponse).Init()
  117. defer func() {
  118. if br.Ret != 200 {
  119. b, _ := json.Marshal(br)
  120. alarm_msg.SendAlarmMsg(string(b), 1)
  121. }
  122. this.Data["json"] = br
  123. this.ServeJSON()
  124. }()
  125. var req request.VerifyCodeReq
  126. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  127. if err != nil {
  128. br.Msg = "参数解析异常!"
  129. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  130. return
  131. }
  132. if req.VerifyType != 1 && req.VerifyType != 2 {
  133. br.Msg = "验证方式有误"
  134. br.ErrMsg = fmt.Sprintf("验证方式异常<%d>", req.VerifyType)
  135. }
  136. code := utils.GetRandDigit(6)
  137. fmt.Println(code)
  138. switch req.VerifyType {
  139. case 1:
  140. if req.AreaCode == "" {
  141. br.Msg = "请选择区号"
  142. return
  143. }
  144. if req.Phone == "" {
  145. br.Msg = "请输入手机号"
  146. return
  147. }
  148. if req.AreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Phone) {
  149. br.Msg = "您的手机号输入有误, 请检查"
  150. return
  151. }
  152. phoneKey := utils.CACHE_ACCESS_PHONE_LOGIN_CODE + req.AreaCode + req.Phone
  153. res, _ := go_redis.RedisInt(phoneKey)
  154. if res > 5 {
  155. br.Msg = "验证码发送太频繁,请稍后重试"
  156. return
  157. }
  158. var ok bool
  159. if req.AreaCode == "86" {
  160. ok = services.SendSmsCode(req.Phone, code)
  161. }
  162. if !ok {
  163. br.Msg = "请检查手机号和区号"
  164. br.ErrMsg = "短信验证码发送失败"
  165. return
  166. } else {
  167. item := new(models.MsgCode)
  168. item.OpenId = ""
  169. item.Code = code
  170. item.Mobile = req.AreaCode + req.Phone
  171. item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
  172. item.CreateTime = time.Now()
  173. err = item.Insert()
  174. if err != nil {
  175. br.Msg = "发送失败"
  176. br.ErrMsg = "发送失败,Err:" + err.Error()
  177. return
  178. }
  179. br.Msg = "发送成功"
  180. isExist := go_redis.IsExist(phoneKey)
  181. if isExist {
  182. go_redis.Incr(phoneKey)
  183. } else {
  184. go_redis.SetNX(phoneKey, 1, time.Minute*15)
  185. }
  186. }
  187. case 2:
  188. if req.Email == "" {
  189. br.Msg = "请输入邮箱"
  190. }
  191. if !utils.ValidateEmailFormatat(req.Email) {
  192. br.Msg = "您的邮箱格式输入有误, 请检查"
  193. return
  194. }
  195. emailKey := utils.CACHE_ACCESS_EMAIL_LOGIN_CODE + req.Email
  196. res, _ := go_redis.RedisInt(emailKey)
  197. if res > 5 {
  198. br.Msg = "验证码发送太频繁,请稍后重试"
  199. return
  200. }
  201. date := time.Now()
  202. content := "尊敬的用户:</br>本次请求的验证码为:" + code + "(为了保障您账号的安全性,请在15分钟内完成验证。)</br>东吴期货研究团队 </br>" + fmt.Sprintf("%d年%02d月%02d日", date.Year(), date.Month(), date.Day())
  203. title := "东吴期货登录验证"
  204. result, err := utils.SendEmailByDw(title, content, req.Email)
  205. if err != nil {
  206. br.Msg = "发送失败"
  207. br.ErrMsg = "发送失败,Err:" + err.Error()
  208. return
  209. }
  210. if result {
  211. item := new(models.MsgCode)
  212. item.OpenId = ""
  213. item.Code = code
  214. item.Mobile = req.Email
  215. item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
  216. item.CreateTime = time.Now()
  217. err = item.Insert()
  218. if err != nil {
  219. br.Msg = "发送失败"
  220. br.ErrMsg = "发送失败,Err:" + err.Error()
  221. return
  222. }
  223. br.Msg = "发送成功"
  224. isExist := go_redis.IsExist(emailKey)
  225. if isExist {
  226. go_redis.Incr(emailKey)
  227. } else {
  228. go_redis.SetNX(emailKey, 1, time.Minute*15)
  229. }
  230. } else {
  231. br.Msg = "发送失败"
  232. }
  233. }
  234. br.Ret = 200
  235. br.Success = true
  236. br.Msg = "发送成功"
  237. }
  238. // @Title 新增报告浏览记录
  239. // @Description 新增报告浏览记录接口
  240. // @Param request body models.ReportRecordReq true "type json string"
  241. // @Success 200 新增成功
  242. // @router /addReportRecord [post]
  243. func (this *UserAuthController) AddReportRecord() {
  244. br := new(models.BaseResponse).Init()
  245. defer func() {
  246. this.Data["json"] = br
  247. this.ServeJSON()
  248. }()
  249. user := this.User
  250. if user == nil {
  251. br.Msg = "请登录"
  252. br.ErrMsg = "请登录,用户信息为空"
  253. br.Ret = 403
  254. return
  255. }
  256. var req request.ReportRecordReq
  257. if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
  258. br.Msg = "参数解析异常!"
  259. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  260. return
  261. }
  262. if req.ReportId <= 0 {
  263. br.Msg = "参数错误"
  264. br.ErrMsg = "参数错误,报告id小于等于0"
  265. return
  266. }
  267. reportResp, err := services.GetReportDetail(req.ReportId, user.UserId)
  268. if err != nil {
  269. br.Msg = "添加阅读记录失败"
  270. br.ErrMsg = "获取研报错误,Err:" + err.Error()
  271. return
  272. }
  273. if reportResp.Ret != 200 {
  274. br.Msg = reportResp.Msg
  275. br.ErrMsg = reportResp.ErrMsg
  276. return
  277. }
  278. reportDetail := reportResp.Data.Report
  279. reportPermissionResp, err := services.GetReportChartPermissionList(req.ReportId)
  280. if err != nil {
  281. br.Msg = "添加阅读记录失败"
  282. br.ErrMsg = "获取研报品种错误,Err:" + err.Error()
  283. return
  284. }
  285. if reportPermissionResp.Ret != 200 {
  286. br.Msg = reportPermissionResp.Msg
  287. br.ErrMsg = reportPermissionResp.ErrMsg
  288. return
  289. }
  290. reportPermissionList := reportPermissionResp.Data
  291. chartPermissionList, err := services.GetChartPermissionAllList()
  292. if err != nil {
  293. br.Msg = "添加阅读记录失败"
  294. br.ErrMsg = "获取研报品种列表失败,Err:" + err.Error()
  295. return
  296. }
  297. chartMap := make(map[int]*models.ChartPermission)
  298. for _, permission := range chartPermissionList.Data {
  299. chartMap[permission.ChartPermissionId] = permission
  300. }
  301. curTime := time.Now()
  302. var insertId int64
  303. if req.RecordId == 0 {
  304. // 如果不存在就新增一条记录
  305. permission1Ids := make([]int, 0)
  306. permission2Ids := make([]int, 0)
  307. permissionNames := make([]string, 0)
  308. for _, item := range reportPermissionList {
  309. curPermission := chartMap[item.ChartPermissionId]
  310. permission1Ids = append(permission1Ids, curPermission.ParentId)
  311. permission2Ids = append(permission2Ids, curPermission.ChartPermissionId)
  312. permissionNames = append(permissionNames, curPermission.PermissionName)
  313. }
  314. permission1Ids = utils.Unique(permission1Ids)
  315. permission2Ids = utils.Unique(permission2Ids)
  316. permissionNames = utils.Unique(permissionNames)
  317. userReadRecord := &models.UserReadRecord{
  318. UserId: user.UserId,
  319. ReportId: req.ReportId,
  320. ReportTitle: reportDetail.Title,
  321. ChartPermissionName: strings.Join(permissionNames, ","),
  322. ClassifyId1: reportDetail.ClassifyIdFirst,
  323. ClassifyName1: reportDetail.ClassifyNameFirst,
  324. ClassifyId2: reportDetail.ClassifyIdSecond,
  325. ClassifyName2: reportDetail.ClassifyNameSecond,
  326. Timestamp: int(curTime.Unix()),
  327. CreateTime: curTime,
  328. CreateDate: curTime.Format(utils.FormatDate),
  329. ReportType: utils.ReportTypeEta,
  330. }
  331. insertId, err = userReadRecord.Insert()
  332. if err != nil {
  333. br.Msg = "添加阅读记录失败"
  334. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  335. return
  336. }
  337. userReadPermission1 := make([]*models.UserReadPermission1, 0)
  338. for _, id := range permission1Ids {
  339. userReadPermission1 = append(userReadPermission1, &models.UserReadPermission1{
  340. UserReadRecordId: int(insertId),
  341. ChartPermissionId: id,
  342. PermissionName: chartMap[id].PermissionName,
  343. })
  344. }
  345. err = models.UserReadPermission1MultiInsert(userReadPermission1)
  346. if err != nil {
  347. br.Msg = "添加阅读记录失败"
  348. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  349. return
  350. }
  351. userReadPermission2 := make([]*models.UserReadPermission2, 0)
  352. for _, id := range permission2Ids {
  353. userReadPermission2 = append(userReadPermission2, &models.UserReadPermission2{
  354. UserReadRecordId: int(insertId),
  355. ChartPermissionId: id,
  356. })
  357. }
  358. err = models.UserReadPermission2MultiInsert(userReadPermission2)
  359. if err != nil {
  360. br.Msg = "添加阅读记录失败"
  361. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  362. return
  363. }
  364. } else {
  365. // 如果存在就计算停留时间
  366. userRecord, err := models.GetUserReadRecordListById(req.RecordId)
  367. if err != nil {
  368. if err.Error() == utils.ErrNoRow() {
  369. br.Msg = "更新阅读记录不存在"
  370. return
  371. }
  372. br.Msg = "更新阅读记录失败"
  373. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  374. return
  375. }
  376. stayTime := curTime.Unix() - int64(userRecord.Timestamp)
  377. stayTimeStr := utils.SecondsToHMS(stayTime)
  378. err = models.UpdateUserReadRecordById(req.RecordId, int(curTime.Unix()), int(stayTime), stayTimeStr)
  379. if err != nil {
  380. br.Msg = "更新阅读记录失败"
  381. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  382. return
  383. }
  384. }
  385. resp := new(response.UserReadRecordResp)
  386. resp.RecordId = insertId
  387. br.Msg = "添加阅读记录成功"
  388. br.Ret = 200
  389. br.Success = true
  390. br.Data = resp
  391. }
  392. // @Title 新增pdf报告浏览记录
  393. // @Description 新增pdf报告浏览记录接口
  394. // @Param request body models.ReportPdfRecordReq true "type json string"
  395. // @Success 200 新增成功
  396. // @router /pdf/addReportRecord [post]
  397. func (this *UserAuthController) AddReportPdfRecord() {
  398. br := new(models.BaseResponse).Init()
  399. defer func() {
  400. this.Data["json"] = br
  401. this.ServeJSON()
  402. }()
  403. user := this.User
  404. if user == nil {
  405. br.Msg = "请登录"
  406. br.ErrMsg = "请登录,用户信息为空"
  407. br.Ret = 403
  408. return
  409. }
  410. var req request.ReportPdfRecordReq
  411. if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
  412. br.Msg = "参数解析异常!"
  413. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  414. return
  415. }
  416. if req.ReportPdfId <= 0 {
  417. br.Msg = "参数错误"
  418. br.ErrMsg = "参数错误,报告id小于等于0"
  419. return
  420. }
  421. reportPdf, err := models.GetReportPdfById(req.ReportPdfId)
  422. if err != nil {
  423. if err.Error() == utils.ErrNoRow() {
  424. br.Msg = "报告不存在或已删除"
  425. return
  426. }
  427. br.Msg = "添加阅读记录失败"
  428. br.ErrMsg = "获取研报错误,Err:" + err.Error()
  429. return
  430. }
  431. reportPermissionResp, err := services.GetChartPermissionListByClassifyId(reportPdf.ClassifyIdSecond, 0)
  432. if err != nil {
  433. br.Msg = "添加阅读记录失败"
  434. br.ErrMsg = "获取研报品种错误,Err:" + err.Error()
  435. return
  436. }
  437. if reportPermissionResp.Ret != 200 {
  438. br.Msg = reportPermissionResp.Msg
  439. br.ErrMsg = reportPermissionResp.ErrMsg
  440. return
  441. }
  442. reportPermissionList := reportPermissionResp.Data
  443. chartPermissionList, err := services.GetChartPermissionAllList()
  444. if err != nil {
  445. br.Msg = "添加阅读记录失败"
  446. br.ErrMsg = "获取研报品种列表失败,Err:" + err.Error()
  447. return
  448. }
  449. chartMap := make(map[int]*models.ChartPermission)
  450. for _, permission := range chartPermissionList.Data {
  451. chartMap[permission.ChartPermissionId] = permission
  452. }
  453. curTime := time.Now()
  454. var insertId int64
  455. if req.RecordId == 0 {
  456. // 如果不存在就新增一条记录
  457. permission1Ids := make([]int, 0)
  458. permission2Ids := make([]int, 0)
  459. permissionNames := make([]string, 0)
  460. for _, item := range reportPermissionList {
  461. curPermission := chartMap[item.ChartPermissionId]
  462. permission1Ids = append(permission1Ids, curPermission.ParentId)
  463. permission2Ids = append(permission2Ids, curPermission.ChartPermissionId)
  464. permissionNames = append(permissionNames, curPermission.PermissionName)
  465. }
  466. permission1Ids = utils.Unique(permission1Ids)
  467. permission2Ids = utils.Unique(permission2Ids)
  468. permissionNames = utils.Unique(permissionNames)
  469. userReadRecord := &models.UserReadRecord{
  470. UserId: user.UserId,
  471. ReportId: reportPdf.ReportPdfId,
  472. ReportTitle: reportPdf.Title,
  473. ChartPermissionName: strings.Join(permissionNames, ","),
  474. ClassifyId1: reportPdf.ClassifyIdFirst,
  475. ClassifyName1: reportPdf.ClassifyNameFirst,
  476. ClassifyId2: reportPdf.ClassifyIdSecond,
  477. ClassifyName2: reportPdf.ClassifyNameSecond,
  478. Timestamp: int(curTime.Unix()),
  479. CreateTime: curTime,
  480. CreateDate: curTime.Format(utils.FormatDate),
  481. ReportType: utils.ReportTypePdf,
  482. }
  483. insertId, err = userReadRecord.Insert()
  484. if err != nil {
  485. br.Msg = "添加阅读记录失败"
  486. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  487. return
  488. }
  489. userReadPermission1 := make([]*models.UserReadPermission1, 0)
  490. for _, id := range permission1Ids {
  491. userReadPermission1 = append(userReadPermission1, &models.UserReadPermission1{
  492. UserReadRecordId: int(insertId),
  493. ChartPermissionId: id,
  494. PermissionName: chartMap[id].PermissionName,
  495. })
  496. }
  497. err = models.UserReadPermission1MultiInsert(userReadPermission1)
  498. if err != nil {
  499. br.Msg = "添加阅读记录失败"
  500. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  501. return
  502. }
  503. userReadPermission2 := make([]*models.UserReadPermission2, 0)
  504. for _, id := range permission2Ids {
  505. userReadPermission2 = append(userReadPermission2, &models.UserReadPermission2{
  506. UserReadRecordId: int(insertId),
  507. ChartPermissionId: id,
  508. })
  509. }
  510. err = models.UserReadPermission2MultiInsert(userReadPermission2)
  511. if err != nil {
  512. br.Msg = "添加阅读记录失败"
  513. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  514. return
  515. }
  516. count, err := models.GetUserReadRecordCountByReportPdfIdAndUserId(req.ReportPdfId, user.UserId)
  517. if err != nil {
  518. br.Msg = "添加阅读记录失败"
  519. br.ErrMsg = "获取阅读记录失败,Err:" + err.Error()
  520. return
  521. }
  522. if count > 1 {
  523. err = models.UpdateReportPdfPv(req.ReportPdfId)
  524. if err != nil {
  525. br.Msg = "添加研报阅读记录失败"
  526. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  527. return
  528. }
  529. } else {
  530. err = models.UpdateReportPdfUvAndPv(req.ReportPdfId)
  531. if err != nil {
  532. br.Msg = "添加研报阅读记录失败"
  533. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  534. return
  535. }
  536. }
  537. } else {
  538. // 如果存在就计算停留时间
  539. userRecord, err := models.GetUserReadRecordListById(req.RecordId)
  540. if err != nil {
  541. if err.Error() == utils.ErrNoRow() {
  542. br.Msg = "更新阅读记录不存在"
  543. return
  544. }
  545. br.Msg = "更新阅读记录失败"
  546. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  547. return
  548. }
  549. stayTime := curTime.Unix() - int64(userRecord.Timestamp)
  550. stayTimeStr := utils.SecondsToHMS(stayTime)
  551. err = models.UpdateUserReadRecordById(req.RecordId, int(curTime.Unix()), int(stayTime), stayTimeStr)
  552. if err != nil {
  553. br.Msg = "更新阅读记录失败"
  554. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  555. return
  556. }
  557. }
  558. resp := new(response.UserReadRecordResp)
  559. resp.RecordId = insertId
  560. br.Msg = "添加阅读记录成功"
  561. br.Ret = 200
  562. br.Success = true
  563. br.Data = resp
  564. }
  565. // AreaCodeList
  566. // @Title 手机号区号列表
  567. // @Description 手机号区号列表
  568. // @Success 200 Ret=200 获取成功
  569. // @router /area_code/list [get]
  570. func (this *UserController) AreaCodeList() {
  571. br := new(models.BaseResponse).Init()
  572. defer func() {
  573. if br.ErrMsg == "" {
  574. br.IsSendEmail = false
  575. }
  576. this.Data["json"] = br
  577. this.ServeJSON()
  578. }()
  579. resp := make([]response.AreaCodeListResp, 0)
  580. confValue, e := models.GetConfigDetailByCode(models.ConfAreaCodeListKey)
  581. if e != nil {
  582. br.Msg = "获取失败"
  583. br.ErrMsg = "获取手机号区号配置失败, Err: " + e.Error()
  584. return
  585. }
  586. if confValue == "" {
  587. br.Msg = "获取失败"
  588. br.ErrMsg = "手机号区号配置为空"
  589. return
  590. }
  591. if e := json.Unmarshal([]byte(confValue), &resp); e != nil {
  592. br.Msg = "获取失败"
  593. br.ErrMsg = "手机号区号配置有误"
  594. return
  595. }
  596. br.Data = resp
  597. br.Ret = 200
  598. br.Success = true
  599. br.Msg = "获取成功"
  600. }
  601. // 用户品种信息
  602. // @Title 用户信息
  603. // @Description 用户信息
  604. // @Success 200 Ret=200 获取成功
  605. // @router /info [get]
  606. func (this *UserAuthController) Info() {
  607. br := new(models.BaseResponse).Init()
  608. defer func() {
  609. this.Data["json"] = br
  610. this.ServeJSON()
  611. }()
  612. user := this.User
  613. if user == nil {
  614. user = &models.User{}
  615. }
  616. seller, err := models.GetSellerById(user.SellerId)
  617. if err != nil && err.Error() != utils.ErrNoRow() {
  618. br.Msg = "查询失败"
  619. br.ErrMsg = "查询失败,系统异常,Err:" + err.Error()
  620. return
  621. }
  622. private, err := services.GetPrivateChartPermissionList()
  623. if err != nil {
  624. br.Msg = "查询失败"
  625. br.ErrMsg = "查询失败,系统异常,Err:" + err.Error()
  626. return
  627. }
  628. if private.Ret != 200 {
  629. br.Msg = private.Msg
  630. br.ErrMsg = private.ErrMsg
  631. return
  632. }
  633. userPermissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  634. if err != nil {
  635. br.Msg = "查看权限失败"
  636. br.ErrMsg = "查看权限失败,系统异常,Err:" + err.Error()
  637. return
  638. }
  639. userPermissionMap := make(map[int]struct{})
  640. for _, v := range userPermissionIds {
  641. userPermissionMap[v] = struct{}{}
  642. }
  643. if seller == nil {
  644. // 添加默认的销售电话
  645. seller = &models.SysUser{
  646. Phone: utils.DefaultPhone,
  647. AreaCode: utils.DefaultAreaCode,
  648. }
  649. }
  650. var hasPermission string
  651. for _, v := range private.Data {
  652. for _, vv := range v.Child {
  653. if _, ok := userPermissionMap[vv.ChartPermissionId]; ok {
  654. hasPermission = "私有权限"
  655. break
  656. }
  657. }
  658. }
  659. if hasPermission == "" || user.Status == utils.UserStatusNo {
  660. hasPermission = "公有权限"
  661. }
  662. miniconf, err := models.GetMiniConf()
  663. if err != nil {
  664. br.Msg = "获取配置失败"
  665. br.ErrMsg = "获取配置失败,系统异常,Err:" + err.Error()
  666. return
  667. }
  668. userView := &models.UserView{
  669. Headimgurl: miniconf["Logo"],
  670. RealName: user.RealName,
  671. Phone: user.Phone,
  672. AreaCode: user.AreaCode,
  673. SellerName: seller.SysRealName,
  674. SellerPhone: seller.Phone,
  675. SellerAreaCode: seller.AreaCode,
  676. HasPermission: hasPermission,
  677. Email: user.Email,
  678. Componey: user.Company,
  679. IsRegistered: user.IsRegistered,
  680. Status: user.Status,
  681. ValidEndTime: user.ValidEndTime.Format(utils.FormatDate),
  682. }
  683. if user.ValidEndTime.IsZero() {
  684. userView.ValidEndTime = ""
  685. }
  686. br.Data = userView
  687. br.Msg = "查询成功"
  688. br.Success = true
  689. br.Ret = 200
  690. }
  691. // PermissionList
  692. // @Title 用户个人权限列表
  693. // @Description 用户个人权限列表
  694. // @Param request body models.LoginReq true "type json string"
  695. // @Success 200 {object} []models.ChartPermissionTreeView
  696. // @router /permission/list [get]
  697. func (this *UserAuthController) PermissionList() {
  698. br := new(models.BaseResponse).Init()
  699. defer func() {
  700. this.Data["json"] = br
  701. this.ServeJSON()
  702. }()
  703. user := this.User
  704. permissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  705. if err != nil {
  706. br.Msg = "查询用户权限失败"
  707. br.ErrMsg = "查询用户权限失败,系统异常,Err:" + err.Error()
  708. return
  709. }
  710. userPermissionMap := make(map[int]struct{})
  711. for _, v := range permissionIds {
  712. userPermissionMap[v] = struct{}{}
  713. }
  714. // 查询公有权限
  715. var publicView []*models.ChartPermissionTreeView
  716. publicResp, err := services.GetPublicChartPermissionList()
  717. if err != nil {
  718. br.Msg = "查询公有权限失败"
  719. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  720. }
  721. if publicResp.Ret != 200 {
  722. br.Msg = publicResp.Msg
  723. br.ErrMsg = publicResp.ErrMsg
  724. return
  725. }
  726. publicView = publicResp.Data
  727. // 如果是正式用户,则查询私有权限
  728. var privateView []*models.ChartPermissionTreeView
  729. if user.Status == utils.UserStatusFormal {
  730. privateResp, err := services.GetPrivateChartPermissionList()
  731. if err != nil {
  732. br.Msg = "查询私有权限失败"
  733. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  734. }
  735. if privateResp.Ret != 200 {
  736. br.Msg = privateResp.Msg
  737. br.ErrMsg = privateResp.ErrMsg
  738. return
  739. }
  740. for _, v := range privateResp.Data {
  741. IsAdd := false
  742. curPermissionView := &models.ChartPermissionTreeView{
  743. ChartPermissionId: v.ChartPermissionId,
  744. PermissionName: v.PermissionName,
  745. IsPublic: v.IsPublic,
  746. Sort: v.Sort,
  747. Child: make([]*models.ChartPermissionTreeView, 0),
  748. }
  749. for _, vv := range v.Child {
  750. if _, ok := userPermissionMap[vv.ChartPermissionId]; ok {
  751. curPermissionView.Child = append(curPermissionView.Child, vv)
  752. IsAdd = true
  753. }
  754. }
  755. if IsAdd {
  756. privateView = append(privateView, curPermissionView)
  757. }
  758. }
  759. }
  760. privateAloneView := make([]*models.ChartPermissionTreeView, 0)
  761. for _, vi := range privateView {
  762. isHas := false
  763. for _, vb := range publicView {
  764. if vi.ChartPermissionId == vb.ChartPermissionId {
  765. isHas = true
  766. vb.Child = append(vb.Child, vi.Child...)
  767. break
  768. }
  769. }
  770. if !isHas {
  771. privateAloneView = append(privateAloneView, vi)
  772. }
  773. }
  774. publicView = append(publicView, privateAloneView...)
  775. for _, vi := range publicView {
  776. vi.ChartPermissionType = utils.PermissionTypeEta
  777. for _, vi := range vi.Child {
  778. vi.ChartPermissionType = utils.PermissionTypeEta
  779. }
  780. }
  781. br.Data = publicView
  782. br.Msg = "查询成功"
  783. br.Success = true
  784. br.Ret = 200
  785. }
  786. // PermissionList
  787. // @Title 用户已购权限列表
  788. // @Description 用户已购权限列表
  789. // @Param request body models.LoginReq true "type json string"
  790. // @Success 200 {object} []models.ChartPermissionTreeView
  791. // @router /purchased/list [get]
  792. func (this *UserAuthController) PurchasedPermission() {
  793. br := new(models.BaseResponse).Init()
  794. defer func() {
  795. this.Data["json"] = br
  796. this.ServeJSON()
  797. }()
  798. user := this.User
  799. permissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  800. if err != nil {
  801. br.Msg = "查询用户权限失败"
  802. br.ErrMsg = "查询用户权限失败,系统异常,Err:" + err.Error()
  803. return
  804. }
  805. permissionMap := make(map[int]struct{})
  806. for _, v := range permissionIds {
  807. permissionMap[v] = struct{}{}
  808. }
  809. // 查询私有权限
  810. privateResp, err := services.GetPrivateChartPermissionList()
  811. if err != nil {
  812. br.Msg = "查询私有权限失败"
  813. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  814. }
  815. if privateResp.Ret != 200 {
  816. br.Msg = privateResp.Msg
  817. br.ErrMsg = privateResp.ErrMsg
  818. return
  819. }
  820. var privateView []*models.ChartPermissionTreeView
  821. for _, v := range privateResp.Data {
  822. IsAdd := false
  823. curPermissionView := &models.ChartPermissionTreeView{
  824. ChartPermissionId: v.ChartPermissionId,
  825. PermissionName: v.PermissionName,
  826. IsPublic: v.IsPublic,
  827. Sort: v.Sort,
  828. Child: make([]*models.ChartPermissionTreeView, 0),
  829. ChartPermissionType: utils.PermissionTypeEta,
  830. }
  831. for _, vv := range v.Child {
  832. if _, ok := permissionMap[vv.ChartPermissionId]; ok && vv.ParentId != 0 {
  833. vv.ChartPermissionType = utils.PermissionTypeEta
  834. curPermissionView.Child = append(curPermissionView.Child, vv)
  835. IsAdd = true
  836. }
  837. }
  838. if IsAdd {
  839. privateView = append(privateView, curPermissionView)
  840. }
  841. }
  842. for _, vi := range privateView {
  843. vi.Child = append(vi.Child, &models.ChartPermissionTreeView{
  844. ChartPermissionId: 0,
  845. PermissionName: utils.MINI_CUSTOM_PERMISSION_NAME,
  846. IsPublic: 0,
  847. Sort: 0,
  848. ChartPermissionType: utils.ReportTypePdf,
  849. })
  850. }
  851. br.Data = privateView
  852. br.Msg = "查询成功"
  853. br.Success = true
  854. br.Ret = 200
  855. }