package model import ( "catface/app/global/variable" "catface/app/utils/data_bind" "log" "strings" "github.com/gin-gonic/gin" "go.uber.org/zap" "gorm.io/gorm" ) func CreateEncounterFactory(sqlType string) *Encounter { return &Encounter{BaseModel: BaseModel{DB: UseDbConn(sqlType)}} } type Encounter struct { // Encounter 或者称为 post,指的就是 Human 单次的记录。 BaseModel // TAG 外键关联 UsersModelId int64 `gorm:"column:user_id" json:"user_id"` UsersModel *UsersModel `json:"users_model,omitempty"` // INFO 由于 Detail 返回空子段有些麻烦,先尝试采用指针。 // AnimalsId string `gorm:"size:20" json:"animals_id"` // 关联对象存在上限 // INFO 还是采取分表,方便查询。 Title string `gorm:"size:20;column:title" json:"title"` Content string `json:"content"` Level uint8 `json:"level" gorm:"column:level;default:1"` Tags string `json:"tags,omitempty" gorm:"column:tags;size:50"` TagsSlice []string `gorm:"-" json:"tags_list,omitempty"` // TAG Avatar 最好是压缩后的备份图像 Avatar string `gorm:"type:varchar(50)" json:"avatar,omitempty"` // 缩略图 url,为 Go 获取 Photo 之后压缩处理后的图像,单独存储。 AvatarHeight uint16 `json:"avatar_height,omitempty"` // 为了方便前端在加载图像前的骨架图 & 瀑布流展示。 AvatarWidth uint16 `json:"avatar_width,omitempty"` Photos string `gorm:"type:varchar(100)" json:"photos,omitempty"` // 图片数组 PhotosSlice []string `gorm:"-" json:"photos_list,omitempty"` // TIP GORM 忽略 // POI Latitude float64 `json:"latitude,omitempty"` // POI 位置相关 Longitude float64 `json:"longitude,omitempty"` // TODO Comment Num 然后去统计? } func (e *Encounter) TableName() string { return "encounters" } func (e *Encounter) InsertDate(c *gin.Context) (int64, bool) { var tmp Encounter if err := data_bind.ShouldBindFormDataToModel(c, &tmp); err == nil { if res := e.Create(&tmp); res.Error == nil { return tmp.Id, true } else { variable.ZapLog.Error("Encounter 数据新增出错", zap.Error(res.Error)) } } else { variable.ZapLog.Error("Encounter 数据绑定出错", zap.Error(err)) } return 0, false } func formatEncounterList(rows *gorm.DB) (temp []EncounterList, err error) { // 获取底层的 sql.Rows 对象 sqlRows, err := rows.Rows() if err != nil { log.Println("获取 sql.Rows 失败:", err) return nil, err } defer sqlRows.Close() for sqlRows.Next() { var item EncounterList var ueLikeInt int dest := []interface{}{ &item.Id, &item.UserId, &item.Title, &item.Avatar, &item.AvatarHeight, &item.AvatarWidth, &item.UpdatedAt, &item.UserName, &item.UserAvatar, &ueLikeInt, } if err := sqlRows.Scan(dest...); err != nil { log.Println("扫描失败:", err) continue } // STAGE: Set link Status item.Like = ueLikeInt == 1 // STAGE: Check Url exist if item.Avatar == "" { if animal_id, ok := CreateEncounterAnimalLinkFactory("").ShowByEncounterIdFirst(int64(item.Id)); ok { if animal := CreateAnimalFactory("").ShowByID(animal_id); animal != nil { item.Avatar = animal.Avatar item.AvatarHeight = int(animal.AvatarHeight) item.AvatarWidth = int(animal.AvatarWidth) item.UseAnimalAvatar = true } } } // append temp = append(temp, item) } return } func (e *Encounter) Show(num, skip, user_id int, animals_id []int) (temp []EncounterList) { // SATGE - 1:build SQL var sqlBuilder strings.Builder // 构建基础查询 sqlBuilder.WriteString(` SELECT e.id, e.user_id, title, avatar, avatar_height, avatar_width, e.updated_at, user_name, user_avatar, EXISTS ( SELECT 1 FROM encounter_likes l WHERE l.user_id = ? AND l.encounter_id = e.id AND l.is_del = 0 ) AS ue_like FROM encounters e JOIN tb_users u ON e.user_id = u.id`) // 动态插入 animals_id 条件 if len(animals_id) > 0 { sqlBuilder.WriteString(` JOIN encounter_animal_links eal ON e.id = eal.encounter_id WHERE eal.animal_id IN (?)`) } // 添加排序和分页 sqlBuilder.WriteString(` ORDER BY e.updated_at DESC LIMIT ? OFFSET ? `) sql := sqlBuilder.String() // 获取到 SQL; // STAGE - 2: Exe SQL var rows *gorm.DB if len(animals_id) > 0 { rows = e.Raw(sql, user_id, animals_id, num, skip) } else { rows = e.Raw(sql, user_id, num, skip) } if rows.Error != nil { log.Println("查询失败:", rows.Error) return nil } var err error if temp, err = formatEncounterList(rows); err != nil { log.Println("格式化数据失败:", err) return nil } return } func (e *Encounter) ShowByID(id int64) (temp *Encounter, err error) { // 1. search encounter if err = e.Where("id = ?", id).First(&temp).Error; err != nil { return } return // // 2. search user data // user := UsersModel{BaseModel: BaseModel{Id: encounter.UsersModelId}} // if err := user.Select("user_name", "user_avatar").First(&user).Error; err != nil { // return // } // // 3. search animals data // animals_id := query_handler.StringToint64Array(encounter.AnimalsId) // var animals []Animal // if err := e.Model(&animals).Select("id", "avatar", "name").Where("id in (?)", animals_id).Find(&animals).Error; err != nil { // return // } // // TODO 4. 然后整合 // return }