package com.xmzs.system.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import com.xmzs.common.chat.constant.OpenAIConst;
|
import com.xmzs.common.chat.entity.chat.ChatCompletion;
|
import com.xmzs.common.core.domain.model.LoginUser;
|
import com.xmzs.common.core.exception.ServiceException;
|
import com.xmzs.common.core.exception.base.BaseException;
|
import com.xmzs.common.satoken.utils.LoginHelper;
|
import com.xmzs.system.domain.ChatToken;
|
import com.xmzs.system.domain.SysUser;
|
import com.xmzs.system.domain.bo.ChatMessageBo;
|
import com.xmzs.system.mapper.SysUserMapper;
|
import com.xmzs.system.service.IChatMessageService;
|
import com.xmzs.system.service.IChatService;
|
import com.xmzs.system.service.IChatTokenService;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
/**
|
* @author hncboy
|
* @date 2023/3/22 19:41
|
* 聊天相关业务实现类
|
*/
|
@Slf4j
|
@Service
|
@RequiredArgsConstructor
|
public class ChatServiceImpl implements IChatService {
|
|
private final SysUserMapper sysUserMapper;
|
|
private final IChatMessageService chatMessageService;
|
|
private final IChatTokenService chatTokenService;
|
|
|
/**
|
* 根据消耗的tokens扣除余额
|
*
|
* @param chatMessageBo
|
*/
|
public void deductToken(ChatMessageBo chatMessageBo) {
|
// 计算总token数
|
ChatToken chatToken = chatTokenService.queryByUserId(chatMessageBo.getUserId(), chatMessageBo.getModelName());
|
if (chatToken == null) {
|
chatToken = new ChatToken();
|
chatToken.setToken(0);
|
}
|
int totalTokens = chatToken.getToken() + chatMessageBo.getTotalTokens();
|
// 如果总token数大于等于1000,进行费用扣除
|
if (totalTokens >= 1000) {
|
// 计算费用
|
int token1 = totalTokens / 1000;
|
int token2 = totalTokens % 1000;
|
if (token2 > 0) {
|
// 保存剩余tokens
|
chatToken.setToken(token2);
|
chatTokenService.editToken(chatToken);
|
} else {
|
chatTokenService.resetToken(chatMessageBo.getUserId(), chatMessageBo.getModelName());
|
}
|
// 扣除用户余额
|
Double numberCost = token1 * ChatCompletion.getModelCost(chatMessageBo.getModelName());
|
deductUserBalance(chatMessageBo.getUserId(), numberCost);
|
chatMessageBo.setDeductCost(numberCost);
|
} else {
|
// 扣除用户余额
|
deductUserBalance(chatMessageBo.getUserId(), 0.0);
|
chatMessageBo.setDeductCost(0d);
|
chatMessageBo.setRemark("不满1kToken,计入下一次!");
|
chatToken.setToken(totalTokens);
|
chatToken.setModelName(chatMessageBo.getModelName());
|
chatToken.setUserId(chatMessageBo.getUserId());
|
chatTokenService.editToken(chatToken);
|
}
|
// 保存消息记录
|
chatMessageService.insertByBo(chatMessageBo);
|
}
|
|
/**
|
* 从用户余额中扣除费用
|
*
|
* @param userId 用户ID
|
* @param numberCost 要扣除的费用
|
*/
|
@Override
|
public void deductUserBalance(Long userId, Double numberCost) {
|
SysUser sysUser = sysUserMapper.selectById(userId);
|
if (sysUser == null) {
|
return;
|
}
|
|
Double userBalance = sysUser.getUserBalance();
|
if (userBalance < numberCost || userBalance < OpenAIConst.GPT4_COST) {
|
throw new ServiceException("余额不足，请联系管理员充值!");
|
}
|
|
sysUserMapper.update(null,
|
new LambdaUpdateWrapper<SysUser>()
|
.set(SysUser::getUserBalance, Math.max(userBalance - numberCost, 0))
|
.eq(SysUser::getUserId, userId));
|
}
|
|
/**
|
* 扣除mj任务费用
|
*
|
* @param prompt
|
* @param cost
|
*/
|
@Override
|
public void mjTaskDeduct(String prompt, double cost) {
|
deductUserBalance(getUserId(), cost);
|
// 保存消息记录
|
ChatMessageBo chatMessageBo = new ChatMessageBo();
|
chatMessageBo.setUserId(getUserId());
|
chatMessageBo.setModelName("mj");
|
chatMessageBo.setContent(prompt);
|
chatMessageBo.setDeductCost(cost);
|
chatMessageBo.setTotalTokens(0);
|
chatMessageService.insertByBo(chatMessageBo);
|
}
|
|
/**
|
* 获取用户Id
|
*
|
* @return
|
*/
|
public Long getUserId() {
|
LoginUser loginUser = LoginHelper.getLoginUser();
|
if (loginUser == null) {
|
throw new BaseException("用户未登录！");
|
}
|
return loginUser.getUserId();
|
}
|
}
|
