mirror of
https://github.com/woodchen-ink/aimodels-prices.git
synced 2025-07-18 05:32:00 +08:00
wood chen
680d684016
- 将价格倍率处理函数从 handlers/prices.go 移动到新的 handlers/rates 包 - 更新 main.go 中的路由配置，使用新的 rates.GetPriceRates 处理函数 - 在 prices.go 中新增 clearPriceCache 时调用 rates.ClearRatesCache - 模块化价格倍率计算逻辑，提高代码组织性和可维护性
444 lines
13 KiB
Go
444 lines
13 KiB
Go
package handlers
|
||
|
||
import (
|
||
"fmt"
|
||
"net/http"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
|
||
"aimodels-prices/database"
|
||
"aimodels-prices/handlers/rates"
|
||
"aimodels-prices/models"
|
||
)
|
||
|
||
func GetPrices(c *gin.Context) {
|
||
// 获取分页和筛选参数
|
||
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
||
pageSize, _ := strconv.Atoi(c.DefaultQuery("pageSize", "20"))
|
||
channelType := c.Query("channel_type") // 厂商筛选参数
|
||
modelType := c.Query("model_type") // 模型类型筛选参数
|
||
|
||
if page < 1 {
|
||
page = 1
|
||
}
|
||
if pageSize < 1 {
|
||
pageSize = 20
|
||
}
|
||
|
||
offset := (page - 1) * pageSize
|
||
|
||
// 构建缓存键
|
||
cacheKey := fmt.Sprintf("prices_page_%d_size_%d_channel_%s_type_%s",
|
||
page, pageSize, channelType, modelType)
|
||
|
||
// 尝试从缓存获取
|
||
if cachedData, found := database.GlobalCache.Get(cacheKey); found {
|
||
if result, ok := cachedData.(gin.H); ok {
|
||
c.JSON(http.StatusOK, result)
|
||
return
|
||
}
|
||
}
|
||
|
||
// 构建查询 - 使用索引优化
|
||
query := database.DB.Model(&models.Price{}).Select("*")
|
||
|
||
// 添加筛选条件
|
||
if channelType != "" {
|
||
query = query.Where("channel_type = ?", channelType)
|
||
}
|
||
if modelType != "" {
|
||
query = query.Where("model_type = ?", modelType)
|
||
}
|
||
|
||
// 获取总数 - 使用缓存优化
|
||
var total int64
|
||
totalCacheKey := fmt.Sprintf("prices_count_channel_%s_type_%s", channelType, modelType)
|
||
|
||
if cachedTotal, found := database.GlobalCache.Get(totalCacheKey); found {
|
||
if t, ok := cachedTotal.(int64); ok {
|
||
total = t
|
||
} else {
|
||
if err := query.Count(&total).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to count prices"})
|
||
return
|
||
}
|
||
database.GlobalCache.Set(totalCacheKey, total, 5*time.Minute)
|
||
}
|
||
} else {
|
||
if err := query.Count(&total).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to count prices"})
|
||
return
|
||
}
|
||
database.GlobalCache.Set(totalCacheKey, total, 5*time.Minute)
|
||
}
|
||
|
||
// 获取分页数据 - 使用索引优化
|
||
var prices []models.Price
|
||
if err := query.Order("created_at DESC").Limit(pageSize).Offset(offset).Find(&prices).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch prices"})
|
||
return
|
||
}
|
||
|
||
result := gin.H{
|
||
"total": total,
|
||
"data": prices,
|
||
}
|
||
|
||
// 存入缓存，有效期5分钟
|
||
database.GlobalCache.Set(cacheKey, result, 5*time.Minute)
|
||
|
||
c.JSON(http.StatusOK, result)
|
||
}
|
||
|
||
func CreatePrice(c *gin.Context) {
|
||
var price models.Price
|
||
if err := c.ShouldBindJSON(&price); err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||
return
|
||
}
|
||
|
||
// 验证模型厂商ID是否存在
|
||
var provider models.Provider
|
||
if err := database.DB.Where("id = ?", price.ChannelType).First(&provider).Error; err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid provider ID"})
|
||
return
|
||
}
|
||
|
||
// 检查同一厂商下是否已存在相同名称的模型
|
||
var count int64
|
||
if err := database.DB.Model(&models.Price{}).Where("channel_type = ? AND model = ? AND status = 'approved'",
|
||
price.ChannelType, price.Model).Count(&count).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to check model existence"})
|
||
return
|
||
}
|
||
if count > 0 {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": "Model with the same name already exists for this provider"})
|
||
return
|
||
}
|
||
|
||
// 设置状态和创建者
|
||
price.Status = "pending"
|
||
|
||
// 创建记录
|
||
if err := database.DB.Create(&price).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create price"})
|
||
return
|
||
}
|
||
|
||
// 清除所有价格相关缓存
|
||
clearPriceCache()
|
||
|
||
c.JSON(http.StatusCreated, price)
|
||
}
|
||
|
||
func UpdatePriceStatus(c *gin.Context) {
|
||
id := c.Param("id")
|
||
var input struct {
|
||
Status string `json:"status" binding:"required,oneof=approved rejected"`
|
||
}
|
||
|
||
if err := c.ShouldBindJSON(&input); err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||
return
|
||
}
|
||
|
||
// 查找价格记录
|
||
var price models.Price
|
||
if err := database.DB.Where("id = ?", id).First(&price).Error; err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "Price not found"})
|
||
return
|
||
}
|
||
|
||
// 开始事务
|
||
tx := database.DB.Begin()
|
||
|
||
if input.Status == "approved" {
|
||
// 如果是批准，将临时字段的值更新到正式字段
|
||
updateMap := map[string]interface{}{
|
||
"status": input.Status,
|
||
"updated_at": time.Now(),
|
||
}
|
||
|
||
// 如果临时字段有值，则更新主字段
|
||
if price.TempModel != nil {
|
||
updateMap["model"] = *price.TempModel
|
||
}
|
||
if price.TempModelType != nil {
|
||
updateMap["model_type"] = *price.TempModelType
|
||
}
|
||
if price.TempBillingType != nil {
|
||
updateMap["billing_type"] = *price.TempBillingType
|
||
}
|
||
if price.TempChannelType != nil {
|
||
updateMap["channel_type"] = *price.TempChannelType
|
||
}
|
||
if price.TempCurrency != nil {
|
||
updateMap["currency"] = *price.TempCurrency
|
||
}
|
||
if price.TempInputPrice != nil {
|
||
updateMap["input_price"] = *price.TempInputPrice
|
||
}
|
||
if price.TempOutputPrice != nil {
|
||
updateMap["output_price"] = *price.TempOutputPrice
|
||
}
|
||
if price.TempPriceSource != nil {
|
||
updateMap["price_source"] = *price.TempPriceSource
|
||
}
|
||
|
||
// 清除所有临时字段
|
||
updateMap["temp_model"] = nil
|
||
updateMap["temp_model_type"] = nil
|
||
updateMap["temp_billing_type"] = nil
|
||
updateMap["temp_channel_type"] = nil
|
||
updateMap["temp_currency"] = nil
|
||
updateMap["temp_input_price"] = nil
|
||
updateMap["temp_output_price"] = nil
|
||
updateMap["temp_price_source"] = nil
|
||
updateMap["updated_by"] = nil
|
||
|
||
if err := tx.Model(&price).Updates(updateMap).Error; err != nil {
|
||
tx.Rollback()
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update price status"})
|
||
return
|
||
}
|
||
} else {
|
||
// 如果是拒绝，清除临时字段
|
||
if err := tx.Model(&price).Updates(map[string]interface{}{
|
||
"status": input.Status,
|
||
"updated_at": time.Now(),
|
||
"temp_model": nil,
|
||
"temp_model_type": nil,
|
||
"temp_billing_type": nil,
|
||
"temp_channel_type": nil,
|
||
"temp_currency": nil,
|
||
"temp_input_price": nil,
|
||
"temp_output_price": nil,
|
||
"temp_price_source": nil,
|
||
"updated_by": nil,
|
||
}).Error; err != nil {
|
||
tx.Rollback()
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update price status"})
|
||
return
|
||
}
|
||
}
|
||
|
||
// 提交事务
|
||
if err := tx.Commit().Error; err != nil {
|
||
tx.Rollback()
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to commit transaction"})
|
||
return
|
||
}
|
||
|
||
// 清除所有价格相关缓存
|
||
clearPriceCache()
|
||
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"message": "Status updated successfully",
|
||
"status": input.Status,
|
||
"updated_at": time.Now(),
|
||
})
|
||
}
|
||
|
||
func UpdatePrice(c *gin.Context) {
|
||
id := c.Param("id")
|
||
var price models.Price
|
||
if err := c.ShouldBindJSON(&price); err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||
return
|
||
}
|
||
|
||
// 验证模型厂商ID是否存在
|
||
var provider models.Provider
|
||
if err := database.DB.Where("id = ?", price.ChannelType).First(&provider).Error; err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid provider ID"})
|
||
return
|
||
}
|
||
|
||
// 检查同一厂商下是否已存在相同名称的模型（排除当前正在编辑的记录）
|
||
var count int64
|
||
if err := database.DB.Model(&models.Price{}).Where("channel_type = ? AND model = ? AND id != ? AND status = 'approved'",
|
||
price.ChannelType, price.Model, id).Count(&count).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to check model existence"})
|
||
return
|
||
}
|
||
if count > 0 {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": "Model with the same name already exists for this provider"})
|
||
return
|
||
}
|
||
|
||
// 获取当前用户
|
||
user, exists := c.Get("user")
|
||
if !exists {
|
||
c.JSON(http.StatusUnauthorized, gin.H{"error": "User not found"})
|
||
return
|
||
}
|
||
currentUser := user.(*models.User)
|
||
|
||
// 查找现有记录
|
||
var existingPrice models.Price
|
||
if err := database.DB.Where("id = ?", id).First(&existingPrice).Error; err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "Price not found"})
|
||
return
|
||
}
|
||
|
||
// 根据用户角色决定更新方式
|
||
if currentUser.Role == "admin" {
|
||
// 管理员直接更新主字段
|
||
existingPrice.Model = price.Model
|
||
existingPrice.ModelType = price.ModelType
|
||
existingPrice.BillingType = price.BillingType
|
||
existingPrice.ChannelType = price.ChannelType
|
||
existingPrice.Currency = price.Currency
|
||
existingPrice.InputPrice = price.InputPrice
|
||
existingPrice.OutputPrice = price.OutputPrice
|
||
existingPrice.PriceSource = price.PriceSource
|
||
existingPrice.Status = "approved"
|
||
existingPrice.UpdatedBy = ¤tUser.Username
|
||
existingPrice.TempModel = nil
|
||
existingPrice.TempModelType = nil
|
||
existingPrice.TempBillingType = nil
|
||
existingPrice.TempChannelType = nil
|
||
existingPrice.TempCurrency = nil
|
||
existingPrice.TempInputPrice = nil
|
||
existingPrice.TempOutputPrice = nil
|
||
existingPrice.TempPriceSource = nil
|
||
|
||
if err := database.DB.Save(&existingPrice).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update price"})
|
||
return
|
||
}
|
||
} else {
|
||
// 普通用户更新临时字段
|
||
existingPrice.TempModel = &price.Model
|
||
existingPrice.TempModelType = &price.ModelType
|
||
existingPrice.TempBillingType = &price.BillingType
|
||
existingPrice.TempChannelType = &price.ChannelType
|
||
existingPrice.TempCurrency = &price.Currency
|
||
existingPrice.TempInputPrice = &price.InputPrice
|
||
existingPrice.TempOutputPrice = &price.OutputPrice
|
||
existingPrice.TempPriceSource = &price.PriceSource
|
||
existingPrice.Status = "pending"
|
||
existingPrice.UpdatedBy = ¤tUser.Username
|
||
|
||
if err := database.DB.Save(&existingPrice).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update price"})
|
||
return
|
||
}
|
||
}
|
||
|
||
// 清除所有价格相关缓存
|
||
clearPriceCache()
|
||
|
||
c.JSON(http.StatusOK, existingPrice)
|
||
}
|
||
|
||
func DeletePrice(c *gin.Context) {
|
||
id := c.Param("id")
|
||
|
||
// 查找价格记录
|
||
var price models.Price
|
||
if err := database.DB.Where("id = ?", id).First(&price).Error; err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "Price not found"})
|
||
return
|
||
}
|
||
|
||
// 删除记录
|
||
if err := database.DB.Delete(&price).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete price"})
|
||
return
|
||
}
|
||
|
||
// 清除所有价格相关缓存
|
||
clearPriceCache()
|
||
|
||
c.JSON(http.StatusOK, gin.H{"message": "Price deleted successfully"})
|
||
}
|
||
|
||
func ApproveAllPrices(c *gin.Context) {
|
||
// 查找所有待审核的价格
|
||
var pendingPrices []models.Price
|
||
if err := database.DB.Where("status = 'pending'").Find(&pendingPrices).Error; err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch pending prices"})
|
||
return
|
||
}
|
||
|
||
// 开始事务
|
||
tx := database.DB.Begin()
|
||
|
||
for _, price := range pendingPrices {
|
||
updateMap := map[string]interface{}{
|
||
"status": "approved",
|
||
"updated_at": time.Now(),
|
||
}
|
||
|
||
// 如果临时字段有值，则更新主字段
|
||
if price.TempModel != nil {
|
||
updateMap["model"] = *price.TempModel
|
||
}
|
||
if price.TempModelType != nil {
|
||
updateMap["model_type"] = *price.TempModelType
|
||
}
|
||
if price.TempBillingType != nil {
|
||
updateMap["billing_type"] = *price.TempBillingType
|
||
}
|
||
if price.TempChannelType != nil {
|
||
updateMap["channel_type"] = *price.TempChannelType
|
||
}
|
||
if price.TempCurrency != nil {
|
||
updateMap["currency"] = *price.TempCurrency
|
||
}
|
||
if price.TempInputPrice != nil {
|
||
updateMap["input_price"] = *price.TempInputPrice
|
||
}
|
||
if price.TempOutputPrice != nil {
|
||
updateMap["output_price"] = *price.TempOutputPrice
|
||
}
|
||
if price.TempPriceSource != nil {
|
||
updateMap["price_source"] = *price.TempPriceSource
|
||
}
|
||
|
||
// 清除所有临时字段
|
||
updateMap["temp_model"] = nil
|
||
updateMap["temp_model_type"] = nil
|
||
updateMap["temp_billing_type"] = nil
|
||
updateMap["temp_channel_type"] = nil
|
||
updateMap["temp_currency"] = nil
|
||
updateMap["temp_input_price"] = nil
|
||
updateMap["temp_output_price"] = nil
|
||
updateMap["temp_price_source"] = nil
|
||
updateMap["updated_by"] = nil
|
||
|
||
if err := tx.Model(&price).Updates(updateMap).Error; err != nil {
|
||
tx.Rollback()
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to approve prices"})
|
||
return
|
||
}
|
||
}
|
||
|
||
// 提交事务
|
||
if err := tx.Commit().Error; err != nil {
|
||
tx.Rollback()
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to commit transaction"})
|
||
return
|
||
}
|
||
|
||
// 清除所有价格相关缓存
|
||
clearPriceCache()
|
||
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"message": "All pending prices approved successfully",
|
||
"count": len(pendingPrices),
|
||
})
|
||
}
|
||
|
||
// clearPriceCache 清除所有价格相关的缓存
|
||
func clearPriceCache() {
|
||
// 由于我们无法精确知道哪些缓存键与价格相关，所以清除所有缓存
|
||
database.GlobalCache.Clear()
|
||
|
||
// 同时清除价格倍率缓存
|
||
rates.ClearRatesCache()
|
||
}
|