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

feat: 优化

Browse Source
This commit is contained in:
sion123
2026-03-15 15:37:32 +08:00
parent 8c5ab8eac4
commit 7b18bde0e8
1 changed files with 85 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -48,110 +48,118 @@ public class AiUsageStatsServiceImpl implements AiUsageStatsService {
@Override @Override
public AiUsageOverviewRespVO getOverview(LocalDateTime startTime, LocalDateTime endTime, String bizType) { public AiUsageOverviewRespVO getOverview(LocalDateTime startTime, LocalDateTime endTime, String bizType) {
// 查询业务类型统计 return TenantUtils.executeIgnore(() -> {
List<Map<String, Object>> bizTypeStatsList = pointRecordMapper.selectBizTypeStats(startTime, endTime); // 查询业务类型统计
List<Map<String, Object>> bizTypeStatsList = pointRecordMapper.selectBizTypeStats(startTime, endTime);
// 计算汇总数据 // 计算汇总数据
long totalCallCount = 0; long totalCallCount = 0;
long totalConsumePoints = 0; long totalConsumePoints = 0;
long totalTokens = 0; long totalTokens = 0;
List<AiUsageOverviewRespVO.BizTypeStats> bizTypeStats = new ArrayList<>(); List<AiUsageOverviewRespVO.BizTypeStats> bizTypeStats = new ArrayList<>();
for (Map<String, Object> stats : bizTypeStatsList) { for (Map<String, Object> stats : bizTypeStatsList) {
String bt = (String) stats.get("bizType"); String bt = (String) stats.get("bizType");
// 如果指定了业务类型，只统计该类型 // 如果指定了业务类型，只统计该类型
if (bizType != null && !bizType.isEmpty() && !bizType.equals(bt)) { if (bizType != null && !bizType.isEmpty() && !bizType.equals(bt)) {
continue; continue;
}
long callCount = getLongValue(stats, "callCount");
long consumePoints = getLongValue(stats, "consumePoints");
long tokens = getLongValue(stats, "totalTokens");
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 callCount = getLongValue(stats, "callCount"); // 查询活跃用户数
long consumePoints = getLongValue(stats, "consumePoints"); Long activeUserCount = pointRecordMapper.selectActiveUserCount(startTime, endTime, bizType);
long tokens = getLongValue(stats, "totalTokens");
totalCallCount += callCount; return AiUsageOverviewRespVO.builder()
totalConsumePoints += consumePoints; .totalCallCount(totalCallCount)
totalTokens += tokens; .totalConsumePoints(totalConsumePoints)
.totalTokens(totalTokens)
bizTypeStats.add(AiUsageOverviewRespVO.BizTypeStats.builder() .activeUserCount(activeUserCount)
.bizType(bt) .bizTypeStats(bizTypeStats)
.bizTypeName(BIZ_TYPE_NAMES.getOrDefault(bt, bt)) .build();
.callCount(callCount) });
.consumePoints(consumePoints)
.totalTokens(tokens)
.build());
}
// 查询活跃用户数
Long activeUserCount = pointRecordMapper.selectActiveUserCount(startTime, endTime, bizType);
return AiUsageOverviewRespVO.builder()
.totalCallCount(totalCallCount)
.totalConsumePoints(totalConsumePoints)
.totalTokens(totalTokens)
.activeUserCount(activeUserCount)
.bizTypeStats(bizTypeStats)
.build();
} }
@Override @Override
public PageResult<AiUsageUserStatsRespVO> getUserStatsPage(AiUsageStatsPageReqVO reqVO) { public PageResult<AiUsageUserStatsRespVO> getUserStatsPage(AiUsageStatsPageReqVO reqVO) {
// 查询用户统计数据 return TenantUtils.executeIgnore(() -> {
List<Map<String, Object>> userStatsList = pointRecordMapper.selectUserStats( // 查询用户统计数据
reqVO.getStartTime(), reqVO.getEndTime(), reqVO.getBizType(), reqVO.getUserId(), reqVO.getMobile()); List<Map<String, Object>> userStatsList = pointRecordMapper.selectUserStats(
reqVO.getStartTime(), reqVO.getEndTime(), reqVO.getBizType(), reqVO.getUserId(), reqVO.getMobile());
// 计算总数 // 计算总数
long total = userStatsList.size(); long total = userStatsList.size();
// 分页处理 // 分页处理
int fromIndex = (reqVO.getPageNo() - 1) * reqVO.getPageSize(); int fromIndex = (reqVO.getPageNo() - 1) * reqVO.getPageSize();
int toIndex = Math.min(fromIndex + reqVO.getPageSize(), userStatsList.size()); int toIndex = Math.min(fromIndex + reqVO.getPageSize(), userStatsList.size());
if (fromIndex >= userStatsList.size()) { if (fromIndex >= userStatsList.size()) {
return new PageResult<>(Collections.emptyList(), total); return new PageResult<>(Collections.emptyList(), total);
} }
List<Map<String, Object>> pagedList = userStatsList.subList(fromIndex, toIndex); List<Map<String, Object>> pagedList = userStatsList.subList(fromIndex, toIndex);
// 转换为 VO // 转换为 VO
List<AiUsageUserStatsRespVO> list = pagedList.stream() List<AiUsageUserStatsRespVO> list = pagedList.stream()
.map(this::convertToUserStatsVO) .map(this::convertToUserStatsVO)
.collect(Collectors.toList()); .collect(Collectors.toList());
return new PageResult<>(list, total); return new PageResult<>(list, total);
});
} }
@Override @Override
public List<AiUsageAppStatsRespVO> getAppStats(LocalDateTime startTime, LocalDateTime endTime, String bizType) { public List<AiUsageAppStatsRespVO> getAppStats(LocalDateTime startTime, LocalDateTime endTime, String bizType) {
// 查询应用统计数据 return TenantUtils.executeIgnore(() -> {
List<Map<String, Object>> appStatsList = pointRecordMapper.selectAppStats( // 查询应用统计数据
startTime, endTime, bizType); List<Map<String, Object>> appStatsList = pointRecordMapper.selectAppStats(
startTime, endTime, bizType);
// 获取服务配置信息 // 获取服务配置信息
Map<String, AiServiceConfigDO> configMap = aiServiceConfigMapper.selectList( Map<String, AiServiceConfigDO> configMap = aiServiceConfigMapper.selectList(
new LambdaQueryWrapperX<AiServiceConfigDO>().eq(AiServiceConfigDO::getStatus, 1)) new LambdaQueryWrapperX<AiServiceConfigDO>().eq(AiServiceConfigDO::getStatus, 1))
.stream() .stream()
.collect(Collectors.toMap(AiServiceConfigDO::getServiceCode, c -> c, (a, b) -> a)); .collect(Collectors.toMap(AiServiceConfigDO::getServiceCode, c -> c, (a, b) -> a));
// 转换为 VO // 转换为 VO
return appStatsList.stream() return appStatsList.stream()
.map(stats -> convertToAppStatsVO(stats, configMap)) .map(stats -> convertToAppStatsVO(stats, configMap))
.collect(Collectors.toList()); .collect(Collectors.toList());
});
} }
@Override @Override
public AiUsageTrendRespVO getTrend(LocalDateTime startTime, LocalDateTime endTime, String bizType, String type) { public AiUsageTrendRespVO getTrend(LocalDateTime startTime, LocalDateTime endTime, String bizType, String type) {
// 查询趋势数据 return TenantUtils.executeIgnore(() -> {
List<Map<String, Object>> trendList = pointRecordMapper.selectTrend( // 查询趋势数据
startTime, endTime, bizType, "day".equals(type) ? "%Y-%m-%d" : "%Y-%m-%d %H:00"); List<Map<String, Object>> trendList = pointRecordMapper.selectTrend(
startTime, endTime, bizType, "day".equals(type) ? "%Y-%m-%d" : "%Y-%m-%d %H:00");
// 转换为 VO // 转换为 VO
List<AiUsageTrendRespVO.TrendItem> items = trendList.stream() List<AiUsageTrendRespVO.TrendItem> items = trendList.stream()
.map(this::convertToTrendItem) .map(this::convertToTrendItem)
.collect(Collectors.toList()); .collect(Collectors.toList());
return AiUsageTrendRespVO.builder() return AiUsageTrendRespVO.builder()
.trendList(items) .trendList(items)
.build(); .build();
});
} }
/** /**