package cn.iocoder.yudao.module.im.controller.admin.manager.statistics;
|
|
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
|
import cn.iocoder.yudao.framework.common.util.collection.MapUtils;
|
import cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils;
|
import cn.iocoder.yudao.module.im.controller.admin.manager.statistics.vo.*;
|
import cn.iocoder.yudao.module.im.service.statistics.ImStatisticsManagerService;
|
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
|
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
|
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.constraints.Max;
|
import jakarta.validation.constraints.Min;
|
import org.springframework.security.access.prepost.PreAuthorize;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.time.LocalDate;
|
import java.time.LocalDateTime;
|
import java.util.*;
|
|
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
|
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
|
|
@Tag(name = "管理后台 - IM 数据看板")
|
@RestController
|
@RequestMapping("/im/manager/statistics")
|
@Validated
|
public class ImStatisticsManagerController {
|
|
/**
|
* 群规模分桶名称的展示顺序
|
*/
|
private static final List<String> GROUP_SIZE_BUCKETS = Arrays.asList("1-9 人", "10-49 人", "50-199 人", "200+ 人");
|
/**
|
* 看板分布默认时间窗口（天）
|
*/
|
private static final int DISTRIBUTION_WINDOW_DAYS = 30;
|
/**
|
* TOP 发送者数量
|
*/
|
private static final int TOP_SENDER_LIMIT = 10;
|
|
@Resource
|
private ImStatisticsManagerService statisticsService;
|
@Resource
|
private AdminUserApi adminUserApi;
|
|
@GetMapping("/overview")
|
@Operation(summary = "获得数据概览")
|
@PreAuthorize("@ss.hasPermission('im:manager:statistics:query')")
|
public CommonResult<ImStatisticsManagerOverviewRespVO> getOverview() {
|
LocalDateTime todayBegin = LocalDate.now().atStartOfDay();
|
LocalDateTime tomorrowBegin = todayBegin.plusDays(1);
|
LocalDateTime yesterdayBegin = todayBegin.minusDays(1);
|
// 周活/月活定义为 N 天滚动窗口（含今天）
|
LocalDateTime weekBegin = todayBegin.minusDays(6);
|
LocalDateTime monthBegin = todayBegin.minusDays(29);
|
return success(new ImStatisticsManagerOverviewRespVO()
|
.setTotalUser(statisticsService.getTotalUserCount())
|
.setNewUserToday(statisticsService.getNewUserCount(todayBegin, tomorrowBegin))
|
.setTotalGroup(statisticsService.getTotalGroupCount())
|
.setNewGroupToday(statisticsService.getNewGroupCount(todayBegin, tomorrowBegin))
|
.setActiveUserDaily(statisticsService.getActiveUserCount(todayBegin, tomorrowBegin))
|
.setActiveUserWeekly(statisticsService.getActiveUserCount(weekBegin, tomorrowBegin))
|
.setActiveUserMonthly(statisticsService.getActiveUserCount(monthBegin, tomorrowBegin))
|
.setPrivateMessageToday(statisticsService.getPrivateMessageCount(todayBegin, tomorrowBegin))
|
.setGroupMessageToday(statisticsService.getGroupMessageCount(todayBegin, tomorrowBegin))
|
.setPrivateMessageYesterday(statisticsService.getPrivateMessageCount(yesterdayBegin, todayBegin))
|
.setGroupMessageYesterday(statisticsService.getGroupMessageCount(yesterdayBegin, todayBegin)));
|
}
|
|
@GetMapping("/message-trend")
|
@Operation(summary = "获得消息趋势（私聊 / 群聊双线）")
|
@Parameter(name = "days", description = "回看天数（含今日）", example = "7")
|
@PreAuthorize("@ss.hasPermission('im:manager:statistics:query')")
|
public CommonResult<ImStatisticsManagerTrendRespVO> getMessageTrend(
|
@RequestParam(value = "days", defaultValue = "7") @Min(1) @Max(90) int days) {
|
List<LocalDateTime> dates = LocalDateTimeUtils.getLatestDays(days);
|
LocalDateTime beginTime = dates.get(0);
|
LocalDateTime endTime = dates.get(days - 1).plusDays(1);
|
Map<LocalDateTime, Long> privateMap = statisticsService.getPrivateMessageDailyCountMap(beginTime, endTime);
|
Map<LocalDateTime, Long> groupMap = statisticsService.getGroupMessageDailyCountMap(beginTime, endTime);
|
// 转换格式
|
Map<String, List<Long>> series = new LinkedHashMap<>();
|
series.put("private", alignSeries(dates, privateMap));
|
series.put("group", alignSeries(dates, groupMap));
|
return success(new ImStatisticsManagerTrendRespVO().setDates(dates).setSeries(series));
|
}
|
|
@GetMapping("/user-trend")
|
@Operation(summary = "获得用户趋势（新增注册 / 日活双线）")
|
@Parameter(name = "days", description = "回看天数（含今日）", example = "7")
|
@PreAuthorize("@ss.hasPermission('im:manager:statistics:query')")
|
public CommonResult<ImStatisticsManagerTrendRespVO> getUserTrend(
|
@RequestParam(value = "days", defaultValue = "7") @Min(1) @Max(90) int days) {
|
List<LocalDateTime> dates = LocalDateTimeUtils.getLatestDays(days);
|
LocalDateTime beginTime = dates.get(0);
|
LocalDateTime endTime = dates.get(days - 1).plusDays(1);
|
Map<LocalDateTime, Long> registerMap = statisticsService.getNewUserDailyCountMap(beginTime, endTime);
|
Map<LocalDateTime, Long> activeMap = statisticsService.getActiveUserDailyCountMap(beginTime, endTime);
|
// 转换格式
|
Map<String, List<Long>> series = new LinkedHashMap<>();
|
series.put("register", alignSeries(dates, registerMap));
|
series.put("active", alignSeries(dates, activeMap));
|
return success(new ImStatisticsManagerTrendRespVO().setDates(dates).setSeries(series));
|
}
|
|
@GetMapping("/message-type-distribution")
|
@Operation(summary = "获得内容类型分布（最近 30 天）")
|
@PreAuthorize("@ss.hasPermission('im:manager:statistics:query')")
|
public CommonResult<List<ImStatisticsManagerMessageTypeRespVO>> getMessageTypeDistribution() {
|
LocalDateTime endTime = LocalDate.now().plusDays(1).atStartOfDay();
|
LocalDateTime beginTime = endTime.minusDays(DISTRIBUTION_WINDOW_DAYS);
|
Map<Integer, Long> typeCountMap = statisticsService.getMessageTypeCountMap(beginTime, endTime);
|
// 转换格式
|
return success(convertList(typeCountMap.entrySet(), entry -> new ImStatisticsManagerMessageTypeRespVO()
|
.setType(entry.getKey()).setValue(entry.getValue())));
|
}
|
|
@GetMapping("/group-size-distribution")
|
@Operation(summary = "获得群规模分布")
|
@PreAuthorize("@ss.hasPermission('im:manager:statistics:query')")
|
public CommonResult<List<ImStatisticsManagerGroupSizeRespVO>> getGroupSizeDistribution() {
|
Map<String, Long> groupSizeMap = statisticsService.getGroupSizeCountMap();
|
// 转换格式
|
return success(convertList(GROUP_SIZE_BUCKETS, bucket -> new ImStatisticsManagerGroupSizeRespVO()
|
.setRange(bucket).setCount(groupSizeMap.getOrDefault(bucket, 0L))));
|
}
|
|
@GetMapping("/top-senders")
|
@Operation(summary = "获得消息 TOP 发送者（最近 30 天）")
|
@PreAuthorize("@ss.hasPermission('im:manager:statistics:query')")
|
public CommonResult<List<ImStatisticsManagerTopSenderRespVO>> getTopSenders() {
|
LocalDateTime endTime = LocalDate.now().plusDays(1).atStartOfDay();
|
LocalDateTime beginTime = endTime.minusDays(DISTRIBUTION_WINDOW_DAYS);
|
Map<Long, Long> topSenderMap = statisticsService.getTopSenderCountMap(beginTime, endTime, TOP_SENDER_LIMIT);
|
// TOP 发送者：批量回填昵称
|
Map<Long, AdminUserRespDTO> userMap = adminUserApi.getUserMap(topSenderMap.keySet());
|
return success(convertList(topSenderMap.entrySet(), entry -> {
|
ImStatisticsManagerTopSenderRespVO item = new ImStatisticsManagerTopSenderRespVO()
|
.setUserId(entry.getKey()).setMessageCount(entry.getValue());
|
MapUtils.findAndThen(userMap, entry.getKey(), user -> item.setNickname(user.getNickname()));
|
return item;
|
}));
|
}
|
|
/**
|
* 把每日聚合 Map 对齐到 dates 序列；缺失天补 0
|
*/
|
private static List<Long> alignSeries(List<LocalDateTime> dates, Map<LocalDateTime, Long> dailyMap) {
|
return convertList(dates, date -> dailyMap.getOrDefault(date, 0L));
|
}
|
|
}
|
