88 lines
2.2 KiB
Go
Executable File
88 lines
2.2 KiB
Go
Executable File
package aitemplatemoduleser
|
|
|
|
import (
|
|
"91porn-server/common/log"
|
|
"91porn-server/models/cache/aitemplatemoduledata"
|
|
"91porn-server/models/commod"
|
|
"91porn-server/models/v/aitemplatemodulemod"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
)
|
|
|
|
type AppQueryListReq struct {
|
|
commod.Page
|
|
}
|
|
|
|
type AppListRes struct {
|
|
Total int64 `json:"total"`
|
|
HasNext bool `json:"hasNext"`
|
|
List []*aitemplatemodulemod.AiTemplateModuleInfo `json:"list"`
|
|
}
|
|
|
|
// GetList 获取列表
|
|
func (p *AppQueryListReq) GetList() AppListRes {
|
|
var res AppListRes
|
|
var err error
|
|
filter := bson.M{}
|
|
|
|
sort := bson.D{{"_id", -1}}
|
|
// 获取列表
|
|
var data []aitemplatemodulemod.AiTemplateModule
|
|
data, res.Total, res.HasNext, err = aitemplatemoduledata.GetListFromCache(filter, int64(p.Skip()), int64(p.Limit()), sort)
|
|
if err != nil {
|
|
log.Error("获取AI模版模块列表列表数据错误", log.Any("Params", *p), log.E(err))
|
|
return res
|
|
}
|
|
res.List = aitemplatemoduledata.FormatAppDataList(data)
|
|
|
|
return res
|
|
}
|
|
|
|
type AppQueryAllListReq struct {
|
|
}
|
|
|
|
type AppAllListRes struct {
|
|
List []*aitemplatemodulemod.AiTemplateModuleInfo `json:"list"`
|
|
}
|
|
|
|
// GetList 获取列表
|
|
func (p *AppQueryAllListReq) GetList() AppAllListRes {
|
|
var res AppAllListRes
|
|
var err error
|
|
filter := bson.M{"status": 1}
|
|
sort := bson.D{{"sortNum", -1}}
|
|
// 获取列表
|
|
data, err := aitemplatemodulemod.GetAll(filter, sort)
|
|
if err != nil {
|
|
log.Error("获取AI模版模块列表列表数据错误", log.Any("Params", *p), log.E(err))
|
|
return res
|
|
}
|
|
|
|
res.List = aitemplatemoduledata.FormatAppDataList(data)
|
|
return res
|
|
}
|
|
|
|
type AppQueryInfoReq struct {
|
|
ID string `json:"id" form:"id"` // id
|
|
}
|
|
type AppQueryInfoRes = *aitemplatemodulemod.AiTemplateModuleInfo
|
|
|
|
// GetInfo 获取详情
|
|
func (p *AppQueryInfoReq) GetInfo() (res AppQueryInfoRes, err error) {
|
|
oid, err := primitive.ObjectIDFromHex(p.ID)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
|
|
var item aitemplatemodulemod.AiTemplateModule
|
|
item, err = aitemplatemoduledata.GetInfoFromCache(oid)
|
|
if err != nil {
|
|
log.Error("获取AI模版模块列表详情数据错误", log.Any("ID", p.ID), log.E(err))
|
|
return
|
|
}
|
|
|
|
res = aitemplatemoduledata.FormatAppData(item)
|
|
return
|
|
}
|