From 373424bd010cfae218570e602199e2ac6754a0ef Mon Sep 17 00:00:00 2001 From: ageerle <ageerle@163.com> Date: 星期六, 24 五月 2025 16:18:18 +0800 Subject: [PATCH] feat(更新日志): --- ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/impl/ChatTokenServiceImpl.java | 17 +++++++++-------- 1 files changed, 9 insertions(+), 8 deletions(-) diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/impl/ChatTokenServiceImpl.java b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/impl/ChatTokenServiceImpl.java index 85d931d..0c29dec 100644 --- a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/impl/ChatTokenServiceImpl.java +++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/impl/ChatTokenServiceImpl.java @@ -4,7 +4,8 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import lombok.RequiredArgsConstructor; import org.ruoyi.domain.ChatToken; -import org.ruoyi.mapper.ChatTokenMapper; +import org.ruoyi.domain.ChatUsageToken; +import org.ruoyi.mapper.ChatUsageTokenMapper; import org.ruoyi.service.IChatTokenService; import org.springframework.stereotype.Service; @@ -18,14 +19,14 @@ @Service public class ChatTokenServiceImpl implements IChatTokenService { - private final ChatTokenMapper baseMapper; + private final ChatUsageTokenMapper baseMapper; @Override - public ChatToken queryByUserId(Long userId, String modelName) { + public ChatUsageToken queryByUserId(Long userId, String modelName) { return baseMapper.selectOne( - new LambdaQueryWrapper<ChatToken>() - .eq(ChatToken::getUserId, userId) - .eq(ChatToken::getModelName, modelName) + new LambdaQueryWrapper<ChatUsageToken>() + .eq(ChatUsageToken::getUserId, userId) + .eq(ChatUsageToken::getModelName, modelName) .last("limit 1") ); } @@ -36,7 +37,7 @@ */ @Override public void resetToken(Long userId,String modelName) { - ChatToken chatToken = queryByUserId(userId, modelName); + ChatUsageToken chatToken = queryByUserId(userId, modelName); chatToken.setToken(0); baseMapper.updateById(chatToken); } @@ -46,7 +47,7 @@ * */ @Override - public void editToken(ChatToken chatToken) { + public void editToken(ChatUsageToken chatToken) { if(chatToken.getId() == null){ baseMapper.insert(chatToken); }else { -- Gitblit v1.9.3