67 lines
3.2 KiB
Go
67 lines
3.2 KiB
Go
package model
|
||
|
||
import (
|
||
"catface/app/global/variable"
|
||
"catface/app/utils/gorm_v2"
|
||
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
func CreateAnimalFactory(sqlType string) *Animal {
|
||
return &Animal{BaseModel: BaseModel{DB: UseDbConn(sqlType)}}
|
||
}
|
||
|
||
type Animal struct {
|
||
BaseModel // 假设 BaseModel 中不需要添加 omitempty 标签
|
||
Name string `gorm:"type:varchar(20)" json:"name,omitempty"` // 名称
|
||
Birthday string `json:"birthday,omitempty"` // 生日
|
||
Gender uint8 `json:"gender,omitempty"` // 性别
|
||
Breed uint8 `json:"breed,omitempty"` // 品种
|
||
Sterilization uint8 `json:"sterilization,omitempty"` // 1 不明 2 未绝育 3 已绝育
|
||
NickName string `gorm:"type:varchar(31)" json:"nick_name,omitempty"` // 别称，辅助查询；存储上采取 , 间隔符的方式; VARCHAR 会比较合适
|
||
Status uint8 `json:"status,omitempty"` // 状态
|
||
Description string `gorm:"column:description;type:varchar(255)" json:"description,omitempty"` // 简明介绍
|
||
Avatar string `gorm:"type:varchar(10)" json:"avatar,omitempty"` // 缩略图 url，为 Go 获取 Photo 之后压缩处理后的图像，单独存储。
|
||
AvatarHeight uint16 `json:"avatar_height,omitempty"` // 为了方便前端在加载图像前的骨架图 & 瀑布流展示。
|
||
AvatarWidth uint16 `json:"avatar_width,omitempty"` // 为了方便前端在加载图像前的骨架图 & 瀑布流展示。
|
||
HeadImg string `gorm:"type:varchar(10)" json:"head_img,omitempty"` // Head 默认处理为正方形。
|
||
Photos string `gorm:"type:varchar(100)" json:"photos,omitempty"` // 图片数组
|
||
Latitude float64 `json:"latitude,omitempty"` // POI 位置相关
|
||
Longitude float64 `json:"longitude,omitempty"` // POI 位置相关
|
||
ActivityRadius uint64 `json:"activity_radius,omitempty"`
|
||
Tags string `json:"tags,omitempty"` // 活动半径
|
||
}
|
||
|
||
func (a *Animal) TableName() string {
|
||
return "animals"
|
||
}
|
||
|
||
func (a *Animal) Show(attrs []string, gender []uint8, breed []uint8, sterilzation []uint8, status []uint8, num int, skip int) (temp []Animal) {
|
||
db := a.DB.Table(a.TableName()).Limit(int(num)).Offset(int(skip)).Select(attrs)
|
||
|
||
// 创建条件映射
|
||
conditions := map[string][]uint8{
|
||
"gender": gender,
|
||
"breed": breed,
|
||
"sterilization": sterilzation,
|
||
"status": status,
|
||
}
|
||
|
||
db = gorm_v2.BuildWhere(db, conditions)
|
||
|
||
err := db.Find(&temp).Error
|
||
if err != nil {
|
||
variable.ZapLog.Error("Animal Show Error", zap.Error(err))
|
||
}
|
||
return
|
||
}
|
||
|
||
func (a *Animal) ShowByID(id int) *Animal {
|
||
var temp Animal
|
||
err := a.DB.Table(a.TableName()).Model(&temp).Where("id = ?", id).Scan(&temp).Error
|
||
if err != nil {
|
||
variable.ZapLog.Error("Animal ShowByID Error", zap.Error(err))
|
||
}
|
||
return &temp
|
||
}
|