Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

models/cache/aitexttonoveldata/aitexttonovel.go

@@ -0,0 +1,88 @@
+package aitexttonoveldata
+
+import (
+	"91porn-server/common/cachev2"
+	"91porn-server/common/db"
+	"91porn-server/models"
+	"91porn-server/models/v/aitexttonovelmod"
+	"time"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+)
+
+type ListRes struct {
+	List    []aitexttonovelmod.AiTextToNovel
+	Count   int64
+	HasNext bool
+}
+
+// GetListFromCache 从缓存获取列表数据
+func GetListFromCache(cond bson.M, skip, limit int64, sort bson.D) ([]aitexttonovelmod.AiTextToNovel, int64, bool, error) {
+	var data ListRes
+	_, err := cachev2.Classes().CacheTime(10*time.Minute).AutoListKey(models.AiTextToNovel).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 = aitexttonovelmod.GetList(cond, skip, limit, sort)
+	return
+}
+
+// GetInfoFromCache 获取数据详情
+func GetInfoFromCache(id primitive.ObjectID) (res aitexttonovelmod.AiTextToNovel, err error) {
+	_, err = cachev2.Classes().CacheTime(30*time.Minute).AutoInfoKey(models.AiTextToNovel, id.Hex()).ResBind(&res).Cache(aitexttonovelmod.GetInfo, id)
+	return
+}
+
+// GetAllFromCache 获取所有数据
+func GetAllFromCache(cond bson.M, sort bson.D) (res []aitexttonovelmod.AiTextToNovel, err error) {
+	_, err = cachev2.Classes().CacheTime(30*time.Minute).AutoListKey(models.AiTextToNovel+"-all").ResBind(&res).Cache(aitexttonovelmod.GetAll, cond, sort)
+	return
+}
+
+// InsertData 插入数据
+func InsertData(t *db.MongoTool, data aitexttonovelmod.AiTextToNovel) (primitive.ObjectID, error) {
+	newID, err := aitexttonovelmod.Insert(t, data)
+	if err != nil {
+		return newID, err
+	}
+	// 清除缓存数据
+	_, err = cachev2.Classes().Table(models.AiTextToNovel).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 := aitexttonovelmod.UpdateByID(t, objID, data)
+	if err != nil {
+		return 0, err
+	}
+
+	// 清理缓存
+	_, _ = cachev2.Classes().Table(models.AiTextToNovel).AutoClear(true, &id)
+
+	return affect, err
+}
+
+// DeleteData 删除数据
+func DeleteData(t *db.MongoTool, id primitive.ObjectID) (err error) {
+	err = aitexttonovelmod.DeleteByID(t, id)
+	if err != nil {
+		return
+	}
+
+	// 清理缓存
+	all := "*"
+	_, _ = cachev2.Classes().Table(models.AiTextToNovel).AutoClear(true, &all)
+	return
+}

models/cache/aitexttonoveldata/data.go

@@ -0,0 +1,49 @@
+package aitexttonoveldata
+
+import "91porn-server/models/v/aitexttonovelmod"
+
+// FormatAppDataList 格式化app列表数据
+func FormatAppDataList(origin []aitexttonovelmod.AiTextToNovel) (res []*aitexttonovelmod.AiTextToNovel) {
+	res = make([]*aitexttonovelmod.AiTextToNovel, len(origin))
+	if len(origin) == 0 {
+		return
+	}
+
+	// 组装返回数据
+	for i, item := range origin {
+		res[i] = FormatAppData(item)
+	}
+
+	return
+}
+
+// FormatAppData 格式化app数据
+func FormatAppData(item aitexttonovelmod.AiTextToNovel) (res *aitexttonovelmod.AiTextToNovel) {
+	if item.ID.IsZero() {
+		return
+	}
+
+	// 组装返回数据
+	res = &aitexttonovelmod.AiTextToNovel{
+		ID:               item.ID,
+		UID:              item.UID,
+		Description:      item.Description,
+		Content:          item.Content,
+		Status:           item.Status,
+		CharacterSetting: item.CharacterSetting,
+		LocationScene:    item.LocationScene,
+		Details:          item.Details,
+		ModelType:        item.ModelType,
+		Coin:             item.Coin,
+		DebitAmountCoin:  item.DebitAmountCoin,
+		DebitIncomeCoin:  item.DebitIncomeCoin,
+		IsFreeTimes:      item.IsFreeTimes,
+		IsHide:           item.IsHide,
+		Remark:           item.Remark,
+		IsDelete:         item.IsDelete,
+		CreatedAt:        item.CreatedAt,
+		UpdatedAt:        item.UpdatedAt,
+	}
+
+	return
+}