12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115 |
- package controllers
- import (
- "encoding/json"
- "eta/eta_mini_api/models"
- "eta/eta_mini_api/models/request"
- "eta/eta_mini_api/models/response"
- "eta/eta_mini_api/services"
- "eta/eta_mini_api/services/go_redis"
- "eta/eta_mini_api/utils"
- "fmt"
- "image/color"
- "strconv"
- "strings"
- "time"
- "github.com/mojocn/base64Captcha"
- )
- type UserController struct {
- BaseCommonController
- }
- type UserAuthController struct {
- BaseAuthController
- }
- // @Title 用户登录接口
- // @Description 用户登录
- // @Param request body models.LoginReq true "type json string"
- // @Success 200 {object} models.LoginResp
- // @router /login [post]
- func (this *UserAuthController) Login() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- session := this.Session
- var req request.LoginReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析失败"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- if req.RealName == "" {
- br.Msg = "请输入姓名"
- return
- }
- if req.SysDepartmentId <= 0 {
- br.Msg = "请选择营业部门"
- return
- }
- switch req.LoginType {
- case 1:
- if req.Phone == "" {
- br.Msg = "请输入手机号"
- br.ErrMsg = "请输入手机号"
- return
- }
- if req.SmsCode == "" {
- br.Msg = "请输入验证码"
- br.ErrMsg = "请输入验证码"
- return
- }
- phone := req.AreaCode + req.Phone
- item, err := models.GetMsgCode(phone, req.SmsCode)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "验证码错误,请重新输入"
- br.ErrMsg = "校验验证码失败,Err:" + err.Error()
- return
- } else {
- br.Msg = "验证码错误,请重新输入"
- br.ErrMsg = "校验验证码失败,Err:" + err.Error()
- return
- }
- }
- if item == nil {
- br.Msg = "验证码错误,请重新输入"
- return
- }
- case 2:
- if req.Email == "" {
- br.Msg = "请输入邮箱"
- br.ErrMsg = "请输入邮箱"
- return
- }
- if req.SmsCode == "" {
- br.Msg = "请输入验证码"
- br.ErrMsg = "请输入验证码"
- return
- }
- item, err := models.GetMsgCode(req.Email, req.SmsCode)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "验证码错误,请重新输入"
- br.ErrMsg = "校验验证码失败,Err:" + err.Error()
- return
- } else {
- br.Msg = "验证码错误,请重新输入"
- br.ErrMsg = "校验验证码失败,Err:" + err.Error()
- return
- }
- }
- if item == nil {
- br.Msg = "验证码错误,请重新输入"
- return
- }
- }
- _, errMsg, err := services.BindUser(session.UnionId, session.OpenId, req.Phone, req.Email, req.AreaCode, req.SysDepartmentId, req.RealName)
- if err != nil && err.Error() != utils.ErrNoRow() {
- br.Msg = "登录失败,系统处理中,请稍后重试"
- if errMsg != "" {
- br.Msg = errMsg
- }
- br.ErrMsg = "登录失败:" + err.Error()
- return
- }
- br.Msg = "登录成功"
- br.Success = true
- br.Ret = 200
- }
- // GenerateCaptcha
- // @Title 生成图形验证码
- // @Description 生成图形验证码
- // @Success 200 Ret=200 获取成功
- // @router /getCaptcha [get]
- func (this *UserController) GenerateCaptcha() {
- br := new(models.BaseResponse).Init()
- defer func() {
- if br.ErrMsg == "" {
- br.IsSendEmail = false
- }
- this.Data["json"] = br
- this.ServeJSON()
- }()
- // 自定义验证码样式
- var driver base64Captcha.Driver
- driverString := base64Captcha.DriverString{
- Height: 60, //高度
- Width: 120, //宽度
- NoiseCount: 0, //干扰数
- ShowLineOptions: 2 | 4, //展示个数
- Length: 4, //长度
- //Source: "1234567890qwertyuioplkjhgfdsazxcvbnm", //验证码随机字符串来源
- Source: "1234567890", //验证码随机字符串来源
- BgColor: &color.RGBA{ // 背景颜色
- R: 0,
- G: 0,
- B: 0,
- A: 0,
- },
- Fonts: []string{"wqy-microhei.ttc"}, // 字体
- }
- driver = driverString.ConvertFonts()
- // 生成验证码
- store := services.CaptchaRedis{}
- captcha := base64Captcha.NewCaptcha(driver, store)
- id, b64s, _, err := captcha.Generate()
- if err != nil {
- br.Msg = "生成失败"
- br.ErrMsg = "生成验证码失败, Err: " + err.Error()
- return
- }
- type CaptchaResult struct {
- Id string
- Base64Blob string
- }
- res := new(CaptchaResult)
- res.Id = id
- res.Base64Blob = b64s
- br.Ret = 200
- br.Success = true
- br.Msg = "获取成功"
- br.Data = res
- }
- // @Title 获取短信/邮箱验证码
- // @Description 用户登录
- // @Param request body models.LoginReq true "type json string"
- // @Success 200 {object} models.LoginResp
- // @router /getVerifyCode [post]
- func (this *UserController) GetVerifyCode() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- var req request.VerifyCodeReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- if req.VerifyType != 1 && req.VerifyType != 2 {
- br.Msg = "验证方式有误"
- br.ErrMsg = fmt.Sprintf("验证方式异常<%d>", req.VerifyType)
- }
- if req.CaptchaId == "" || req.CaptchaCode == "" {
- br.Msg = "请输入图形验证码"
- return
- }
- code := utils.GetRandDigit(6)
- fmt.Println(code)
- switch req.VerifyType {
- case 1:
- if req.AreaCode == "" {
- br.Msg = "请选择区号"
- return
- }
- if req.Phone == "" {
- br.Msg = "请输入手机号"
- return
- }
- if req.AreaCode == utils.TelAreaCodeHome && !utils.ValidateMobileFormatat(req.Phone) {
- br.Msg = "您的手机号输入有误, 请检查"
- return
- }
- phoneKey := utils.CACHE_ACCESS_PHONE_LOGIN_CODE + req.AreaCode + req.Phone
- res, _ := go_redis.RedisInt(phoneKey)
- if res >= 5 {
- br.Msg = "验证码发送太频繁,请稍后重试"
- return
- }
- phoneCountKey := utils.CACHE_ACCESS_PHONE_COUNT_LOGIN_CODE + req.AreaCode + req.Phone
- resCount, _ := go_redis.RedisInt(phoneCountKey)
- if resCount >= utils.VerifyCodeSendLimit {
- br.Msg = fmt.Sprintf("一天最多获取%s次,已超限", strconv.Itoa(utils.VerifyCodeSendLimit))
- return
- }
- store := services.CaptchaRedis{}
- var ok bool
- ok = store.Verify(req.CaptchaId, req.CaptchaCode, true)
- if !ok {
- br.Msg = "图形验证码错误"
- return
- }
- if req.AreaCode == "86" {
- ok = services.SendSmsCode(req.Phone, code)
- }
- if !ok {
- br.Msg = "请检查手机号和区号"
- br.ErrMsg = "短信验证码发送失败"
- return
- } else {
- item := new(models.MsgCode)
- item.OpenId = ""
- item.Code = code
- item.Mobile = req.AreaCode + req.Phone
- item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
- item.CreateTime = time.Now()
- err = item.Insert()
- if err != nil {
- br.Msg = "发送失败"
- br.ErrMsg = "发送失败,Err:" + err.Error()
- return
- }
- br.Msg = "发送成功"
- phoneVerifyCahcheSvc := &services.VerifyCacheIncrService{}
- err = phoneVerifyCahcheSvc.VerifyCacheIncr(phoneKey, 15*int(time.Minute.Seconds()))
- if err != nil {
- utils.FileLog.Info("验证码手机号临时缓存失败", err.Error())
- }
- err = phoneVerifyCahcheSvc.VerifyCacheIncr(phoneCountKey, int(utils.SetKeyExpireToday().Seconds()))
- if err != nil {
- utils.FileLog.Info("验证码手机号当日缓存失败", err.Error())
- }
- }
- case 2:
- if req.Email == "" {
- br.Msg = "请输入邮箱"
- }
- if !utils.ValidateEmailFormatat(req.Email) {
- br.Msg = "您的邮箱格式输入有误, 请检查"
- return
- }
- emailKey := utils.CACHE_ACCESS_EMAIL_LOGIN_CODE + req.Email
- res, _ := go_redis.RedisInt(emailKey)
- if res >= 5 {
- br.Msg = "验证码发送太频繁,请稍后重试"
- return
- }
- emailCountKey := utils.CACHE_ACCESS_EMAIL_COUNT_LOGIN_CODE + req.Email
- resCount, _ := go_redis.RedisInt(emailCountKey)
- if resCount >= utils.VerifyCodeSendLimit {
- br.Msg = fmt.Sprintf("一天最多获取%s次,已超限", strconv.Itoa(utils.VerifyCodeSendLimit))
- return
- }
- store := services.CaptchaRedis{}
- var ok bool
- ok = store.Verify(req.CaptchaId, req.CaptchaCode, true)
- if !ok {
- br.Msg = "图形验证码错误"
- return
- }
- date := time.Now()
- content := "尊敬的用户:</br>本次请求的验证码为:" + code + "(为了保障您账号的安全性,请在15分钟内完成验证。)</br>东吴期货研究团队 </br>" + fmt.Sprintf("%d年%02d月%02d日", date.Year(), date.Month(), date.Day())
- title := "东吴期货登录验证"
- result, err := utils.SendEmailByDw(title, content, req.Email)
- if err != nil {
- br.Msg = "发送失败"
- br.ErrMsg = "发送失败,Err:" + err.Error()
- return
- }
- if result {
- item := new(models.MsgCode)
- item.OpenId = ""
- item.Code = code
- item.Mobile = req.Email
- item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
- item.CreateTime = time.Now()
- err = item.Insert()
- if err != nil {
- br.Msg = "发送失败"
- br.ErrMsg = "发送失败,Err:" + err.Error()
- return
- }
- br.Msg = "发送成功"
- emailVerifyCahcheSvc := &services.VerifyCacheIncrService{}
- err = emailVerifyCahcheSvc.VerifyCacheIncr(emailKey, 15*int(time.Minute.Seconds()))
- if err != nil {
- utils.FileLog.Info("验证码邮箱临时缓存失败, err:", err.Error())
- }
- err = emailVerifyCahcheSvc.VerifyCacheIncr(emailCountKey, int(utils.SetKeyExpireToday().Seconds()))
- if err != nil {
- utils.FileLog.Info("验证码邮箱当日缓存失败, err:", err.Error())
- }
- } else {
- br.Msg = "发送失败"
- }
- }
- br.Ret = 200
- br.Success = true
- br.Msg = "发送成功"
- }
- // @Title 新增报告浏览记录
- // @Description 新增报告浏览记录接口
- // @Param request body models.ReportRecordReq true "type json string"
- // @Success 200 新增成功
- // @router /addReportRecord [post]
- func (this *UserAuthController) AddReportRecord() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- user := this.User
- if user == nil {
- br.Msg = "请登录"
- br.ErrMsg = "请登录,用户信息为空"
- br.Ret = 403
- return
- }
- var req request.ReportRecordReq
- if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- if req.ReportId <= 0 {
- br.Msg = "参数错误"
- br.ErrMsg = "参数错误,报告id小于等于0"
- return
- }
- reportResp, err := services.GetReportDetail(req.ReportId, user.UserId)
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "获取研报错误,Err:" + err.Error()
- return
- }
- if reportResp.Ret != 200 {
- br.Msg = reportResp.Msg
- br.ErrMsg = reportResp.ErrMsg
- return
- }
- reportDetail := reportResp.Data.Report
- reportPermissionResp, err := services.GetReportChartPermissionList(req.ReportId)
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "获取研报品种错误,Err:" + err.Error()
- return
- }
- if reportPermissionResp.Ret != 200 {
- br.Msg = reportPermissionResp.Msg
- br.ErrMsg = reportPermissionResp.ErrMsg
- return
- }
- reportPermissionList := reportPermissionResp.Data
- chartPermissionList, err := services.GetChartPermissionAllList()
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "获取研报品种列表失败,Err:" + err.Error()
- return
- }
- chartMap := make(map[int]*models.ChartPermission)
- for _, permission := range chartPermissionList.Data {
- chartMap[permission.ChartPermissionId] = permission
- }
- curTime := time.Now()
- var insertId int64
- if req.RecordId == 0 {
- // 如果不存在就新增一条记录
- permission1Ids := make([]int, 0)
- permission2Ids := make([]int, 0)
- permissionNames := make([]string, 0)
- for _, item := range reportPermissionList {
- curPermission := chartMap[item.ChartPermissionId]
- permission1Ids = append(permission1Ids, curPermission.ParentId)
- permission2Ids = append(permission2Ids, curPermission.ChartPermissionId)
- permissionNames = append(permissionNames, curPermission.PermissionName)
- }
- permission1Ids = utils.Unique(permission1Ids)
- permission2Ids = utils.Unique(permission2Ids)
- permissionNames = utils.Unique(permissionNames)
- userReadRecord := &models.UserReadRecord{
- UserId: user.UserId,
- ReportId: req.ReportId,
- ReportTitle: reportDetail.Title,
- ChartPermissionName: strings.Join(permissionNames, ","),
- ClassifyId1: reportDetail.ClassifyIdFirst,
- ClassifyName1: reportDetail.ClassifyNameFirst,
- ClassifyId2: reportDetail.ClassifyIdSecond,
- ClassifyName2: reportDetail.ClassifyNameSecond,
- ClassifyId3: reportDetail.ClassifyIdThird,
- ClassifyName3: reportDetail.ClassifyNameThird,
- Timestamp: int(curTime.Unix()),
- CreateTime: curTime,
- CreateDate: curTime.Format(utils.FormatDate),
- ReportType: utils.ReportTypeEta,
- }
- insertId, err = userReadRecord.Insert()
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
- return
- }
- userReadPermission1 := make([]*models.UserReadPermission1, 0)
- for _, id := range permission1Ids {
- userReadPermission1 = append(userReadPermission1, &models.UserReadPermission1{
- UserReadRecordId: int(insertId),
- ChartPermissionId: id,
- PermissionName: chartMap[id].PermissionName,
- })
- }
- err = models.UserReadPermission1MultiInsert(userReadPermission1)
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
- return
- }
- userReadPermission2 := make([]*models.UserReadPermission2, 0)
- for _, id := range permission2Ids {
- userReadPermission2 = append(userReadPermission2, &models.UserReadPermission2{
- UserReadRecordId: int(insertId),
- ChartPermissionId: id,
- })
- }
- err = models.UserReadPermission2MultiInsert(userReadPermission2)
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
- return
- }
- } else {
- // 如果存在就计算停留时间
- userRecord, err := models.GetUserReadRecordListById(req.RecordId)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "更新阅读记录不存在"
- return
- }
- br.Msg = "更新阅读记录失败"
- br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
- return
- }
- stayTime := curTime.Unix() - int64(userRecord.Timestamp)
- stayTimeStr := utils.SecondsToHMS(stayTime)
- err = models.UpdateUserReadRecordById(req.RecordId, int(curTime.Unix()), int(stayTime), stayTimeStr)
- if err != nil {
- br.Msg = "更新阅读记录失败"
- br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
- return
- }
- }
- resp := new(response.UserReadRecordResp)
- resp.RecordId = insertId
- br.Msg = "添加阅读记录成功"
- br.Ret = 200
- br.Success = true
- br.Data = resp
- }
- // @Title 新增pdf报告浏览记录
- // @Description 新增pdf报告浏览记录接口
- // @Param request body models.ReportPdfRecordReq true "type json string"
- // @Success 200 新增成功
- // @router /pdf/addReportRecord [post]
- func (this *UserAuthController) AddReportPdfRecord() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- user := this.User
- if user == nil {
- br.Msg = "请登录"
- br.ErrMsg = "请登录,用户信息为空"
- br.Ret = 403
- return
- }
- var req request.ReportPdfRecordReq
- if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- if req.ReportPdfId <= 0 {
- br.Msg = "参数错误"
- br.ErrMsg = "参数错误,报告id小于等于0"
- return
- }
- reportPdf, err := models.GetReportPdfById(req.ReportPdfId)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "报告不存在或已删除"
- return
- }
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "获取研报错误,Err:" + err.Error()
- return
- }
- reportClassifyId := services.GetReportPdfClassify(reportPdf)
- reportPermissionResp, err := services.GetChartPermissionListByClassifyId(reportClassifyId, 0)
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "获取研报品种错误,Err:" + err.Error()
- return
- }
- if reportPermissionResp.Ret != 200 {
- br.Msg = reportPermissionResp.Msg
- br.ErrMsg = reportPermissionResp.ErrMsg
- return
- }
- reportPermissionList := reportPermissionResp.Data
- chartPermissionList, err := services.GetChartPermissionAllList()
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "获取研报品种列表失败,Err:" + err.Error()
- return
- }
- chartMap := make(map[int]*models.ChartPermission)
- for _, permission := range chartPermissionList.Data {
- chartMap[permission.ChartPermissionId] = permission
- }
- curTime := time.Now()
- var insertId int64
- if req.RecordId == 0 {
- // 如果不存在就新增一条记录
- permission1Ids := make([]int, 0)
- permission2Ids := make([]int, 0)
- permissionNames := make([]string, 0)
- for _, item := range reportPermissionList {
- curPermission := chartMap[item.ChartPermissionId]
- permission1Ids = append(permission1Ids, curPermission.ParentId)
- permission2Ids = append(permission2Ids, curPermission.ChartPermissionId)
- permissionNames = append(permissionNames, curPermission.PermissionName)
- }
- permission1Ids = utils.Unique(permission1Ids)
- permission2Ids = utils.Unique(permission2Ids)
- permissionNames = utils.Unique(permissionNames)
- userReadRecord := &models.UserReadRecord{
- UserId: user.UserId,
- ReportId: reportPdf.ReportPdfId,
- ReportTitle: reportPdf.Title,
- ChartPermissionName: strings.Join(permissionNames, ","),
- ClassifyId1: reportPdf.ClassifyIdFirst,
- ClassifyName1: reportPdf.ClassifyNameFirst,
- ClassifyId2: reportPdf.ClassifyIdSecond,
- ClassifyName2: reportPdf.ClassifyNameSecond,
- Timestamp: int(curTime.Unix()),
- CreateTime: curTime,
- CreateDate: curTime.Format(utils.FormatDate),
- ReportType: utils.ReportTypePdf,
- }
- insertId, err = userReadRecord.Insert()
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
- return
- }
- userReadPermission1 := make([]*models.UserReadPermission1, 0)
- for _, id := range permission1Ids {
- userReadPermission1 = append(userReadPermission1, &models.UserReadPermission1{
- UserReadRecordId: int(insertId),
- ChartPermissionId: id,
- PermissionName: chartMap[id].PermissionName,
- })
- }
- err = models.UserReadPermission1MultiInsert(userReadPermission1)
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
- return
- }
- userReadPermission2 := make([]*models.UserReadPermission2, 0)
- for _, id := range permission2Ids {
- userReadPermission2 = append(userReadPermission2, &models.UserReadPermission2{
- UserReadRecordId: int(insertId),
- ChartPermissionId: id,
- })
- }
- err = models.UserReadPermission2MultiInsert(userReadPermission2)
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "添加阅读记录失败,Err:" + err.Error()
- return
- }
- count, err := models.GetUserReadRecordCountByReportPdfIdAndUserId(req.ReportPdfId, user.UserId)
- if err != nil {
- br.Msg = "添加阅读记录失败"
- br.ErrMsg = "获取阅读记录失败,Err:" + err.Error()
- return
- }
- if count > 1 {
- err = models.UpdateReportPdfPv(req.ReportPdfId)
- if err != nil {
- br.Msg = "添加研报阅读记录失败"
- br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
- return
- }
- } else {
- err = models.UpdateReportPdfUvAndPv(req.ReportPdfId)
- if err != nil {
- br.Msg = "添加研报阅读记录失败"
- br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
- return
- }
- }
- } else {
- // 如果存在就计算停留时间
- userRecord, err := models.GetUserReadRecordListById(req.RecordId)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "更新阅读记录不存在"
- return
- }
- br.Msg = "更新阅读记录失败"
- br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
- return
- }
- stayTime := curTime.Unix() - int64(userRecord.Timestamp)
- stayTimeStr := utils.SecondsToHMS(stayTime)
- err = models.UpdateUserReadRecordById(req.RecordId, int(curTime.Unix()), int(stayTime), stayTimeStr)
- if err != nil {
- br.Msg = "更新阅读记录失败"
- br.ErrMsg = "更新阅读记录失败,Err:" + err.Error()
- return
- }
- }
- resp := new(response.UserReadRecordResp)
- resp.RecordId = insertId
- br.Msg = "添加阅读记录成功"
- br.Ret = 200
- br.Success = true
- br.Data = resp
- }
- // AreaCodeList
- // @Title 手机号区号列表
- // @Description 手机号区号列表
- // @Success 200 Ret=200 获取成功
- // @router /area_code/list [get]
- func (this *UserController) AreaCodeList() {
- br := new(models.BaseResponse).Init()
- defer func() {
- if br.ErrMsg == "" {
- br.IsSendEmail = false
- }
- this.Data["json"] = br
- this.ServeJSON()
- }()
- resp := make([]response.AreaCodeListResp, 0)
- confValue, e := models.GetConfigDetailByCode(models.ConfAreaCodeListKey)
- if e != nil {
- br.Msg = "获取失败"
- br.ErrMsg = "获取手机号区号配置失败, Err: " + e.Error()
- return
- }
- if confValue == "" {
- br.Msg = "获取失败"
- br.ErrMsg = "手机号区号配置为空"
- return
- }
- if e := json.Unmarshal([]byte(confValue), &resp); e != nil {
- br.Msg = "获取失败"
- br.ErrMsg = "手机号区号配置有误"
- return
- }
- br.Data = resp
- br.Ret = 200
- br.Success = true
- br.Msg = "获取成功"
- }
- // 用户品种信息
- // @Title 用户信息
- // @Description 用户信息
- // @Success 200 Ret=200 获取成功
- // @router /info [get]
- func (this *UserAuthController) Info() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- user := this.User
- if user == nil {
- user = &models.User{}
- }
- var department *models.SysDepartment
- var err error
- if user.SellerDepartmentId > 0 {
- department, err = models.GetSysDepartmentById(user.SellerDepartmentId)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "查询失败"
- br.ErrMsg = "部门不存在,系统异常,Err:" + err.Error()
- return
- }
- br.Msg = "查询失败"
- br.ErrMsg = "查询失败,系统异常,Err:" + err.Error()
- return
- }
- } else {
- department = new(models.SysDepartment)
- }
- private, err := services.GetPrivateChartPermissionList()
- if err != nil {
- br.Msg = "查询失败"
- br.ErrMsg = "查询失败,系统异常,Err:" + err.Error()
- return
- }
- if private.Ret != 200 {
- br.Msg = private.Msg
- br.ErrMsg = private.ErrMsg
- return
- }
- userPermissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
- if err != nil {
- br.Msg = "查看权限失败"
- br.ErrMsg = "查看权限失败,系统异常,Err:" + err.Error()
- return
- }
- userPermissionMap := make(map[int]struct{})
- for _, v := range userPermissionIds {
- userPermissionMap[v] = struct{}{}
- }
- var hasPermission string
- for _, v := range private.Data {
- for _, vv := range v.Child {
- if _, ok := userPermissionMap[vv.ChartPermissionId]; ok {
- hasPermission = "私有权限"
- break
- }
- }
- }
- if hasPermission == "" || user.Status == utils.UserStatusNo {
- hasPermission = "公有权限"
- }
- miniconf, err := models.GetMiniConf()
- if err != nil {
- br.Msg = "获取配置失败"
- br.ErrMsg = "获取配置失败,系统异常,Err:" + err.Error()
- return
- }
- userView := &models.UserView{
- Headimgurl: miniconf["Logo"],
- RealName: user.RealName,
- Phone: user.Phone,
- AreaCode: user.AreaCode,
- SellerDepartmentName: department.SysDepartmentName,
- HasPermission: hasPermission,
- Email: user.Email,
- Componey: user.Company,
- IsRegistered: user.IsRegistered,
- Status: user.Status,
- ValidEndTime: user.ValidEndTime.Format(utils.FormatDate),
- }
- // 固定电话
- userView.SellerDepartmentPhone = append(userView.SellerDepartmentPhone,
- []string{"021-63123065", "021-63123067"}...)
- if user.ValidEndTime.IsZero() {
- userView.ValidEndTime = ""
- }
- br.Data = userView
- br.Msg = "查询成功"
- br.Success = true
- br.Ret = 200
- }
- // PermissionList
- // @Title 用户个人权限列表
- // @Description 用户个人权限列表
- // @Param request body models.LoginReq true "type json string"
- // @Success 200 {object} []models.ChartPermissionTreeView
- // @router /permission/list [get]
- func (this *UserAuthController) PermissionList() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- user := this.User
- permissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
- if err != nil {
- br.Msg = "查询用户权限失败"
- br.ErrMsg = "查询用户权限失败,系统异常,Err:" + err.Error()
- return
- }
- userPermissionMap := make(map[int]struct{})
- for _, v := range permissionIds {
- userPermissionMap[v] = struct{}{}
- }
- // 查询公有权限
- var publicView []*models.ChartPermissionTreeView
- publicResp, err := services.GetPublicChartPermissionList()
- if err != nil {
- br.Msg = "查询公有权限失败"
- br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
- }
- if publicResp.Ret != 200 {
- br.Msg = publicResp.Msg
- br.ErrMsg = publicResp.ErrMsg
- return
- }
- publicView = publicResp.Data
- // 如果是正式用户,则查询私有权限
- var privateView []*models.ChartPermissionTreeView
- if user.Status == utils.UserStatusFormal {
- privateResp, err := services.GetPrivateChartPermissionList()
- if err != nil {
- br.Msg = "查询私有权限失败"
- br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
- }
- if privateResp.Ret != 200 {
- br.Msg = privateResp.Msg
- br.ErrMsg = privateResp.ErrMsg
- return
- }
- for _, v := range privateResp.Data {
- IsAdd := false
- curPermissionView := &models.ChartPermissionTreeView{
- ChartPermissionId: v.ChartPermissionId,
- PermissionName: v.PermissionName,
- IsPublic: v.IsPublic,
- Sort: v.Sort,
- Child: make([]*models.ChartPermissionTreeView, 0),
- }
- for _, vv := range v.Child {
- if _, ok := userPermissionMap[vv.ChartPermissionId]; ok {
- curPermissionView.Child = append(curPermissionView.Child, vv)
- IsAdd = true
- }
- }
- if IsAdd {
- privateView = append(privateView, curPermissionView)
- }
- }
- }
- privateAloneView := make([]*models.ChartPermissionTreeView, 0)
- for _, vi := range privateView {
- isHas := false
- for _, vb := range publicView {
- if vi.ChartPermissionId == vb.ChartPermissionId {
- isHas = true
- vb.Child = append(vb.Child, vi.Child...)
- break
- }
- }
- if !isHas {
- privateAloneView = append(privateAloneView, vi)
- }
- }
- publicView = append(publicView, privateAloneView...)
- for _, vi := range publicView {
- vi.ChartPermissionType = utils.PermissionTypeEta
- for _, vi := range vi.Child {
- vi.ChartPermissionType = utils.PermissionTypeEta
- }
- }
- br.Data = publicView
- br.Msg = "查询成功"
- br.Success = true
- br.Ret = 200
- }
- // PermissionList
- // @Title 用户已购权限列表
- // @Description 用户已购权限列表
- // @Param request body models.LoginReq true "type json string"
- // @Success 200 {object} []models.ChartPermissionTreeView
- // @router /purchased/list [get]
- func (this *UserAuthController) PurchasedPermission() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- user := this.User
- permissionIds, err := models.GetUserChartPermissionIdByUserId(user.UserId)
- if err != nil {
- br.Msg = "查询用户权限失败"
- br.ErrMsg = "查询用户权限失败,系统异常,Err:" + err.Error()
- return
- }
- permissionMap := make(map[int]struct{})
- for _, v := range permissionIds {
- permissionMap[v] = struct{}{}
- }
- // 查询私有权限
- privateResp, err := services.GetPrivateChartPermissionList()
- if err != nil {
- br.Msg = "查询私有权限失败"
- br.ErrMsg = "查询私有权限失败,系统异常,Err:" + err.Error()
- }
- if privateResp.Ret != 200 {
- br.Msg = privateResp.Msg
- br.ErrMsg = privateResp.ErrMsg
- return
- }
- var privateView []*models.ChartPermissionTreeView
- for _, v := range privateResp.Data {
- IsAdd := false
- curPermissionView := &models.ChartPermissionTreeView{
- ChartPermissionId: v.ChartPermissionId,
- PermissionName: v.PermissionName,
- IsPublic: v.IsPublic,
- Sort: v.Sort,
- Child: make([]*models.ChartPermissionTreeView, 0),
- ChartPermissionType: utils.PermissionTypeEta,
- }
- for _, vv := range v.Child {
- if _, ok := permissionMap[vv.ChartPermissionId]; ok && vv.ParentId != 0 {
- vv.ChartPermissionType = utils.PermissionTypeEta
- curPermissionView.Child = append(curPermissionView.Child, vv)
- IsAdd = true
- }
- }
- if IsAdd {
- privateView = append(privateView, curPermissionView)
- }
- }
- for _, vi := range privateView {
- vi.Child = append(vi.Child, &models.ChartPermissionTreeView{
- ChartPermissionId: -1,
- PermissionName: utils.MINI_CUSTOM_PERMISSION_NAME,
- IsPublic: 0,
- Sort: 0,
- ChartPermissionType: utils.ReportTypePdf,
- })
- }
- br.Data = privateView
- br.Msg = "查询成功"
- br.Success = true
- br.Ret = 200
- }
- // UserInfoSave
- // @Title 用户信息补充
- // @Description 用户信息补充
- // @Param request body request.UserEidtReq true "type json string"
- // @Success Ret=200 保存成功
- // @router /info/save [post]
- func (this *UserAuthController) UserInfoSave() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- var req request.UserInfoSaveReq
- err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
- if err != nil {
- br.Msg = "参数解析异常!"
- br.ErrMsg = "参数解析失败,Err:" + err.Error()
- return
- }
- user := this.User
- if user == nil {
- br.Msg = "用户不存在"
- return
- }
- if user.SellerDepartmentId > 0 {
- req.SysDepartmentId = user.SellerDepartmentId
- }
- if user.RealName != "" {
- req.RealName = user.RealName
- }
- msg, err := services.SaveUserInfo(user.UserId, req.SysDepartmentId, req.RealName)
- if err != nil {
- if msg == "" {
- br.Msg = "保存失败"
- } else {
- br.Msg = msg
- }
- br.ErrMsg = "保存失败,系统异常,Err:" + err.Error()
- return
- }
- if msg != "" {
- br.Msg = msg
- return
- }
- br.Msg = "用户信息保存成功"
- br.Success = true
- br.Ret = 200
- }
- // ApplyPermission
- // @Title 用户申请认证
- // @Description 用户申请认证
- // @Success Ret=200 获取成功
- // @router /apply/permission [post]
- func (this *UserAuthController) ApplyPermission() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- user := this.User
- if user.ApplyStatus == 1 {
- br.Ret = 200
- br.Msg = "您已提交过认证申请,请等待!"
- br.Success = true
- return
- }
- user.ApplyStatus = 1
- user.ApplyTime = time.Now()
- err := user.Update([]string{"apply_status", "apply_time"})
- if err != nil {
- br.Msg = "提交认证申请失败"
- br.ErrMsg = "提交认证申请失败,系统异常,Err:" + err.Error()
- return
- }
- br.Msg = "认证申请,需要1~3个工作日"
- br.Success = true
- br.Ret = 200
- }
- // DepartmentList
- // @Title 营业部门列表
- // @Description 营业部门列表
- // @Success 200 {object} []*models.SysDepartment
- // @router /department/list [get]
- func (this *UserController) DepartmentList() {
- br := new(models.BaseResponse).Init()
- defer func() {
- this.Data["json"] = br
- this.ServeJSON()
- }()
- department, err := models.GetSysDepartmentByName("营业部")
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- br.Msg = "营业部不存在"
- return
- }
- br.Msg = "获取营业部失败"
- br.ErrMsg = "获取营业部失败,系统异常,Err:" + err.Error()
- return
- }
- depList, err := models.GetChildSysDepartmentByParentId(department.SysDepartmentId)
- if err != nil {
- br.Msg = "获取营业部门失败"
- br.ErrMsg = "获取子部门失败,系统异常,Err:" + err.Error()
- return
- }
- br.Msg = "获取成功"
- br.Data = depList
- br.Success = true
- br.Ret = 200
- }
|