sion/sionrui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 优化

Browse Source
This commit is contained in:
sion123
2026-03-07 23:39:57 +08:00
parent f3c2bae6bf
commit 2aa459889a
24 changed files with 1575 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/api/file/FileApi.java
View File

@@ -72,4 +72,11 @@ public interface FileApi {
*/
String getMasterFileDomain();
/**
* 根据OSS URL删除文件
*
* @param ossUrl OSS文件完整URL
*/
void deleteFileByUrl(@NotEmpty(message = "URL 不能为空") String ossUrl);
}

9
yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/api/file/FileApiImpl.java
View File

@@ -49,4 +49,13 @@ public class FileApiImpl implements FileApi {
return null;
}
@Override
public void deleteFileByUrl(String ossUrl) {
try {
fileService.deleteFileByUrl(ossUrl);
} catch (Exception e) {
throw new RuntimeException("删除文件失败: " + e.getMessage(), e);
}
}
}

36
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/client/DifyClient.java
View File

@@ -174,7 +174,24 @@ public class DifyClient {
return switch (eventType) {
case "message", "agent_message" -> DifyChatRespVO.message(answer, conversationId);
case "workflow_finished", "message_end" -> DifyChatRespVO.done(conversationId, null);
case "workflow_finished", "message_end" -> {
// 解析 token 使用信息
@SuppressWarnings("unchecked")
Map<String, Object> metadata = (Map<String, Object>) data.get("metadata");
if (metadata != null) {
@SuppressWarnings("unchecked")
Map<String, Object> usage = (Map<String, Object>) metadata.get("usage");
if (usage != null) {
Integer promptTokens = parseTokenCount(usage.get("prompt_tokens"));
Integer completionTokens = parseTokenCount(usage.get("completion_tokens"));
Integer totalTokens = parseTokenCount(usage.get("total_tokens"));
log.debug("[parseSSEEvent] message_end with tokens: prompt={}, completion={}, total={}",
promptTokens, completionTokens, totalTokens);
yield DifyChatRespVO.doneWithTokens(conversationId, null, promptTokens, completionTokens, totalTokens);
}
}
yield DifyChatRespVO.done(conversationId, null);
}
case "error" -> DifyChatRespVO.error(answer);
// chatflow 节点事件，尝试提取内容
case "node_finished" -> {
@@ -273,4 +290,21 @@ public class DifyClient {
.block();
}
/**
* 解析 token 数量（处理 Number 类型）
*/
private Integer parseTokenCount(Object value) {
if (value == null) {
return 0;
}
if (value instanceof Number) {
return ((Number) value).intValue();
}
try {
return Integer.parseInt(value.toString());
} catch (NumberFormatException e) {
return 0;
}
}
}

76
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java
View File

@@ -49,6 +49,8 @@ public class DifyServiceImpl implements DifyService {
AtomicLong pendingRecordId = new AtomicLong();
// 用于存储会话ID
AtomicReference<String> conversationIdRef = new AtomicReference<>(reqVO.getConversationId());
// 用于存储 token 使用信息
AtomicReference<DifyChatRespVO> tokenUsageRef = new AtomicReference<>();
// Dify 用户标识（固定格式）
String difyUserId = "user-" + userId;
@@ -70,12 +72,13 @@ public class DifyServiceImpl implements DifyService {
// 3. 预检积分
pointsService.checkPoints(userId, config.getConsumePoints());
// 4. 创建预扣记录
// 4. 创建预扣记录（带 serviceCode）
Long recordId = pointsService.createPendingDeduct(
userId,
config.getConsumePoints(),
"dify_chat",
reqVO.getAgentId().toString()
reqVO.getAgentId().toString(),
config.getServiceCode()
);
pendingRecordId.set(recordId);
@@ -95,13 +98,29 @@ public class DifyServiceImpl implements DifyService {
if (resp.getConversationId() != null) {
conversationIdRef.set(resp.getConversationId());
}
// 捕获 token 使用信息
if (resp.getTotalTokens() != null && resp.getTotalTokens() > 0) {
tokenUsageRef.set(resp);
}
})
// 7. 流结束时确认扣费
// 7. 流结束时确认扣费（带 token）
.doOnComplete(() -> {
if (pendingRecordId.get() > 0) {
try {
pointsService.confirmPendingDeduct(pendingRecordId.get());
log.info("[chatStream] 流结束，确认扣费，记录ID: {}", pendingRecordId.get());
DifyChatRespVO tokenUsage = tokenUsageRef.get();
if (tokenUsage != null) {
pointsService.confirmPendingDeductWithTokens(
pendingRecordId.get(),
tokenUsage.getInputTokens(),
tokenUsage.getOutputTokens(),
tokenUsage.getTotalTokens()
);
log.info("[chatStream] 流结束，确认扣费（带token），记录ID: {}, tokens: {}",
pendingRecordId.get(), tokenUsage.getTotalTokens());
} else {
pointsService.confirmPendingDeduct(pendingRecordId.get());
log.info("[chatStream] 流结束，确认扣费（无token），记录ID: {}", pendingRecordId.get());
}
} catch (Exception e) {
log.error("[chatStream] 确认扣费失败", e);
}
@@ -133,6 +152,15 @@ public class DifyServiceImpl implements DifyService {
})
// 10. 在最后添加 done 事件
.concatWith(Mono.defer(() -> {
DifyChatRespVO tokenUsage = tokenUsageRef.get();
if (tokenUsage != null) {
return Mono.just(DifyChatRespVO.doneWithTokens(
conversationIdRef.get(), null,
tokenUsage.getInputTokens(),
tokenUsage.getOutputTokens(),
tokenUsage.getTotalTokens()
));
}
return Mono.just(DifyChatRespVO.done(conversationIdRef.get(), null));
}))
.onErrorResume(e -> {
@@ -147,6 +175,8 @@ public class DifyServiceImpl implements DifyService {
AtomicLong pendingRecordId = new AtomicLong();
// 用于存储会话ID
AtomicReference<String> conversationIdRef = new AtomicReference<>("");
// 用于存储 token 使用信息
AtomicReference<DifyChatRespVO> tokenUsageRef = new AtomicReference<>();
// Dify 用户标识（固定格式）
String difyUserId = "user-" + userId;
@@ -168,12 +198,13 @@ public class DifyServiceImpl implements DifyService {
// 3. 预检积分
pointsService.checkPoints(userId, config.getConsumePoints());
// 4. 创建预扣记录
// 4. 创建预扣记录（带 serviceCode）
Long recordId = pointsService.createPendingDeduct(
userId,
config.getConsumePoints(),
"forecast_rewrite",
reqVO.getModelType()
reqVO.getModelType(),
config.getServiceCode()
);
pendingRecordId.set(recordId);
@@ -213,13 +244,29 @@ public class DifyServiceImpl implements DifyService {
if (resp.getConversationId() != null) {
conversationIdRef.set(resp.getConversationId());
}
// 捕获 token 使用信息
if (resp.getTotalTokens() != null && resp.getTotalTokens() > 0) {
tokenUsageRef.set(resp);
}
})
// 8. 流结束时确认扣费
// 8. 流结束时确认扣费（带 token）
.doOnComplete(() -> {
if (pendingRecordId.get() > 0) {
try {
pointsService.confirmPendingDeduct(pendingRecordId.get());
log.info("[rewriteStream] 流结束，确认扣费，记录ID: {}", pendingRecordId.get());
DifyChatRespVO tokenUsage = tokenUsageRef.get();
if (tokenUsage != null) {
pointsService.confirmPendingDeductWithTokens(
pendingRecordId.get(),
tokenUsage.getInputTokens(),
tokenUsage.getOutputTokens(),
tokenUsage.getTotalTokens()
);
log.info("[rewriteStream] 流结束，确认扣费（带token），记录ID: {}, tokens: {}",
pendingRecordId.get(), tokenUsage.getTotalTokens());
} else {
pointsService.confirmPendingDeduct(pendingRecordId.get());
log.info("[rewriteStream] 流结束，确认扣费（无token），记录ID: {}", pendingRecordId.get());
}
} catch (Exception e) {
log.error("[rewriteStream] 确认扣费失败", e);
}
@@ -250,6 +297,15 @@ public class DifyServiceImpl implements DifyService {
})
// 11. 在最后添加 done 事件
.concatWith(Mono.defer(() -> {
DifyChatRespVO tokenUsage = tokenUsageRef.get();
if (tokenUsage != null) {
return Mono.just(DifyChatRespVO.doneWithTokens(
conversationIdRef.get(), null,
tokenUsage.getInputTokens(),
tokenUsage.getOutputTokens(),
tokenUsage.getTotalTokens()
));
}
return Mono.just(DifyChatRespVO.done(conversationIdRef.get(), null));
}))
.onErrorResume(e -> {

21
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/vo/DifyChatRespVO.java
View File

@@ -31,6 +31,15 @@ public class DifyChatRespVO {
@Schema(description = "错误信息")
private String errorMessage;
@Schema(description = "输入token数")
private Integer inputTokens;
@Schema(description = "输出token数")
private Integer outputTokens;
@Schema(description = "总token数")
private Integer totalTokens;
/** 事件类型常量 */
public static final String EVENT_MESSAGE = "message";
public static final String EVENT_DONE = "done";
@@ -52,6 +61,18 @@ public class DifyChatRespVO {
.build();
}
public static DifyChatRespVO doneWithTokens(String conversationId, Integer consumePoints,
Integer inputTokens, Integer outputTokens, Integer totalTokens) {
return DifyChatRespVO.builder()
.event(EVENT_DONE)
.conversationId(conversationId)
.consumePoints(consumePoints)
.inputTokens(inputTokens)
.outputTokens(outputTokens)
.totalTokens(totalTokens)
.build();
}
public static DifyChatRespVO error(String errorMessage) {
return DifyChatRespVO.builder()
.event(EVENT_ERROR)

8
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/mix/service/MixTaskServiceImpl.java
View File

@@ -6,7 +6,7 @@ import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder;
import cn.iocoder.yudao.framework.tenant.core.util.TenantUtils;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.module.infra.service.file.FileService;
import cn.iocoder.yudao.module.infra.api.file.FileApi;
import com.alibaba.ttl.TtlRunnable;
import cn.iocoder.yudao.module.tik.mix.client.IceClient;
import cn.iocoder.yudao.module.tik.mix.constants.MixTaskConstants;
@@ -45,7 +45,7 @@ public class MixTaskServiceImpl implements MixTaskService {
private final IceClient iceClient;
@Resource
private FileService fileService;
private FileApi fileApi;
@Override
@Transactional(rollbackFor = Exception.class)
@@ -115,7 +115,7 @@ public class MixTaskServiceImpl implements MixTaskService {
log.info("[MixTask][删除OSS文件] taskId={}, fileCount={}", id, outputUrls.size());
for (String url : outputUrls) {
try {
fileService.deleteFileByUrl(url);
fileApi.deleteFileByUrl(url);
} catch (Exception e) {
log.error("[MixTask][删除单个OSS文件失败] taskId={}, url={}", id, url, e);
}
@@ -432,7 +432,7 @@ public class MixTaskServiceImpl implements MixTaskService {
return null;
}
// 使用FileService生成签名URL
return fileService.presignGetUrl(ossUrl, expirationSeconds);
return fileApi.presignGetUrl(ossUrl, expirationSeconds);
} catch (Exception e) {
log.error("[MixTask][生成签名URL失败] url={}", ossUrl, e);
return null;

74
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/controller/AiUsageStatsController.java Normal file
View File

@@ -0,0 +1,74 @@
package cn.iocoder.yudao.module.tik.muye.aiusagestats.controller;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.module.tik.muye.aiusagestats.service.AiUsageStatsService;
import cn.iocoder.yudao.module.tik.muye.aiusagestats.vo.*;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.validation.Valid;
import org.springframework.format.annotation.DateTimeFormat;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import java.time.LocalDateTime;
import java.util.List;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
/**
* AI 使用统计 Controller
*/
@Tag(name = "管理后台 - AI 使用统计")
@RestController
@RequestMapping("/admin-api/muye/ai-usage-stats")
@Validated
public class AiUsageStatsController {
@Resource
private AiUsageStatsService aiUsageStatsService;
@GetMapping("/overview")
@Operation(summary = "获取概览统计")
public CommonResult<AiUsageOverviewRespVO> getOverview(
@Parameter(description = "开始时间") @RequestParam(required = false)
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime startTime,
@Parameter(description = "结束时间") @RequestParam(required = false)
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime endTime,
@Parameter(description = "业务类型") @RequestParam(required = false) String bizType) {
return success(aiUsageStatsService.getOverview(startTime, endTime, bizType));
}
@GetMapping("/user-stats")
@Operation(summary = "获取用户统计分页")
public CommonResult<PageResult<AiUsageUserStatsRespVO>> getUserStatsPage(@Valid AiUsageStatsPageReqVO reqVO) {
return success(aiUsageStatsService.getUserStatsPage(reqVO));
}
@GetMapping("/app-stats")
@Operation(summary = "获取应用统计列表")
public CommonResult<List<AiUsageAppStatsRespVO>> getAppStats(
@Parameter(description = "开始时间") @RequestParam(required = false)
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime startTime,
@Parameter(description = "结束时间") @RequestParam(required = false)
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime endTime,
@Parameter(description = "业务类型") @RequestParam(required = false) String bizType) {
return success(aiUsageStatsService.getAppStats(startTime, endTime, bizType));
}
@GetMapping("/trend")
@Operation(summary = "获取趋势数据")
public CommonResult<AiUsageTrendRespVO> getTrend(
@Parameter(description = "开始时间") @RequestParam(required = false)
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime startTime,
@Parameter(description = "结束时间") @RequestParam(required = false)
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime endTime,
@Parameter(description = "业务类型") @RequestParam(required = false) String bizType,
@Parameter(description = "类型（day-按天，hour-按小时）") @RequestParam(defaultValue = "day") String type) {
return success(aiUsageStatsService.getTrend(startTime, endTime, bizType, type));
}
}

66
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/service/AiUsageStatsService.java Normal file
View File

@@ -0,0 +1,66 @@
package cn.iocoder.yudao.module.tik.muye.aiusagestats.service;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.module.tik.muye.aiusagestats.vo.*;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;
import java.util.List;
import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
/**
* AI 使用统计 Service 接口
*/
public interface AiUsageStatsService {
/**
* 获取概览统计
*
* @param startTime 开始时间
* @param endTime 结束时间
* @param bizType 业务类型（可选）
* @return 概览统计
*/
AiUsageOverviewRespVO getOverview(
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime startTime,
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime endTime,
String bizType);
/**
* 获取用户统计分页
*
* @param reqVO 查询条件
* @return 用户统计分页
*/
PageResult<AiUsageUserStatsRespVO> getUserStatsPage(AiUsageStatsPageReqVO reqVO);
/**
* 获取应用统计列表
*
* @param startTime 开始时间
* @param endTime 结束时间
* @param bizType 业务类型（可选）
* @return 应用统计列表
*/
List<AiUsageAppStatsRespVO> getAppStats(
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime startTime,
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime endTime,
String bizType);
/**
* 获取趋势数据
*
* @param startTime 开始时间
* @param endTime 结束时间
* @param bizType 业务类型（可选）
* @param type 类型（day-按天，hour-按小时）
* @return 趋势数据
*/
AiUsageTrendRespVO getTrend(
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime startTime,
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) LocalDateTime endTime,
String bizType,
String type);
}

243
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/service/AiUsageStatsServiceImpl.java Normal file
View File

@@ -0,0 +1,243 @@
package cn.iocoder.yudao.module.tik.muye.aiusagestats.service;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.tik.muye.aiusagestats.vo.*;
import cn.iocoder.yudao.module.tik.muye.aiserviceconfig.dal.AiServiceConfigDO;
import cn.iocoder.yudao.module.tik.muye.aiserviceconfig.mapper.AiServiceConfigMapper;
import cn.iocoder.yudao.module.tik.muye.pointrecord.dal.PointRecordDO;
import cn.iocoder.yudao.module.tik.muye.pointrecord.mapper.PointRecordMapper;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;
import java.time.LocalDateTime;
import java.util.*;
import java.util.stream.Collectors;
/**
* AI 使用统计 Service 实现类
*/
@Service
@Validated
@Slf4j
public class AiUsageStatsServiceImpl implements AiUsageStatsService {
@Resource
private PointRecordMapper pointRecordMapper;
@Resource
private AiServiceConfigMapper aiServiceConfigMapper;
/**
* 业务类型名称映射
*/
private static final Map<String, String> BIZ_TYPE_NAMES = new HashMap<>();
static {
BIZ_TYPE_NAMES.put("dify_chat", "AI对话");
BIZ_TYPE_NAMES.put("forecast_rewrite", "文案改写");
BIZ_TYPE_NAMES.put("voice_tts", "语音合成");
BIZ_TYPE_NAMES.put("digital_human", "数字人");
BIZ_TYPE_NAMES.put("tikhub_fetch", "TikHub获取");
BIZ_TYPE_NAMES.put("signin", "签到");
BIZ_TYPE_NAMES.put("recharge", "充值");
BIZ_TYPE_NAMES.put("exchange", "兑换");
BIZ_TYPE_NAMES.put("admin", "后台调整");
BIZ_TYPE_NAMES.put("gift", "礼包赠送");
}
@Override
public AiUsageOverviewRespVO getOverview(LocalDateTime startTime, LocalDateTime endTime, String bizType) {
// 查询业务类型统计
List<Map<String, Object>> bizTypeStatsList = pointRecordMapper.selectBizTypeStats(startTime, endTime);
// 计算汇总数据
long totalCallCount = 0;
long totalConsumePoints = 0;
long totalTokens = 0;
List<AiUsageOverviewRespVO.BizTypeStats> bizTypeStats = new ArrayList<>();
for (Map<String, Object> stats : bizTypeStatsList) {
String bt = (String) stats.get("biz_type");
// 如果指定了业务类型，只统计该类型
if (bizType != null && !bizType.isEmpty() && !bizType.equals(bt)) {
continue;
}
long callCount = getLongValue(stats, "call_count");
long consumePoints = getLongValue(stats, "consume_points");
long tokens = getLongValue(stats, "total_tokens");
totalCallCount += callCount;
totalConsumePoints += consumePoints;
totalTokens += tokens;
bizTypeStats.add(AiUsageOverviewRespVO.BizTypeStats.builder()
.bizType(bt)
.bizTypeName(BIZ_TYPE_NAMES.getOrDefault(bt, bt))
.callCount(callCount)
.consumePoints(consumePoints)
.totalTokens(tokens)
.build());
}
// 查询活跃用户数
Long activeUserCount = pointRecordMapper.selectCount(
new LambdaQueryWrapperX<PointRecordDO>()
.betweenIfPresent(PointRecordDO::getCreateTime, startTime, endTime)
.eqIfPresent(PointRecordDO::getBizType, bizType)
.eq(PointRecordDO::getStatus, "confirmed")
.eq(PointRecordDO::getType, "decrease")
.select(PointRecordDO::getUserId)
.groupBy(PointRecordDO::getUserId)
);
return AiUsageOverviewRespVO.builder()
.totalCallCount(totalCallCount)
.totalConsumePoints(totalConsumePoints)
.totalTokens(totalTokens)
.activeUserCount(activeUserCount)
.bizTypeStats(bizTypeStats)
.build();
}
@Override
public PageResult<AiUsageUserStatsRespVO> getUserStatsPage(AiUsageStatsPageReqVO reqVO) {
// 查询用户统计数据
List<Map<String, Object>> userStatsList = pointRecordMapper.selectUserStats(
reqVO.getStartTime(), reqVO.getEndTime(), reqVO.getBizType(), reqVO.getUserId());
// 计算总数
long total = userStatsList.size();
// 分页处理
int fromIndex = (reqVO.getPageNo() - 1) * reqVO.getPageSize();
int toIndex = Math.min(fromIndex + reqVO.getPageSize(), userStatsList.size());
if (fromIndex >= userStatsList.size()) {
return new PageResult<>(Collections.emptyList(), total);
}
List<Map<String, Object>> pagedList = userStatsList.subList(fromIndex, toIndex);
// 转换为 VO
List<AiUsageUserStatsRespVO> list = pagedList.stream()
.map(this::convertToUserStatsVO)
.collect(Collectors.toList());
return new PageResult<>(list, total);
}
@Override
public List<AiUsageAppStatsRespVO> getAppStats(LocalDateTime startTime, LocalDateTime endTime, String bizType) {
// 查询应用统计数据
List<Map<String, Object>> appStatsList = pointRecordMapper.selectAppStats(
startTime, endTime, bizType);
// 获取服务配置信息
Map<String, AiServiceConfigDO> configMap = new HashMap<>();
List<AiServiceConfigDO> configs = aiServiceConfigMapper.selectList(
new LambdaQueryWrapperX<AiServiceConfigDO>().eq(AiServiceConfigDO::getStatus, 1));
for (AiServiceConfigDO config : configs) {
configMap.put(config.getServiceCode(), config);
}
// 转换为 VO
return appStatsList.stream()
.map(stats -> convertToAppStatsVO(stats, configMap))
.collect(Collectors.toList());
}
@Override
public AiUsageTrendRespVO getTrend(LocalDateTime startTime, LocalDateTime endTime, String bizType, String type) {
// 查询趋势数据
List<Map<String, Object>> trendList = pointRecordMapper.selectTrend(
startTime, endTime, bizType, "day".equals(type) ? "%Y-%m-%d" : "%Y-%m-%d %H:00");
// 转换为 VO
List<AiUsageTrendRespVO.TrendItem> items = trendList.stream()
.map(this::convertToTrendItem)
.collect(Collectors.toList());
return AiUsageTrendRespVO.builder()
.trendList(items)
.build();
}
/**
* 转换为用户统计 VO
*/
private AiUsageUserStatsRespVO convertToUserStatsVO(Map<String, Object> stats) {
Long callCount = getLongValue(stats, "call_count");
Long consumePoints = getLongValue(stats, "consume_points");
return AiUsageUserStatsRespVO.builder()
.userId(getLongValue(stats, "user_id"))
.mobile((String) stats.get("mobile"))
.callCount(callCount)
.consumePoints(consumePoints)
.inputTokens(getLongValue(stats, "input_tokens"))
.outputTokens(getLongValue(stats, "output_tokens"))
.totalTokens(getLongValue(stats, "total_tokens"))
.avgPointsPerCall(callCount > 0 ? (double) consumePoints / callCount : 0.0)
.build();
}
/**
* 转换为应用统计 VO
*/
private AiUsageAppStatsRespVO convertToAppStatsVO(Map<String, Object> stats,
Map<String, AiServiceConfigDO> configMap) {
String serviceCode = (String) stats.get("service_code");
Long callCount = getLongValue(stats, "call_count");
Long consumePoints = getLongValue(stats, "consume_points");
Long totalTokens = getLongValue(stats, "total_tokens");
AiServiceConfigDO config = configMap.get(serviceCode);
return AiUsageAppStatsRespVO.builder()
.serviceCode(serviceCode)
.serviceName(config != null ? config.getServiceName() : serviceCode)
.platform(config != null ? config.getPlatform() : "")
.callCount(callCount)
.consumePoints(consumePoints)
.inputTokens(getLongValue(stats, "input_tokens"))
.outputTokens(getLongValue(stats, "output_tokens"))
.totalTokens(totalTokens)
.avgPointsPerCall(callCount > 0 ? (double) consumePoints / callCount : 0.0)
.avgTokensPerCall(callCount > 0 ? (double) totalTokens / callCount : 0.0)
.build();
}
/**
* 转换为趋势数据项
*/
private AiUsageTrendRespVO.TrendItem convertToTrendItem(Map<String, Object> stats) {
return AiUsageTrendRespVO.TrendItem.builder()
.time((String) stats.get("time"))
.callCount(getLongValue(stats, "call_count"))
.consumePoints(getLongValue(stats, "consume_points"))
.totalTokens(getLongValue(stats, "total_tokens"))
.build();
}
/**
* 安全获取 Long 值
*/
private Long getLongValue(Map<String, Object> map, String key) {
Object value = map.get(key);
if (value == null) {
return 0L;
}
if (value instanceof Number) {
return ((Number) value).longValue();
}
try {
return Long.parseLong(value.toString());
} catch (NumberFormatException e) {
return 0L;
}
}
}

49
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/vo/AiUsageAppStatsRespVO.java Normal file
View File

@@ -0,0 +1,49 @@
package cn.iocoder.yudao.module.tik.muye.aiusagestats.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* AI 使用应用统计响应 VO
*/
@Schema(description = "AI 使用应用统计响应")
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class AiUsageAppStatsRespVO {
@Schema(description = "服务标识")
private String serviceCode;
@Schema(description = "服务名称")
private String serviceName;
@Schema(description = "平台")
private String platform;
@Schema(description = "调用次数")
private Long callCount;
@Schema(description = "消耗积分")
private Long consumePoints;
@Schema(description = "输入 Token 数")
private Long inputTokens;
@Schema(description = "输出 Token 数")
private Long outputTokens;
@Schema(description = "总 Token 数")
private Long totalTokens;
@Schema(description = "平均每次消耗积分")
private Double avgPointsPerCall;
@Schema(description = "平均每次 Token 数")
private Double avgTokensPerCall;
}

60
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/vo/AiUsageOverviewRespVO.java Normal file
View File

@@ -0,0 +1,60 @@
package cn.iocoder.yudao.module.tik.muye.aiusagestats.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.util.List;
/**
* AI 使用概览统计响应 VO
*/
@Schema(description = "AI 使用概览统计响应")
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class AiUsageOverviewRespVO {
@Schema(description = "总调用次数")
private Long totalCallCount;
@Schema(description = "总消耗积分")
private Long totalConsumePoints;
@Schema(description = "总 Token 数")
private Long totalTokens;
@Schema(description = "活跃用户数")
private Long activeUserCount;
@Schema(description = "业务类型统计列表")
private List<BizTypeStats> bizTypeStats;
/**
* 业务类型统计
*/
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public static class BizTypeStats {
@Schema(description = "业务类型")
private String bizType;
@Schema(description = "业务类型名称")
private String bizTypeName;
@Schema(description = "调用次数")
private Long callCount;
@Schema(description = "消耗积分")
private Long consumePoints;
@Schema(description = "Token 数")
private Long totalTokens;
}
}

40
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/vo/AiUsageStatsPageReqVO.java Normal file
View File

@@ -0,0 +1,40 @@
package cn.iocoder.yudao.module.tik.muye.aiusagestats.vo;
import cn.iocoder.yudao.framework.common.pojo.SortablePageParam;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.ToString;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;
import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
/**
* AI 使用统计查询请求 VO
*/
@Schema(description = "AI 使用统计查询请求")
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
public class AiUsageStatsPageReqVO extends SortablePageParam {
@Schema(description = "开始时间")
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
private LocalDateTime startTime;
@Schema(description = "结束时间")
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
private LocalDateTime endTime;
@Schema(description = "业务类型")
private String bizType;
@Schema(description = "用户ID")
private Long userId;
@Schema(description = "服务标识")
private String serviceCode;
}

45
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/vo/AiUsageTrendRespVO.java Normal file
View File

@@ -0,0 +1,45 @@
package cn.iocoder.yudao.module.tik.muye.aiusagestats.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.util.List;
/**
* AI 使用趋势统计响应 VO
*/
@Schema(description = "AI 使用趋势统计响应")
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class AiUsageTrendRespVO {
@Schema(description = "趋势数据列表")
private List<TrendItem> trendList;
/**
* 趋势数据项
*/
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public static class TrendItem {
@Schema(description = "时间（日期或小时）")
private String time;
@Schema(description = "调用次数")
private Long callCount;
@Schema(description = "消耗积分")
private Long consumePoints;
@Schema(description = "Token 数")
private Long totalTokens;
}
}

43
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/vo/AiUsageUserStatsRespVO.java Normal file
View File

@@ -0,0 +1,43 @@
package cn.iocoder.yudao.module.tik.muye.aiusagestats.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* AI 使用用户统计响应 VO
*/
@Schema(description = "AI 使用用户统计响应")
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class AiUsageUserStatsRespVO {
@Schema(description = "用户ID")
private Long userId;
@Schema(description = "手机号")
private String mobile;
@Schema(description = "调用次数")
private Long callCount;
@Schema(description = "消耗积分")
private Long consumePoints;
@Schema(description = "输入 Token 数")
private Long inputTokens;
@Schema(description = "输出 Token 数")
private Long outputTokens;
@Schema(description = "总 Token 数")
private Long totalTokens;
@Schema(description = "平均每次消耗积分")
private Double avgPointsPerCall;
}

44
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/memberuserprofile/service/MemberUserProfileServiceImpl.java
View File

@@ -1,6 +1,5 @@
package cn.iocoder.yudao.module.tik.muye.memberuserprofile.service;
import cn.iocoder.yudao.framework.common.exception.ErrorCode;
import cn.iocoder.yudao.module.member.api.user.MemberUserApi;
import cn.iocoder.yudao.module.member.api.user.dto.MemberUserRespDTO;
@@ -8,14 +7,15 @@ import cn.iocoder.yudao.module.tik.muye.memberuserprofile.dal.MemberUserProfileD
import cn.iocoder.yudao.module.tik.muye.memberuserprofile.mapper.MemberUserProfileMapper;
import cn.iocoder.yudao.module.tik.muye.memberuserprofile.vo.MemberUserProfilePageReqVO;
import cn.iocoder.yudao.module.tik.muye.memberuserprofile.vo.MemberUserProfileSaveReqVO;
import org.springframework.stereotype.Service;
import jakarta.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import java.time.LocalDateTime;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.*;
import java.time.LocalDateTime;
import java.util.List;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
@@ -96,11 +96,7 @@ public class MemberUserProfileServiceImpl implements MemberUserProfileService {
profile.setUserId(String.valueOf(userId));
// 获取用户手机号
MemberUserRespDTO user = memberUserApi.getUser(userId);
if (user != null) {
profile.setMobile(user.getMobile());
} else {
profile.setMobile("");
}
profile.setMobile(user != null ? user.getMobile() : "");
profile.setRegisterTime(LocalDateTime.now());
profile.setLastLoginTime(LocalDateTime.now());
profile.setTotalPoints(0);
@@ -136,28 +132,24 @@ public class MemberUserProfileServiceImpl implements MemberUserProfileService {
@Override
public boolean increaseUsedStorage(String userId, long fileSizeBytes) {
// 确保档案存在（兼容旧用户）
createIfAbsent(Long.parseLong(userId));
// 将字节转换为GB（保留6位小数）
BigDecimal storageGb = new BigDecimal(fileSizeBytes).divide(GB_TO_BYTES, 6, RoundingMode.HALF_UP);
String storageGbStr = storageGb.toPlainString();
int affectedRows = memberUserProfileMapper.updateStorageIncrease(userId, storageGbStr);
return affectedRows > 0;
String storageGbStr = convertBytesToGb(fileSizeBytes);
return memberUserProfileMapper.updateStorageIncrease(userId, storageGbStr) > 0;
}
@Override
public boolean decreaseUsedStorage(String userId, long fileSizeBytes) {
// 确保档案存在（兼容旧用户）
createIfAbsent(Long.parseLong(userId));
String storageGbStr = convertBytesToGb(fileSizeBytes);
return memberUserProfileMapper.updateStorageDecrease(userId, storageGbStr) > 0;
}
// 将字节转换为GB（保留6位小数）
/**
* 将字节转换为GB字符串
*/
private String convertBytesToGb(long fileSizeBytes) {
BigDecimal storageGb = new BigDecimal(fileSizeBytes).divide(GB_TO_BYTES, 6, RoundingMode.HALF_UP);
String storageGbStr = storageGb.toPlainString();
int affectedRows = memberUserProfileMapper.updateStorageDecrease(userId, storageGbStr);
return affectedRows > 0;
return storageGb.toPlainString();
}
@Override

16
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/pointrecord/dal/PointRecordDO.java
View File

@@ -64,5 +64,21 @@ public class PointRecordDO extends BaseDO {
* 状态：pending-预扣 confirmed-已确认 canceled-已取消
*/
private String status;
/**
* AI服务标识(对应muye_ai_service_config.service_code)
*/
private String serviceCode;
/**
* 输入token数
*/
private Integer inputTokens;
/**
* 输出token数
*/
private Integer outputTokens;
/**
* 总token数
*/
private Integer totalTokens;
}

90
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/pointrecord/mapper/PointRecordMapper.java
View File

@@ -1,5 +1,6 @@
package cn.iocoder.yudao.module.tik.muye.pointrecord.mapper;
import java.time.LocalDateTime;
import java.util.*;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
@@ -8,6 +9,8 @@ import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.module.tik.muye.pointrecord.dal.PointRecordDO;
import cn.iocoder.yudao.module.tik.muye.pointrecord.vo.PointRecordPageReqVO;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import org.apache.ibatis.annotations.Update;
/**
@@ -41,4 +44,89 @@ public interface PointRecordMapper extends BaseMapperX<PointRecordDO> {
"WHERE status = 'pending' AND create_time < DATE_SUB(NOW(), INTERVAL 30 MINUTE)")
int cancelExpiredPendingRecords();
}
/**
* 按业务类型统计
*/
@Select("<script>" +
"SELECT biz_type, " +
" COUNT(*) as call_count, " +
" SUM(ABS(point_amount)) as consume_points, " +
" SUM(COALESCE(total_tokens, 0)) as total_tokens " +
"FROM muye_point_record " +
"WHERE status = 'confirmed' AND type = 'decrease' " +
"<if test='startTime != null'> AND create_time &gt;= #{startTime}</if>" +
"<if test='endTime != null'> AND create_time &lt;= #{endTime}</if>" +
"GROUP BY biz_type " +
"ORDER BY consume_points DESC" +
"</script>")
List<Map<String, Object>> selectBizTypeStats(@Param("startTime") LocalDateTime startTime,
@Param("endTime") LocalDateTime endTime);
/**
* 按用户统计
*/
@Select("<script>" +
"SELECT user_id, mobile, " +
" COUNT(*) as call_count, " +
" SUM(ABS(point_amount)) as consume_points, " +
" SUM(COALESCE(input_tokens, 0)) as input_tokens, " +
" SUM(COALESCE(output_tokens, 0)) as output_tokens, " +
" SUM(COALESCE(total_tokens, 0)) as total_tokens " +
"FROM muye_point_record " +
"WHERE status = 'confirmed' AND type = 'decrease' " +
"<if test='startTime != null'> AND create_time &gt;= #{startTime}</if>" +
"<if test='endTime != null'> AND create_time &lt;= #{endTime}</if>" +
"<if test='bizType != null and bizType != \"\"'> AND biz_type = #{bizType}</if>" +
"<if test='userId != null'> AND user_id = #{userId}</if>" +
"GROUP BY user_id, mobile " +
"ORDER BY consume_points DESC" +
"</script>")
List<Map<String, Object>> selectUserStats(@Param("startTime") LocalDateTime startTime,
@Param("endTime") LocalDateTime endTime,
@Param("bizType") String bizType,
@Param("userId") Long userId);
/**
* 按应用统计
*/
@Select("<script>" +
"SELECT service_code, " +
" COUNT(*) as call_count, " +
" SUM(ABS(point_amount)) as consume_points, " +
" SUM(COALESCE(input_tokens, 0)) as input_tokens, " +
" SUM(COALESCE(output_tokens, 0)) as output_tokens, " +
" SUM(COALESCE(total_tokens, 0)) as total_tokens " +
"FROM muye_point_record " +
"WHERE status = 'confirmed' AND type = 'decrease' AND service_code IS NOT NULL AND service_code != '' " +
"<if test='startTime != null'> AND create_time &gt;= #{startTime}</if>" +
"<if test='endTime != null'> AND create_time &lt;= #{endTime}</if>" +
"<if test='bizType != null and bizType != \"\"'> AND biz_type = #{bizType}</if>" +
"GROUP BY service_code " +
"ORDER BY consume_points DESC" +
"</script>")
List<Map<String, Object>> selectAppStats(@Param("startTime") LocalDateTime startTime,
@Param("endTime") LocalDateTime endTime,
@Param("bizType") String bizType);
/**
* 按时间趋势统计
*/
@Select("<script>" +
"SELECT DATE_FORMAT(create_time, #{dateFormat}) as time, " +
" COUNT(*) as call_count, " +
" SUM(ABS(point_amount)) as consume_points, " +
" SUM(COALESCE(total_tokens, 0)) as total_tokens " +
"FROM muye_point_record " +
"WHERE status = 'confirmed' AND type = 'decrease' " +
"<if test='startTime != null'> AND create_time &gt;= #{startTime}</if>" +
"<if test='endTime != null'> AND create_time &lt;= #{endTime}</if>" +
"<if test='bizType != null and bizType != \"\"'> AND biz_type = #{bizType}</if>" +
"GROUP BY DATE_FORMAT(create_time, #{dateFormat}) " +
"ORDER BY time ASC" +
"</script>")
List<Map<String, Object>> selectTrend(@Param("startTime") LocalDateTime startTime,
@Param("endTime") LocalDateTime endTime,
@Param("bizType") String bizType,
@Param("dateFormat") String dateFormat);
}

22
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/points/service/PointsService.java
View File

@@ -66,4 +66,26 @@ public interface PointsService {
*/
void cancelPendingDeduct(Long recordId);
/**
* 创建预扣（带服务标识）
*
* @param userId 用户ID
* @param points 预扣积分数量
* @param bizType 业务类型
* @param bizId 业务关联ID
* @param serviceCode AI服务标识
* @return 预扣记录ID
*/
Long createPendingDeduct(String userId, Integer points, String bizType, String bizId, String serviceCode);
/**
* 确认预扣并更新 token 信息
*
* @param recordId 预扣记录ID
* @param inputTokens 输入token数
* @param outputTokens 输出token数
* @param totalTokens 总token数
*/
void confirmPendingDeductWithTokens(Long recordId, Integer inputTokens, Integer outputTokens, Integer totalTokens);
}

77
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/points/service/PointsServiceImpl.java
View File

@@ -191,4 +191,81 @@ public class PointsServiceImpl implements PointsService {
recordId, record.getUserId(), Math.abs(record.getPointAmount()));
}
@Override
@Transactional(rollbackFor = Exception.class)
public Long createPendingDeduct(String userId, Integer points, String bizType, String bizId, String serviceCode) {
// 1. 预检积分
checkPoints(userId, points);
// 2. 查询当前余额
MemberUserProfileDO profile = memberUserProfileMapper.selectByUserId(userId);
// 2.1 获取用户手机号
MemberUserDO user = memberUserService.getUser(Long.parseLong(userId));
String mobile = user != null ? user.getMobile() : "";
// 3. 创建预扣记录（待确认状态）
PointRecordDO record = PointRecordDO.builder()
.userId(Long.parseLong(userId))
.mobile(mobile)
.type("decrease")
.pointAmount(-points)
.balance(profile.getRemainingPoints())
.reason(bizType + "(预扣)")
.bizType(bizType)
.bizId(bizId != null ? bizId : UUID.randomUUID().toString())
.serviceCode(serviceCode)
.inputTokens(0)
.outputTokens(0)
.totalTokens(0)
.status(STATUS_PENDING)
.build();
pointRecordMapper.insert(record);
log.info("[createPendingDeduct] 用户 {} 创建预扣 {} 积分，业务类型 {}，服务标识 {}，记录ID {}",
userId, points, bizType, serviceCode, record.getId());
return record.getId();
}
@Override
@Transactional(rollbackFor = Exception.class)
public void confirmPendingDeductWithTokens(Long recordId, Integer inputTokens, Integer outputTokens, Integer totalTokens) {
// 1. 查询预扣记录
PointRecordDO record = pointRecordMapper.selectById(recordId);
if (record == null) {
throw exception(POINTS_PENDING_NOT_FOUND);
}
// 2. 校验状态
if (!STATUS_PENDING.equals(record.getStatus())) {
throw exception(POINTS_PENDING_ALREADY_CONFIRMED);
}
// 3. 获取扣减信息
String userId = record.getUserId().toString();
Integer points = Math.abs(record.getPointAmount());
// 4. 原子扣减积分
int affectedRows = memberUserProfileMapper.updatePointsDeduct(userId, points);
if (affectedRows == 0) {
log.warn("[confirmPendingDeductWithTokens] 积分扣减失败，可能余额不足，记录ID {}", recordId);
throw exception(POINTS_DEDUCT_FAILED);
}
// 5. 查询扣减后余额
MemberUserProfileDO profile = memberUserProfileMapper.selectByUserId(userId);
// 6. 更新预扣记录状态和 token 信息
record.setStatus(STATUS_CONFIRMED);
record.setBalance(profile.getRemainingPoints());
record.setReason(record.getReason().replace("(预扣)", ""));
record.setInputTokens(inputTokens != null ? inputTokens : 0);
record.setOutputTokens(outputTokens != null ? outputTokens : 0);
record.setTotalTokens(totalTokens != null ? totalTokens : 0);
pointRecordMapper.updateById(record);
log.info("[confirmPendingDeductWithTokens] 确认预扣记录 {}，用户 {} 扣减 {} 积分，tokens: {}/{}/{}",
recordId, userId, points, inputTokens, outputTokens, totalTokens);
}
}

26
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/voice/service/TikUserVoiceServiceImpl.java
View File

@@ -113,8 +113,17 @@ public class TikUserVoiceServiceImpl implements TikUserVoiceService {
public Long createVoice(AppTikUserVoiceCreateReqVO createReqVO) {
Long userId = SecurityFrameworkUtils.getLoginUserId();
// 1. 校验文件是否存在且属于voice分类
FileDO fileDO = fileMapper.selectById(createReqVO.getFileId());
// 1. 前端传入的是 userFileId（tik_user_file.id），先查询用户文件记录
TikUserFileDO userFile = userFileMapper.selectOne(new LambdaQueryWrapperX<TikUserFileDO>()
.eq(TikUserFileDO::getId, createReqVO.getFileId())
.eq(TikUserFileDO::getFileCategory, "voice")
.eq(TikUserFileDO::getUserId, userId));
if (userFile == null) {
throw exception(VOICE_FILE_NOT_EXISTS, "文件不存在或不属于voice分类");
}
// 2. 通过 userFile.fileId（infra_file.id）查询实际文件信息
FileDO fileDO = fileMapper.selectById(userFile.getFileId());
if (fileDO == null) {
throw exception(VOICE_FILE_NOT_EXISTS);
}
@@ -126,15 +135,6 @@ public class TikUserVoiceServiceImpl implements TikUserVoiceService {
String.format("音频文件过大（%.1fMB），请上传小于5MB的音频文件", sizeMB));
}
// 验证文件分类是否为voice（通过tik_user_file表查询）
TikUserFileDO userFile = userFileMapper.selectOne(new LambdaQueryWrapperX<TikUserFileDO>()
.eq(TikUserFileDO::getFileId, createReqVO.getFileId())
.eq(TikUserFileDO::getFileCategory, "voice")
.eq(TikUserFileDO::getUserId, userId));
if (userFile == null) {
throw exception(VOICE_FILE_NOT_EXISTS, "文件不存在或不属于voice分类");
}
// 2. 校验名称是否重复
TikUserVoiceDO existingVoice = voiceMapper.selectOne(new LambdaQueryWrapperX<TikUserVoiceDO>()
.eq(TikUserVoiceDO::getUserId, userId)
@@ -144,11 +144,11 @@ public class TikUserVoiceServiceImpl implements TikUserVoiceService {
throw exception(VOICE_NAME_DUPLICATE);
}
// 3. 创建配音记录
// 3. 创建配音记录（fileId 存储 infra_file.id，用于后续查询文件URL）
TikUserVoiceDO voice = new TikUserVoiceDO()
.setUserId(userId)
.setName(createReqVO.getName())
.setFileId(createReqVO.getFileId())
.setFileId(userFile.getFileId()) // 存储 infra_file.id
.setLanguage(StrUtil.blankToDefault(createReqVO.getLanguage(), "zh-CN"))
.setGender(StrUtil.blankToDefault(createReqVO.getGender(), "female"))
.setNote(createReqVO.getNote())

2
yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/voice/vo/AppTikUserVoiceCreateReqVO.java
View File

@@ -19,7 +19,7 @@ public class AppTikUserVoiceCreateReqVO {
@NotBlank(message = "配音名称不能为空")
private String name;
@Schema(description = "音频文件编号（关联 infra_file.id）", requiredMode = Schema.RequiredMode.REQUIRED, example = "1")
@Schema(description = "用户文件编号（关联 tik_user_file.id，上传文件后返回的 userFileId）", requiredMode = Schema.RequiredMode.REQUIRED, example = "1")
@NotNull(message = "音频文件编号不能为空")
private Long fileId;

92
yudao-ui-admin-vue3/src/api/muye/aiusagestats/index.ts Normal file
View File

@@ -0,0 +1,92 @@
import request from '@/config/axios'
export interface AiUsageOverview {
totalCallCount: number
totalConsumePoints: number
totalTokens: number
activeUserCount: number
bizTypeStats: BizTypeStats[]
}
export interface BizTypeStats {
bizType: string
bizTypeName: string
callCount: number
consumePoints: number
totalTokens: number
}
export interface AiUsageUserStats {
userId: number
mobile: string
callCount: number
consumePoints: number
inputTokens: number
outputTokens: number
totalTokens: number
avgPointsPerCall: number
}
export interface AiUsageAppStats {
serviceCode: string
serviceName: string
platform: string
callCount: number
consumePoints: number
inputTokens: number
outputTokens: number
totalTokens: number
avgPointsPerCall: number
avgTokensPerCall: number
}
export interface AiUsageTrend {
trendList: TrendItem[]
}
export interface TrendItem {
time: string
callCount: number
consumePoints: number
totalTokens: number
}
// 获取概览统计
export const getAiUsageOverview = (params: {
startTime?: string
endTime?: string
bizType?: string
}) => {
return request.get({ url: '/muye/ai-usage-stats/overview', params })
}
// 获取用户统计分页
export const getAiUsageUserStatsPage = (params: {
pageNo: number
pageSize: number
startTime?: string
endTime?: string
bizType?: string
userId?: number
}) => {
return request.get({ url: '/muye/ai-usage-stats/user-stats', params })
}
// 获取应用统计列表
export const getAiUsageAppStats = (params: {
startTime?: string
endTime?: string
bizType?: string
}) => {
return request.get({ url: '/muye/ai-usage-stats/app-stats', params })
}
// 获取趋势数据
export const getAiUsageTrend = (params: {
startTime?: string
endTime?: string
bizType?: string
type?: string
}) => {
return request.get({ url: '/muye/ai-usage-stats/trend', params })
}

474
yudao-ui-admin-vue3/src/views/muye/aiusage/index.vue Normal file
View File

@@ -0,0 +1,474 @@
<template>
<ContentWrap>
<!-- 筛选条件 -->
<el-form :model="queryParams" :inline="true" class="-mb-15px">
<el-form-item label="时间范围">
<el-date-picker
v-model="queryParams.timeRange"
type="daterange"
value-format="YYYY-MM-DD HH:mm:ss"
:shortcuts="timeShortcuts"
range-separator="-"
start-placeholder="开始时间"
end-placeholder="结束时间"
class="!w-240px"
@change="handleQuery"
/>
</el-form-item>
<el-form-item label="业务类型">
<el-select v-model="queryParams.bizType" placeholder="全部" clearable class="!w-180px" @change="handleQuery">
<el-option label="全部" value="" />
<el-option label="AI对话" value="dify_chat" />
<el-option label="文案改写" value="forecast_rewrite" />
<el-option label="语音合成" value="voice_tts" />
<el-option label="数字人" value="digital_human" />
</el-select>
</el-form-item>
</el-form>
</ContentWrap>
<!-- Tab 切换 -->
<ContentWrap>
<el-tabs v-model="activeTab" @tab-change="handleTabChange">
<!-- 概览 Tab -->
<el-tab-pane label="概览" name="overview">
<!-- 统计卡片 -->
<el-row :gutter="16" class="mb-16px">
<el-col :span="6">
<el-card shadow="hover">
<div class="stat-card">
<div class="stat-icon bg-blue">
<Icon icon="ep:phone" />
</div>
<div class="stat-content">
<div class="stat-label">总调用次数</div>
<div class="stat-value">{{ overviewData.totalCallCount?.toLocaleString() || 0 }}</div>
</div>
</div>
</el-card>
</el-col>
<el-col :span="6">
<el-card shadow="hover">
<div class="stat-card">
<div class="stat-icon bg-orange">
<Icon icon="ep:coin" />
</div>
<div class="stat-content">
<div class="stat-label">总消耗积分</div>
<div class="stat-value">{{ overviewData.totalConsumePoints?.toLocaleString() || 0 }}</div>
</div>
</div>
</el-card>
</el-col>
<el-col :span="6">
<el-card shadow="hover">
<div class="stat-card">
<div class="stat-icon bg-green">
<Icon icon="ep:data-analysis" />
</div>
<div class="stat-content">
<div class="stat-label">总 Token 数</div>
<div class="stat-value">{{ overviewData.totalTokens?.toLocaleString() || 0 }}</div>
</div>
</div>
</el-card>
</el-col>
<el-col :span="6">
<el-card shadow="hover">
<div class="stat-card">
<div class="stat-icon bg-purple">
<Icon icon="ep:user" />
</div>
<div class="stat-content">
<div class="stat-label">活跃用户数</div>
<div class="stat-value">{{ overviewData.activeUserCount?.toLocaleString() || 0 }}</div>
</div>
</div>
</el-card>
</el-col>
</el-row>
<!-- 图表区域 -->
<el-row :gutter="16">
<el-col :span="16">
<el-card shadow="hover">
<template #header>
<div class="card-header">
<span>消耗趋势</span>
</div>
</template>
<div ref="trendChartRef" class="chart-container"></div>
</el-card>
</el-col>
<el-col :span="8">
<el-card shadow="hover">
<template #header>
<div class="card-header">
<span>业务类型占比</span>
</div>
</template>
<div ref="pieChartRef" class="chart-container"></div>
</el-card>
</el-col>
</el-row>
<!-- 业务类型统计表格 -->
<el-card shadow="hover" class="mt-16px">
<template #header>
<div class="card-header">
<span>业务类型统计</span>
</div>
</template>
<el-table :data="overviewData.bizTypeStats || []" :stripe="true">
<el-table-column label="业务类型" align="center" prop="bizTypeName" />
<el-table-column label="调用次数" align="center" prop="callCount">
<template #default="{ row }">{{ row.callCount?.toLocaleString() }}</template>
</el-table-column>
<el-table-column label="消耗积分" align="center" prop="consumePoints">
<template #default="{ row }">{{ row.consumePoints?.toLocaleString() }}</template>
</el-table-column>
<el-table-column label="Token 数" align="center" prop="totalTokens">
<template #default="{ row }">{{ row.totalTokens?.toLocaleString() || '-' }}</template>
</el-table-column>
</el-table>
</el-card>
</el-tab-pane>
<!-- 用户统计 Tab -->
<el-tab-pane label="用户统计" name="user">
<el-table v-loading="userLoading" :data="userStatsList" :stripe="true">
<el-table-column label="用户ID" align="center" prop="userId" />
<el-table-column label="手机号" align="center" prop="mobile" />
<el-table-column label="调用次数" align="center" prop="callCount">
<template #default="{ row }">{{ row.callCount?.toLocaleString() }}</template>
</el-table-column>
<el-table-column label="消耗积分" align="center" prop="consumePoints">
<template #default="{ row }">{{ row.consumePoints?.toLocaleString() }}</template>
</el-table-column>
<el-table-column label="输入Token" align="center" prop="inputTokens">
<template #default="{ row }">{{ row.inputTokens?.toLocaleString() || '-' }}</template>
</el-table-column>
<el-table-column label="输出Token" align="center" prop="outputTokens">
<template #default="{ row }">{{ row.outputTokens?.toLocaleString() || '-' }}</template>
</el-table-column>
<el-table-column label="总Token" align="center" prop="totalTokens">
<template #default="{ row }">{{ row.totalTokens?.toLocaleString() || '-' }}</template>
</el-table-column>
<el-table-column label="平均积分/次" align="center" prop="avgPointsPerCall">
<template #default="{ row }">{{ row.avgPointsPerCall?.toFixed(2) }}</template>
</el-table-column>
</el-table>
<Pagination
v-model:page="userPageParams.pageNo"
v-model:limit="userPageParams.pageSize"
:total="userPageParams.total"
@pagination="getUserStatsList"
/>
</el-tab-pane>
<!-- 应用统计 Tab -->
<el-tab-pane label="应用统计" name="app">
<el-table v-loading="appLoading" :data="appStatsList" :stripe="true">
<el-table-column label="服务标识" align="center" prop="serviceCode" />
<el-table-column label="服务名称" align="center" prop="serviceName" />
<el-table-column label="平台" align="center" prop="platform" />
<el-table-column label="调用次数" align="center" prop="callCount">
<template #default="{ row }">{{ row.callCount?.toLocaleString() }}</template>
</el-table-column>
<el-table-column label="消耗积分" align="center" prop="consumePoints">
<template #default="{ row }">{{ row.consumePoints?.toLocaleString() }}</template>
</el-table-column>
<el-table-column label="总Token" align="center" prop="totalTokens">
<template #default="{ row }">{{ row.totalTokens?.toLocaleString() || '-' }}</template>
</el-table-column>
<el-table-column label="平均积分/次" align="center" prop="avgPointsPerCall">
<template #default="{ row }">{{ row.avgPointsPerCall?.toFixed(2) }}</template>
</el-table-column>
<el-table-column label="平均Token/次" align="center" prop="avgTokensPerCall">
<template #default="{ row }">{{ row.avgTokensPerCall?.toFixed(2) }}</template>
</el-table-column>
</el-table>
</el-tab-pane>
</el-tabs>
</ContentWrap>
</template>
<script setup lang="ts">
import { ref, reactive, onMounted, onUnmounted, nextTick } from 'vue'
import * as echarts from 'echarts'
import {
getAiUsageOverview,
getAiUsageUserStatsPage,
getAiUsageAppStats,
getAiUsageTrend,
type AiUsageOverview,
type AiUsageUserStats,
type AiUsageAppStats
} from '@/api/muye/aiusagestats'
defineOptions({ name: 'AiUsageStats' })
// 时间快捷选项
const timeShortcuts = [
{ text: '今天', value: () => {
const today = new Date()
today.setHours(0, 0, 0, 0)
return [today, new Date()]
}},
{ text: '本周', value: () => {
const today = new Date()
const day = today.getDay() || 7
const monday = new Date(today)
monday.setDate(today.getDate() - day + 1)
monday.setHours(0, 0, 0, 0)
return [monday, new Date()]
}},
{ text: '本月', value: () => {
const today = new Date()
const firstDay = new Date(today.getFullYear(), today.getMonth(), 1)
return [firstDay, new Date()]
}},
{ text: '最近7天', value: () => {
const end = new Date()
const start = new Date()
start.setDate(start.getDate() - 7)
return [start, end]
}},
{ text: '最近30天', value: () => {
const end = new Date()
const start = new Date()
start.setDate(start.getDate() - 30)
return [start, end]
}}
]
// 查询参数
const queryParams = reactive({
timeRange: [] as string[],
bizType: ''
})
// Tab
const activeTab = ref('overview')
// 概览数据
const overviewData = ref<AiUsageOverview>({} as AiUsageOverview)
// 用户统计
const userLoading = ref(false)
const userStatsList = ref<AiUsageUserStats[]>([])
const userPageParams = reactive({
pageNo: 1,
pageSize: 10,
total: 0
})
// 应用统计
const appLoading = ref(false)
const appStatsList = ref<AiUsageAppStats[]>([])
// 图表
const trendChartRef = ref<HTMLElement>()
const pieChartRef = ref<HTMLElement>()
let trendChart: echarts.ECharts | null = null
let pieChart: echarts.ECharts | null = null
// 获取概览数据
const getOverviewData = async () => {
const [startTime, endTime] = queryParams.timeRange || []
const res = await getAiUsageOverview({
startTime,
endTime,
bizType: queryParams.bizType
})
overviewData.value = res.data || {}
}
// 获取趋势数据并绘制图表
const getTrendData = async () => {
const [startTime, endTime] = queryParams.timeRange || []
const res = await getAiUsageTrend({
startTime,
endTime,
bizType: queryParams.bizType,
type: 'day'
})
await nextTick()
if (!trendChart && trendChartRef.value) {
trendChart = echarts.init(trendChartRef.value)
}
if (trendChart && res.data?.trendList) {
const data = res.data.trendList
trendChart.setOption({
tooltip: { trigger: 'axis' },
legend: { data: ['调用次数', '消耗积分', 'Token数'] },
xAxis: { type: 'category', data: data.map((d: any) => d.time) },
yAxis: [
{ type: 'value', name: '次数/积分' },
{ type: 'value', name: 'Token' }
],
series: [
{ name: '调用次数', type: 'line', data: data.map((d: any) => d.callCount) },
{ name: '消耗积分', type: 'line', data: data.map((d: any) => d.consumePoints) },
{ name: 'Token数', type: 'line', yAxisIndex: 1, data: data.map((d: any) => d.totalTokens) }
]
})
}
}
// 绘制饼图
const drawPieChart = () => {
if (!pieChart && pieChartRef.value) {
pieChart = echarts.init(pieChartRef.value)
}
if (pieChart && overviewData.value.bizTypeStats) {
const data = overviewData.value.bizTypeStats.map((item: any) => ({
name: item.bizTypeName,
value: item.consumePoints
}))
pieChart.setOption({
tooltip: { trigger: 'item', formatter: '{b}: {c} ({d}%)' },
legend: { orient: 'vertical', left: 'left' },
series: [{
type: 'pie',
radius: '50%',
data,
emphasis: { itemStyle: { shadowBlur: 10, shadowOffsetX: 0, shadowColor: 'rgba(0, 0, 0, 0.5)' }}
}]
})
}
}
// 获取用户统计列表
const getUserStatsList = async () => {
userLoading.value = true
try {
const [startTime, endTime] = queryParams.timeRange || []
const res = await getAiUsageUserStatsPage({
pageNo: userPageParams.pageNo,
pageSize: userPageParams.pageSize,
startTime,
endTime,
bizType: queryParams.bizType
})
userStatsList.value = res.data?.list || []
userPageParams.total = res.data?.total || 0
} finally {
userLoading.value = false
}
}
// 获取应用统计列表
const getAppStatsList = async () => {
appLoading.value = true
try {
const [startTime, endTime] = queryParams.timeRange || []
const res = await getAiUsageAppStats({
startTime,
endTime,
bizType: queryParams.bizType
})
appStatsList.value = res.data || []
} finally {
appLoading.value = false
}
}
// 查询
const handleQuery = async () => {
await getOverviewData()
await getTrendData()
drawPieChart()
if (activeTab.value === 'user') {
getUserStatsList()
} else if (activeTab.value === 'app') {
getAppStatsList()
}
}
// Tab 切换
const handleTabChange = (tab: string) => {
if (tab === 'user') {
getUserStatsList()
} else if (tab === 'app') {
getAppStatsList()
}
}
// 窗口大小变化时重绘图表
const handleResize = () => {
trendChart?.resize()
pieChart?.resize()
}
onMounted(async () => {
// 设置默认时间范围（最近7天）
const end = new Date()
const start = new Date()
start.setDate(start.getDate() - 7)
queryParams.timeRange = [
start.toISOString().slice(0, 19).replace('T', ' '),
end.toISOString().slice(0, 19).replace('T', ' ')
]
await handleQuery()
window.addEventListener('resize', handleResize)
})
onUnmounted(() => {
window.removeEventListener('resize', handleResize)
trendChart?.dispose()
pieChart?.dispose()
})
</script>
<style scoped lang="scss">
.stat-card {
display: flex;
align-items: center;
padding: 8px;
.stat-icon {
width: 48px;
height: 48px;
border-radius: 8px;
display: flex;
align-items: center;
justify-content: center;
font-size: 24px;
color: #fff;
margin-right: 16px;
&.bg-blue { background: linear-gradient(135deg, #409eff, #3375b9); }
&.bg-orange { background: linear-gradient(135deg, #e6a23c, #cf9236); }
&.bg-green { background: linear-gradient(135deg, #67c23a, #529b2e); }
&.bg-purple { background: linear-gradient(135deg, #909399, #73767a); }
}
.stat-content {
.stat-label {
font-size: 14px;
color: #909399;
margin-bottom: 4px;
}
.stat-value {
font-size: 24px;
font-weight: 600;
color: #303133;
}
}
}
.card-header {
font-size: 16px;
font-weight: 500;
}
.chart-container {
height: 300px;
}
</style>

23
yudao-ui-admin-vue3/src/views/muye/pointrecord/index.vue
View File

@@ -247,43 +247,42 @@ const openForm = (type: string, id?: number) => {
/** 删除按钮操作 */
const handleDelete = async (id: number) => {
try {
// 删除的二次确认
await message.delConfirm()
// 发起删除
await PointRecordApi.deletePointRecord(id)
message.success(t('common.delSuccess'))
// 刷新列表
await getList()
} catch {}
} catch {
// 用户取消操作
}
}
/** 批量删除积分记录 */
const handleDeleteBatch = async () => {
try {
// 删除的二次确认
await message.delConfirm()
await PointRecordApi.deletePointRecordList(checkedIds.value);
checkedIds.value = [];
await PointRecordApi.deletePointRecordList(checkedIds.value)
checkedIds.value = []
message.success(t('common.delSuccess'))
await getList();
} catch {}
await getList()
} catch {
// 用户取消操作
}
}
const checkedIds = ref<number[]>([])
const handleRowCheckboxChange = (records: PointRecord[]) => {
checkedIds.value = records.map((item) => item.id!);
checkedIds.value = records.map((item) => item.id!)
}
/** 导出按钮操作 */
const handleExport = async () => {
try {
// 导出的二次确认
await message.exportConfirm()
// 发起导出
exportLoading.value = true
const data = await PointRecordApi.exportPointRecord(queryParams)
download.excel(data, '积分记录.xls')
} catch {
// 用户取消操作
} finally {
exportLoading.value = false
}