90 lines
2.6 KiB
Go
Executable File
90 lines
2.6 KiB
Go
Executable File
package aitexttoimagedata
|
|
|
|
import (
|
|
"time"
|
|
|
|
"91porn-server/common/cachev2"
|
|
"91porn-server/common/db"
|
|
"91porn-server/models"
|
|
"91porn-server/models/v/aitexttoimagemod"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
)
|
|
|
|
type ListRes struct {
|
|
List []aitexttoimagemod.AiTextToImage
|
|
Count int64
|
|
HasNext bool
|
|
}
|
|
|
|
// GetListFromCache 从缓存获取列表数据
|
|
func GetListFromCache(cond bson.M, skip, limit int64, sort bson.D) ([]aitexttoimagemod.AiTextToImage, int64, bool, error) {
|
|
var data ListRes
|
|
_, err := cachev2.Classes().CacheTime(3*time.Minute).AutoListKey(models.AiTextToImage).ResBind(&data).Cache(GetListData, cond, skip, limit, sort)
|
|
return data.List, data.Count, data.HasNext, err
|
|
}
|
|
|
|
// GetListData 获取列表数据并组装适合缓存格式返回
|
|
func GetListData(cond bson.M, skip, limit int64, sort bson.D) (res ListRes, err error) {
|
|
res.List, res.Count, res.HasNext, err = aitexttoimagemod.GetList(cond, skip, limit, sort)
|
|
return
|
|
}
|
|
|
|
// GetInfoFromCache 获取数据详情
|
|
func GetInfoFromCache(id primitive.ObjectID) (res aitexttoimagemod.AiTextToImage, err error) {
|
|
_, err = cachev2.Classes().CacheTime(5*time.Minute).AutoInfoKey(models.AiTextToImage, id.Hex()).ResBind(&res).Cache(aitexttoimagemod.GetInfo, id)
|
|
return
|
|
}
|
|
|
|
// GetAllFromCache 获取所有数据
|
|
func GetAllFromCache(cond bson.M, sort bson.D) (res []aitexttoimagemod.AiTextToImage, err error) {
|
|
_, err = cachev2.Classes().CacheTime(8*time.Minute).AutoListKey(models.AiTextToImage+"-all").ResBind(&res).Cache(aitexttoimagemod.GetAll, cond, sort)
|
|
return
|
|
}
|
|
|
|
// InsertData 插入数据
|
|
func InsertData(t *db.MongoTool, data aitexttoimagemod.AiTextToImage) (primitive.ObjectID, error) {
|
|
newID, err := aitexttoimagemod.Insert(t, data)
|
|
if err != nil {
|
|
return newID, err
|
|
}
|
|
// 清除缓存数据
|
|
_, err = cachev2.Classes().Table(models.AiTextToImage).AutoClear(true, nil)
|
|
if err != nil {
|
|
return newID, err
|
|
}
|
|
return newID, nil
|
|
}
|
|
|
|
// UpdateData 更新数据
|
|
func UpdateData(t *db.MongoTool, id string, data map[string]interface{}) (int64, error) {
|
|
objID, err := primitive.ObjectIDFromHex(id)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
affect, err := aitexttoimagemod.UpdateByID(t, objID, data)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
// 清理缓存
|
|
_, _ = cachev2.Classes().Table(models.AiTextToImage).AutoClear(true, &id)
|
|
|
|
return affect, err
|
|
}
|
|
|
|
// DeleteData 删除数据
|
|
func DeleteData(t *db.MongoTool, id primitive.ObjectID) (err error) {
|
|
err = aitexttoimagemod.DeleteByID(t, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// 清理缓存
|
|
all := "*"
|
|
_, _ = cachev2.Classes().Table(models.AiTextToImage).AutoClear(true, &all)
|
|
return
|
|
}
|