150 lines
3.4 KiB
Go
150 lines
3.4 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"hepa-calc-api/api/dao"
|
|
"hepa-calc-api/api/dto"
|
|
"hepa-calc-api/extend/aliyun"
|
|
"hepa-calc-api/global"
|
|
"math/rand"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type PublicService struct {
|
|
}
|
|
|
|
// GetPhoneCode 获取手机验证码
|
|
func (r *PublicService) GetPhoneCode(scene int, phone string) (bool, error) {
|
|
var sendCodeCount int // // 获取验证码最大次数
|
|
var code string // 验证码
|
|
var templateCode string // 短信模版
|
|
|
|
// 登录获取验证码
|
|
if scene == 1 {
|
|
// 验证发送次数
|
|
res, _ := global.Redis.Get(context.Background(), "login_code_count_"+phone).Result()
|
|
if res != "" {
|
|
sendCodeCount, err := strconv.Atoi(res)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if sendCodeCount > 3 {
|
|
// 超出规定时间内最大获取次数
|
|
return false, errors.New("手机号超出规定时间内最大获取次数，请您稍后再试")
|
|
}
|
|
}
|
|
|
|
// 生成随机数
|
|
rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
code = strconv.Itoa(rand.Intn(9000) + 1000)
|
|
|
|
// 模版
|
|
templateCode = "SMS_215344868"
|
|
|
|
sendCodeCount = sendCodeCount + 1
|
|
}
|
|
|
|
if code == "" || templateCode == "" {
|
|
return false, errors.New("验证码发送失败，请您稍后再试")
|
|
}
|
|
|
|
// 发送验证码
|
|
templateParam := make(map[string]interface{})
|
|
templateParam["code"] = code
|
|
err := aliyun.SendSms(phone, templateCode, "获取验证码", templateParam)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
// 记录发送次数
|
|
if sendCodeCount != 0 {
|
|
_, err = global.Redis.Set(context.Background(), "login_code_count_"+phone, time.Now().Unix(), 60*5*time.Second).Result()
|
|
if err != nil {
|
|
return false, errors.New("验证码发送失败，请您稍后再试")
|
|
}
|
|
}
|
|
|
|
// 设置验证码有效期
|
|
_, err = global.Redis.Set(context.Background(), "login_code_"+phone, code, 60*5*time.Second).Result()
|
|
if err != nil {
|
|
return false, errors.New("验证码发送失败，请您稍后再试")
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// GetIndex 获取首页数据
|
|
func (r *PublicService) GetIndex(userId int64) (g *dto.IndexDto, err error) {
|
|
// 获取疾病分类列表
|
|
baseClassDao := dao.BaseClassDao{}
|
|
|
|
maps := make(map[string]interface{})
|
|
maps["class_status"] = 1
|
|
baseClasss, err := baseClassDao.GetBaseClassOrderList(maps)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
questionService := &QuestionService{}
|
|
|
|
// 获取算一算热榜-人气数最高的9个
|
|
hotQuestions, err := questionService.GetHotList()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 获取为你推荐-后台指定的推广
|
|
recommendQuestions, err := questionService.GetRecommendList()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 获取猜你喜欢
|
|
guessUserLikes, err := questionService.GetGuessUserLIkeList(userId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 处理返回值
|
|
g = &dto.IndexDto{}
|
|
|
|
g.LoadBaseClass(baseClasss)
|
|
|
|
g.LoadHotQuestionList(hotQuestions)
|
|
|
|
g.LoadRecommendQuestionList(recommendQuestions)
|
|
|
|
g.LoadGuessUserLikeList(guessUserLikes)
|
|
|
|
return g, nil
|
|
}
|
|
|
|
// GetUserIP 获取用户ip
|
|
func (r *PublicService) GetUserIP(h *http.Request) string {
|
|
forwarded := h.Header.Get("X-FORWARDED-FOR")
|
|
if forwarded != "" {
|
|
return forwarded
|
|
}
|
|
return h.RemoteAddr
|
|
}
|
|
|
|
// GetOssSign 获取oss签名
|
|
func (a *PublicService) GetOssSign(scene int) (*aliyun.GetOssSignResponse, error) {
|
|
var dir string
|
|
if scene == 1 {
|
|
dir = dir + "user/avatar/"
|
|
}
|
|
|
|
if dir == "" {
|
|
return nil, errors.New("获取签名失败")
|
|
}
|
|
|
|
// 生成签名
|
|
ossSign, _ := aliyun.GetOssSign(dir)
|
|
return ossSign, nil
|
|
}
|