Havoc412
4d2f304ab7
- 优化了 prefer 模式下的查询流程，减少了冗余代码 - 重构了 getPreferCatsId 函数，增加了 skip 参数支持分页查询 - 修改了 Encounter 模型中的 EncounteredCats 方法，支持 DISTINCT 查询和分页 - 提高了查询效率，减少了数据库访问次数
238 lines
9.1 KiB
Go
238 lines
9.1 KiB
Go
package web
|
||
|
||
import (
|
||
"catface/app/global/consts"
|
||
"catface/app/global/errcode"
|
||
"catface/app/global/variable"
|
||
"catface/app/http/validator/core/data_transfer"
|
||
"catface/app/model"
|
||
"catface/app/model_es"
|
||
"catface/app/service/animals/curd"
|
||
"catface/app/service/upload_file"
|
||
"catface/app/utils/query_handler"
|
||
"catface/app/utils/redis_factory"
|
||
"catface/app/utils/response"
|
||
"os"
|
||
"path/filepath"
|
||
"strconv"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
type Animals struct { // INFO 起到一个标记的作用，这样 web.xxx 的时候不同模块就不会命名冲突了。
|
||
}
|
||
|
||
func (a *Animals) List(context *gin.Context) {
|
||
// 1. Get Params
|
||
attrs := context.GetString(consts.ValidatorPrefix + "attrs")
|
||
gender := context.GetString(consts.ValidatorPrefix + "gender")
|
||
breed := context.GetString(consts.ValidatorPrefix + "breed")
|
||
sterilization := context.GetString(consts.ValidatorPrefix + "sterilization")
|
||
status := context.GetString(consts.ValidatorPrefix + "status")
|
||
department := context.GetString(consts.ValidatorPrefix + "department")
|
||
num := int(context.GetFloat64(consts.ValidatorPrefix + "num"))
|
||
skip := int(context.GetFloat64(consts.ValidatorPrefix + "skip"))
|
||
userId := context.GetFloat64(consts.ValidatorPrefix + "user_id")
|
||
|
||
mode := context.GetString(consts.ValidatorPrefix + "mode")
|
||
|
||
// TAG prefer MODE 查询模式。
|
||
var redis_preferCatsId []int64
|
||
var key int64
|
||
var animalsWithLike []model.AnimalWithLikeList
|
||
if mode == consts.AnimalPreferMode {
|
||
key = int64(context.GetFloat64(consts.ValidatorPrefix + "key"))
|
||
|
||
redisClient := redis_factory.GetOneRedisClient()
|
||
defer redisClient.ReleaseOneRedisClient()
|
||
if key != 0 {
|
||
redis_preferCatsId, _ = redisClient.Int64sFromList(redisClient.Execute("lrange", key, 0, -1))
|
||
} else {
|
||
key = variable.SnowFlake.GetId()
|
||
}
|
||
|
||
if len(redis_preferCatsId) == skip {
|
||
preferCatsId, preferCats, _ := getPreferCatsId(int(userId), num, skip, attrs)
|
||
if len(preferCatsId) > 0 {
|
||
redis_preferCatsId = append(redis_preferCatsId, preferCatsId...)
|
||
animalsWithLike = append(animalsWithLike, preferCats...)
|
||
}
|
||
|
||
if _, err := redisClient.String(redisClient.Execute("lpush", key, redis_preferCatsId)); err != nil {
|
||
}
|
||
}
|
||
}
|
||
|
||
// 计算还需要多少动物
|
||
num -= len(animalsWithLike)
|
||
skip = max(0, skip-len(redis_preferCatsId))
|
||
if num > 0 {
|
||
additionalAnimals := curd.CreateAnimalsCurdFactory().List(attrs, gender, breed, sterilization, status, department, redis_preferCatsId, num, skip, int(userId))
|
||
// 将 additionalAnimals 整合到 animalsWithLike 的后面
|
||
animalsWithLike = append(animalsWithLike, additionalAnimals...)
|
||
}
|
||
|
||
if animalsWithLike != nil {
|
||
response.Success(context, consts.CurdStatusOkMsg, gin.H{
|
||
"animals": animalsWithLike,
|
||
"key": key,
|
||
})
|
||
} else {
|
||
response.Fail(context, errcode.AnimalNoFind, errcode.ErrMsg[errcode.AnimalNoFind], errcode.ErrMsgForUser[errcode.AnimalNoFind])
|
||
}
|
||
}
|
||
|
||
// UPDATE 就先简单一些，主要就依靠 encounter - animal_id 来获取一个目标。
|
||
func getPreferCatsId(userId, num, skip int, attrs string) (ids []int64, list []model.AnimalWithLikeList, err error) {
|
||
// STAGE - 1 模块一，无视过滤条件，获取路遇“过”的 id 列表；先获取 ID，然后再去查询细节信息。
|
||
ids, err = model.CreateEncounterFactory("").EncounteredCats(userId, num, skip)
|
||
|
||
if err == nil && len(ids) > 0 {
|
||
attrsSlice := query_handler.StringToStringArray(attrs)
|
||
attrsSlice = append(attrsSlice, "id")
|
||
|
||
animalMap := make(map[int64]model.Animal, len(ids))
|
||
animals := model.CreateAnimalFactory("").ShowByIDs(ids, attrsSlice...)
|
||
|
||
for _, v := range animals {
|
||
animalMap[v.Id] = v
|
||
}
|
||
|
||
// 根据 preferCatsId 的顺序重构最终结果列表
|
||
for _, id := range ids {
|
||
if animal, ok := animalMap[id]; ok {
|
||
list = append(list, model.AnimalWithLikeList{Animal: animal})
|
||
}
|
||
}
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
// v0.1
|
||
// func (a *Animals) Detail(context *gin.Context) {
|
||
// // 1. Get Params
|
||
// anmId, err := strconv.Atoi(context.Param("anm_id"))
|
||
// // 2. Select & Filter
|
||
// var animal model.Animal
|
||
// err = variable.GormDbMysql.Table("animals").Model(&animal).Where("id = ?", anmId).Scan(&animal).Error // TIP GORM.First 采取默认的
|
||
// if err != nil {
|
||
// response.Fail(context, errcode.ErrAnimalSqlFind, errcode.ErrMsg[errcode.ErrAnimalSqlFind], err) // UPDATE consts ?
|
||
// } else {
|
||
// response.Success(context, consts.CurdStatusOkMsg, animal)
|
||
// }
|
||
// }
|
||
|
||
func (a *Animals) Detail(context *gin.Context) {
|
||
// 1. Get Params
|
||
anmId := context.Param("anm_id")
|
||
|
||
animal := curd.CreateAnimalsCurdFactory().Detail(anmId)
|
||
if animal != nil {
|
||
response.Success(context, consts.CurdStatusOkMsg, animal)
|
||
} else {
|
||
response.Fail(context, errcode.AnimalNoFind, errcode.ErrMsg[errcode.AnimalNoFind], "")
|
||
}
|
||
}
|
||
|
||
func (a *Animals) Create(context *gin.Context) {
|
||
userId := strconv.Itoa(int(context.GetFloat64(consts.ValidatorPrefix + "user_id")))
|
||
// STAGE-1 Get Params
|
||
photos := data_transfer.GetStringSlice(context, "photos")
|
||
if len(photos) > 0 {
|
||
avatar := photos[0]
|
||
avatarWidth := variable.ConfigYml.GetFloat64("FileUploadSetting.AvatarWidth")
|
||
|
||
srcPath := filepath.Join(variable.ConfigYml.GetString("FileUploadSetting.UploadFileSavePath"), "catsPhotos", "hum_"+userId, avatar)
|
||
dstPath := filepath.Join(variable.ConfigYml.GetString("FileUploadSetting.UploadFileSavePath"), "catsAvatar", avatar)
|
||
avatarHeight, err := upload_file.ResizeImage(srcPath, dstPath, int(avatarWidth))
|
||
if err != nil {
|
||
response.Fail(context, consts.FilesUploadFailCode, consts.FilesUploadFailMsg, "")
|
||
return
|
||
}
|
||
context.Set(consts.ValidatorPrefix+"avatar", avatar)
|
||
context.Set(consts.ValidatorPrefix+"avatar_height", float64(avatarHeight))
|
||
context.Set(consts.ValidatorPrefix+"avatar_width", float64(avatarWidth))
|
||
}
|
||
poi := context.GetStringMap(consts.ValidatorPrefix + "poi")
|
||
if poi != nil {
|
||
// 感觉这里就是获取信息之后，然后解析后再存储，方便后续 Model 直接绑定到数据。
|
||
latitude := poi["latitude"].(float64)
|
||
longitude := poi["longitude"].(float64)
|
||
context.Set(consts.ValidatorPrefix+"latitude", latitude)
|
||
context.Set(consts.ValidatorPrefix+"longitude", longitude)
|
||
}
|
||
health := context.GetStringMap(consts.ValidatorPrefix + "health")
|
||
if health != nil {
|
||
sterilization := health["sterilization"].(float64)
|
||
vaccination := health["vaccination"].(float64)
|
||
deworming := health["deworming"].(float64)
|
||
context.Set(consts.ValidatorPrefix+"sterilization", sterilization)
|
||
context.Set(consts.ValidatorPrefix+"vaccination", vaccination)
|
||
context.Set(consts.ValidatorPrefix+"deworming", deworming)
|
||
}
|
||
extra := context.GetStringMap(consts.ValidatorPrefix + "extra")
|
||
var nickNames []string
|
||
var tags []string
|
||
if extra != nil {
|
||
context.Set(consts.ValidatorPrefix+"nick_names", extra["nick_names"])
|
||
context.Set(consts.ValidatorPrefix+"tags", extra["tags"])
|
||
nickNames = data_transfer.GetStringSlice(context, "nick_names")
|
||
tags = data_transfer.GetStringSlice(context, "tags")
|
||
context.Set(consts.ValidatorPrefix+"nick_names_list", nickNames)
|
||
context.Set(consts.ValidatorPrefix+"nick_names", nickNames)
|
||
context.Set(consts.ValidatorPrefix+"tags", tags) // UPDATE 有点冗余，但是不用复杂代码；
|
||
}
|
||
// STAGE-2
|
||
if res, err := data_transfer.ConvertSliceToString(photos); err == nil {
|
||
context.Set(consts.ValidatorPrefix+"photos", res)
|
||
} else {
|
||
response.Fail(context, consts.ValidatorParamsCheckFailCode, consts.ValidatorParamsCheckFailMsg, "")
|
||
return
|
||
}
|
||
if res, err := data_transfer.ConvertSliceToString(nickNames); err == nil {
|
||
context.Set(consts.ValidatorPrefix+"nick_names", res)
|
||
} else {
|
||
response.Fail(context, consts.ValidatorParamsCheckFailCode, consts.ValidatorParamsCheckFailMsg, "")
|
||
return
|
||
}
|
||
if res, err := data_transfer.ConvertSliceToString(tags); err == nil {
|
||
context.Set(consts.ValidatorPrefix+"tags", res)
|
||
} else {
|
||
response.Fail(context, consts.ValidatorParamsCheckFailCode, consts.ValidatorParamsCheckFailMsg, "")
|
||
return
|
||
}
|
||
// STAGE-3
|
||
if animal, ok := model.CreateAnimalFactory("").InsertDate(context); ok {
|
||
// 转移 photos 到 anm；采用 rename dir 的方式
|
||
oldName := filepath.Join(variable.ConfigYml.GetString("FileUploadSetting.UploadFileSavePath"), "catsPhotos", "hum_"+userId)
|
||
newName := filepath.Join(variable.ConfigYml.GetString("FileUploadSetting.UploadFileSavePath"), "catsPhotos", "anm_"+strconv.FormatInt(animal.Id, 10))
|
||
err := os.Rename(oldName, newName)
|
||
if err != nil {
|
||
// TODO 特殊返回，成功了一半？或者需要清空原有的操作？不过感觉这一步几乎不会出错。
|
||
// TODO 或许直接采用 go 会比较好呢？
|
||
}
|
||
|
||
// 2. 将部分数据插入 ES；
|
||
go model_es.CreateAnimalESFactory(&animal).InsertDocument()
|
||
|
||
response.Success(context, consts.CurdStatusOkMsg, gin.H{
|
||
"anm_id": animal.Id,
|
||
})
|
||
} else {
|
||
response.Fail(context, consts.CurdCreatFailCode, consts.CurdCreatFailMsg+",新增错误", "")
|
||
}
|
||
}
|
||
|
||
func (a *Animals) Name(context *gin.Context) {
|
||
attrs := context.GetString(consts.ValidatorPrefix + "attrs")
|
||
name := context.GetString(consts.ValidatorPrefix + "name")
|
||
|
||
animals := curd.CreateAnimalsCurdFactory().ShowByName(attrs, name)
|
||
if animals != nil {
|
||
response.Success(context, consts.CurdStatusOkMsg, animals)
|
||
} else {
|
||
response.Fail(context, errcode.AnimalNoFind, errcode.ErrMsg[errcode.AnimalNoFind], "")
|
||
}
|
||
}
|