user.go 32 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139
  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/go_redis"
  9. "eta/eta_mini_api/utils"
  10. "fmt"
  11. "image/color"
  12. "strconv"
  13. "strings"
  14. "time"
  15. "github.com/mojocn/base64Captcha"
  16. )
  17. type UserController struct {
  18. BaseCommonController
  19. }
  20. type UserAuthController struct {
  21. BaseAuthController
  22. }
  23. // @Title 用户登录接口
  24. // @Description 用户登录
  25. // @Param request body models.LoginReq true "type json string"
  26. // @Success 200 {object} models.LoginResp
  27. // @router /login [post]
  28. func (this *UserAuthController) Login() {
  29. br := new(models.BaseResponse).Init()
  30. defer func() {
  31. this.Data["json"] = br
  32. this.ServeJSON()
  33. }()
  34. session := this.Session
  35. var req request.LoginReq
  36. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  37. if err != nil {
  38. br.Msg = "参数解析失败"
  39. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  40. return
  41. }
  42. if req.RealName == "" {
  43. br.Msg = "请输入姓名"
  44. return
  45. }
  46. if req.SysDepartmentId <= 0 {
  47. br.Msg = "请选择营业部门"
  48. return
  49. }
  50. switch req.LoginType {
  51. case 1:
  52. if req.Phone == "" {
  53. br.Msg = "请输入手机号"
  54. br.ErrMsg = "请输入手机号"
  55. return
  56. }
  57. if req.SmsCode == "" {
  58. br.Msg = "请输入验证码"
  59. br.ErrMsg = "请输入验证码"
  60. return
  61. }
  62. req.Email = ""
  63. phone := req.AreaCode + req.Phone
  64. item, err := models.GetMsgCode(phone, req.SmsCode)
  65. if err != nil {
  66. if err.Error() == utils.ErrNoRow() {
  67. br.Msg = "验证码错误,请重新输入"
  68. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  69. return
  70. } else {
  71. br.Msg = "验证码错误,请重新输入"
  72. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  73. return
  74. }
  75. }
  76. if item == nil {
  77. br.Msg = "验证码错误,请重新输入"
  78. return
  79. }
  80. case 2:
  81. if req.Email == "" {
  82. br.Msg = "请输入邮箱"
  83. br.ErrMsg = "请输入邮箱"
  84. return
  85. }
  86. if req.SmsCode == "" {
  87. br.Msg = "请输入验证码"
  88. br.ErrMsg = "请输入验证码"
  89. return
  90. }
  91. req.Phone = ""
  92. item, err := models.GetMsgCode(req.Email, req.SmsCode)
  93. if err != nil {
  94. if err.Error() == utils.ErrNoRow() {
  95. br.Msg = "验证码错误,请重新输入"
  96. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  97. return
  98. } else {
  99. br.Msg = "验证码错误,请重新输入"
  100. br.ErrMsg = "校验验证码失败,Err:" + err.Error()
  101. return
  102. }
  103. }
  104. if item == nil {
  105. br.Msg = "验证码错误,请重新输入"
  106. return
  107. }
  108. }
  109. _, errMsg, err := services.BindUser(session.UnionId, session.OpenId, req.Phone, req.Email, req.AreaCode, req.SysDepartmentId, req.RealName)
  110. if err != nil && err.Error() != utils.ErrNoRow() {
  111. br.Msg = "登录失败,系统处理中,请稍后重试"
  112. if errMsg != "" {
  113. br.Msg = errMsg
  114. }
  115. br.ErrMsg = "登录失败:" + err.Error()
  116. return
  117. }
  118. br.Msg = "登录成功"
  119. br.Success = true
  120. br.Ret = 200
  121. }
  122. // GenerateCaptcha
  123. // @Title 生成图形验证码
  124. // @Description 生成图形验证码
  125. // @Success 200 Ret=200 获取成功
  126. // @router /getCaptcha [get]
  127. func (this *UserController) GenerateCaptcha() {
  128. br := new(models.BaseResponse).Init()
  129. defer func() {
  130. if br.ErrMsg == "" {
  131. br.IsSendEmail = false
  132. }
  133. this.Data["json"] = br
  134. this.ServeJSON()
  135. }()
  136. // 自定义验证码样式
  137. var driver base64Captcha.Driver
  138. driverString := base64Captcha.DriverString{
  139. Height: 60, //高度
  140. Width: 120, //宽度
  141. NoiseCount: 0, //干扰数
  142. ShowLineOptions: 2 | 4, //展示个数
  143. Length: 4, //长度
  144. //Source: "1234567890qwertyuioplkjhgfdsazxcvbnm", //验证码随机字符串来源
  145. Source: "1234567890", //验证码随机字符串来源
  146. BgColor: &color.RGBA{ // 背景颜色
  147. R: 0,
  148. G: 0,
  149. B: 0,
  150. A: 0,
  151. },
  152. Fonts: []string{"wqy-microhei.ttc"}, // 字体
  153. }
  154. driver = driverString.ConvertFonts()
  155. // 生成验证码
  156. store := services.CaptchaRedis{}
  157. captcha := base64Captcha.NewCaptcha(driver, store)
  158. id, b64s, _, err := captcha.Generate()
  159. if err != nil {
  160. br.Msg = "生成失败"
  161. br.ErrMsg = "生成验证码失败, Err: " + err.Error()
  162. return
  163. }
  164. type CaptchaResult struct {
  165. Id string
  166. Base64Blob string
  167. }
  168. res := new(CaptchaResult)
  169. res.Id = id
  170. res.Base64Blob = b64s
  171. br.Ret = 200
  172. br.Success = true
  173. br.Msg = "获取成功"
  174. br.Data = res
  175. }
  176. // @Title 获取短信/邮箱验证码
  177. // @Description 用户登录
  178. // @Param request body models.LoginReq true "type json string"
  179. // @Success 200 {object} models.LoginResp
  180. // @router /getVerifyCode [post]
  181. func (this *UserController) GetVerifyCode() {
  182. br := new(models.BaseResponse).Init()
  183. defer func() {
  184. this.Data["json"] = br
  185. this.ServeJSON()
  186. }()
  187. var req request.VerifyCodeReq
  188. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  189. if err != nil {
  190. br.Msg = "参数解析异常!"
  191. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  192. return
  193. }
  194. if req.VerifyType != 1 && req.VerifyType != 2 {
  195. br.Msg = "验证方式有误"
  196. br.ErrMsg = fmt.Sprintf("验证方式异常<%d>", req.VerifyType)
  197. }
  198. if req.CaptchaId == "" || req.CaptchaCode == "" {
  199. br.Msg = "请输入图形验证码"
  200. return
  201. }
  202. code := utils.GetRandDigit(6)
  203. fmt.Println(code)
  204. switch req.VerifyType {
  205. case 1:
  206. if req.AreaCode == "" {
  207. br.Msg = "请选择区号"
  208. return
  209. }
  210. if req.Phone == "" {
  211. br.Msg = "请输入手机号"
  212. return
  213. }
  214. if req.AreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Phone) {
  215. br.Msg = "您的手机号输入有误, 请检查"
  216. return
  217. }
  218. phoneKey := utils.CACHE_ACCESS_PHONE_LOGIN_CODE + req.AreaCode + req.Phone
  219. res, _ := go_redis.RedisInt(phoneKey)
  220. if res >= 5 {
  221. br.Msg = "验证码发送太频繁,请稍后重试"
  222. return
  223. }
  224. phoneCountKey := utils.CACHE_ACCESS_PHONE_COUNT_LOGIN_CODE + req.AreaCode + req.Phone
  225. resCount, _ := go_redis.RedisInt(phoneCountKey)
  226. if resCount >= utils.VerifyCodeSendLimit {
  227. br.Msg = fmt.Sprintf("一天最多获取%s次,已超限", strconv.Itoa(utils.VerifyCodeSendLimit))
  228. return
  229. }
  230. store := services.CaptchaRedis{}
  231. var ok bool
  232. ok = store.Verify(req.CaptchaId, req.CaptchaCode, true)
  233. if !ok {
  234. br.Msg = "图形验证码错误"
  235. return
  236. }
  237. if req.AreaCode == "86" {
  238. smsClient, e := services.NewSmsClient()
  239. if e != nil {
  240. br.Msg = "短信发送失败"
  241. br.ErrMsg = "短信发送失败,Err:" + e.Error()
  242. return
  243. }
  244. //ok = services.SendSmsCode(req.Phone, code)
  245. var smsReq services.UserLoginSmsCodeReq
  246. smsReq.Mobile = req.Phone
  247. smsReq.TelAreaCode = req.AreaCode
  248. smsReq.VerifyCode = code
  249. smsResult, e := smsClient.SendUserLoginCode(smsReq)
  250. if e != nil {
  251. br.Msg = "短信发送失败"
  252. br.ErrMsg = "短信发送失败,Err:" + e.Error()
  253. return
  254. }
  255. ok = smsResult.Success
  256. if !ok {
  257. br.Msg = "短信发送失败"
  258. br.ErrMsg = "短信发送失败," + smsResult.Message
  259. return
  260. }
  261. }
  262. if !ok {
  263. br.Msg = "请检查手机号和区号"
  264. br.ErrMsg = "短信验证码发送失败"
  265. return
  266. } else {
  267. item := new(models.MsgCode)
  268. item.OpenId = ""
  269. item.Code = code
  270. item.Mobile = req.AreaCode + req.Phone
  271. item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
  272. item.CreateTime = time.Now()
  273. err = item.Insert()
  274. if err != nil {
  275. br.Msg = "发送失败"
  276. br.ErrMsg = "发送失败,Err:" + err.Error()
  277. return
  278. }
  279. br.Msg = "发送成功"
  280. phoneVerifyCahcheSvc := &services.VerifyCacheIncrService{}
  281. err = phoneVerifyCahcheSvc.VerifyCacheIncr(phoneKey, 15*int(time.Minute.Seconds()))
  282. if err != nil {
  283. utils.FileLog.Info("验证码手机号临时缓存失败", err.Error())
  284. }
  285. err = phoneVerifyCahcheSvc.VerifyCacheIncr(phoneCountKey, int(utils.SetKeyExpireToday().Seconds()))
  286. if err != nil {
  287. utils.FileLog.Info("验证码手机号当日缓存失败", err.Error())
  288. }
  289. }
  290. case 2:
  291. if req.Email == "" {
  292. br.Msg = "请输入邮箱"
  293. }
  294. if !utils.ValidateEmailFormatat(req.Email) {
  295. br.Msg = "您的邮箱格式输入有误, 请检查"
  296. return
  297. }
  298. emailKey := utils.CACHE_ACCESS_EMAIL_LOGIN_CODE + req.Email
  299. res, _ := go_redis.RedisInt(emailKey)
  300. if res >= 5 {
  301. br.Msg = "验证码发送太频繁,请稍后重试"
  302. return
  303. }
  304. emailCountKey := utils.CACHE_ACCESS_EMAIL_COUNT_LOGIN_CODE + req.Email
  305. resCount, _ := go_redis.RedisInt(emailCountKey)
  306. if resCount >= utils.VerifyCodeSendLimit {
  307. br.Msg = fmt.Sprintf("一天最多获取%s次,已超限", strconv.Itoa(utils.VerifyCodeSendLimit))
  308. return
  309. }
  310. store := services.CaptchaRedis{}
  311. var ok bool
  312. ok = store.Verify(req.CaptchaId, req.CaptchaCode, true)
  313. if !ok {
  314. br.Msg = "图形验证码错误"
  315. return
  316. }
  317. date := time.Now()
  318. content := "尊敬的用户:</br>本次请求的验证码为:" + code + "(为了保障您账号的安全性,请在15分钟内完成验证。)</br>东吴期货研究团队 </br>" + fmt.Sprintf("%d年%02d月%02d日", date.Year(), date.Month(), date.Day())
  319. title := "东吴期货登录验证"
  320. result, err := utils.SendEmailByDw(title, content, req.Email)
  321. if err != nil {
  322. br.Msg = "发送失败"
  323. br.ErrMsg = "发送失败,Err:" + err.Error()
  324. return
  325. }
  326. if result {
  327. item := new(models.MsgCode)
  328. item.OpenId = ""
  329. item.Code = code
  330. item.Mobile = req.Email
  331. item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
  332. item.CreateTime = time.Now()
  333. err = item.Insert()
  334. if err != nil {
  335. br.Msg = "发送失败"
  336. br.ErrMsg = "发送失败,Err:" + err.Error()
  337. return
  338. }
  339. br.Msg = "发送成功"
  340. emailVerifyCahcheSvc := &services.VerifyCacheIncrService{}
  341. err = emailVerifyCahcheSvc.VerifyCacheIncr(emailKey, 15*int(time.Minute.Seconds()))
  342. if err != nil {
  343. utils.FileLog.Info("验证码邮箱临时缓存失败, err:", err.Error())
  344. }
  345. err = emailVerifyCahcheSvc.VerifyCacheIncr(emailCountKey, int(utils.SetKeyExpireToday().Seconds()))
  346. if err != nil {
  347. utils.FileLog.Info("验证码邮箱当日缓存失败, err:", err.Error())
  348. }
  349. } else {
  350. br.Msg = "发送失败"
  351. }
  352. }
  353. br.Ret = 200
  354. br.Success = true
  355. br.Msg = "发送成功"
  356. }
  357. // @Title 新增报告浏览记录
  358. // @Description 新增报告浏览记录接口
  359. // @Param request body models.ReportRecordReq true "type json string"
  360. // @Success 200 新增成功
  361. // @router /addReportRecord [post]
  362. func (this *UserAuthController) AddReportRecord() {
  363. br := new(models.BaseResponse).Init()
  364. defer func() {
  365. this.Data["json"] = br
  366. this.ServeJSON()
  367. }()
  368. user := this.User
  369. if user == nil {
  370. br.Msg = "请登录"
  371. br.ErrMsg = "请登录,用户信息为空"
  372. br.Ret = 403
  373. return
  374. }
  375. var req request.ReportRecordReq
  376. if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
  377. br.Msg = "参数解析异常!"
  378. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  379. return
  380. }
  381. if req.ReportId <= 0 {
  382. br.Msg = "参数错误"
  383. br.ErrMsg = "参数错误,报告id小于等于0"
  384. return
  385. }
  386. reportResp, err := services.GetReportDetail(req.ReportId, user.UserId)
  387. if err != nil {
  388. br.Msg = "添加阅读记录失败"
  389. br.ErrMsg = "获取研报错误,Err:" + err.Error()
  390. return
  391. }
  392. if reportResp.Ret != 200 {
  393. br.Msg = reportResp.Msg
  394. br.ErrMsg = reportResp.ErrMsg
  395. return
  396. }
  397. reportDetail := reportResp.Data.Report
  398. reportPermissionResp, err := services.GetReportChartPermissionList(req.ReportId)
  399. if err != nil {
  400. br.Msg = "添加阅读记录失败"
  401. br.ErrMsg = "获取研报品种错误,Err:" + err.Error()
  402. return
  403. }
  404. if reportPermissionResp.Ret != 200 {
  405. br.Msg = reportPermissionResp.Msg
  406. br.ErrMsg = reportPermissionResp.ErrMsg
  407. return
  408. }
  409. reportPermissionList := reportPermissionResp.Data
  410. chartPermissionList, err := services.GetChartPermissionAllList()
  411. if err != nil {
  412. br.Msg = "添加阅读记录失败"
  413. br.ErrMsg = "获取研报品种列表失败,Err:" + err.Error()
  414. return
  415. }
  416. chartMap := make(map[int]*models.ChartPermission)
  417. for _, permission := range chartPermissionList.Data {
  418. chartMap[permission.ChartPermissionId] = permission
  419. }
  420. curTime := time.Now()
  421. var insertId int64
  422. if req.RecordId == 0 {
  423. // 如果不存在就新增一条记录
  424. permission1Ids := make([]int, 0)
  425. permission2Ids := make([]int, 0)
  426. permissionNames := make([]string, 0)
  427. for _, item := range reportPermissionList {
  428. curPermission := chartMap[item.ChartPermissionId]
  429. permission1Ids = append(permission1Ids, curPermission.ParentId)
  430. permission2Ids = append(permission2Ids, curPermission.ChartPermissionId)
  431. permissionNames = append(permissionNames, curPermission.PermissionName)
  432. }
  433. permission1Ids = utils.Unique(permission1Ids)
  434. permission2Ids = utils.Unique(permission2Ids)
  435. permissionNames = utils.Unique(permissionNames)
  436. userReadRecord := &models.UserReadRecord{
  437. UserId: user.UserId,
  438. ReportId: req.ReportId,
  439. ReportTitle: reportDetail.Title,
  440. ChartPermissionName: strings.Join(permissionNames, ","),
  441. ClassifyId1: reportDetail.ClassifyIdFirst,
  442. ClassifyName1: reportDetail.ClassifyNameFirst,
  443. ClassifyId2: reportDetail.ClassifyIdSecond,
  444. ClassifyName2: reportDetail.ClassifyNameSecond,
  445. ClassifyId3: reportDetail.ClassifyIdThird,
  446. ClassifyName3: reportDetail.ClassifyNameThird,
  447. Timestamp: int(curTime.Unix()),
  448. CreateTime: curTime,
  449. CreateDate: curTime.Format(utils.FormatDate),
  450. ReportType: utils.ReportTypeEta,
  451. }
  452. insertId, err = userReadRecord.Insert()
  453. if err != nil {
  454. br.Msg = "添加阅读记录失败"
  455. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  456. return
  457. }
  458. userReadPermission1 := make([]*models.UserReadPermission1, 0)
  459. for _, id := range permission1Ids {
  460. userReadPermission1 = append(userReadPermission1, &models.UserReadPermission1{
  461. UserReadRecordId: int(insertId),
  462. ChartPermissionId: id,
  463. PermissionName: chartMap[id].PermissionName,
  464. })
  465. }
  466. err = models.UserReadPermission1MultiInsert(userReadPermission1)
  467. if err != nil {
  468. br.Msg = "添加阅读记录失败"
  469. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  470. return
  471. }
  472. userReadPermission2 := make([]*models.UserReadPermission2, 0)
  473. for _, id := range permission2Ids {
  474. userReadPermission2 = append(userReadPermission2, &models.UserReadPermission2{
  475. UserReadRecordId: int(insertId),
  476. ChartPermissionId: id,
  477. })
  478. }
  479. err = models.UserReadPermission2MultiInsert(userReadPermission2)
  480. if err != nil {
  481. br.Msg = "添加阅读记录失败"
  482. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  483. return
  484. }
  485. } else {
  486. // 如果存在就计算停留时间
  487. userRecord, err := models.GetUserReadRecordListById(req.RecordId)
  488. if err != nil {
  489. if err.Error() == utils.ErrNoRow() {
  490. br.Msg = "更新阅读记录不存在"
  491. return
  492. }
  493. br.Msg = "更新阅读记录失败"
  494. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  495. return
  496. }
  497. stayTime := curTime.Unix() - int64(userRecord.Timestamp)
  498. stayTimeStr := utils.SecondsToHMS(stayTime)
  499. err = models.UpdateUserReadRecordById(req.RecordId, int(curTime.Unix()), int(stayTime), stayTimeStr)
  500. if err != nil {
  501. br.Msg = "更新阅读记录失败"
  502. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  503. return
  504. }
  505. }
  506. resp := new(response.UserReadRecordResp)
  507. resp.RecordId = insertId
  508. br.Msg = "添加阅读记录成功"
  509. br.Ret = 200
  510. br.Success = true
  511. br.Data = resp
  512. }
  513. // @Title 新增pdf报告浏览记录
  514. // @Description 新增pdf报告浏览记录接口
  515. // @Param request body models.ReportPdfRecordReq true "type json string"
  516. // @Success 200 新增成功
  517. // @router /pdf/addReportRecord [post]
  518. func (this *UserAuthController) AddReportPdfRecord() {
  519. br := new(models.BaseResponse).Init()
  520. defer func() {
  521. this.Data["json"] = br
  522. this.ServeJSON()
  523. }()
  524. user := this.User
  525. if user == nil {
  526. br.Msg = "请登录"
  527. br.ErrMsg = "请登录,用户信息为空"
  528. br.Ret = 403
  529. return
  530. }
  531. var req request.ReportPdfRecordReq
  532. if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
  533. br.Msg = "参数解析异常!"
  534. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  535. return
  536. }
  537. if req.ReportPdfId <= 0 {
  538. br.Msg = "参数错误"
  539. br.ErrMsg = "参数错误,报告id小于等于0"
  540. return
  541. }
  542. reportPdf, err := models.GetReportPdfById(req.ReportPdfId)
  543. if err != nil {
  544. if err.Error() == utils.ErrNoRow() {
  545. br.Msg = "报告不存在或已删除"
  546. return
  547. }
  548. br.Msg = "添加阅读记录失败"
  549. br.ErrMsg = "获取研报错误,Err:" + err.Error()
  550. return
  551. }
  552. reportClassifyId := services.GetReportPdfClassify(reportPdf)
  553. reportPermissionResp, err := services.GetChartPermissionListByClassifyId(reportClassifyId, 0)
  554. if err != nil {
  555. br.Msg = "添加阅读记录失败"
  556. br.ErrMsg = "获取研报品种错误,Err:" + err.Error()
  557. return
  558. }
  559. if reportPermissionResp.Ret != 200 {
  560. br.Msg = reportPermissionResp.Msg
  561. br.ErrMsg = reportPermissionResp.ErrMsg
  562. return
  563. }
  564. reportPermissionList := reportPermissionResp.Data
  565. chartPermissionList, err := services.GetChartPermissionAllList()
  566. if err != nil {
  567. br.Msg = "添加阅读记录失败"
  568. br.ErrMsg = "获取研报品种列表失败,Err:" + err.Error()
  569. return
  570. }
  571. chartMap := make(map[int]*models.ChartPermission)
  572. for _, permission := range chartPermissionList.Data {
  573. chartMap[permission.ChartPermissionId] = permission
  574. }
  575. curTime := time.Now()
  576. var insertId int64
  577. if req.RecordId == 0 {
  578. // 如果不存在就新增一条记录
  579. permission1Ids := make([]int, 0)
  580. permission2Ids := make([]int, 0)
  581. permissionNames := make([]string, 0)
  582. for _, item := range reportPermissionList {
  583. curPermission := chartMap[item.ChartPermissionId]
  584. permission1Ids = append(permission1Ids, curPermission.ParentId)
  585. permission2Ids = append(permission2Ids, curPermission.ChartPermissionId)
  586. permissionNames = append(permissionNames, curPermission.PermissionName)
  587. }
  588. permission1Ids = utils.Unique(permission1Ids)
  589. permission2Ids = utils.Unique(permission2Ids)
  590. permissionNames = utils.Unique(permissionNames)
  591. userReadRecord := &models.UserReadRecord{
  592. UserId: user.UserId,
  593. ReportId: reportPdf.ReportPdfId,
  594. ReportTitle: reportPdf.Title,
  595. ChartPermissionName: strings.Join(permissionNames, ","),
  596. ClassifyId1: reportPdf.ClassifyIdFirst,
  597. ClassifyName1: reportPdf.ClassifyNameFirst,
  598. ClassifyId2: reportPdf.ClassifyIdSecond,
  599. ClassifyName2: reportPdf.ClassifyNameSecond,
  600. Timestamp: int(curTime.Unix()),
  601. CreateTime: curTime,
  602. CreateDate: curTime.Format(utils.FormatDate),
  603. ReportType: utils.ReportTypePdf,
  604. }
  605. insertId, err = userReadRecord.Insert()
  606. if err != nil {
  607. br.Msg = "添加阅读记录失败"
  608. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  609. return
  610. }
  611. userReadPermission1 := make([]*models.UserReadPermission1, 0)
  612. for _, id := range permission1Ids {
  613. userReadPermission1 = append(userReadPermission1, &models.UserReadPermission1{
  614. UserReadRecordId: int(insertId),
  615. ChartPermissionId: id,
  616. PermissionName: chartMap[id].PermissionName,
  617. })
  618. }
  619. err = models.UserReadPermission1MultiInsert(userReadPermission1)
  620. if err != nil {
  621. br.Msg = "添加阅读记录失败"
  622. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  623. return
  624. }
  625. userReadPermission2 := make([]*models.UserReadPermission2, 0)
  626. for _, id := range permission2Ids {
  627. userReadPermission2 = append(userReadPermission2, &models.UserReadPermission2{
  628. UserReadRecordId: int(insertId),
  629. ChartPermissionId: id,
  630. })
  631. }
  632. err = models.UserReadPermission2MultiInsert(userReadPermission2)
  633. if err != nil {
  634. br.Msg = "添加阅读记录失败"
  635. br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
  636. return
  637. }
  638. count, err := models.GetUserReadRecordCountByReportPdfIdAndUserId(req.ReportPdfId, user.UserId)
  639. if err != nil {
  640. br.Msg = "添加阅读记录失败"
  641. br.ErrMsg = "获取阅读记录失败,Err:" + err.Error()
  642. return
  643. }
  644. if count > 1 {
  645. err = models.UpdateReportPdfPv(req.ReportPdfId)
  646. if err != nil {
  647. br.Msg = "添加研报阅读记录失败"
  648. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  649. return
  650. }
  651. } else {
  652. err = models.UpdateReportPdfUvAndPv(req.ReportPdfId)
  653. if err != nil {
  654. br.Msg = "添加研报阅读记录失败"
  655. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  656. return
  657. }
  658. }
  659. } else {
  660. // 如果存在就计算停留时间
  661. userRecord, err := models.GetUserReadRecordListById(req.RecordId)
  662. if err != nil {
  663. if err.Error() == utils.ErrNoRow() {
  664. br.Msg = "更新阅读记录不存在"
  665. return
  666. }
  667. br.Msg = "更新阅读记录失败"
  668. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  669. return
  670. }
  671. stayTime := curTime.Unix() - int64(userRecord.Timestamp)
  672. stayTimeStr := utils.SecondsToHMS(stayTime)
  673. err = models.UpdateUserReadRecordById(req.RecordId, int(curTime.Unix()), int(stayTime), stayTimeStr)
  674. if err != nil {
  675. br.Msg = "更新阅读记录失败"
  676. br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
  677. return
  678. }
  679. }
  680. resp := new(response.UserReadRecordResp)
  681. resp.RecordId = insertId
  682. br.Msg = "添加阅读记录成功"
  683. br.Ret = 200
  684. br.Success = true
  685. br.Data = resp
  686. }
  687. // AreaCodeList
  688. // @Title 手机号区号列表
  689. // @Description 手机号区号列表
  690. // @Success 200 Ret=200 获取成功
  691. // @router /area_code/list [get]
  692. func (this *UserController) AreaCodeList() {
  693. br := new(models.BaseResponse).Init()
  694. defer func() {
  695. if br.ErrMsg == "" {
  696. br.IsSendEmail = false
  697. }
  698. this.Data["json"] = br
  699. this.ServeJSON()
  700. }()
  701. resp := make([]response.AreaCodeListResp, 0)
  702. confValue, e := models.GetConfigDetailByCode(models.ConfAreaCodeListKey)
  703. if e != nil {
  704. br.Msg = "获取失败"
  705. br.ErrMsg = "获取手机号区号配置失败, Err: " + e.Error()
  706. return
  707. }
  708. if confValue == "" {
  709. br.Msg = "获取失败"
  710. br.ErrMsg = "手机号区号配置为空"
  711. return
  712. }
  713. if e := json.Unmarshal([]byte(confValue), &resp); e != nil {
  714. br.Msg = "获取失败"
  715. br.ErrMsg = "手机号区号配置有误"
  716. return
  717. }
  718. br.Data = resp
  719. br.Ret = 200
  720. br.Success = true
  721. br.Msg = "获取成功"
  722. }
  723. // 用户品种信息
  724. // @Title 用户信息
  725. // @Description 用户信息
  726. // @Success 200 Ret=200 获取成功
  727. // @router /info [get]
  728. func (this *UserAuthController) Info() {
  729. br := new(models.BaseResponse).Init()
  730. defer func() {
  731. this.Data["json"] = br
  732. this.ServeJSON()
  733. }()
  734. user := this.User
  735. if user == nil {
  736. user = &models.User{}
  737. }
  738. var department *models.SysDepartment
  739. var err error
  740. if user.SellerDepartmentId > 0 {
  741. department, err = models.GetSysDepartmentById(user.SellerDepartmentId)
  742. if err != nil {
  743. if err.Error() == utils.ErrNoRow() {
  744. br.Msg = "查询失败"
  745. br.ErrMsg = "部门不存在,系统异常,Err:" + err.Error()
  746. return
  747. }
  748. br.Msg = "查询失败"
  749. br.ErrMsg = "查询失败,系统异常,Err:" + err.Error()
  750. return
  751. }
  752. } else {
  753. department = new(models.SysDepartment)
  754. }
  755. private, err := services.GetPrivateChartPermissionList()
  756. if err != nil {
  757. br.Msg = "查询失败"
  758. br.ErrMsg = "查询失败,系统异常,Err:" + err.Error()
  759. return
  760. }
  761. if private.Ret != 200 {
  762. br.Msg = private.Msg
  763. br.ErrMsg = private.ErrMsg
  764. return
  765. }
  766. userPermissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  767. if err != nil {
  768. br.Msg = "查看权限失败"
  769. br.ErrMsg = "查看权限失败,系统异常,Err:" + err.Error()
  770. return
  771. }
  772. userPermissionMap := make(map[int]struct{})
  773. for _, v := range userPermissionIds {
  774. userPermissionMap[v] = struct{}{}
  775. }
  776. var hasPermission string
  777. for _, v := range private.Data {
  778. for _, vv := range v.Child {
  779. if _, ok := userPermissionMap[vv.ChartPermissionId]; ok {
  780. hasPermission = "私有权限"
  781. break
  782. }
  783. }
  784. }
  785. if hasPermission == "" || user.Status == utils.UserStatusNo {
  786. hasPermission = "公有权限"
  787. }
  788. miniconf, err := models.GetMiniConf()
  789. if err != nil {
  790. br.Msg = "获取配置失败"
  791. br.ErrMsg = "获取配置失败,系统异常,Err:" + err.Error()
  792. return
  793. }
  794. userView := &models.UserView{
  795. Headimgurl: miniconf["Logo"],
  796. RealName: user.RealName,
  797. Phone: user.Phone,
  798. AreaCode: user.AreaCode,
  799. SellerDepartmentName: department.SysDepartmentName,
  800. HasPermission: hasPermission,
  801. Email: user.Email,
  802. Componey: user.Company,
  803. IsRegistered: user.IsRegistered,
  804. Status: user.Status,
  805. ValidEndTime: user.ValidEndTime.Format(utils.FormatDate),
  806. }
  807. // 固定电话
  808. userView.SellerDepartmentPhone = append(userView.SellerDepartmentPhone,
  809. []string{"021-63123065", "021-63123067"}...)
  810. if user.ValidEndTime.IsZero() {
  811. userView.ValidEndTime = ""
  812. }
  813. br.Data = userView
  814. br.Msg = "查询成功"
  815. br.Success = true
  816. br.Ret = 200
  817. }
  818. // PermissionList
  819. // @Title 用户个人权限列表
  820. // @Description 用户个人权限列表
  821. // @Param request body models.LoginReq true "type json string"
  822. // @Success 200 {object} []models.ChartPermissionTreeView
  823. // @router /permission/list [get]
  824. func (this *UserAuthController) PermissionList() {
  825. br := new(models.BaseResponse).Init()
  826. defer func() {
  827. this.Data["json"] = br
  828. this.ServeJSON()
  829. }()
  830. user := this.User
  831. permissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  832. if err != nil {
  833. br.Msg = "查询用户权限失败"
  834. br.ErrMsg = "查询用户权限失败,系统异常,Err:" + err.Error()
  835. return
  836. }
  837. userPermissionMap := make(map[int]struct{})
  838. for _, v := range permissionIds {
  839. userPermissionMap[v] = struct{}{}
  840. }
  841. // 查询公有权限
  842. var publicView []*models.ChartPermissionTreeView
  843. publicResp, err := services.GetPublicChartPermissionList()
  844. if err != nil {
  845. br.Msg = "查询公有权限失败"
  846. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  847. }
  848. if publicResp.Ret != 200 {
  849. br.Msg = publicResp.Msg
  850. br.ErrMsg = publicResp.ErrMsg
  851. return
  852. }
  853. publicView = publicResp.Data
  854. // 如果是正式用户,则查询私有权限
  855. var privateView []*models.ChartPermissionTreeView
  856. if user.Status == utils.UserStatusFormal {
  857. privateResp, err := services.GetPrivateChartPermissionList()
  858. if err != nil {
  859. br.Msg = "查询私有权限失败"
  860. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  861. }
  862. if privateResp.Ret != 200 {
  863. br.Msg = privateResp.Msg
  864. br.ErrMsg = privateResp.ErrMsg
  865. return
  866. }
  867. for _, v := range privateResp.Data {
  868. IsAdd := false
  869. curPermissionView := &models.ChartPermissionTreeView{
  870. ChartPermissionId: v.ChartPermissionId,
  871. PermissionName: v.PermissionName,
  872. IsPublic: v.IsPublic,
  873. Sort: v.Sort,
  874. Child: make([]*models.ChartPermissionTreeView, 0),
  875. }
  876. for _, vv := range v.Child {
  877. if _, ok := userPermissionMap[vv.ChartPermissionId]; ok {
  878. curPermissionView.Child = append(curPermissionView.Child, vv)
  879. IsAdd = true
  880. }
  881. }
  882. if IsAdd {
  883. privateView = append(privateView, curPermissionView)
  884. }
  885. }
  886. }
  887. privateAloneView := make([]*models.ChartPermissionTreeView, 0)
  888. for _, vi := range privateView {
  889. isHas := false
  890. for _, vb := range publicView {
  891. if vi.ChartPermissionId == vb.ChartPermissionId {
  892. isHas = true
  893. vb.Child = append(vb.Child, vi.Child...)
  894. break
  895. }
  896. }
  897. if !isHas {
  898. privateAloneView = append(privateAloneView, vi)
  899. }
  900. }
  901. publicView = append(publicView, privateAloneView...)
  902. for _, vi := range publicView {
  903. vi.ChartPermissionType = utils.PermissionTypeEta
  904. for _, vi := range vi.Child {
  905. vi.ChartPermissionType = utils.PermissionTypeEta
  906. }
  907. }
  908. br.Data = publicView
  909. br.Msg = "查询成功"
  910. br.Success = true
  911. br.Ret = 200
  912. }
  913. // PermissionList
  914. // @Title 用户已购权限列表
  915. // @Description 用户已购权限列表
  916. // @Param request body models.LoginReq true "type json string"
  917. // @Success 200 {object} []models.ChartPermissionTreeView
  918. // @router /purchased/list [get]
  919. func (this *UserAuthController) PurchasedPermission() {
  920. br := new(models.BaseResponse).Init()
  921. defer func() {
  922. this.Data["json"] = br
  923. this.ServeJSON()
  924. }()
  925. user := this.User
  926. permissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
  927. if err != nil {
  928. br.Msg = "查询用户权限失败"
  929. br.ErrMsg = "查询用户权限失败,系统异常,Err:" + err.Error()
  930. return
  931. }
  932. permissionMap := make(map[int]struct{})
  933. for _, v := range permissionIds {
  934. permissionMap[v] = struct{}{}
  935. }
  936. // 查询私有权限
  937. privateResp, err := services.GetPrivateChartPermissionList()
  938. if err != nil {
  939. br.Msg = "查询私有权限失败"
  940. br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
  941. }
  942. if privateResp.Ret != 200 {
  943. br.Msg = privateResp.Msg
  944. br.ErrMsg = privateResp.ErrMsg
  945. return
  946. }
  947. var privateView []*models.ChartPermissionTreeView
  948. for _, v := range privateResp.Data {
  949. IsAdd := false
  950. curPermissionView := &models.ChartPermissionTreeView{
  951. ChartPermissionId: v.ChartPermissionId,
  952. PermissionName: v.PermissionName,
  953. IsPublic: v.IsPublic,
  954. Sort: v.Sort,
  955. Child: make([]*models.ChartPermissionTreeView, 0),
  956. ChartPermissionType: utils.PermissionTypeEta,
  957. }
  958. for _, vv := range v.Child {
  959. if _, ok := permissionMap[vv.ChartPermissionId]; ok && vv.ParentId != 0 {
  960. vv.ChartPermissionType = utils.PermissionTypeEta
  961. curPermissionView.Child = append(curPermissionView.Child, vv)
  962. IsAdd = true
  963. }
  964. }
  965. if IsAdd {
  966. privateView = append(privateView, curPermissionView)
  967. }
  968. }
  969. for _, vi := range privateView {
  970. vi.Child = append(vi.Child, &models.ChartPermissionTreeView{
  971. ChartPermissionId: -1,
  972. PermissionName: utils.MINI_CUSTOM_PERMISSION_NAME,
  973. IsPublic: 0,
  974. Sort: 0,
  975. ChartPermissionType: utils.ReportTypePdf,
  976. })
  977. }
  978. br.Data = privateView
  979. br.Msg = "查询成功"
  980. br.Success = true
  981. br.Ret = 200
  982. }
  983. // UserInfoSave
  984. // @Title 用户信息补充
  985. // @Description 用户信息补充
  986. // @Param request body request.UserEidtReq true "type json string"
  987. // @Success Ret=200 保存成功
  988. // @router /info/save [post]
  989. func (this *UserAuthController) UserInfoSave() {
  990. br := new(models.BaseResponse).Init()
  991. defer func() {
  992. this.Data["json"] = br
  993. this.ServeJSON()
  994. }()
  995. var req request.UserInfoSaveReq
  996. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  997. if err != nil {
  998. br.Msg = "参数解析异常!"
  999. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1000. return
  1001. }
  1002. user := this.User
  1003. if user == nil {
  1004. br.Msg = "用户不存在"
  1005. return
  1006. }
  1007. if user.SellerDepartmentId > 0 {
  1008. req.SysDepartmentId = user.SellerDepartmentId
  1009. }
  1010. if user.RealName != "" {
  1011. req.RealName = user.RealName
  1012. }
  1013. msg, err := services.SaveUserInfo(user.UserId, req.SysDepartmentId, req.RealName)
  1014. if err != nil {
  1015. if msg == "" {
  1016. br.Msg = "保存失败"
  1017. } else {
  1018. br.Msg = msg
  1019. }
  1020. br.ErrMsg = "保存失败,系统异常,Err:" + err.Error()
  1021. return
  1022. }
  1023. if msg != "" {
  1024. br.Msg = msg
  1025. return
  1026. }
  1027. br.Msg = "用户信息保存成功"
  1028. br.Success = true
  1029. br.Ret = 200
  1030. }
  1031. // ApplyPermission
  1032. // @Title 用户申请认证
  1033. // @Description 用户申请认证
  1034. // @Success Ret=200 获取成功
  1035. // @router /apply/permission [post]
  1036. func (this *UserAuthController) ApplyPermission() {
  1037. br := new(models.BaseResponse).Init()
  1038. defer func() {
  1039. this.Data["json"] = br
  1040. this.ServeJSON()
  1041. }()
  1042. user := this.User
  1043. if user.ApplyStatus == 1 {
  1044. br.Ret = 200
  1045. br.Msg = "您已提交过认证申请,请等待!"
  1046. br.Success = true
  1047. return
  1048. }
  1049. user.ApplyStatus = 1
  1050. user.ApplyTime = time.Now()
  1051. err := user.Update([]string{"apply_status", "apply_time"})
  1052. if err != nil {
  1053. br.Msg = "提交认证申请失败"
  1054. br.ErrMsg = "提交认证申请失败,系统异常,Err:" + err.Error()
  1055. return
  1056. }
  1057. br.Msg = "认证申请,需要1~3个工作日"
  1058. br.Success = true
  1059. br.Ret = 200
  1060. }
  1061. // DepartmentList
  1062. // @Title 营业部门列表
  1063. // @Description 营业部门列表
  1064. // @Success 200 {object} []*models.SysDepartment
  1065. // @router /department/list [get]
  1066. func (this *UserController) DepartmentList() {
  1067. br := new(models.BaseResponse).Init()
  1068. defer func() {
  1069. this.Data["json"] = br
  1070. this.ServeJSON()
  1071. }()
  1072. department, err := models.GetSysDepartmentByName("营业部")
  1073. if err != nil {
  1074. if err.Error() == utils.ErrNoRow() {
  1075. br.Msg = "营业部不存在"
  1076. return
  1077. }
  1078. br.Msg = "获取营业部失败"
  1079. br.ErrMsg = "获取营业部失败,系统异常,Err:" + err.Error()
  1080. return
  1081. }
  1082. depList, err := models.GetChildSysDepartmentByParentId(department.SysDepartmentId)
  1083. if err != nil {
  1084. br.Msg = "获取营业部门失败"
  1085. br.ErrMsg = "获取子部门失败,系统异常,Err:" + err.Error()
  1086. return
  1087. }
  1088. br.Msg = "获取成功"
  1089. br.Data = depList
  1090. br.Success = true
  1091. br.Ret = 200
  1092. }