89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
package controller
|
||
|
||
import (
|
||
"net/http"
|
||
"strings"
|
||
|
||
"github.com/QuantumNous/new-api/service"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
func GetChannelAffinityCacheStats(c *gin.Context) {
|
||
stats := service.GetChannelAffinityCacheStats()
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"success": true,
|
||
"message": "",
|
||
"data": stats,
|
||
})
|
||
}
|
||
|
||
func ClearChannelAffinityCache(c *gin.Context) {
|
||
all := strings.TrimSpace(c.Query("all"))
|
||
ruleName := strings.TrimSpace(c.Query("rule_name"))
|
||
|
||
if all == "true" {
|
||
deleted := service.ClearChannelAffinityCacheAll()
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"success": true,
|
||
"message": "",
|
||
"data": gin.H{
|
||
"deleted": deleted,
|
||
},
|
||
})
|
||
return
|
||
}
|
||
|
||
if ruleName == "" {
|
||
c.JSON(http.StatusBadRequest, gin.H{
|
||
"success": false,
|
||
"message": "缺少参数：rule_name，或使用 all=true 清空全部",
|
||
})
|
||
return
|
||
}
|
||
|
||
deleted, err := service.ClearChannelAffinityCacheByRuleName(ruleName)
|
||
if err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{
|
||
"success": false,
|
||
"message": err.Error(),
|
||
})
|
||
return
|
||
}
|
||
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"success": true,
|
||
"message": "",
|
||
"data": gin.H{
|
||
"deleted": deleted,
|
||
},
|
||
})
|
||
}
|
||
|
||
func GetChannelAffinityUsageCacheStats(c *gin.Context) {
|
||
ruleName := strings.TrimSpace(c.Query("rule_name"))
|
||
usingGroup := strings.TrimSpace(c.Query("using_group"))
|
||
keyFp := strings.TrimSpace(c.Query("key_fp"))
|
||
|
||
if ruleName == "" {
|
||
c.JSON(http.StatusBadRequest, gin.H{
|
||
"success": false,
|
||
"message": "missing param: rule_name",
|
||
})
|
||
return
|
||
}
|
||
if keyFp == "" {
|
||
c.JSON(http.StatusBadRequest, gin.H{
|
||
"success": false,
|
||
"message": "missing param: key_fp",
|
||
})
|
||
return
|
||
}
|
||
|
||
stats := service.GetChannelAffinityUsageCacheStats(ruleName, usingGroup, keyFp)
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"success": true,
|
||
"message": "",
|
||
"data": stats,
|
||
})
|
||
}
|