mirror of
https://github.com/coaidev/coai.git
synced 2025-05-20 21:40:15 +09:00
39 lines
980 B
Go
39 lines
980 B
Go
package manager
|
|
|
|
import (
|
|
"chat/channel"
|
|
"chat/globals"
|
|
"chat/utils"
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"time"
|
|
)
|
|
|
|
type CacheProps struct {
|
|
Message []globals.Message `json:"message" required:"true"`
|
|
Model string `json:"model" required:"true"`
|
|
Reversible bool `json:"reversible"`
|
|
}
|
|
|
|
type CacheData struct {
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
func ExtractCacheData(c *gin.Context, props *CacheProps) *CacheData {
|
|
hash := utils.Md5Encrypt(utils.Marshal(props))
|
|
data, err := utils.GetCacheFromContext(c).Get(c, fmt.Sprintf(":niodata:%s", hash)).Result()
|
|
if err == nil && data != "" {
|
|
return utils.UnmarshalForm[CacheData](data)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func SaveCacheData(c *gin.Context, props *CacheProps, data *CacheData) {
|
|
if channel.ChargeInstance.IsBilling(props.Model) {
|
|
return
|
|
}
|
|
|
|
hash := utils.Md5Encrypt(utils.Marshal(props))
|
|
utils.GetCacheFromContext(c).Set(c, fmt.Sprintf(":niodata:%s", hash), utils.Marshal(data), time.Hour*12)
|
|
}
|