38 Commits
Author SHA1 Message Date
Zhou Weidong aa5954ef6a chore(merge): 合并 main 的印尼语敏感词库种子到 develop 2026-10-29 09:21:19 +08:00
Zhou Weidong 963bf5c0da test(playback): 补充合并预写缓冲与迟到条目丢弃用例 2026-10-28 19:02:33 +08:00
Zhou Weidong a71d2701d1 fix(playback): 登录合并前预写设备缓冲,迟到条目不得复活已删记录
- 新增 flushIdentity:把某身份缓冲里未落库的进度立即写进权威表;
  mergeDeviceIntoUser 先调用它再扫库,否则登录前 2 秒内的上报会被漏掉,
  稍后落库还会生成一条登录用户看不到的匿名孤儿行;合并完清掉设备侧缓冲。
- planOne 增加判定:逻辑删除会刷新 updated_at,上报时间不晚于它的条目直接丢弃,
  防止删除瞬间已 drain 的在途引用把 deleted 写回 0。
- 去掉 save(Long userId, ...) 上多余的 @Transactional(缓冲路径白开事务)。
2026-10-28 19:02:32 +08:00
Zhou Weidong 025fbe73f0 perf(playback): 上报改用 Lua 单次往返写入缓冲
原先 HSET、EXPIRE、LPUSH、LTRIM 分四次往返,进程若死在写状态与入队之间,
该剧进度只留在 HASH 里永远不落库、12 小时后被 TTL 清掉。合并成一个脚本后
热路径只需 1 次往返,脚本顺带返回 HLEN 是否超 30 决定要不要补裁剪。
2026-10-28 19:02:32 +08:00
Zhou Weidong 0d161f318f chore(merge): 合并 origin/develop 的看门狗周期巡检与 admin-api Redis 配置
watchdog.ps1 冲突取本地实现:计数落盘用 Import/Export 函数,另含 Webhook
三级兜底与告警节流,为远程内联版本的超集。
2026-10-28 18:53:11 +08:00
Zhou Weidong 2eac1fe31d test(playback): 补充缓冲删除双写、已删剧不读回与失效条目丢弃用例 2026-10-28 18:51:52 +08:00
Zhou Weidong 09b0f89e8d fix(deploy): 看门狗失败计数落盘并补齐告警 Webhook 兜底
计划任务每 2 分钟起新进程,进程内计数永远到不了阈值,告警静默哑火;计数改
JSON 落盘跨进程累加,达阈值后只在首达与每 30 轮提醒。Webhook 支持参数、环境
变量、deploy/feishu-webhook.txt 三级兜底,密钥文件不入库。
2026-10-28 18:51:52 +08:00
Zhou Weidong 8fb1f52777 fix(config): 调度线程池扩到 4 条,避免落库与其它定时任务互相排队 2026-10-28 18:51:52 +08:00
Zhou Weidong 44b7abcd0f fix(playback): 删除观看历史时同步清理进度缓冲
缓冲已是读接口的权威来源,只删库行会被读覆盖层读回、并被下一次落库把行复活。
缓冲补 remove/clear(HDEL/DEL),删除路径先清缓冲再删库行,读覆盖层按库行
deleted 过滤;落库时缓冲状态已失效的条目直接丢弃,不再整批退回队列。
2026-10-28 18:51:52 +08:00
Zhou Weidong 181b338780 fix(ops): 修复管理端 Redis 未配置与看门狗未真正周期巡检
- admin-api 的 application.yml 缺少 spring.data.redis 配置,容器内默认连
  127.0.0.1 导致 /actuator/health 返回 DOWN(redis 组件不可用);
  补齐后与 app-api 共用 REDIS_HOST/REDIS_PORT,健康检查恢复 UP
- 计划任务原先用 BootTrigger 挂 Repetition,Windows 不生效(Next Run Time
  为 N/A,9 小时只跑过一次),看门狗形同虚设;改为 TimeTrigger + PT2M
  无限重复 + StartWhenAvailable,实测每 2 分钟稳定巡检
- watchdog.ps1 的连续失败计数改为落盘 watchdog-state.json,避免每轮进程
  重启后清零导致「连续 3 次失败告警」永不触发
2026-10-28 18:23:48 +08:00
Zhou Weidong 657190d5a8 chore(deploy): 加入服务看门狗脚本与 Windows 计划任务注册文件
- watchdog.ps1:每 2 分钟健康检查,Docker/DB/Redis/API/Grafana 任一
  不健康自动重启,连续 3 次失败发飞书告警,C 盘超 85% 也告警
- backend-watchdog-task.xml:开机触发、2 分钟循环的计划任务配置
2026-10-28 15:00:58 +08:00
Zhou Weidong 3e2d6a06d9 perf(playback): 观看进度上报异步化(Redis write-behind 批量落库)
- POST /app/watch/progress 只写 Redis 缓冲立即返回(p95 75ms→2ms),
  WatchProgressFlusher 每 2s 把 watch:pg:dirty 队列按身份合并批量 UPSERT,
  Redis 异常自动回退同步写库,接口契约不变
- 读接口用缓冲覆盖 DB 行保证读己之写;Caffeine 缓存剧集校验免去上报路径两条 DB 查询
- 落库按「新增进度/新增流水/更新行」分组攒批(MyBatis BATCH 只合并连续同形状语句),
  配合 reWriteBatchedInserts 多值 INSERT,12.4 万积压 45 秒消化完,
  1650 QPS 持续压测队列稳态归零
2026-10-28 14:59:09 +08:00
Zhou Weidong fc9e429187 feat(monitoring): 接入 Loki + Promtail 日志采集,Grafana 可直接搜索容器日志 2026-10-28 10:38:41 +08:00
Zhou Weidong 5d139783c9 fix(media): 缩略图唯一索引改为部分索引,避免软删行占用 episode_id 导致登记冲突 2026-10-27 21:07:10 +08:00
Zhou Weidong ddd2406f32 fix(build): Docker 构建阶段固定阿里云 Maven 中央镜像
容器内从 Maven Central 拉依赖时 maven-filtering:3.3.1 解析失败,
且 dependency:go-offline 被 || true 静默吞掉,导致发布时才暴露。
构建阶段写入 settings.xml 走阿里云 central 镜像,并让 go-offline 失败即中断。
2026-10-27 20:14:19 +08:00
Vicky 272aa62434 feat: phase1 UI prep — P1 安全(XFF防护), P2 规模化(平台索引+region保护), P3 文档补全
P1 安全: WebUtils.getClientIp 改为 remoteAddr 优先 + 可信代理 CIDR 配置
P2 规模化: DISTINCT ON 查询补匹配排序的部分索引 (migrate-k113)
P2 一致性: syncCountryByIp regionCode 同语言/时区/币种加保护,不覆盖用户自定义
P3 文档: api-phase1.md 新增 platform 参数说明, IpRegionService 新增集成测试 + VERSION
新增: CountryProfile 52 国映射 + IpRegionService 离线 ip2region 解析
平台筛选: app-users/favorites/likes/watch-history 按最近一次会话 ANDROID/IOS 筛选
2026-10-27 15:45:15 +08:00
jianghaifeng a41ab18636 perf(admin): 优化观看记录列表并统一时间格式
- 新增 watched_at 索引,默认倒序翻页走 Merge Append 而不是全表排序
- 装配只取展示所需列,会话查询不再拉 refresh_token_hash
- 关键词预检索加 2000 条上限
- admin-api 的 LocalDateTime 统一输出 yyyy-MM-dd HH:mm:ss
2026-10-27 14:39:13 +08:00
jianghaifeng 8999ff395d feat(admin): 收藏/点赞/观看记录支持按客户端平台筛选
- 设备画像按 device_id 命中,登录用户按最近一次登录会话的平台命中
- 修正 orderBy 先于动态条件导致 SQL 拼接错误的问题
2026-10-27 14:27:10 +08:00
jianghaifeng 229f1d0198 refactor(device): 设备平台只记 iOS/Android
去掉系统版本、机型与版本号上报,表结构收敛为 device_id + platform。
2026-10-27 14:00:12 +08:00
jianghaifeng 2dadca6288 chore(merge): 合并设备画像与后台系统展示 2026-10-27 13:53:49 +08:00
Vicky 2cf8f9a1c3 feat(device): 按 device_id 记录客户端系统与机型
- 新增 app_device_profile 表与 upsert,匿名请求同样建档
- app-api 设备画像过滤器,同设备 60 秒节流写一次
- 收藏/点赞/观看列表优先用设备画像,缺失回退最近会话
2026-10-27 13:53:47 +08:00
jianghaifeng d9a1670a72 chore(merge): 合并互动计数校准 2026-10-27 13:29:09 +08:00
Vicky 5ff4626b33 chore(sql): 校准被压测数据污染的互动计数
剧集点赞数与剧收藏数改回按实际未删除互动行重算,可重复执行。
2026-10-27 13:29:09 +08:00
jianghaifeng 6bbde0b100 chore(merge): 合并剧级点赞历史数据清理 2026-10-27 13:28:15 +08:00
Vicky 884324b8fd chore(sql): 清理历史遗留的剧级点赞数据
客户端点赞入口只作用于剧集,13 条 DRAMA 记录为旧模型遗留;
软删并同步校准 drama.like_count。
2026-10-27 13:28:15 +08:00
jianghaifeng 7baaeddf26 chore(merge): 合并点赞筛选说明文档 2026-10-27 13:25:52 +08:00
Vicky 99d49a1765 docs(admin): 说明点赞列表三种目标与 episodeId 筛选边界
- 明确客户端点赞落 EPISODE,DRAMA 为历史数据
- 说明 episodeId 只命中该集及该集评论,不含对短剧本身的点赞
2026-10-27 13:25:51 +08:00
jianghaifeng 92f7606f65 chore(merge): 合并收藏与点赞列表查询加固 2026-10-27 13:23:23 +08:00
Vicky c2ba3df47c fix(admin): 加固收藏与点赞列表查询
- 关键词/剧名预检索加 2000 条上限,避免 IN 列表超限并给出可操作提示
- 新增 created_at 分页索引迁移,避免无条件列表退化为全表排序
- dramaPublished 改为三态,短剧已不存在时不再误报未发布
2026-10-27 13:23:21 +08:00
Zhou Weidong 226dc275bb merge(admin): 合并收藏与点赞只读查询 2026-10-27 12:32:23 +08:00
Zhou Weidong 7a2ff3f80e feat(admin): 新增收藏与点赞只读查询接口 2026-10-27 12:32:23 +08:00
Zhou Weidong 23321a0b0c feat(favorites): 支持设备批量取消收藏 2026-10-27 10:33:59 +08:00
Zhou Weidong 5d44ac239b fix(watch): 修复匿名进度查询参数类型 2026-10-27 10:01:21 +08:00
Zhou Weidong 848c746505 feat(watch): 支持设备观看与登录合并 2026-10-27 09:54:47 +08:00
Zhou Weidong 4b0fbd73d3 feat(interaction): 支持游客互动与登录合并 2026-10-26 22:34:14 +08:00
Zhou Weidong 2fe24b9db7 feat(interaction): 新增分集计数统计接口 2026-10-26 21:31:17 +08:00
Zhou Weidong 6abb09e5c2 chore(merge): 将 release/v1.0.0 同步到 main 2026-10-23 13:07:42 +08:00
Zhou Weidong 8b37e23951 feat: 印尼语敏感词库种子数据(id-ID 增量补全 78 词) 2026-10-19 18:43:01 +08:00
118 changed files with 5970 additions and 276 deletions

No files matched your search

+4
View File
@@ -19,3 +19,7 @@ data/uploads/*
# 部署用的真实环境变量(含密钥),只提交 .env.example
.env
dump.rdb
# 看门狗飞书告警 Webhook,只存在于部署机
deploy/feishu-webhook.txt
+15 -1
View File
@@ -15,6 +15,20 @@
FROM maven:3.9-eclipse-temurin-21 AS build
WORKDIR /build
# Maven Central 在国内网络下拉依赖不稳定(构建时曾出现 maven-filtering:3.3.1 解析失败),
# 构建阶段固定走阿里云 central 镜像;其它仓库仍直连。
RUN mkdir -p /root/.m2 \
&& printf '%s\n' \
'<settings xmlns="http://maven.apache.org/SETTINGS/1.0.0">' \
' <mirrors>' \
' <mirror>' \
' <id>aliyun-central</id>' \
' <mirrorOf>central</mirrorOf>' \
' <url>https://maven.aliyun.com/repository/central</url>' \
' </mirror>' \
' </mirrors>' \
'</settings>' > /root/.m2/settings.xml
# 先只拷各模块 pom 再拉依赖:依赖层可复用,改业务代码不会重新下载依赖
COPY pom.xml .
COPY fruitshorts-common/pom.xml fruitshorts-common/
@@ -30,7 +44,7 @@ COPY fruitshorts-module-creator/pom.xml fruitshorts-module-creator/
COPY fruitshorts-admin-api/pom.xml fruitshorts-admin-api/
COPY fruitshorts-app-api/pom.xml fruitshorts-app-api/
COPY fruitshorts-transcode-worker/pom.xml fruitshorts-transcode-worker/
RUN mvn -B -q -DskipTests dependency:go-offline || true
RUN mvn -B -q -DskipTests dependency:go-offline
COPY . .
ARG MODULE=fruitshorts-app-api
+38
View File
@@ -0,0 +1,38 @@
<?xml version="1.0" encoding="UTF-16"?>
<Task version="1.3" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">
<RegistrationInfo>
<Description>FruitShorts backend watchdog: health check every 2 minutes and restart unhealthy services.</Description>
</RegistrationInfo>
<Principals>
<Principal id="Author">
<UserId>S-1-5-18</UserId>
<RunLevel>HighestAvailable</RunLevel>
</Principal>
</Principals>
<Settings>
<MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>
<DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>
<StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>
<StartWhenAvailable>true</StartWhenAvailable>
<ExecutionTimeLimit>PT10M</ExecutionTimeLimit>
</Settings>
<Triggers>
<!-- BootTrigger 上挂 <Repetition> 在 Windows 计划任务里不生效(导入后 Next Run Time 为 N/A,
实际只在开机跑一次)。改成 TimeTrigger + 无限重复,配合 StartWhenAvailable
保证开机后若错过也会立即补跑;这样才真正做到每 2 分钟巡检一次。 -->
<TimeTrigger>
<StartBoundary>2026-01-01T00:00:00</StartBoundary>
<Repetition>
<Interval>PT2M</Interval>
<Duration>P3650D</Duration>
<StopAtDurationEnd>false</StopAtDurationEnd>
</Repetition>
</TimeTrigger>
</Triggers>
<Actions Context="Author">
<Exec>
<Command>powershell.exe</Command>
<Arguments>-NoProfile -ExecutionPolicy Bypass -File "C:\Users\Administrator\Desktop\ReelShorts\FruitServer\deploy\watchdog.ps1" -ComposeDir "C:\Users\Administrator\Desktop\ReelShorts\FruitServer" -LogDir "C:\Users\Administrator\Desktop\ReelShorts\FruitServer\logs"</Arguments>
</Exec>
</Actions>
</Task>
+215
View File
@@ -0,0 +1,215 @@
# Author: fruit.com.cn
#
# 服务看门狗:每 2 分钟检查一次,挂了自动拉起,连续失败发飞书告警。
# 注册为 Windows 计划任务后无人值守运行。
param(
[string]$ComposeDir = "C:\FruitServer",
[string]$LogDir = "C:\FruitServer\logs",
[string]$FeishuWebhook = "",
[int]$MaxRestarts = 3,
[string]$FrontendTask = "FruitShorts-Web-3005"
)
$ErrorActionPreference = "Continue"
$ComposeFile = Join-Path $ComposeDir "docker-compose.yml"
$LogFile = Join-Path $LogDir "watchdog-$(Get-Date -Format 'yyyy-MM-dd').log"
# 连续失败计数落盘文件:计划任务每 2 分钟起一个新进程,计数只放进程内永远累积不到
# $MaxRestarts,告警会静默哑火。
$StateFile = Join-Path $LogDir "watchdog-state.json"
function Write-Log($msg) {
$line = "$(Get-Date -Format 'yyyy-MM-dd HH:mm:ss') $msg"
Add-Content -Path $LogFile -Value $line -ErrorAction SilentlyContinue
}
function Send-Alert($title, $msg) {
if (-not $FeishuWebhook) { return }
$body = @{
msg_type = "text"
content = @{ text = "[$title] $msg" }
} | ConvertTo-Json -Depth 3
try {
Invoke-RestMethod -Uri $FeishuWebhook -Method Post -Body $body -ContentType "application/json" -TimeoutSec 10
} catch {
Write-Log "飞书告警发送失败: $_"
}
}
# 连续失败计数的读写:JSON 落盘,跨进程保留
function Import-FailCounts {
$counts = @{}
if (-not (Test-Path $StateFile)) { return $counts }
try {
$saved = Get-Content -Path $StateFile -Raw | ConvertFrom-Json
foreach ($prop in $saved.PSObject.Properties) {
$counts[$prop.Name] = [int]$prop.Value
}
} catch {
Write-Log "看门狗计数文件解析失败,各服务按 0 重新计数: $_"
}
return $counts
}
function Export-FailCounts($counts) {
try {
$counts | ConvertTo-Json | Set-Content -Path $StateFile -Encoding UTF8
} catch {
Write-Log "看门狗计数文件写入失败: $_"
}
}
function Test-Docker {
docker info --format "{{.ServerVersion}}" 2>$null | Out-Null
return ($LASTEXITCODE -eq 0)
}
function Test-Http([string]$url) {
try {
$response = Invoke-WebRequest -Uri $url -TimeoutSec 10 -UseBasicParsing -ErrorAction Stop
return ($response.StatusCode -ge 200 -and $response.StatusCode -lt 400)
} catch {
return $false
}
}
function Test-Container([string]$container) {
$status = docker inspect -f "{{.State.Status}}" $container 2>$null
return ($status -eq "running")
}
function Invoke-Compose([string[]]$composeArgs) {
& docker compose -f $ComposeFile @composeArgs 2>&1 | Out-Null
return $LASTEXITCODE
}
# 确保日志目录存在
New-Item -ItemType Directory -Force -Path $LogDir | Out-Null
# 告警 Webhook:命令行参数 > 环境变量 > deploy 目录下的 feishu-webhook.txt。
# 计划任务注册命令里不写密钥(密钥不该进版本库),由部署机本地放文件或配环境变量。
if (-not $FeishuWebhook) { $FeishuWebhook = $env:FRUITSHORTS_FEISHU_WEBHOOK }
if (-not $FeishuWebhook) {
$webhookFile = Join-Path $PSScriptRoot "feishu-webhook.txt"
if (Test-Path $webhookFile) {
$FeishuWebhook = (Get-Content -Path $webhookFile -Raw -ErrorAction SilentlyContinue)
if ($FeishuWebhook) { $FeishuWebhook = $FeishuWebhook.Trim() }
}
}
if (-not $FeishuWebhook) {
Write-Log "未配置飞书 Webhook(参数/环境变量/feishu-webhook.txt 均为空),告警只写本地日志"
}
# Docker Desktop 登录后偶尔尚未就绪,先等待并尝试拉起。
if (-not (Test-Docker)) {
Write-Log "Docker 未就绪,尝试启动 Docker Desktop"
$dockerExe = "C:\Program Files\Docker\Docker\Docker Desktop.exe"
if (Test-Path $dockerExe) {
Start-Process -FilePath $dockerExe
}
for ($i = 0; $i -lt 18; $i++) {
Start-Sleep -Seconds 5
if (Test-Docker) { break }
}
}
if (-not (Test-Docker)) {
Write-Log "Docker 在 90 秒内未就绪,跳过本轮检查"
Send-Alert "服务异常" "Docker Desktop 无法启动,请人工检查桌面机。"
exit 1
}
# 健康检查定义:名称 → 检查方式
$checks = @(
@{ Name = "app-api"; Type = "http"; Url = "http://127.0.0.1:8081/healthz"; Service = "app-api"; Container = "fruitshorts-app-api" },
@{ Name = "admin-api"; Type = "http"; Url = "http://127.0.0.1:8080/healthz"; Service = "admin-api"; Container = "fruitshorts-admin-api" },
@{ Name = "frontend"; Type = "frontend"; Url = "http://127.0.0.1:3005/dist-dev/" },
@{ Name = "Grafana"; Type = "http"; Url = "http://127.0.0.1:3000/api/health"; Service = "grafana"; Container = "fruitshorts-grafana" },
@{ Name = "PostgreSQL"; Type = "docker"; Service = "db"; Container = "fruitshorts-db" },
@{ Name = "Redis"; Type = "docker"; Service = "redis"; Container = "fruitshorts-redis" },
@{ Name = "Prometheus"; Type = "http"; Url = "http://127.0.0.1:9090/-/healthy"; Service = "prometheus"; Container = "fruitshorts-prometheus" }
)
# 跟踪每个服务的连续失败次数(跨进程从状态文件恢复)
$failCounts = Import-FailCounts
foreach ($check in $checks) {
$name = $check.Name
$healthy = $false
if ($check.Type -eq "http") {
$healthy = Test-Http $check.Url
} elseif ($check.Type -eq "frontend") {
$healthy = Test-Http $check.Url
} else {
$healthy = Test-Container $check.Container
}
if ($healthy) {
if ($failCounts[$name] -gt 0) {
Write-Log "$name 恢复正常(此前连续失败 $($failCounts[$name]) 次)"
Send-Alert "服务恢复" "$name 已恢复正常"
}
$failCounts[$name] = 0
} else {
if ($null -eq $failCounts[$name]) {
$failCounts[$name] = 0
}
$failCounts[$name] = $failCounts[$name] + 1
$count = $failCounts[$name]
Write-Log "$name 不健康(第 $count 次)"
if ($check.Type -eq "frontend") {
Write-Log "$name 重启计划任务 $FrontendTask"
Stop-ScheduledTask -TaskName $FrontendTask -ErrorAction SilentlyContinue
Start-Sleep -Seconds 2
Start-ScheduledTask -TaskName $FrontendTask -ErrorAction SilentlyContinue
Start-Sleep -Seconds 6
$healthy = Test-Http $check.Url
} elseif ($check.Type -eq "http") {
Write-Log "$name 重启 Compose 服务 $($check.Service)"
Invoke-Compose @("restart", $check.Service) | Out-Null
Start-Sleep -Seconds 12
$healthy = Test-Http $check.Url
if (-not $healthy) {
Write-Log "$name 重启后仍不健康,执行 compose up -d"
Invoke-Compose @("up", "-d", $check.Service) | Out-Null
}
} else {
Write-Log "$name 拉起容器 $($check.Container)"
docker start $check.Container 2>&1 | Out-Null
Start-Sleep -Seconds 5
$healthy = Test-Container $check.Container
if (-not $healthy) {
Write-Log "$name 启动失败,执行 compose up -d"
Invoke-Compose @("up", "-d", $check.Service) | Out-Null
}
}
# 连续失败达到上限 → 告警。计数会跨进程累积,所以只在首次达阈值和
# 之后每 30 轮(约 1 小时)提醒一次,避免每 2 分钟刷屏。
if (-not $healthy -and $count -ge $MaxRestarts -and
($count -eq $MaxRestarts -or $count % 30 -eq 0)) {
Send-Alert "服务异常" "$name 已连续 $count 次不健康,已尝试重启。请人工介入检查。"
Write-Log "$name 连续 $count 次失败,已发告警"
} elseif ($healthy) {
Write-Log "$name 已自动恢复"
}
}
}
# 磁盘检查:超过 85% 告警
$drive = (Get-PSDrive C -ErrorAction SilentlyContinue)
if ($drive) {
$pct = [math]::Round(($drive.Used / ($drive.Used + $drive.Free)) * 100)
if ($pct -gt 85) {
Write-Log "磁盘使用率 ${pct}%(超过 85%)"
Send-Alert "磁盘告警" "C 盘使用率 ${pct}%,请清理日志或扩容"
}
}
# 本轮结果落盘,供下一轮(2 分钟后的新进程)继续累加
Export-FailCounts $failCounts
+36 -2
View File
@@ -71,8 +71,10 @@ services:
env_file: .env
environment:
SERVER_PORT: "8081"
# 默认连编排里的 db;要连外部库就在 .env 里覆盖 DB_URL
DB_URL: ${DB_URL:-jdbc:postgresql://db:5432/${DB_NAME:-fruitshorts}}
# 默认连编排里的 db;要连外部库就在 .env 里覆盖 DB_URL。
# reWriteBatchedInserts=true:pgjdbc 把 JDBC 批内多条 INSERT 重写成多值
# INSERT,观看进度缓冲落库(watch:pg:dirty 消费任务)依赖它跑满吞吐。
DB_URL: ${DB_URL:-jdbc:postgresql://db:5432/${DB_NAME:-fruitshorts}?reWriteBatchedInserts=true}
# 限流/缓存用 Redis;容器内 127.0.0.1 是容器自身,必须指向编排里的 redis
REDIS_HOST: ${REDIS_HOST:-redis}
REDIS_PORT: ${REDIS_PORT:-6379}
@@ -197,9 +199,40 @@ services:
GF_SECURITY_ADMIN_USER: ${GRAFANA_ADMIN_USER:-admin}
GF_SECURITY_ADMIN_PASSWORD: ${GRAFANA_ADMIN_PASSWORD:?Grafana 管理员密码必填,见 .env.example}
GF_USERS_ALLOW_SIGN_UP: "false"
# 新用户默认界面语言:简体中文
GF_USERS_LANGUAGE: "zh-Hans"
TZ: Asia/Shanghai
depends_on:
- prometheus
- loki
# ─────────── 日志(Loki + Promtail)───────────
# Loki:日志存储与查询引擎;Promtail:采集容器 stdout/stderr 推给 Loki。
# Grafana 里通过 Loki 数据源可直接搜索/过滤各容器的请求日志。
loki:
image: grafana/loki:latest
container_name: fruitshorts-loki
restart: unless-stopped
command: -config.file=/etc/loki/local-config.yaml
volumes:
- ./loki-config.yml:/etc/loki/local-config.yaml:ro
- loki-data:/loki
expose:
- "3100"
# 镜像为 distroless(无 shell/wget),无法在容器内做 HTTP 探活;
# 靠 restart: unless-stopped 自动拉起,Grafana 数据源状态可反映可用性。
promtail:
image: grafana/promtail:latest
container_name: fruitshorts-promtail
restart: unless-stopped
command: -config.file=/etc/promtail/config.yml
volumes:
- ./promtail-config.yml:/etc/promtail/config.yml:ro
# 读取容器日志需要 Docker socket
- /var/run/docker.sock:/var/run/docker.sock:ro
depends_on:
- loki
# 黑盒探活:从外部探测接口可用性与耗时
blackbox:
@@ -227,3 +260,4 @@ volumes:
redis-data:
prometheus-data:
grafana-data:
loki-data:
+120 -21
View File
@@ -115,6 +115,8 @@ POST /app/auth/register
| 密码强度 | 8–64 位,且**同时包含字母与数字**(不强制特殊字符:低端机输入成本高,拦机器注册靠频控) |
| 注册频控 | 同 IP / 同设备 24h 默认上限分别为 5 / 3 个,超出返回 `429`(`TOO_MANY_REQUESTS`) |
| 邮箱验证状态 | 免码注册时 `app_user_credential.verified = 0`(保留"待验证"),**不阻塞登录**;只有**验证码登录**会把 `verified` 置 1,密码登录不改变该状态 |
| 国家 / 语言 / 时区 / 币种 | 优先按客户端 IP(`X-Forwarded-For` / `X-Real-IP`,离线 ip2region)解析国家;内网或解析失败才用请求体 `countryCode`(ISO-2),最后兜底 `ID`。`language` / `timezone` / `currency` 随国家映射,不再写死 `id-ID` / `IDR` |
| 登录时修正国家 | `/app/auth/login`、`/app/auth/google` 会用 IP 重新解析国家并更新 `country_code` / `region_code`(历史账号下次登录即自愈);`language` / `timezone` / `currency` 只在仍是旧国家映射值时才同步,用户自己改过就保留 |
频控口径落库在 `app_user.register_ip` / `app_user.register_device_id` 上(按 24h 计数,不新建计数表)。
@@ -286,7 +288,8 @@ app.register.max-per-device-per-day
| 方法 | 路径 | 鉴权 | 说明 |
| --- | --- | --- | --- |
| GET | `/app/episodes/{id}/play` | 可选 | 签发播放票据与签名地址 |
| GET | `/app/episodes/{id}/play` | 免费集可选;锁定集必须登录 | 签发播放票据与签名地址 |
| GET | `/app/episodes/{id}/stats` | 可选 | 播放页轻量计数快照,不签发地址、不写票据缓存 |
```json
{
@@ -310,8 +313,10 @@ app.register.max-per-device-per-day
}
```
`accessType` 取值:`FREE`(免费集)、`ENTITLED`(已购权益)、`COIN_REQUIRED`(需解锁,
此时 `playUrl` 为空、`unlockCoin` 有值)、`NO_MEDIA`(还没上传片源或未转码完成)。
`accessType` 取值:`FREE`(免费集)、`ENTITLED`(已购权益)、`COIN_REQUIRED`(已登录但
尚未解锁,此时 `playUrl` 为空、`unlockCoin` 有值)、`NO_MEDIA`(已登录用户访问的剧集还
没上传片源或未转码完成)。未登录请求锁定集在查询片源前直接返回 `UNAUTHORIZED`,不会
按设备身份取票或降级为匿名播放。
**判定顺序由服务端决定,客户端不得自行推断可否播放。** 签名默认 15 分钟有效;
开启 `fruitshorts.media.signature-required` 后,`/files/episode/**` 没带有效签名会返回 403。
@@ -320,23 +325,86 @@ app.register.max-per-device-per-day
`sprite_XXX.jpg#xywh=x,y,w,h`。有 READY `media_thumbnail` 时返回签名地址,
无产物时为 `null`;该字段不影响 `playUrl`、`qualities` 和 HLS/原文件回落。
**`GET /app/episodes/{id}/stats`(播放页计数刷新)**
只读返回本集点赞数、本集未删除评论数、所属剧收藏数,以及当前用户对
本集/所属剧的互动状态。已登录时按账号读取;匿名请求时按 `X-Device-Id`
读取设备互动状态,未带设备号时 `liked`、`favorited` 均为 `false`。
接口不生成播放签名地址、不写 `playback_ticket` 缓存,计数变化后立即可读。
```json
{
"code": 0,
"message": "ok",
"data": {
"episodeId": "12",
"dramaId": "5",
"likeCount": "45",
"commentCount": "8",
"favoriteCount": "1800",
"liked": true,
"favorited": false
}
}
```
`likeCount`、`commentCount`、`favoriteCount` 与其他计数一样按 Long 字符串下发。
### 1.5 观看记录与收藏(已实现)
| 方法 | 路径 | 说明 |
| --- | --- | --- |
| POST | `/app/watch/progress` | 上报进度(建议每 10s 一次,切集时补一次) |
| GET | `/app/watch/progress?dramaId=` | 某剧续看位置 |
| GET | `/app/watch/recent` | 最近观看分页;`pageIndex` 默认 1,`pageSize` 默认 20、上限 100 |
| DELETE | `/app/watch/progress/{dramaId}` | 删除某剧记录 |
| DELETE | `/app/watch/progress` | 清空记录 |
| POST | `/app/favorites/toggle` | 收藏/取消收藏,返回切换后状态 |
| GET | `/app/favorites` | 收藏分页;`pageIndex` 默认 1,`pageSize` 默认 20、上限 100 |
| DELETE | `/app/favorites/{dramaId}` | 取消收藏 |
| POST | `/app/favorites/remove` | 批量取消收藏,请求体 `{ "dramaIds": ["12", "15"] }` |
| 方法 | 路径 | 鉴权 | 说明 |
| --- | --- | --- | --- |
| POST | `/app/watch/progress` | 可选登录 | 上报进度;匿名按 `X-Device-Id`,建议每 10s 一次,切集时补一次 |
| GET | `/app/watch/progress?dramaId=` | 可选登录 | 某剧续看位置,匿名按设备读取 |
| GET | `/app/watch/recent` | 可选登录 | 最近观看分页;`pageIndex` 默认 1,`pageSize` 默认 20、上限 100 |
| DELETE | `/app/watch/progress/{dramaId}` | 可选登录 | 删除当前账号或设备的某剧记录 |
| DELETE | `/app/watch/progress` | 可选登录 | 清空当前账号或设备的记录 |
| POST | `/app/favorites/toggle` | 可选登录 | 收藏/取消收藏,返回切换后状态;匿名按 `X-Device-Id` |
| GET | `/app/favorites` | 可选登录 | 收藏分页;匿名按设备读取,`pageIndex` 默认 1,`pageSize` 默认 20、上限 100 |
| DELETE | `/app/favorites/{dramaId}` | 可选登录 | 取消收藏;匿名按 `X-Device-Id` |
| POST | `/app/favorites/remove` | 可选登录 | 批量取消收藏;匿名按 `X-Device-Id`,返回实际删除数量 |
进度合并规则:同一集内只前进不后退;换集直接采纳新集;`finished` 只在接近片尾或客户端显式上报时置位。
这样用户换设备不会看到进度倒退。
收藏与观看数据的身份解析顺序统一为登录 `userId` → `X-Device-Id` → 请求参数或请求体
`deviceId`,全部缺失时返回业务码 `200007`。匿名读取不按设备限流;上报、删除和清空等写操作
沿用设备级限流。登录或注册成功后,同一事务会把设备收藏、观看进度和观看流水合并到账号:
同一部剧的进度先比较集号,同集再比较播放位置,最后用更新时间兜底,只保留更靠前的一条;
观看流水按剧集去重并保留更靠后的位置。`playCount` 只取最终保留记录,不重复累加。
#### 批量取消收藏
`POST /app/favorites/remove`
```json
{
"dramaIds": ["12", "15"],
"deviceId": "optional-device-id"
}
```
| 字段 | 类型 | 必填 | 说明 |
| --- | --- | --- | --- |
| `dramaIds` | array&lt;Long&gt; | 是 | 待取消的短剧 ID,最多 200 条,服务端去重后删除 |
| `deviceId` | string | 否 | 未登录且未携带 `X-Device-Id` 时的设备号兜底 |
身份解析优先登录用户,其次 `X-Device-Id`,最后请求体 `deviceId`。登录用户按
`user_id` 删除,匿名用户按 `device_id` 删除;匿名写操作沿用设备级限流。
所有删除和剧级收藏数回退在单一事务内完成,其中任一步失败会整体回滚。
响应 `data` 为实际删除的收藏关系数:
```json
{
"code": 0,
"message": "ok",
"data": 2
}
```
空列表返回业务码 `400`;超过 200 条同样返回 `400`。
#### 收藏分页
`GET /app/favorites?pageIndex=1&pageSize=20`
@@ -446,13 +514,22 @@ app.register.max-per-device-per-day
| 方法 | 路径 | 鉴权 | 说明 |
| --- | --- | --- | --- |
| POST | `/app/comments` | 登录 | 发表评论/回复:过敏感词后按策略发布或转待审 |
| POST | `/app/comments` | 可选登录 | 发表评论/回复;匿名按 `X-Device-Id`,过敏感词后按策略发布或转待审 |
| GET | `/app/comments?dramaId=&cursor=&size=` | 匿名 | 游标翻页;不传 `rootId` 返回顶层评论,传了返回该楼层回复 |
| GET | `/app/comments/count?dramaId=&episodeId=` | 匿名 | 传 `episodeId` 返回该集评论数;仅传 `dramaId` 返回全剧兼容汇总 |
| DELETE | `/app/comments/{id}` | 登录 | 删除自己的评论(逻辑置为 `deleted`) |
| POST | `/app/likes/toggle` | 登录 | 点赞/取消点赞,`targetType` 支持 `DRAMA` / `EPISODE` / `COMMENT`;新客户端点赞按集 |
| DELETE | `/app/comments/{id}` | 可选登录 | 删除自己的评论(匿名按设备);逻辑置为 `deleted` |
| POST | `/app/likes/toggle` | 可选登录 | 点赞/取消点赞,`targetType` 支持 `DRAMA` / `EPISODE` / `COMMENT`;匿名按 `X-Device-Id` |
| POST | `/app/reports` | 登录 | 举报评论/用户/短剧;同一目标只保留一条待处理工单 |
**非付费互动身份与合并**:点赞、收藏、评论的身份解析顺序为登录 `userId` →
`X-Device-Id` → 请求体 `deviceId`。三者都没有时返回业务码 `200007`。匿名写入按设备号
每分钟限流;评论同时做长度、空白、短时间重复内容和频率校验。登录或注册成功后,服务端
在同一登录事务内把该设备未归属的点赞、收藏、评论、观看进度和观看流水迁移到账号;
账号已有同内容时保留账号记录并删除设备重复记录,计数同步回退,不重复累加。
匿名评论展示为「游客」,头像为空占位。解锁、金币/VIP 权益、锁定集正片票据不在本规则内,
仍必须登录。
**敏感词三档处置**:`BLOCK` 直接拒绝并返回原因、`REPLACE` 替换后放行、`REVIEW` 转人工待审。
审核策略由 `fruitshorts.interaction.comment-audit-mode` 控制:
`MACHINE_FIRST`(机审通过即发布,当前默认)或 `PRE_REVIEW`(全部先审后发,生产建议)。
@@ -788,15 +865,36 @@ GET /app/notices?pageIndex=1&pageSize=20
| 方法 | 路径 | 权限点 | 说明 |
| --- | --- | --- | --- |
| GET | `/admin/app-users/page` | `app-user:list` | 检索:keyword(uid/昵称/手机号/邮箱)、status、countryCode、registerSource |
| GET | `/admin/app-users/page` | `app-user:list` | 检索:keyword(uid/昵称/手机号/邮箱)、status、countryCode、registerSource、platform(ANDROID/IOS,按用户**最近一次**会话平台筛选) |
| GET | `/admin/app-users/{id}` | `app-user:list` | 详情 |
| GET | `/admin/app-users/{id}/ban-records` | `app-user:list` | 封禁与申诉记录 |
| GET | `/admin/app-users/{id}/watch-history` | `app-user:list` | 观看流水明细(带剧名/集号,客服核对用) |
| GET | `/admin/watch-history/page` | `watch-history:list` | 跨用户观看记录分页:keyword、userId、dramaTitle、dramaId、episodeId、source、finished、startTime、endTime |
| GET | `/admin/watch-history/page` | `watch-history:list` | 跨用户观看记录分页:keyword、userId、dramaTitle、dramaId、episodeId、source、finished、startTime、endTime、platform(ANDROID/IOS,按**最近一次会话**) |
| GET | `/admin/favorites/page` | `favorite:list` | 收藏只读分页,含游客;支持 platform 筛选(ANDROID/IOS,按**最近一次会话**) |
| GET | `/admin/likes/page` | `like:list` | 点赞只读分页,含游客及多态目标;支持 platform 筛选(ANDROID/IOS,按**最近一次会话**) |
| PUT | `/admin/app-users/{id}/status` | `app-user:update` | 改状态:normal/limited/banned |
状态处置会**立即吊销该用户全部会话**并写入封禁记录,避免"封了还能用"。
收藏和点赞分页参数:`pageIndex` 默认 1,`pageSize` 默认 10、最大 100;
`keyword` 模糊匹配 uid/昵称,`userId`、`deviceId`、`dramaId` 精确匹配,
`dramaTitle` 模糊匹配;`startTime/endTime` 按创建时间闭区间过滤,
格式 `yyyy-MM-dd HH:mm:ss`。点赞另支持 `targetType`(DRAMA/EPISODE/COMMENT)、
`targetId` 和 `episodeId` 精确过滤,短剧和剧集过滤会解析评论及剧集所属关系。
按 `createdAt DESC, id DESC` 排序,用户/短剧预检索无结果返回 total=0、list=[]。
点赞的三种目标语义:客户端点赞按钮作用于当前播放剧集,落库为 `targetType=EPISODE`;
`DRAMA` 是早期按整部剧点赞的历史数据;`COMMENT` 为评论点赞。筛选行为:
`dramaTitle` / `dramaId` 同时命中「该剧本身的点赞」与「该剧下各集的点赞」,
`episodeId` 只命中「该集本身的点赞」与该集下的评论点赞,**不包含对短剧本身的点赞**;
不传 `episodeId` 时不受此限制。
收藏行字段:`id,userId,uid,nickname,deviceId,anonymous,dramaId,dramaTitle,dramaCover,
dramaStatus,dramaPublished,createdAt`。点赞行字段:`id,userId,uid,nickname,deviceId,
anonymous,targetType,targetId,dramaId,dramaTitle,dramaCover,episodeId,episodeNo,episodeTitle,
commentSummary,createdAt`。游客的 userId 为空、anonymous 为 true;评论摘要最多 30 字。
关联资料逻辑删除后展示字段为空,不排除原始互动记录。
### 2.4 下载管理(已实现)
| 方法 | 路径 | 权限点 | 说明 |
@@ -1001,8 +1099,9 @@ GET /app/notices?pageIndex=1&pageSize=20
只接受 `bizType=EPISODE` 且状态为 `ready` / `uploaded` 的产物。
**收费配置**:`previewSeconds` 落在 `drama_episode.preview_seconds`(0 = 不限制),
与 `is_free` / `unlock_coin` 一起就是播放与下载付费判定读的字段,**改完立即生效**(实测:未登录请求付费集返回
`{"accessType":"COIN_REQUIRED","unlockCoin":50,"previewSeconds":15}`;免费集直接返回签名播放地址)。
与 `is_free` / `unlock_coin` 一起就是播放与下载付费判定读的字段,**改完立即生效**(实测:未登录请求锁定集直接
返回 `UNAUTHORIZED`;已登录未解锁用户返回 `{"accessType":"COIN_REQUIRED","unlockCoin":50,"previewSeconds":15}`;
免费集直接返回签名播放地址)。
定价沿用「可编辑状态」口径:`draft` / `offline` 可改,已上线要改价需先申请下线(避免创作者悄悄改已上线内容的价格);
`dramaId` 以路径为准,body 里传别人的剧 id 无效,列表里带别人的 `episodeId` 会被忽略(只作用于自己剧的分集)。
@@ -0,0 +1,28 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.config;
import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer;
import org.springframework.boot.autoconfigure.jackson.Jackson2ObjectMapperBuilderCustomizer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import java.time.format.DateTimeFormatter;
/**
* 后台接口的时间格式统一成 `yyyy-MM-dd HH:mm:ss`。
*
* <p>application.yml 里的 `spring.jackson.date-format` 只作用于 java.util.Date,
* LocalDateTime 仍会输出 ISO 的 `2026-10-27T14:14:12.644`,后台表格里会露出一串带 T 的原文。
* 该配置只放在 admin-api,C 端保持 ISO,避免影响客户端解析。
*/
@Configuration
public class AdminJacksonTimeConfig {
private static final DateTimeFormatter DATE_TIME = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
@Bean
public Jackson2ObjectMapperBuilderCustomizer adminLocalDateTimeFormat() {
return builder -> builder.serializers(new LocalDateTimeSerializer(DATE_TIME));
}
}
@@ -32,12 +32,13 @@ import java.util.List;
public class AuthController {
private final AuthService authService;
private final WebUtils webUtils;
@Operation(summary = "管理端登录")
@PostMapping("/login")
@OperationLog(module = "认证", action = "登录")
public Result<LoginRespVO> login(@Valid @RequestBody LoginReqVO req, HttpServletRequest request) {
return Result.ok(authService.login(req, WebUtils.getClientIp(request)));
return Result.ok(authService.login(req, webUtils.clientIp(request)));
}
/** 当前用户的菜单路由树,结构与前端 element-plus-admin 的动态路由一致。 */
@@ -0,0 +1,34 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.controller;
import com.fruitshorts.admin.service.FavoriteAdminService;
import com.fruitshorts.admin.vo.AdminFavoriteVO;
import com.fruitshorts.admin.vo.FavoriteQueryReqVO;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.pojo.Result;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import lombok.RequiredArgsConstructor;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
/** 后台收藏只读查询入口。 */
@Tag(name = "后台-收藏列表")
@RestController
@RequestMapping("/admin/favorites")
@RequiredArgsConstructor
public class FavoriteAdminController {
private final FavoriteAdminService service;
/** 按用户、设备、内容和时间分页查询收藏。 */
@Operation(summary = "收藏列表分页查询")
@GetMapping("/page")
@PreAuthorize("@ss.hasPermission('favorite:list')")
public Result<PageResult<AdminFavoriteVO>> page(@Valid FavoriteQueryReqVO req) {
return Result.ok(service.page(req));
}
}
@@ -0,0 +1,34 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.controller;
import com.fruitshorts.admin.service.LikeAdminService;
import com.fruitshorts.admin.vo.AdminLikeVO;
import com.fruitshorts.admin.vo.LikeQueryReqVO;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.pojo.Result;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import lombok.RequiredArgsConstructor;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
/** 后台点赞只读查询入口。 */
@Tag(name = "后台-点赞列表")
@RestController
@RequestMapping("/admin/likes")
@RequiredArgsConstructor
public class LikeAdminController {
private final LikeAdminService service;
/** 按用户、设备、内容和时间分页查询点赞。 */
@Operation(summary = "点赞列表分页查询")
@GetMapping("/page")
@PreAuthorize("@ss.hasPermission('like:list')")
public Result<PageResult<AdminLikeVO>> page(@Valid LikeQueryReqVO req) {
return Result.ok(service.page(req));
}
}
@@ -3,7 +3,9 @@
package com.fruitshorts.admin.service;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.fruitshorts.user.dal.entity.AppDeviceProfile;
import com.fruitshorts.user.dal.entity.AppUserSession;
import com.fruitshorts.user.dal.mapper.AppDeviceProfileMapper;
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
@@ -13,16 +15,19 @@ import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
/**
* 按用户最近会话解析系统。同一设备优先,否则用仍有效的最近一次登录。
* 按设备画像解析系统,缺画像时回退到用户最近会话。
* 设备画像不依赖登录态,游客行为也能显示系统。
*/
@Service
@RequiredArgsConstructor
public class AppSessionSystemService {
private final AppUserSessionMapper sessionMapper;
private final AppDeviceProfileMapper profileMapper;
/** 按用户分组,组内按最近活跃倒序。 */
public Map<Long, List<AppUserSession>> groupByUser(Collection<Long> userIds) {
@@ -31,6 +36,9 @@ public class AppSessionSystemService {
return grouped;
}
List<AppUserSession> sessions = sessionMapper.selectList(Wrappers.<AppUserSession>lambdaQuery()
.select(AppUserSession::getId, AppUserSession::getUserId, AppUserSession::getDeviceId,
AppUserSession::getPlatform, AppUserSession::getOsVersion,
AppUserSession::getRevokedAt, AppUserSession::getLastActiveAt)
.in(AppUserSession::getUserId, userIds)
.orderByDesc(AppUserSession::getLastActiveAt)
.orderByDesc(AppUserSession::getId));
@@ -40,6 +48,49 @@ public class AppSessionSystemService {
return grouped;
}
/** 按 device_id 批量取设备画像,未建档的设备不出现在结果里。 */
public Map<String, AppDeviceProfile> deviceProfiles(Collection<String> deviceIds) {
Map<String, AppDeviceProfile> result = new HashMap<>();
if (deviceIds == null || deviceIds.isEmpty()) {
return result;
}
List<String> keys = deviceIds.stream()
.filter(StringUtils::hasText).map(String::trim).distinct().toList();
if (keys.isEmpty()) {
return result;
}
for (AppDeviceProfile profile : profileMapper.selectList(
Wrappers.<AppDeviceProfile>lambdaQuery().in(AppDeviceProfile::getDeviceId, keys))) {
result.put(profile.getDeviceId(), profile);
}
return result;
}
/**
* 平台筛选的命中范围:设备平台直接给 device_id;登录用户的行为行不带 device_id,
* 只能按该用户登录会话的平台判定,两者取并集。
*/
public record PlatformScope(List<String> deviceIds, List<Long> userIds) {
public boolean empty() {
return deviceIds.isEmpty() && userIds.isEmpty();
}
}
/** 按平台解析命中范围;未传平台返回 null 表示不限制。 */
public PlatformScope platformScope(String platform) {
if (!StringUtils.hasText(platform)) {
return null;
}
String value = platform.trim().toUpperCase(Locale.ROOT);
List<String> deviceIds = profileMapper.selectList(Wrappers.<AppDeviceProfile>lambdaQuery()
.select(AppDeviceProfile::getDeviceId)
.eq(AppDeviceProfile::getPlatform, value))
.stream().map(AppDeviceProfile::getDeviceId).toList();
// 只认「最近一次登录」的平台,与列表里展示的系统保持一致
List<Long> userIds = sessionMapper.selectUserIdsByLatestPlatform(value);
return new PlatformScope(deviceIds, userIds);
}
/** 观看记录优先匹配同一设备,其次取未吊销会话。 */
public AppUserSession match(List<AppUserSession> sessions, String deviceId) {
if (sessions == null || sessions.isEmpty()) {
@@ -0,0 +1,147 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.service;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fruitshorts.admin.vo.EngagementQueryReqVO;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.mapper.DramaMapper;
import com.fruitshorts.user.dal.entity.AppDeviceProfile;
import com.fruitshorts.user.dal.entity.AppUser;
import com.fruitshorts.user.dal.entity.AppUserSession;
import com.fruitshorts.user.dal.mapper.AppDeviceProfileMapper;
import com.fruitshorts.user.dal.mapper.AppUserMapper;
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Objects;
import java.util.function.Function;
import java.util.stream.Collectors;
/** 后台互动列表的分页、预检索及批量资料读取。 */
@Component
@RequiredArgsConstructor
public class EngagementAdminLookup {
private static final int DEFAULT_PAGE_SIZE = 10;
private static final int MAX_PAGE_SIZE = 100;
/**
* 关键词预检索的 ID 上限:命中过多时 IN 列表会超出数据库参数上限,
* 也说明关键词过于宽泛,直接提示使用者补充条件而不是截断结果。
*/
private static final int MAX_LOOKUP_SIZE = 2000;
private static final int LOOKUP_PROBE = MAX_LOOKUP_SIZE + 1;
private final AppUserMapper userMapper;
private final DramaMapper dramaMapper;
private final AppDeviceProfileMapper deviceProfileMapper;
private final AppUserSessionMapper sessionMapper;
private final AppSessionSystemService sessionSystem;
/** 规范化页码和页大小。 */
public <T> Page<T> page(EngagementQueryReqVO req) {
int index = req.getPageIndex() == null ? 1 : Math.max(1, req.getPageIndex());
int size = req.getPageSize() == null || req.getPageSize() < 1
? DEFAULT_PAGE_SIZE : Math.min(MAX_PAGE_SIZE, req.getPageSize());
return new Page<>(index, size);
}
/** 按 uid 或昵称预检索,未传关键词时不限制用户。 */
public List<Long> userIds(String keyword) {
if (!StringUtils.hasText(keyword)) {
return null;
}
String value = keyword.trim();
List<Long> ids = userMapper.selectList(Wrappers.<AppUser>lambdaQuery().select(AppUser::getId)
.and(q -> q.like(AppUser::getUid, value).or().like(AppUser::getNickname, value))
.last("LIMIT " + LOOKUP_PROBE))
.stream().map(AppUser::getId).toList();
return ensureLookupSize(ids, "uid / 昵称");
}
/** 按剧名预检索,未传剧名时不限制短剧。 */
public List<Long> dramaIds(String title) {
if (!StringUtils.hasText(title)) {
return null;
}
List<Long> ids = dramaMapper.selectList(Wrappers.<Drama>lambdaQuery().select(Drama::getId)
.like(Drama::getTitle, title.trim())
.last("LIMIT " + LOOKUP_PROBE)).stream().map(Drama::getId).toList();
return ensureLookupSize(ids, "短剧名称");
}
/** 命中数量超限时给出可操作的提示,避免把整张表拉进 IN 条件。 */
private List<Long> ensureLookupSize(List<Long> ids, String field) {
if (ids.size() > MAX_LOOKUP_SIZE) {
throw new ServiceException(ErrorCode.BAD_REQUEST,
field + "命中超过 " + MAX_LOOKUP_SIZE + " 条,请使用更精确的筛选条件");
}
return ids;
}
/** 区分无筛选和有筛选但无匹配。 */
public boolean noMatch(List<Long> ids) {
return ids != null && ids.isEmpty();
}
/** 按非空去重 ID 批量读取用户,避免游客触发空 IN。 */
public Map<Long, AppUser> users(Collection<Long> ids) {
List<Long> keys = keys(ids);
return keys.isEmpty() ? Map.of() : userMapper.selectBatchIds(keys).stream()
.collect(Collectors.toMap(AppUser::getId, Function.identity()));
}
/** 按非空去重 ID 批量读取短剧。 */
public Map<Long, Drama> dramas(Collection<Long> ids) {
List<Long> keys = keys(ids);
return keys.isEmpty() ? Map.of() : dramaMapper.selectBatchIds(keys).stream()
.collect(Collectors.toMap(Drama::getId, Function.identity()));
}
/** 按 device_id 批量读取设备画像;游客也有记录,没建档的不返回。 */
public Map<String, AppDeviceProfile> deviceProfiles(Collection<String> deviceIds) {
List<String> keys = deviceIds.stream().filter(StringUtils::hasText)
.map(String::trim).distinct().toList();
return keys.isEmpty() ? Map.of() : deviceProfileMapper.selectList(
Wrappers.<AppDeviceProfile>lambdaQuery().in(AppDeviceProfile::getDeviceId, keys))
.stream().collect(Collectors.toMap(AppDeviceProfile::getDeviceId,
Function.identity(), (first, second) -> first));
}
/**
* 每个用户取最近一次登录会话。
* 已登录用户的互动行 device_id 为空(写入时会清空),只能退回会话取系统。
*/
public Map<Long, AppUserSession> latestSessions(Collection<Long> userIds) {
List<Long> keys = keys(userIds);
if (keys.isEmpty()) {
return Map.of();
}
Map<Long, AppUserSession> result = new HashMap<>();
for (AppUserSession session : sessionMapper.selectList(Wrappers.<AppUserSession>lambdaQuery()
.select(AppUserSession::getId, AppUserSession::getUserId,
AppUserSession::getPlatform)
.in(AppUserSession::getUserId, keys)
.orderByDesc(AppUserSession::getLastActiveAt)
.orderByDesc(AppUserSession::getId))) {
result.putIfAbsent(session.getUserId(), session);
}
return result;
}
/** 按平台解析命中范围;未传平台返回 null 表示不限制。 */
public AppSessionSystemService.PlatformScope platformScope(String platform) {
return sessionSystem.platformScope(platform);
}
private List<Long> keys(Collection<Long> ids) {
return ids.stream().filter(Objects::nonNull).distinct().toList();
}
}
@@ -0,0 +1,117 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.service;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.fruitshorts.admin.vo.AdminFavoriteVO;
import com.fruitshorts.admin.vo.FavoriteQueryReqVO;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.enums.ContentStatus;
import com.fruitshorts.playback.dal.entity.UserFavorite;
import com.fruitshorts.playback.dal.mapper.UserFavoriteMapper;
import com.fruitshorts.user.dal.entity.AppDeviceProfile;
import com.fruitshorts.user.dal.entity.AppUser;
import com.fruitshorts.user.dal.entity.AppUserSession;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import java.util.List;
import java.util.Map;
/** 后台收藏只读分页及跨域资料装配。 */
@Service
@RequiredArgsConstructor
public class FavoriteAdminService {
private final UserFavoriteMapper favoriteMapper;
private final EngagementAdminLookup lookup;
/** 分页查询收藏,预检索无匹配时直接返回空页。 */
public PageResult<AdminFavoriteVO> page(FavoriteQueryReqVO req) {
List<Long> userIds = lookup.userIds(req.getKeyword());
if (lookup.noMatch(userIds)) {
return PageResult.empty();
}
List<Long> dramaIds = lookup.dramaIds(req.getDramaTitle());
if (lookup.noMatch(dramaIds)) {
return PageResult.empty();
}
var scope = lookup.platformScope(req.getPlatform());
if (scope != null && scope.empty()) {
return PageResult.empty();
}
var query = Wrappers.<UserFavorite>lambdaQuery()
.eq(req.getUserId() != null, UserFavorite::getUserId, req.getUserId())
.in(userIds != null, UserFavorite::getUserId, userIds)
.eq(StringUtils.hasText(req.getDeviceId()), UserFavorite::getDeviceId, req.getDeviceId())
.eq(req.getDramaId() != null, UserFavorite::getDramaId, req.getDramaId())
.in(dramaIds != null, UserFavorite::getDramaId, dramaIds)
.ge(req.getStartTime() != null, UserFavorite::getCreatedAt, req.getStartTime())
.le(req.getEndTime() != null, UserFavorite::getCreatedAt, req.getEndTime());
// orderBy 必须最后追加,否则条件会被拼到 ORDER BY 后面
if (scope != null) {
query.and(inner -> {
if (!scope.deviceIds().isEmpty()) {
inner.in(UserFavorite::getDeviceId, scope.deviceIds());
}
if (!scope.userIds().isEmpty()) {
if (!scope.deviceIds().isEmpty()) {
inner.or();
}
inner.in(UserFavorite::getUserId, scope.userIds());
}
});
}
query.orderByDesc(UserFavorite::getCreatedAt, UserFavorite::getId);
var result = favoriteMapper.selectPage(lookup.page(req), query);
return new PageResult<>(result.getTotal(), assemble(result.getRecords()));
}
private List<AdminFavoriteVO> assemble(List<UserFavorite> rows) {
if (rows.isEmpty()) {
return List.of();
}
Map<Long, AppUser> users = lookup.users(rows.stream().map(UserFavorite::getUserId).toList());
Map<Long, Drama> dramas = lookup.dramas(rows.stream().map(UserFavorite::getDramaId).toList());
Map<String, AppDeviceProfile> profiles =
lookup.deviceProfiles(rows.stream().map(UserFavorite::getDeviceId).toList());
Map<Long, AppUserSession> sessions =
lookup.latestSessions(rows.stream().map(UserFavorite::getUserId).toList());
return rows.stream().map(row -> toVo(row, row.getUserId() == null ? null : users.get(row.getUserId()),
row.getDramaId() == null ? null : dramas.get(row.getDramaId()),
row.getDeviceId() == null ? null : profiles.get(row.getDeviceId()),
row.getUserId() == null ? null : sessions.get(row.getUserId())))
.toList();
}
private AdminFavoriteVO toVo(UserFavorite row, AppUser user, Drama drama,
AppDeviceProfile profile, AppUserSession session) {
AdminFavoriteVO vo = new AdminFavoriteVO();
vo.setId(row.getId());
vo.setUserId(row.getUserId());
vo.setDeviceId(row.getDeviceId());
vo.setAnonymous(row.getUserId() == null);
if (profile != null) {
vo.setPlatform(profile.getPlatform());
}
// 已登录用户的互动行不带 device_id,设备平台查不到时退回最近一次登录会话
if (!StringUtils.hasText(vo.getPlatform()) && session != null) {
vo.setPlatform(session.getPlatform());
}
vo.setDramaId(row.getDramaId());
vo.setCreatedAt(row.getCreatedAt());
if (user != null) {
vo.setUid(user.getUid());
vo.setNickname(user.getNickname());
}
if (drama != null) {
vo.setDramaTitle(drama.getTitle());
vo.setDramaCover(drama.getCoverUrl());
vo.setDramaStatus(drama.getStatus());
vo.setDramaPublished(ContentStatus.PUBLISHED.getCode().equals(drama.getStatus()));
}
// 短剧读不到(已删除或数据异常)时保持 dramaPublished 为 null,
// 与「未发布」区分开,前端只在明确 false 时才展示未发布标记。
return vo;
}
}
@@ -0,0 +1,210 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.service;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.fruitshorts.admin.vo.AdminLikeVO;
import com.fruitshorts.admin.vo.LikeQueryReqVO;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.dal.mapper.DramaEpisodeMapper;
import com.fruitshorts.interaction.dal.entity.ContentComment;
import com.fruitshorts.interaction.dal.entity.ContentLike;
import com.fruitshorts.interaction.dal.mapper.ContentCommentMapper;
import com.fruitshorts.interaction.dal.mapper.ContentLikeMapper;
import com.fruitshorts.user.dal.entity.AppDeviceProfile;
import com.fruitshorts.user.dal.entity.AppUser;
import com.fruitshorts.user.dal.entity.AppUserSession;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.function.Function;
import java.util.stream.Collectors;
/** 后台点赞只读分页及多态目标批量装配。 */
@Service
@RequiredArgsConstructor
public class LikeAdminService {
private static final String DRAMA = "DRAMA";
private static final String EPISODE = "EPISODE";
private static final String COMMENT = "COMMENT";
private static final int SUMMARY_LENGTH = 30;
/**
* 点赞目标是多态列(target_type + target_id),按短剧/剧集筛选必须跨表回溯,
* 无法用 LambdaQueryWrapper 表达,只能写原生片段。
* 维护约定:表名、逻辑删除列与 {@code drama_episode} / {@code content_comment}
* 实体上的 @TableName / @TableLogic 保持一致,改表结构时需要同步修改这里。
*/
private static final String DRAMA_MATCH =
"(target_type = 'DRAMA' AND target_id = {0}) OR "
+ "(target_type = 'EPISODE' AND EXISTS (SELECT 1 FROM drama_episode e "
+ "WHERE e.id = target_id AND e.deleted = 0 AND e.drama_id = {0})) OR "
+ "(target_type = 'COMMENT' AND EXISTS (SELECT 1 FROM content_comment c "
+ "WHERE c.id = target_id AND c.deleted = 0 AND c.drama_id = {0}))";
private static final String EPISODE_MATCH =
"(target_type = 'EPISODE' AND target_id = {0}) OR "
+ "(target_type = 'COMMENT' AND EXISTS (SELECT 1 FROM content_comment c "
+ "WHERE c.id = target_id AND c.deleted = 0 AND c.episode_id = {0}))";
private final ContentLikeMapper likeMapper;
private final ContentCommentMapper commentMapper;
private final DramaEpisodeMapper episodeMapper;
private final EngagementAdminLookup lookup;
/** 分页查询点赞;短剧及剧集条件在分页之前作用于各类目标。 */
public PageResult<AdminLikeVO> page(LikeQueryReqVO req) {
List<Long> users = lookup.userIds(req.getKeyword());
if (lookup.noMatch(users)) {
return PageResult.empty();
}
List<Long> dramas = lookup.dramaIds(req.getDramaTitle());
if (lookup.noMatch(dramas)) {
return PageResult.empty();
}
var query = buildQuery(req, users);
var scope = lookup.platformScope(req.getPlatform());
if (scope != null && scope.empty()) {
return PageResult.empty();
}
if (scope != null) {
query.and(inner -> {
if (!scope.deviceIds().isEmpty()) {
inner.in(ContentLike::getDeviceId, scope.deviceIds());
}
if (!scope.userIds().isEmpty()) {
if (!scope.deviceIds().isEmpty()) {
inner.or();
}
inner.in(ContentLike::getUserId, scope.userIds());
}
});
}
if (req.getDramaId() != null) {
query.and(q -> q.apply(DRAMA_MATCH, req.getDramaId()));
}
if (dramas != null) {
query.and(q -> {
for (Long id : dramas) {
q.or().apply(DRAMA_MATCH, id);
}
});
}
if (req.getEpisodeId() != null) {
query.and(q -> q.apply(EPISODE_MATCH, req.getEpisodeId()));
}
// orderBy 必须最后追加,否则条件会被拼到 ORDER BY 后面
query.orderByDesc(ContentLike::getCreatedAt, ContentLike::getId);
var result = likeMapper.selectPage(lookup.page(req), query);
return new PageResult<>(result.getTotal(), assemble(result.getRecords()));
}
private LambdaQueryWrapper<ContentLike> buildQuery(LikeQueryReqVO req, List<Long> users) {
return Wrappers.<ContentLike>lambdaQuery()
.eq(req.getUserId() != null, ContentLike::getUserId, req.getUserId())
.in(users != null, ContentLike::getUserId, users)
.eq(StringUtils.hasText(req.getDeviceId()), ContentLike::getDeviceId, req.getDeviceId())
.eq(StringUtils.hasText(req.getTargetType()), ContentLike::getTargetType, req.getTargetType())
.eq(req.getTargetId() != null, ContentLike::getTargetId, req.getTargetId())
.ge(req.getStartTime() != null, ContentLike::getCreatedAt, req.getStartTime())
.le(req.getEndTime() != null, ContentLike::getCreatedAt, req.getEndTime());
}
private List<AdminLikeVO> assemble(List<ContentLike> rows) {
if (rows.isEmpty()) {
return List.of();
}
Map<Long, AppUser> users = lookup.users(rows.stream().map(ContentLike::getUserId).toList());
Map<Long, ContentComment> comments = comments(targetIds(rows, COMMENT));
List<Long> episodeIds = new ArrayList<>(targetIds(rows, EPISODE));
comments.values().stream().map(ContentComment::getEpisodeId).filter(Objects::nonNull)
.forEach(episodeIds::add);
Map<Long, DramaEpisode> episodes = episodes(episodeIds);
List<Long> dramaIds = new ArrayList<>(targetIds(rows, DRAMA));
comments.values().stream().map(ContentComment::getDramaId).filter(Objects::nonNull).forEach(dramaIds::add);
episodes.values().stream().map(DramaEpisode::getDramaId).filter(Objects::nonNull).forEach(dramaIds::add);
Map<Long, Drama> dramas = lookup.dramas(dramaIds);
Map<String, AppDeviceProfile> profiles =
lookup.deviceProfiles(rows.stream().map(ContentLike::getDeviceId).toList());
Map<Long, AppUserSession> sessions =
lookup.latestSessions(rows.stream().map(ContentLike::getUserId).toList());
return rows.stream()
.map(row -> toVo(row, users, comments, episodes, dramas, profiles, sessions))
.toList();
}
private List<Long> targetIds(List<ContentLike> rows, String type) {
return rows.stream().filter(row -> type.equals(row.getTargetType())).map(ContentLike::getTargetId)
.filter(Objects::nonNull).distinct().toList();
}
private Map<Long, ContentComment> comments(Collection<Long> ids) {
return ids.isEmpty() ? Map.of() : commentMapper.selectBatchIds(ids).stream()
.collect(Collectors.toMap(ContentComment::getId, Function.identity()));
}
private Map<Long, DramaEpisode> episodes(Collection<Long> ids) {
List<Long> keys = ids.stream().filter(Objects::nonNull).distinct().toList();
return keys.isEmpty() ? Map.of() : episodeMapper.selectBatchIds(keys).stream()
.collect(Collectors.toMap(DramaEpisode::getId, Function.identity()));
}
private AdminLikeVO toVo(ContentLike row, Map<Long, AppUser> users, Map<Long, ContentComment> comments,
Map<Long, DramaEpisode> episodes, Map<Long, Drama> dramas,
Map<String, AppDeviceProfile> profiles,
Map<Long, AppUserSession> sessions) {
AdminLikeVO vo = new AdminLikeVO();
vo.setId(row.getId());
vo.setUserId(row.getUserId());
vo.setDeviceId(row.getDeviceId());
vo.setAnonymous(row.getUserId() == null);
AppDeviceProfile profile = row.getDeviceId() == null ? null : profiles.get(row.getDeviceId());
if (profile != null) {
vo.setPlatform(profile.getPlatform());
}
// 已登录用户的互动行不带 device_id,设备平台查不到时退回最近一次登录会话
if (!StringUtils.hasText(vo.getPlatform()) && row.getUserId() != null) {
AppUserSession session = sessions.get(row.getUserId());
if (session != null) {
vo.setPlatform(session.getPlatform());
}
}
vo.setTargetType(row.getTargetType());
vo.setTargetId(row.getTargetId());
vo.setCreatedAt(row.getCreatedAt());
AppUser user = row.getUserId() == null ? null : users.get(row.getUserId());
if (user != null) {
vo.setUid(user.getUid());
vo.setNickname(user.getNickname());
}
ContentComment comment = COMMENT.equals(row.getTargetType()) ? comments.get(row.getTargetId()) : null;
Long episodeId = EPISODE.equals(row.getTargetType()) ? row.getTargetId()
: comment == null ? null : comment.getEpisodeId();
DramaEpisode episode = episodeId == null ? null : episodes.get(episodeId);
Long dramaId = DRAMA.equals(row.getTargetType()) ? row.getTargetId()
: comment != null ? comment.getDramaId() : episode == null ? null : episode.getDramaId();
vo.setEpisodeId(episodeId);
vo.setDramaId(dramaId);
if (comment != null && comment.getContent() != null) {
String content = comment.getContent();
int count = content.codePointCount(0, content.length());
vo.setCommentSummary(content.substring(0, content.offsetByCodePoints(0, Math.min(count, SUMMARY_LENGTH))));
}
if (episode != null) {
vo.setEpisodeNo(episode.getEpNo());
vo.setEpisodeTitle(episode.getTitle());
}
Drama drama = dramaId == null ? null : dramas.get(dramaId);
if (drama != null) {
vo.setDramaTitle(drama.getTitle());
vo.setDramaCover(drama.getCoverUrl());
}
return vo;
}
}
@@ -8,6 +8,8 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fruitshorts.admin.vo.AdminWatchHistoryVO;
import com.fruitshorts.admin.vo.WatchHistoryQueryReqVO;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
@@ -16,6 +18,7 @@ import com.fruitshorts.content.dal.mapper.DramaMapper;
import com.fruitshorts.content.enums.ContentStatus;
import com.fruitshorts.playback.dal.entity.WatchHistory;
import com.fruitshorts.playback.dal.mapper.WatchHistoryMapper;
import com.fruitshorts.user.dal.entity.AppDeviceProfile;
import com.fruitshorts.user.dal.entity.AppUser;
import com.fruitshorts.user.dal.entity.AppUserSession;
import com.fruitshorts.user.dal.mapper.AppUserMapper;
@@ -27,6 +30,7 @@ import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
/**
* 后台观看记录分页与跨域装配。
@@ -42,6 +46,8 @@ public class WatchHistoryAdminService {
private static final int DEFAULT_PAGE_SIZE = 10;
private static final int MAX_PAGE_SIZE = 100;
private static final long FINISHED_TOLERANCE_MS = 1000L;
private static final int MAX_LOOKUP_SIZE = 2000;
private static final int LOOKUP_PROBE = MAX_LOOKUP_SIZE + 1;
private final WatchHistoryMapper watchHistoryMapper;
private final AppUserMapper appUserMapper;
@@ -65,14 +71,19 @@ public class WatchHistoryAdminService {
if (matchedDramaIds != null && matchedDramaIds.isEmpty()) {
return emptyPage();
}
var scope = sessionSystem.platformScope(req.getPlatform());
if (scope != null && scope.empty()) {
return emptyPage();
}
IPage<WatchHistory> result = watchHistoryMapper.selectPage(new Page<>(pageIndex, pageSize),
buildQuery(req, matchedUserIds, matchedDramaIds));
buildQuery(req, matchedUserIds, matchedDramaIds, scope));
return new PageResult<>(result.getTotal(), toVoList(result.getRecords()));
}
private LambdaQueryWrapper<WatchHistory> buildQuery(WatchHistoryQueryReqVO req,
List<Long> matchedUserIds,
List<Long> matchedDramaIds) {
List<Long> matchedDramaIds,
AppSessionSystemService.PlatformScope scope) {
LambdaQueryWrapper<WatchHistory> wrapper = Wrappers.<WatchHistory>lambdaQuery()
.eq(req.getUserId() != null, WatchHistory::getUserId, req.getUserId())
.in(matchedUserIds != null, WatchHistory::getUserId, matchedUserIds)
@@ -82,6 +93,19 @@ public class WatchHistoryAdminService {
.eq(StringUtils.hasText(req.getSource()), WatchHistory::getSource, req.getSource())
.ge(req.getStartTime() != null, WatchHistory::getWatchedAt, req.getStartTime())
.le(req.getEndTime() != null, WatchHistory::getWatchedAt, req.getEndTime());
if (scope != null) {
wrapper.and(inner -> {
if (!scope.deviceIds().isEmpty()) {
inner.in(WatchHistory::getDeviceId, scope.deviceIds());
}
if (!scope.userIds().isEmpty()) {
if (!scope.deviceIds().isEmpty()) {
inner.or();
}
inner.in(WatchHistory::getUserId, scope.userIds());
}
});
}
applyFinished(wrapper, req.getFinished());
return wrapper.orderByDesc(WatchHistory::getWatchedAt)
.orderByDesc(WatchHistory::getId);
@@ -100,24 +124,37 @@ public class WatchHistoryAdminService {
private List<Long> searchUserIds(String keyword) {
String value = keyword.trim();
return appUserMapper.selectList(Wrappers.<AppUser>lambdaQuery()
List<Long> ids = appUserMapper.selectList(Wrappers.<AppUser>lambdaQuery()
.select(AppUser::getId)
.and(query -> query.like(AppUser::getUid, value)
.or()
.like(AppUser::getNickname, value)))
.like(AppUser::getNickname, value))
.last("LIMIT " + LOOKUP_PROBE))
.stream()
.map(AppUser::getId)
.toList();
return ensureLookupSize(ids, "uid / 昵称");
}
private List<Long> searchDramaIds(String dramaTitle) {
String value = dramaTitle.trim();
return dramaMapper.selectList(Wrappers.<Drama>lambdaQuery()
List<Long> ids = dramaMapper.selectList(Wrappers.<Drama>lambdaQuery()
.select(Drama::getId)
.like(Drama::getTitle, value))
.like(Drama::getTitle, value)
.last("LIMIT " + LOOKUP_PROBE))
.stream()
.map(Drama::getId)
.toList();
return ensureLookupSize(ids, "短剧名称");
}
/** 关键词命中过多时直接提示收窄条件,避免把整张表塞进 IN。 */
private List<Long> ensureLookupSize(List<Long> ids, String field) {
if (ids.size() > MAX_LOOKUP_SIZE) {
throw new ServiceException(ErrorCode.BAD_REQUEST,
field + "命中超过 " + MAX_LOOKUP_SIZE + " 条,请使用更精确的筛选条件");
}
return ids;
}
private List<AdminWatchHistoryVO> toVoList(List<WatchHistory> rows) {
@@ -129,19 +166,28 @@ public class WatchHistoryAdminService {
Map<Long, DramaEpisode> episodes = loadEpisodes(rows);
Map<Long, List<AppUserSession>> sessions = sessionSystem.groupByUser(
rows.stream().map(WatchHistory::getUserId).distinct().toList());
Map<String, AppDeviceProfile> profiles = sessionSystem.deviceProfiles(
rows.stream().map(WatchHistory::getDeviceId).toList());
List<AdminWatchHistoryVO> list = new ArrayList<>(rows.size());
for (WatchHistory row : rows) {
list.add(toVo(row, users.get(row.getUserId()), dramas.get(row.getDramaId()),
episodes.get(row.getEpisodeId()),
sessionSystem.match(sessions.get(row.getUserId()), row.getDeviceId())));
sessionSystem.match(sessions.get(row.getUserId()), row.getDeviceId()),
profiles.get(row.getDeviceId())));
}
return list;
}
private Map<Long, AppUser> loadUsers(List<WatchHistory> rows) {
Map<Long, AppUser> result = new HashMap<>();
List<Long> ids = rows.stream().map(WatchHistory::getUserId).distinct().toList();
for (AppUser user : appUserMapper.selectBatchIds(ids)) {
List<Long> ids = rows.stream().map(WatchHistory::getUserId).filter(Objects::nonNull).distinct().toList();
if (ids.isEmpty()) {
return result;
}
// 列表只需要身份展示字段,避免把整行用户资料拉回来
for (AppUser user : appUserMapper.selectList(Wrappers.<AppUser>lambdaQuery()
.select(AppUser::getId, AppUser::getUid, AppUser::getNickname)
.in(AppUser::getId, ids))) {
result.put(user.getId(), user);
}
return result;
@@ -149,8 +195,13 @@ public class WatchHistoryAdminService {
private Map<Long, Drama> loadDramas(List<WatchHistory> rows) {
Map<Long, Drama> result = new HashMap<>();
List<Long> ids = rows.stream().map(WatchHistory::getDramaId).distinct().toList();
for (Drama drama : dramaMapper.selectBatchIds(ids)) {
List<Long> ids = rows.stream().map(WatchHistory::getDramaId).filter(Objects::nonNull).distinct().toList();
if (ids.isEmpty()) {
return result;
}
for (Drama drama : dramaMapper.selectList(Wrappers.<Drama>lambdaQuery()
.select(Drama::getId, Drama::getTitle, Drama::getStatus)
.in(Drama::getId, ids))) {
result.put(drama.getId(), drama);
}
return result;
@@ -158,15 +209,22 @@ public class WatchHistoryAdminService {
private Map<Long, DramaEpisode> loadEpisodes(List<WatchHistory> rows) {
Map<Long, DramaEpisode> result = new HashMap<>();
List<Long> ids = rows.stream().map(WatchHistory::getEpisodeId).distinct().toList();
for (DramaEpisode episode : dramaEpisodeMapper.selectBatchIds(ids)) {
List<Long> ids = rows.stream().map(WatchHistory::getEpisodeId).filter(Objects::nonNull).distinct().toList();
if (ids.isEmpty()) {
return result;
}
for (DramaEpisode episode : dramaEpisodeMapper.selectList(Wrappers.<DramaEpisode>lambdaQuery()
.select(DramaEpisode::getId, DramaEpisode::getDramaId,
DramaEpisode::getEpNo, DramaEpisode::getTitle)
.in(DramaEpisode::getId, ids))) {
result.put(episode.getId(), episode);
}
return result;
}
private AdminWatchHistoryVO toVo(WatchHistory row, AppUser user, Drama drama,
DramaEpisode episode, AppUserSession session) {
DramaEpisode episode, AppUserSession session,
AppDeviceProfile profile) {
AdminWatchHistoryVO vo = new AdminWatchHistoryVO();
vo.setId(row.getId());
vo.setUserId(row.getUserId());
@@ -177,7 +235,10 @@ public class WatchHistoryAdminService {
vo.setSource(row.getSource());
vo.setDeviceId(row.getDeviceId());
vo.setWatchedAt(row.getWatchedAt());
if (session != null) {
// 设备平台不依赖登录态,优先;没有记录(老数据)再回退到登录会话
if (profile != null && StringUtils.hasText(profile.getPlatform())) {
vo.setPlatform(profile.getPlatform());
} else if (session != null) {
vo.setPlatform(session.getPlatform());
vo.setOsVersion(session.getOsVersion());
}
@@ -0,0 +1,26 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.vo;
import lombok.Data;
import java.time.LocalDateTime;
/** 后台收藏只读列表行,保留游客设备信息。 */
@Data
public class AdminFavoriteVO {
private Long id;
private Long userId;
private String uid;
private String nickname;
private String deviceId;
/** 客户端平台:ANDROID / IOS / WEB,来自设备平台记录,游客同样有值。 */
private String platform;
private Boolean anonymous;
private Long dramaId;
private String dramaTitle;
private String dramaCover;
private String dramaStatus;
/** true=已发布,false=未发布,null=短剧已不存在(不代表未发布)。 */
private Boolean dramaPublished;
private LocalDateTime createdAt;
}
@@ -0,0 +1,29 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.vo;
import lombok.Data;
import java.time.LocalDateTime;
/** 后台点赞只读列表行,保留游客设备信息。 */
@Data
public class AdminLikeVO {
private Long id;
private Long userId;
private String uid;
private String nickname;
private String deviceId;
/** 客户端平台:ANDROID / IOS / WEB,来自设备平台记录,游客同样有值。 */
private String platform;
private Boolean anonymous;
private Long dramaId;
private String dramaTitle;
private String dramaCover;
private String targetType;
private Long targetId;
private Long episodeId;
private Integer episodeNo;
private String episodeTitle;
private String commentSummary;
private LocalDateTime createdAt;
}
@@ -0,0 +1,27 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.vo;
import lombok.Data;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;
/** 收藏与点赞列表的公共检索条件。 */
@Data
public class EngagementQueryReqVO {
private static final int DEFAULT_PAGE_SIZE = 10;
private static final String TIME_PATTERN = "yyyy-MM-dd HH:mm:ss";
private Integer pageIndex = 1;
private Integer pageSize = DEFAULT_PAGE_SIZE;
private String keyword;
private Long userId;
private String deviceId;
/** 客户端平台筛选:ANDROID / IOS / WEB,留空不限制。 */
private String platform;
private String dramaTitle;
private Long dramaId;
@DateTimeFormat(pattern = TIME_PATTERN)
private LocalDateTime startTime;
@DateTimeFormat(pattern = TIME_PATTERN)
private LocalDateTime endTime;
}
@@ -0,0 +1,12 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.vo;
import lombok.Data;
import lombok.EqualsAndHashCode;
/** 后台收藏分页检索条件。 */
@Data
@EqualsAndHashCode(callSuper = true)
public class FavoriteQueryReqVO extends EngagementQueryReqVO {
}
@@ -0,0 +1,17 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.vo;
import jakarta.validation.constraints.Pattern;
import lombok.Data;
import lombok.EqualsAndHashCode;
/** 后台点赞分页检索条件。 */
@Data
@EqualsAndHashCode(callSuper = true)
public class LikeQueryReqVO extends EngagementQueryReqVO {
@Pattern(regexp = "DRAMA|EPISODE|COMMENT", message = "目标类型必须为 DRAMA、EPISODE 或 COMMENT")
private String targetType;
private Long targetId;
private Long episodeId;
}
@@ -20,6 +20,8 @@ public class WatchHistoryQueryReqVO {
private String dramaTitle;
private Long dramaId;
private Long episodeId;
/** 客户端平台筛选:ANDROID / IOS / WEB,留空不限制。 */
private String platform;
private String source;
private Boolean finished;
@@ -14,6 +14,12 @@ spring:
url: ${DB_URL:jdbc:postgresql://127.0.0.1:5432/fruitshorts}
username: ${DB_USER:fruitshorts}
password: ${DB_PASSWORD:fruitshorts}
data:
redis:
# 与 app-api 共用同一个 Redis 实例;容器内必须显式指定 host,
# 否则 Spring Boot 默认连 127.0.0.1(容器内即自身)导致健康检查报 DOWN。
host: ${REDIS_HOST:127.0.0.1}
port: ${REDIS_PORT:6379}
jackson:
date-format: yyyy-MM-dd HH:mm:ss
time-zone: Asia/Shanghai
@@ -40,6 +46,11 @@ mybatis-plus:
datacenter-id: 1
fruitshorts:
web:
# 可信代理 CIDR 列表(逗号分隔)。默认空 = 永不解析 XFF,直接用 request.getRemoteAddr()。
# 配了反代(Nginx/K8s ingress)时填上其 IP 段,才会从 XFF 取最右侧可信一跳。
# 生产若开启了 Tomcat RemoteIpValve(prod profile 已配),保持空即可。
trusted-proxies: ${FRUITSHORTS_TRUSTED_PROXIES:}
media:
# 媒资目录:容器里指向挂载卷;本地默认 ./data/uploads
root-dir: ${MEDIA_ROOT_DIR:./data/uploads}
@@ -0,0 +1,45 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.controller;
import com.fruitshorts.admin.service.FavoriteAdminService;
import com.fruitshorts.admin.service.LikeAdminService;
import com.fruitshorts.admin.support.ControllerContractSupport;
import com.fruitshorts.admin.vo.AdminFavoriteVO;
import com.fruitshorts.common.pojo.PageResult;
import org.junit.jupiter.api.Test;
import java.util.List;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
/** 收藏和点赞分页路由及响应契约。 */
class EngagementControllerContractTest {
@Test
void should_return_guest_fields_for_favorites() throws Exception {
FavoriteAdminService service = mock(FavoriteAdminService.class);
AdminFavoriteVO guest = new AdminFavoriteVO();
guest.setId(1L);
guest.setDeviceId("guest");
guest.setAnonymous(true);
when(service.page(any())).thenReturn(new PageResult<>(1L, List.of(guest)));
ControllerContractSupport.mockMvcOf(new FavoriteAdminController(service))
.perform(get("/admin/favorites/page"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.data.total").value(1))
.andExpect(jsonPath("$.data.list[0].deviceId").value("guest"))
.andExpect(jsonPath("$.data.list[0].anonymous").value(true));
}
@Test
void should_return_empty_like_page() throws Exception {
LikeAdminService service = mock(LikeAdminService.class);
when(service.page(any())).thenReturn(PageResult.empty());
ControllerContractSupport.mockMvcOf(new LikeAdminController(service))
.perform(get("/admin/likes/page").param("targetType", "COMMENT"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.data.total").value(0))
.andExpect(jsonPath("$.data.list").isEmpty());
}
}
@@ -13,7 +13,7 @@ import static org.assertj.core.api.Assertions.assertThat;
/** 观看记录匹配登录系统。 */
class AppSessionSystemServiceTest {
private final AppSessionSystemService service = new AppSessionSystemService(null);
private final AppSessionSystemService service = new AppSessionSystemService(null, null);
@Test
void should_use_same_device_session_when_watch_history_has_device() {
@@ -0,0 +1,105 @@
// Author: fruit.com.cn
package com.fruitshorts.admin.service;
import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fruitshorts.admin.vo.FavoriteQueryReqVO;
import com.fruitshorts.admin.vo.LikeQueryReqVO;
import com.fruitshorts.content.dal.mapper.DramaEpisodeMapper;
import com.fruitshorts.interaction.dal.entity.ContentLike;
import com.fruitshorts.interaction.dal.mapper.ContentCommentMapper;
import com.fruitshorts.interaction.dal.mapper.ContentLikeMapper;
import com.fruitshorts.playback.dal.entity.UserFavorite;
import com.fruitshorts.playback.dal.mapper.UserFavoriteMapper;
import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import java.util.List;
import java.util.Map;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;
/** 收藏及点赞分页的游客、空结果和多态目标检验。 */
class EngagementAdminServiceTest {
@BeforeAll
static void initMetadata() {
MybatisConfiguration config = new MybatisConfiguration();
MapperBuilderAssistant assistant = new MapperBuilderAssistant(config, "");
TableInfoHelper.initTableInfo(assistant, UserFavorite.class);
TableInfoHelper.initTableInfo(assistant, ContentLike.class);
}
@Test
void should_enable_logic_delete_for_both_interaction_entities() {
assertThat(TableInfoHelper.getTableInfo(UserFavorite.class).isWithLogicDelete()).isTrue();
assertThat(TableInfoHelper.getTableInfo(ContentLike.class).isWithLogicDelete()).isTrue();
assertThat(TableInfoHelper.getTableInfo(UserFavorite.class).getLogicDeleteSql(true, true))
.contains("deleted").contains("0");
assertThat(TableInfoHelper.getTableInfo(ContentLike.class).getLogicDeleteSql(true, true))
.contains("deleted").contains("0");
}
@Test
void should_return_empty_without_query_when_keyword_has_no_match() {
EngagementAdminLookup lookup = mock(EngagementAdminLookup.class);
UserFavoriteMapper mapper = mock(UserFavoriteMapper.class);
when(lookup.userIds("missing")).thenReturn(List.of());
when(lookup.noMatch(List.of())).thenReturn(true);
FavoriteQueryReqVO req = new FavoriteQueryReqVO();
req.setKeyword("missing");
var result = new FavoriteAdminService(mapper, lookup).page(req);
assertThat(result.getTotal()).isZero();
assertThat(result.getList()).isEmpty();
verifyNoInteractions(mapper);
}
@Test
void should_keep_drama_published_unknown_when_drama_missing() {
EngagementAdminLookup lookup = mock(EngagementAdminLookup.class);
UserFavoriteMapper mapper = mock(UserFavoriteMapper.class);
when(lookup.page(any())).thenReturn(new Page<>(1, 10));
UserFavorite row = new UserFavorite();
row.setId(1L);
row.setDeviceId("guest-device");
row.setDramaId(2L);
Page<UserFavorite> page = new Page<>(1, 10);
page.setTotal(1);
page.setRecords(List.of(row));
when(mapper.selectPage(any(Page.class), any(Wrapper.class))).thenReturn(page);
when(lookup.users(any())).thenReturn(Map.of());
when(lookup.dramas(any())).thenReturn(Map.of());
var result = new FavoriteAdminService(mapper, lookup).page(new FavoriteQueryReqVO());
assertThat(result.getList().get(0).getAnonymous()).isTrue();
assertThat(result.getList().get(0).getDeviceId()).isEqualTo("guest-device");
// 短剧已不存在时不能误报成「未发布」
assertThat(result.getList().get(0).getDramaPublished()).isNull();
}
@Test
void should_preserve_guest_drama_like_without_episode() {
EngagementAdminLookup lookup = mock(EngagementAdminLookup.class);
ContentLikeMapper mapper = mock(ContentLikeMapper.class);
when(lookup.page(any())).thenReturn(new Page<>(1, 10));
ContentLike row = new ContentLike();
row.setId(1L);
row.setTargetType("DRAMA");
row.setTargetId(2L);
row.setDeviceId("guest-device");
Page<ContentLike> page = new Page<>(1, 10);
page.setTotal(1);
page.setRecords(List.of(row));
when(mapper.selectPage(any(Page.class), any(Wrapper.class))).thenReturn(page);
when(lookup.users(any())).thenReturn(Map.of());
when(lookup.dramas(any())).thenReturn(Map.of());
var service = new LikeAdminService(mapper, mock(ContentCommentMapper.class),
mock(DramaEpisodeMapper.class), lookup);
var result = service.page(new LikeQueryReqVO());
assertThat(result.getList().get(0).getAnonymous()).isTrue();
assertThat(result.getList().get(0).getDramaId()).isEqualTo(2L);
assertThat(result.getList().get(0).getEpisodeId()).isNull();
}
}
@@ -199,9 +199,10 @@ class WatchHistoryAdminServiceTest {
}
private void stubBatchData() {
given(appUserMapper.selectBatchIds(any())).willReturn(List.of(user(7L, "Alice")));
given(dramaMapper.selectBatchIds(any())).willReturn(List.of(drama(5L, "Signal")));
given(dramaEpisodeMapper.selectBatchIds(any())).willReturn(List.of(episode(11L)));
// 装配阶段按需列查询,而不是 selectBatchIds,桩法要跟着改
given(appUserMapper.selectList(any())).willReturn(List.of(user(7L, "Alice")));
given(dramaMapper.selectList(any())).willReturn(List.of(drama(5L, "Signal")));
given(dramaEpisodeMapper.selectList(any())).willReturn(List.of(episode(11L)));
}
@SuppressWarnings("unchecked")
@@ -30,7 +30,9 @@ public final class ControllerContractSupport {
/** 组装一个只含指定控制器与全局异常处理的 MockMvc。 */
@SuppressWarnings("unchecked")
public static MockMvc mockMvcOf(Object controller) {
AccessDeniedRecorder recorder = new AccessDeniedRecorder(Mockito.mock(ObjectProvider.class));
AccessDeniedRecorder recorder = new AccessDeniedRecorder(
Mockito.mock(ObjectProvider.class),
Mockito.mock(ObjectProvider.class));
return MockMvcBuilders.standaloneSetup(controller)
.setControllerAdvice(new GlobalExceptionHandler(recorder))
// 与 Spring Boot 的 Jackson 配置对齐:时间按 ISO 字符串输出而不是数组,
@@ -24,6 +24,7 @@ import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.util.StringUtils;
/** 客户端账号:验证码、注册、登录、刷新、登出。 */
@Tag(name = "客户端-账号")
@@ -34,13 +35,14 @@ public class AppAuthController {
private final AppOtpService otpService;
private final AppAuthService authService;
private final WebUtils webUtils;
/** 发送验证码。频控在服务端,客户端不需要自己限流。 */
@Operation(summary = "发送验证码(频控在服务端)")
@RateLimit(limit = 20, windowSeconds = 60, key = "otp")
@PostMapping("/otp")
public Result<Void> sendOtp(@Valid @RequestBody OtpSendReqVO req, HttpServletRequest request) {
otpService.send(req, WebUtils.getClientIp(request));
otpService.send(req, webUtils.clientIp(request));
return Result.ok();
}
@@ -48,7 +50,8 @@ public class AppAuthController {
@PostMapping("/register")
public Result<TokenRespVO> register(@Valid @RequestBody RegisterReqVO req,
HttpServletRequest request) {
return Result.ok(authService.register(req, WebUtils.getClientIp(request)));
applyHeaderDeviceId(req, request);
return Result.ok(authService.register(req, webUtils.clientIp(request)));
}
@Operation(summary = "登录(验证码或密码)")
@@ -56,13 +59,19 @@ public class AppAuthController {
@PostMapping("/login")
public Result<TokenRespVO> login(@Valid @RequestBody LoginReqVO req,
HttpServletRequest request) {
return Result.ok(authService.login(req, WebUtils.getClientIp(request)));
applyHeaderDeviceId(req, request);
return Result.ok(authService.login(req, webUtils.clientIp(request)));
}
@Operation(summary = "Google 登录(未配置 clientId 时返回明确错误)")
@PostMapping("/google")
public Result<TokenRespVO> google(@Valid @RequestBody GoogleLoginReqVO req) {
return Result.ok(authService.googleLogin(req.getIdToken()));
public Result<TokenRespVO> google(@Valid @RequestBody GoogleLoginReqVO req,
HttpServletRequest request) {
String headerDeviceId = request.getHeader("X-Device-Id");
String deviceId = StringUtils.hasText(headerDeviceId)
? headerDeviceId : req.getDeviceId();
return Result.ok(authService.googleLogin(req.getIdToken(), deviceId,
webUtils.clientIp(request)));
}
@Operation(summary = "刷新令牌(旧 refresh token 立即失效)")
@@ -78,4 +87,16 @@ public class AppAuthController {
authService.logout(SecurityUtils.getUserIdOrNull(), req == null ? new LogoutReqVO() : req);
return Result.ok();
}
private void applyHeaderDeviceId(RegisterReqVO req, HttpServletRequest request) {
if (!StringUtils.hasText(req.getDeviceId())) {
req.setDeviceId(request.getHeader("X-Device-Id"));
}
}
private void applyHeaderDeviceId(LoginReqVO req, HttpServletRequest request) {
if (!StringUtils.hasText(req.getDeviceId())) {
req.setDeviceId(request.getHeader("X-Device-Id"));
}
}
}
@@ -6,6 +6,9 @@ import com.fruitshorts.common.pojo.CursorResult;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.EpisodeService;
import com.fruitshorts.common.pojo.Result;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.framework.interaction.InteractionIdentityResolver;
import com.fruitshorts.app.service.AnonymousInteractionRateLimiter;
import com.fruitshorts.framework.security.SecurityUtils;
import com.fruitshorts.framework.web.WebUtils;
import com.fruitshorts.interaction.service.CommentService;
@@ -33,7 +36,7 @@ import java.util.List;
*
* <p>列表走游标翻页:`cursor` 传上一页最后一条的 id,返回 `nextCursor` 与 `hasMore`。
* 不传 `rootId` 时返回顶层评论,传了则返回该楼层的回复。
* 列表允许匿名浏览;发布与删除需要登录。
* 列表允许匿名浏览;发布与删除支持登录用户或设备身份。
*/
@Tag(name = "客户端-评论")
@RestController
@@ -43,13 +46,21 @@ public class AppCommentController {
private final CommentService commentService;
private final EpisodeService episodeService;
private final InteractionIdentityResolver identityResolver;
private final AnonymousInteractionRateLimiter anonymousRateLimiter;
private final WebUtils webUtils;
@Operation(summary = "发布评论")
@PostMapping
public Result<CommentRespVO> publish(@Valid @RequestBody CommentPublishReqVO req,
HttpServletRequest request) {
return Result.ok(commentService.publish(SecurityUtils.getAppUserId(), req,
WebUtils.getClientIp(request), request.getHeader("X-Device-Id")));
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
req.getDeviceId());
anonymousRateLimiter.check(identity, "comment");
return Result.ok(commentService.publish(
identity, req, webUtils.clientIp(request)));
}
@Operation(summary = "评论列表(游标翻页,可匿名浏览)")
@@ -58,14 +69,18 @@ public class AppCommentController {
@RequestParam(required = false) Long episodeId,
@RequestParam(required = false) Long rootId,
@RequestParam(required = false) Long cursor,
@RequestParam(required = false) Integer size) {
@RequestParam(required = false) Integer size,
HttpServletRequest request) {
CommentQueryReqVO query = new CommentQueryReqVO();
query.setDramaId(dramaId);
query.setEpisodeId(episodeId);
query.setRootId(rootId);
query.setCursor(cursor);
query.setSize(size);
return Result.ok(commentService.page(SecurityUtils.getUserIdOrNull(), query));
return Result.ok(commentService.page(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
query));
}
@Operation(summary = "评论总数:传 episodeId 按集统计,仅 dramaId 时全剧统计(可匿名访问)")
@@ -85,8 +100,13 @@ public class AppCommentController {
@Operation(summary = "删除自己的评论")
@DeleteMapping("/{id}")
public Result<Void> delete(@PathVariable Long id) {
commentService.delete(SecurityUtils.getAppUserId(), id);
public Result<Void> delete(@PathVariable Long id, HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
"");
anonymousRateLimiter.check(identity, "comment");
commentService.delete(identity, id);
return Result.ok();
}
}
@@ -3,7 +3,9 @@
package com.fruitshorts.app.controller;
import com.fruitshorts.app.service.AppCatalogService;
import com.fruitshorts.app.service.AppEpisodeStatsService;
import com.fruitshorts.app.vo.AppDramaDetailVO;
import com.fruitshorts.app.vo.AppEpisodeStatsVO;
import com.fruitshorts.common.pojo.Result;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.EpisodeService;
@@ -34,6 +36,8 @@ public class AppDramaController {
private final PlaybackService playbackService;
private final EpisodeService episodeService;
private final CommentService commentService;
private final AppEpisodeStatsService appEpisodeStatsService;
private final WebUtils webUtils;
/** 详情:包含剧集列表,客户端进详情页一次请求即可渲染 */
@Operation(summary = "短剧详情(含剧集列表,进详情页一次请求即可渲染)")
@@ -53,10 +57,21 @@ public class AppDramaController {
SecurityUtils.getUserIdOrNull(),
id,
request.getHeader("X-Device-Id"),
WebUtils.getClientIp(request));
webUtils.clientIp(request));
DramaEpisode episode = episodeService.getExisting(id);
vo.setLikeCount(episode.getLikeCount() == null ? 0L : episode.getLikeCount());
vo.setCommentCount(commentService.countByEpisodes(List.of(id)).getOrDefault(id, 0L));
return Result.ok(vo);
}
/** 轻量计数快照:只读互动状态,不签发播放地址、不写票据缓存。 */
@Operation(summary = "播放页轻量计数与当前用户互动状态")
@GetMapping("/episodes/{id}/stats")
public Result<AppEpisodeStatsVO> stats(
@PathVariable Long id, HttpServletRequest request) {
return Result.ok(appEpisodeStatsService.get(
id,
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id")));
}
}
@@ -25,11 +25,12 @@ import org.springframework.web.bind.annotation.RestController;
public class AppEventController {
private final AppEventService eventService;
private final WebUtils webUtils;
@Operation(summary = "埋点事件批量上报")
@PostMapping("/events")
public Result<Void> report(@Valid @RequestBody AppEventReqVO req, HttpServletRequest request) {
eventService.save(req, SecurityUtils.getUserIdOrNull(), WebUtils.getClientIp(request));
eventService.save(req, SecurityUtils.getUserIdOrNull(), webUtils.clientIp(request));
return Result.ok();
}
}
@@ -6,6 +6,9 @@ import com.fruitshorts.app.service.AppCatalogService;
import com.fruitshorts.app.vo.AppDramaCardVO;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.pojo.Result;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.framework.interaction.InteractionIdentityResolver;
import com.fruitshorts.app.service.AnonymousInteractionRateLimiter;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.framework.security.SecurityUtils;
import com.fruitshorts.playback.service.FavoriteService;
@@ -14,6 +17,7 @@ import com.fruitshorts.playback.vo.FavoriteToggleReqVO;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
@@ -33,12 +37,21 @@ public class AppFavoriteController {
private final FavoriteService favoriteService;
private final AppCatalogService appCatalogService;
private final InteractionIdentityResolver identityResolver;
private final AnonymousInteractionRateLimiter anonymousRateLimiter;
/** 切换收藏,返回切换后的状态:true 为已收藏。 */
@Operation(summary = "切换收藏(返回切换后的状态)")
@PostMapping("/toggle")
public Result<Boolean> toggle(@Valid @RequestBody FavoriteToggleReqVO req) {
return Result.ok(favoriteService.toggle(SecurityUtils.getAppUserId(), req.getDramaId()));
public Result<Boolean> toggle(
@Valid @RequestBody FavoriteToggleReqVO req,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
req.getDeviceId());
anonymousRateLimiter.check(identity, "favorite");
return Result.ok(favoriteService.toggle(identity, req.getDramaId()));
}
/** 收藏列表,卡片结构与首页一致。 */
@@ -46,17 +59,46 @@ public class AppFavoriteController {
@GetMapping
public Result<PageResult<AppDramaCardVO>> list(
@RequestParam(defaultValue = "1") int pageIndex,
@RequestParam(defaultValue = "20") int pageSize) {
@RequestParam(defaultValue = "20") int pageSize,
@RequestParam(required = false) String deviceId,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
deviceId);
PageResult<Drama> page = favoriteService.page(
SecurityUtils.getAppUserId(), pageIndex, pageSize);
identity, pageIndex, pageSize);
return Result.ok(new PageResult<>(
page.getTotal(), appCatalogService.toCards(page.getList())));
}
@Operation(summary = "取消收藏")
@DeleteMapping("/{dramaId}")
public Result<Void> remove(@PathVariable Long dramaId) {
favoriteService.remove(SecurityUtils.getAppUserId(), dramaId);
public Result<Void> remove(@PathVariable Long dramaId, HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
"");
anonymousRateLimiter.check(identity, "favorite");
if (identity.authenticated()) {
favoriteService.remove(identity.userId(), dramaId);
} else {
favoriteService.removeDevice(identity.deviceId(), dramaId);
}
return Result.ok();
}
/** 批量取消收藏,返回实际删除的关系数。 */
@Operation(summary = "批量取消收藏")
@PostMapping("/remove")
public Result<Integer> removeBatch(
@Valid @RequestBody FavoriteRemoveReqVO req,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
req.getDeviceId());
anonymousRateLimiter.check(identity, "favorite");
return Result.ok(favoriteService.removeBatch(identity, req.getDramaIds()));
}
}
@@ -3,7 +3,10 @@
package com.fruitshorts.app.controller;
import com.fruitshorts.common.pojo.Result;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.framework.interaction.InteractionIdentityResolver;
import com.fruitshorts.framework.security.SecurityUtils;
import com.fruitshorts.app.service.AnonymousInteractionRateLimiter;
import com.fruitshorts.interaction.service.LikeService;
import com.fruitshorts.interaction.service.ReportService;
import com.fruitshorts.interaction.vo.LikeToggleReqVO;
@@ -12,6 +15,7 @@ import com.fruitshorts.interaction.vo.ReportCreateReqVO;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
@@ -27,12 +31,21 @@ public class AppInteractionController {
private final LikeService likeService;
private final ReportService reportService;
private final InteractionIdentityResolver identityResolver;
private final AnonymousInteractionRateLimiter anonymousRateLimiter;
/** 点赞/取消点赞,返回切换后的状态与最新计数。 */
@Operation(summary = "点赞 / 取消点赞")
@PostMapping("/likes/toggle")
public Result<LikeToggleRespVO> toggleLike(@Valid @RequestBody LikeToggleReqVO req) {
return Result.ok(likeService.toggle(SecurityUtils.getAppUserId(), req));
public Result<LikeToggleRespVO> toggleLike(
@Valid @RequestBody LikeToggleReqVO req,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
req.getDeviceId());
anonymousRateLimiter.check(identity, "like");
return Result.ok(likeService.toggle(identity, req));
}
/** 提交举报:同一用户对同一目标只保留一条待处理工单。 */
@@ -4,6 +4,9 @@ package com.fruitshorts.app.controller;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.pojo.Result;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.app.service.AnonymousInteractionRateLimiter;
import com.fruitshorts.framework.interaction.InteractionIdentityResolver;
import com.fruitshorts.framework.security.SecurityUtils;
import com.fruitshorts.playback.service.WatchProgressService;
import com.fruitshorts.playback.vo.ProgressRespVO;
@@ -11,6 +14,7 @@ import com.fruitshorts.playback.vo.ProgressSaveReqVO;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
@@ -29,19 +33,34 @@ import org.springframework.web.bind.annotation.RestController;
public class AppWatchController {
private final WatchProgressService progressService;
private final InteractionIdentityResolver identityResolver;
private final AnonymousInteractionRateLimiter anonymousRateLimiter;
/** 上报进度:客户端每 10 秒一次,切集时立即上报一次。 */
@Operation(summary = "上报观看进度(客户端每 10 秒一次)")
@PostMapping("/progress")
public Result<ProgressRespVO> save(@Valid @RequestBody ProgressSaveReqVO req) {
return Result.ok(progressService.save(SecurityUtils.getAppUserId(), req));
public Result<ProgressRespVO> save(@Valid @RequestBody ProgressSaveReqVO req,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
req.getDeviceId());
anonymousRateLimiter.check(identity, "watch");
return Result.ok(progressService.save(identity, req));
}
/** 某部剧的续看位置,进详情页时用它决定"续看第 N 集"。 */
@Operation(summary = "某部剧的续看位置")
@GetMapping("/progress")
public Result<ProgressRespVO> ofDrama(@RequestParam Long dramaId) {
return Result.ok(progressService.ofDrama(SecurityUtils.getAppUserId(), dramaId));
public Result<ProgressRespVO> ofDrama(
@RequestParam Long dramaId,
@RequestParam(required = false) String deviceId,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
deviceId);
return Result.ok(progressService.ofDrama(identity, dramaId));
}
/** 最近观看分页,契约与 /app/favorites 一致:pageIndex/pageSize。 */
@@ -49,22 +68,43 @@ public class AppWatchController {
@GetMapping("/recent")
public Result<PageResult<ProgressRespVO>> recent(
@RequestParam(defaultValue = "1") int pageIndex,
@RequestParam(defaultValue = "20") int pageSize) {
@RequestParam(defaultValue = "20") int pageSize,
@RequestParam(required = false) String deviceId,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
deviceId);
return Result.ok(progressService.recentPage(
SecurityUtils.getAppUserId(), pageIndex, pageSize));
identity, pageIndex, pageSize));
}
@Operation(summary = "删除某部剧的观看记录")
@DeleteMapping("/progress/{dramaId}")
public Result<Void> remove(@PathVariable Long dramaId) {
progressService.remove(SecurityUtils.getAppUserId(), dramaId);
public Result<Void> remove(
@PathVariable Long dramaId,
@RequestParam(required = false) String deviceId,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
deviceId);
anonymousRateLimiter.check(identity, "watch");
progressService.remove(identity, dramaId);
return Result.ok();
}
@Operation(summary = "清空观看记录")
@DeleteMapping("/progress")
public Result<Void> clear() {
progressService.clear(SecurityUtils.getAppUserId());
public Result<Void> clear(
@RequestParam(required = false) String deviceId,
HttpServletRequest request) {
InteractionIdentity identity = identityResolver.resolve(
SecurityUtils.getUserIdOrNull(),
request.getHeader("X-Device-Id"),
deviceId);
anonymousRateLimiter.check(identity, "watch");
progressService.clear(identity);
return Result.ok();
}
}
@@ -0,0 +1,36 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.framework.redis.RedisService;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import java.time.Duration;
/** 未登录非付费互动的设备级固定窗口限流。 */
@Component
@RequiredArgsConstructor
public class AnonymousInteractionRateLimiter {
private final RedisService redisService;
@Value("${fruitshorts.interaction.anonymous-write-per-minute:30}")
private int writesPerMinute = 30;
public void check(InteractionIdentity identity, String action) {
if (identity.authenticated()) {
return;
}
long count = redisService.increment(
"app:anonymous-interaction:" + action + ":" + identity.deviceId(),
Duration.ofMinutes(1));
if (count > writesPerMinute) {
throw new ServiceException(ErrorCode.TOO_MANY_REQUESTS, "操作过于频繁,请稍后再试");
}
}
}
@@ -0,0 +1,92 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.app.vo.AppEpisodeStatsVO;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.DramaService;
import com.fruitshorts.content.service.EpisodeService;
import com.fruitshorts.interaction.service.CommentService;
import com.fruitshorts.interaction.service.LikeService;
import com.fruitshorts.playback.service.FavoriteService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
/** 播放页只读计数快照;固定维度批量查询,不签发播放地址或票据。 */
@Service
@RequiredArgsConstructor
public class AppEpisodeStatsService {
private final EpisodeService episodeService;
private final DramaService dramaService;
private final CommentService commentService;
private final LikeService likeService;
private final FavoriteService favoriteService;
public AppEpisodeStatsVO get(Long episodeId, Long userId, String deviceId) {
Map<Long, AppEpisodeStatsVO> stats = batch(List.of(episodeId), userId, deviceId);
AppEpisodeStatsVO result = stats.get(episodeId);
if (result == null) {
throw new ServiceException(ErrorCode.DATA_NOT_FOUND, "剧集不存在或未上线");
}
return result;
}
public AppEpisodeStatsVO get(Long episodeId, Long userId) {
return get(episodeId, userId, "");
}
/** 每类数据只查一次,供后续多集统计复用。 */
public Map<Long, AppEpisodeStatsVO> batch(Collection<Long> episodeIds, Long userId) {
return batch(episodeIds, userId, "");
}
public Map<Long, AppEpisodeStatsVO> batch(
Collection<Long> episodeIds, Long userId, String deviceId) {
if (episodeIds == null || episodeIds.isEmpty()) {
return Map.of();
}
List<DramaEpisode> episodes = episodeService.listPublishedByIds(episodeIds);
if (episodes.isEmpty()) {
return Map.of();
}
List<Long> normalizedIds = episodes.stream().map(DramaEpisode::getId).toList();
List<Long> dramaIds = episodes.stream().map(DramaEpisode::getDramaId).distinct().toList();
Map<Long, Long> commentCounts = commentService.countByEpisodes(normalizedIds);
Map<Long, Drama> dramas = new HashMap<>();
for (Drama drama : dramaService.listPublishedByIds(dramaIds)) {
dramas.put(drama.getId(), drama);
}
Set<Long> likedIds = likeService.likedTargetIds(
userId, deviceId, LikeService.TYPE_EPISODE, normalizedIds);
Set<Long> favoriteDramaIds = favoriteService.favoriteDramaIds(
userId, deviceId, dramaIds);
Map<Long, AppEpisodeStatsVO> result = new HashMap<>();
for (DramaEpisode episode : episodes) {
Drama drama = dramas.get(episode.getDramaId());
if (drama == null) {
continue;
}
AppEpisodeStatsVO vo = new AppEpisodeStatsVO();
vo.setEpisodeId(episode.getId());
vo.setDramaId(episode.getDramaId());
vo.setLikeCount(episode.getLikeCount() == null ? 0L : episode.getLikeCount());
vo.setCommentCount(commentCounts.getOrDefault(episode.getId(), 0L));
vo.setFavoriteCount(drama.getFavoriteCount() == null ? 0L : drama.getFavoriteCount());
vo.setLiked(likedIds.contains(episode.getId()));
vo.setFavorited(favoriteDramaIds.contains(episode.getDramaId()));
result.put(episode.getId(), vo);
}
return result;
}
}
@@ -0,0 +1,24 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.user.event.AppInteractionMergeRequested;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import org.springframework.transaction.event.TransactionPhase;
import org.springframework.transaction.event.TransactionalEventListener;
/** 在登录事务提交前执行合并,任一步失败都会整体回滚登录。 */
@Component
@RequiredArgsConstructor
public class InteractionAccountMergeListener {
private final InteractionAccountMergeService mergeService;
@TransactionalEventListener(
phase = TransactionPhase.BEFORE_COMMIT,
fallbackExecution = false)
public void onMergeRequested(AppInteractionMergeRequested event) {
mergeService.merge(event.userId(), event.deviceId());
}
}
@@ -0,0 +1,54 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.interaction.service.CommentService;
import com.fruitshorts.interaction.service.LikeService;
import com.fruitshorts.playback.service.FavoriteService;
import com.fruitshorts.playback.service.WatchProgressService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
/** 登录后的设备互动归并;三项迁移共享一个数据库事务。 */
@Slf4j
@Service
@RequiredArgsConstructor
public class InteractionAccountMergeService {
private final LikeService likeService;
private final FavoriteService favoriteService;
private final CommentService commentService;
private final WatchProgressService watchProgressService;
@Transactional(rollbackFor = Exception.class)
public void merge(Long userId, String deviceId) {
LikeService.MergeResult likes = likeService.mergeDeviceIntoUser(userId, deviceId);
FavoriteService.MergeResult favorites =
favoriteService.mergeDeviceIntoUser(userId, deviceId);
CommentService.MergeResult comments =
commentService.mergeDeviceIntoUser(userId, deviceId);
WatchProgressService.MergeResult watch =
watchProgressService.mergeDeviceIntoUser(userId, deviceId);
log.info(
"interaction-merge userId={} deviceId={} "
+ "likeMigrated={} likeDuplicates={} "
+ "favoriteMigrated={} favoriteDuplicates={} "
+ "commentMigrated={} commentDuplicates={} "
+ "watchProgressMigrated={} watchProgressIgnored={} "
+ "watchHistoryMigrated={} watchHistoryIgnored={}",
userId,
deviceId,
likes.migrated(),
likes.duplicates(),
favorites.migrated(),
favorites.duplicates(),
comments.migrated(),
comments.duplicates(),
watch.progressMigrated(),
watch.progressIgnored(),
watch.historyMigrated(),
watch.historyIgnored());
}
}
@@ -0,0 +1,18 @@
// Author: fruit.com.cn
package com.fruitshorts.app.vo;
import lombok.Data;
/** 播放页轻量计数快照,不触发票据签发。 */
@Data
public class AppEpisodeStatsVO {
private Long episodeId;
private Long dramaId;
private Long likeCount;
private Long commentCount;
private Long favoriteCount;
private Boolean liked;
private Boolean favorited;
}
@@ -11,4 +11,7 @@ public class GoogleLoginReqVO {
@NotBlank(message = "缺少 Google idToken")
private String idToken;
/** 旧客户端未统一发送请求头时,仍可用于合并设备互动。 */
private String deviceId;
}
@@ -0,0 +1,66 @@
// Author: fruit.com.cn
package com.fruitshorts.app.web;
import com.fruitshorts.user.service.DeviceProfileService;
import jakarta.servlet.FilterChain;
import jakarta.servlet.ServletException;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.filter.OncePerRequestFilter;
import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* 收集客户端上报的平台(iOS / Android)。
*
* <p>客户端在请求头带 `X-Client-Platform`,这里按 device_id 建档,
* 后台列表据此区分两端;游客请求同样入库,这是与旧会话方案的关键差异。
* 同一设备 60 秒内只写一次,避免每个请求都打库。
*/
@Slf4j
@Component
@RequiredArgsConstructor
public class DeviceProfileFilter extends OncePerRequestFilter {
private static final long THROTTLE_MS = 60_000L;
private static final int MAX_TRACKED_DEVICES = 20_000;
private final DeviceProfileService deviceProfileService;
private final Map<String, Long> lastWriteAt = new ConcurrentHashMap<>();
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response,
FilterChain filterChain) throws ServletException, IOException {
try {
collect(request);
} catch (RuntimeException ex) {
// 画像只服务于后台展示,任何异常都不能影响业务请求
log.warn("device-profile collect failed reason={}", ex.getMessage());
}
filterChain.doFilter(request, response);
}
private void collect(HttpServletRequest request) {
String deviceId = request.getHeader("X-Device-Id");
if (!StringUtils.hasText(deviceId)) {
return;
}
long now = System.currentTimeMillis();
Long last = lastWriteAt.get(deviceId);
if (last != null && now - last < THROTTLE_MS) {
return;
}
if (lastWriteAt.size() > MAX_TRACKED_DEVICES) {
lastWriteAt.clear();
}
lastWriteAt.put(deviceId, now);
deviceProfileService.touch(deviceId, request.getHeader("X-Client-Platform"));
}
}
@@ -23,6 +23,13 @@ fruitshorts:
cors:
# 生产只允许显式配置的来源;没配就等于空,浏览器侧会被拦住(比"*"安全)
allowed-origins: ${CORS_ALLOWED_ORIGINS:}
watch-progress:
# 观看进度写缓冲:POST 先写 Redis 立即返回,后台任务批量合并进 watch_progress。
# 消化能力 = batch-size / interval ≈ 5000 条/秒,覆盖 30 万 DAU 峰值 ~3700 条/秒上报。
# 关掉(false)即回退同步写库,接口契约不变。
buffer: true
flush-interval-ms: 2000
flush-batch-size: 10000
springdoc:
# 生产只留机器可读契约,关掉 UI(doc.html 不对外开放)
@@ -11,13 +11,22 @@ spring:
datasource:
driver-class-name: org.postgresql.Driver
# 地址必须外置:不同环境(本机 / 测试 / 生产)连的库不一样,
# 生产请用 DB_URL 注入(prod profile 会校验它非空)
url: ${DB_URL:jdbc:postgresql://127.0.0.1:5432/fruitshorts}
# 生产请用 DB_URL 注入(prod profile 会校验它非空)。
# reWriteBatchedInserts:pgjdbc 把批内多条 INSERT 重写成多值 INSERT,
# 观看进度缓冲落库依赖它才能跑满批量吞吐。
url: ${DB_URL:jdbc:postgresql://127.0.0.1:5432/fruitshorts?reWriteBatchedInserts=true}
username: ${DB_USER:fruitshorts}
password: ${DB_PASSWORD:fruitshorts}
jackson:
date-format: yyyy-MM-dd HH:mm:ss
time-zone: Asia/Shanghai
task:
scheduling:
# 默认调度池只有 1 条线程:观看进度落库(每 2s)会和其他定时任务互相排队,
# 落库一旦拖慢就会连带推迟对时延敏感的任务,必须给池子多几条线程。
pool:
size: 4
thread-name-prefix: fruitshorts-sched-
data:
redis:
host: ${REDIS_HOST:127.0.0.1}
@@ -47,6 +56,11 @@ springdoc:
path: /v3/api-docs
fruitshorts:
web:
# 可信代理 CIDR 列表(逗号分隔)。默认空 = 永不解析 XFF,直接用 request.getRemoteAddr()。
# 配了反代(Nginx/K8s ingress)时填上其 IP 段,才会从 XFF 取最右侧可信一跳。
# 生产若开启了 Tomcat RemoteIpValve(prod profile 已配),保持空即可。
trusted-proxies: ${FRUITSHORTS_TRUSTED_PROXIES:}
auth:
# 仅作 sys_config 缺失/非法时的兜底;运行时优先读取后台「系统配置」
register-max-per-ip-per-day: ${REGISTER_MAX_PER_IP_PER_DAY:5}
@@ -4,6 +4,7 @@ package com.fruitshorts.app.controller;
import com.fruitshorts.app.support.ControllerContractSupport;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.framework.web.WebUtils;
import com.fruitshorts.user.service.AppAuthService;
import com.fruitshorts.user.service.AppOtpService;
import com.fruitshorts.user.vo.TokenRespVO;
@@ -32,13 +33,15 @@ class AppAuthControllerContractTest {
private AppOtpService otpService;
private AppAuthService authService;
private WebUtils webUtils;
private MockMvc mockMvc;
@BeforeEach
void setUp() {
otpService = mock(AppOtpService.class);
authService = mock(AppAuthService.class);
mockMvc = ControllerContractSupport.mockMvcOf(new AppAuthController(otpService, authService));
webUtils = mock(WebUtils.class);
mockMvc = ControllerContractSupport.mockMvcOf(new AppAuthController(otpService, authService, webUtils));
}
@Test
@@ -2,6 +2,7 @@
package com.fruitshorts.app.controller;
import com.fruitshorts.app.service.AppCatalogService;
import com.fruitshorts.app.service.AppEpisodeStatsService;
import com.fruitshorts.app.support.ControllerContractSupport;
import com.fruitshorts.app.vo.AppBannerVO;
import com.fruitshorts.app.vo.AppCategoryVO;
@@ -49,7 +50,9 @@ class AppCatalogControllerContractTest {
catalogMvc = ControllerContractSupport.mockMvcOf(new AppCatalogController(catalogService));
dramaMvc = ControllerContractSupport.mockMvcOf(
new AppDramaController(catalogService, mock(PlaybackService.class),
mock(EpisodeService.class), mock(CommentService.class)));
mock(EpisodeService.class), mock(CommentService.class),
mock(AppEpisodeStatsService.class),
mock(com.fruitshorts.framework.web.WebUtils.class)));
}
@Test
@@ -6,6 +6,8 @@ import com.fruitshorts.app.support.ControllerContractSupport;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.EpisodeService;
import com.fruitshorts.interaction.service.CommentService;
import com.fruitshorts.framework.interaction.InteractionIdentityResolver;
import com.fruitshorts.app.service.AnonymousInteractionRateLimiter;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.test.web.servlet.MockMvc;
@@ -32,7 +34,12 @@ class AppCommentControllerContractTest {
commentService = mock(CommentService.class);
episodeService = mock(EpisodeService.class);
mockMvc = ControllerContractSupport.mockMvcOf(
new AppCommentController(commentService, episodeService));
new AppCommentController(
commentService,
episodeService,
mock(InteractionIdentityResolver.class),
mock(AnonymousInteractionRateLimiter.class),
mock(com.fruitshorts.framework.web.WebUtils.class)));
}
@Test
@@ -3,9 +3,21 @@
package com.fruitshorts.app.controller;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fruitshorts.app.service.AppEpisodeStatsService;
import com.fruitshorts.app.service.AppCatalogService;
import com.fruitshorts.app.vo.AppEpisodeStatsVO;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.pojo.CursorResult;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.interaction.service.LikeService;
import com.fruitshorts.playback.service.FavoriteService;
import com.fruitshorts.playback.service.PlaybackService;
import com.fruitshorts.playback.service.WatchProgressService;
import com.fruitshorts.playback.vo.ProgressRespVO;
import com.fruitshorts.content.service.EpisodeService;
import com.fruitshorts.framework.interaction.InteractionIdentityResolver;
import com.fruitshorts.app.service.AnonymousInteractionRateLimiter;
import com.fruitshorts.framework.log.AccessDeniedRecorder;
import com.fruitshorts.framework.security.JwtAuthenticationFilter;
import com.fruitshorts.framework.security.JwtTokenProvider;
@@ -34,8 +46,12 @@ import org.springframework.web.servlet.config.annotation.EnableWebMvc;
import java.util.List;
import static org.mockito.BDDMockito.given;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyList;
import static org.mockito.Mockito.mock;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.delete;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
@@ -52,12 +68,39 @@ class AppCommentCountAnonymousSecurityTest {
private CommentService commentService;
@Autowired
private AppCatalogService catalogService;
@Autowired
private AppEpisodeStatsService episodeStatsService;
@Autowired
private FavoriteService favoriteService;
@Autowired
private WatchProgressService watchProgressService;
@BeforeEach
void setUp() {
given(commentService.countByDrama(7L)).willReturn(9L);
given(catalogService.feed("ANIME", null, null, 20))
.willReturn(new CursorResult<>(List.of(), null, false));
AppEpisodeStatsVO stats = new AppEpisodeStatsVO();
stats.setEpisodeId(11L);
stats.setDramaId(7L);
stats.setLikeCount(12L);
stats.setCommentCount(3L);
stats.setFavoriteCount(8L);
stats.setLiked(true);
stats.setFavorited(false);
given(episodeStatsService.get(11L, null, "device-a")).willReturn(stats);
given(favoriteService.page(any(InteractionIdentity.class), any(Integer.class), any(Integer.class)))
.willReturn(new PageResult<>(0L, List.of()));
given(favoriteService.removeBatch(any(InteractionIdentity.class), anyList()))
.willReturn(1);
given(catalogService.toCards(List.of())).willReturn(List.of());
given(watchProgressService.recentPage(
any(InteractionIdentity.class), any(Integer.class), any(Integer.class)))
.willReturn(new PageResult<>(0L, List.of()));
given(watchProgressService.ofDrama(any(InteractionIdentity.class), any(Long.class)))
.willReturn(null);
given(watchProgressService.save(any(InteractionIdentity.class), any()))
.willReturn(new ProgressRespVO());
}
@Test
@@ -78,6 +121,93 @@ class AppCommentCountAnonymousSecurityTest {
.andExpect(jsonPath("$.data.list").isArray());
}
@Test
void should_allow_anonymous_non_paid_interactions_through_security_chain() throws Exception {
mockMvc.perform(post("/app/likes/toggle")
.header("X-Device-Id", "device-a")
.contentType("application/json")
.content("{\"targetType\":\"EPISODE\",\"targetId\":\"11\"}"))
.andExpect(status().isOk());
mockMvc.perform(post("/app/favorites/toggle")
.header("X-Device-Id", "device-a")
.contentType("application/json")
.content("{\"dramaId\":\"7\"}"))
.andExpect(status().isOk());
mockMvc.perform(post("/app/comments")
.header("X-Device-Id", "device-a")
.contentType("application/json")
.content("{\"dramaId\":\"7\",\"content\":\"hello\"}"))
.andExpect(status().isOk());
}
@Test
void should_allow_anonymous_episode_stats_through_security_chain() throws Exception {
mockMvc.perform(get("/app/episodes/11/stats")
.header("X-Device-Id", "device-a"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0))
.andExpect(jsonPath("$.data.episodeId").value(11))
.andExpect(jsonPath("$.data.liked").value(true))
.andExpect(jsonPath("$.data.favoriteCount").value(8));
}
@Test
void should_allow_anonymous_library_and_watch_operations() throws Exception {
mockMvc.perform(get("/app/favorites").header("X-Device-Id", "device-a"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0));
mockMvc.perform(get("/app/watch/recent").header("X-Device-Id", "device-a"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0));
mockMvc.perform(get("/app/watch/progress")
.header("X-Device-Id", "device-a")
.param("dramaId", "7"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0));
mockMvc.perform(post("/app/watch/progress")
.header("X-Device-Id", "device-a")
.contentType("application/json")
.content("{\"episodeId\":11,\"positionMs\":1000}"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0));
mockMvc.perform(delete("/app/watch/progress/7")
.header("X-Device-Id", "device-a"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0));
mockMvc.perform(delete("/app/watch/progress")
.header("X-Device-Id", "device-a"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0));
}
@Test
void should_allow_anonymous_favorite_batch_removal_through_security_chain()
throws Exception {
mockMvc.perform(post("/app/favorites/remove")
.header("X-Device-Id", "device-a")
.contentType("application/json")
.content("{\"dramaIds\":[\"7\"]}"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0))
.andExpect(jsonPath("$.data").value(1));
}
@Test
void should_keep_paid_and_entitlement_operations_login_required() throws Exception {
mockMvc.perform(post("/app/episodes/11/unlock"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(ErrorCode.UNAUTHORIZED.getCode()));
mockMvc.perform(get("/app/wallet"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(ErrorCode.UNAUTHORIZED.getCode()));
mockMvc.perform(post("/app/orders"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(ErrorCode.UNAUTHORIZED.getCode()));
mockMvc.perform(get("/app/orders"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(ErrorCode.UNAUTHORIZED.getCode()));
}
@Configuration
@EnableWebMvc
@EnableWebSecurity
@@ -90,8 +220,105 @@ class AppCommentCountAnonymousSecurityTest {
}
@Bean
AppCommentController appCommentController(CommentService commentService) {
return new AppCommentController(commentService, mock(EpisodeService.class));
EpisodeService episodeService() {
return mock(EpisodeService.class);
}
@Bean
AppCommentController appCommentController(
CommentService commentService,
EpisodeService episodeService,
InteractionIdentityResolver identityResolver,
AnonymousInteractionRateLimiter rateLimiter) {
return new AppCommentController(
commentService,
episodeService,
identityResolver,
rateLimiter,
mock(com.fruitshorts.framework.web.WebUtils.class));
}
@Bean
LikeService likeService() {
return mock(LikeService.class);
}
@Bean
FavoriteService favoriteService() {
return mock(FavoriteService.class);
}
@Bean
WatchProgressService watchProgressService() {
return mock(WatchProgressService.class);
}
@Bean
AppInteractionController appInteractionController(
LikeService likeService,
InteractionIdentityResolver identityResolver,
AnonymousInteractionRateLimiter rateLimiter) {
return new AppInteractionController(
likeService,
mock(com.fruitshorts.interaction.service.ReportService.class),
identityResolver,
rateLimiter);
}
@Bean
AppFavoriteController appFavoriteController(
FavoriteService favoriteService,
InteractionIdentityResolver identityResolver,
AnonymousInteractionRateLimiter rateLimiter) {
return new AppFavoriteController(
favoriteService,
mock(AppCatalogService.class),
identityResolver,
rateLimiter);
}
@Bean
AppWatchController appWatchController(
WatchProgressService watchProgressService,
InteractionIdentityResolver identityResolver,
AnonymousInteractionRateLimiter rateLimiter) {
return new AppWatchController(
watchProgressService,
identityResolver,
rateLimiter);
}
@Bean
AppEpisodeStatsService appEpisodeStatsService() {
return mock(AppEpisodeStatsService.class);
}
@Bean
AppDramaController appDramaController(
AppCatalogService appCatalogService,
EpisodeService episodeService,
CommentService commentService,
AppEpisodeStatsService appEpisodeStatsService) {
return new AppDramaController(
appCatalogService,
mock(PlaybackService.class),
episodeService,
commentService,
appEpisodeStatsService,
mock(com.fruitshorts.framework.web.WebUtils.class));
}
@Bean
InteractionIdentityResolver identityResolver() {
return new InteractionIdentityResolver();
}
@Bean
AnonymousInteractionRateLimiter anonymousInteractionRateLimiter() {
AnonymousInteractionRateLimiter limiter =
mock(AnonymousInteractionRateLimiter.class);
Mockito.doNothing().when(limiter).check(Mockito.any(), Mockito.anyString());
return limiter;
}
@Bean
@@ -115,7 +342,7 @@ class AppCommentCountAnonymousSecurityTest {
RestAuthEntryPoint restAuthEntryPoint(ObjectMapper objectMapper) {
return new RestAuthEntryPoint(
objectMapper,
new AccessDeniedRecorder(mock(ObjectProvider.class)));
new AccessDeniedRecorder(mock(ObjectProvider.class), mock(ObjectProvider.class)));
}
@Bean
@@ -132,4 +359,5 @@ class AppCommentCountAnonymousSecurityTest {
.build();
}
}
}
@@ -6,8 +6,11 @@ import com.fruitshorts.app.service.AppCatalogService;
import com.fruitshorts.app.support.ControllerContractSupport;
import com.fruitshorts.app.vo.AppDramaCardVO;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.framework.security.LoginUser;
import com.fruitshorts.framework.interaction.InteractionIdentityResolver;
import com.fruitshorts.app.service.AnonymousInteractionRateLimiter;
import com.fruitshorts.playback.service.FavoriteService;
import com.fruitshorts.playback.service.WatchProgressService;
import com.fruitshorts.playback.vo.ProgressRespVO;
@@ -19,12 +22,16 @@ import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.test.web.servlet.MockMvc;
import java.time.LocalDateTime;
import java.util.Collections;
import java.util.List;
import static org.mockito.BDDMockito.given;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
@@ -44,10 +51,17 @@ class AppLibraryPagingContractTest {
favoriteService = mock(FavoriteService.class);
catalogService = mock(AppCatalogService.class);
progressService = mock(WatchProgressService.class);
InteractionIdentityResolver identityResolver = new InteractionIdentityResolver();
AnonymousInteractionRateLimiter rateLimiter =
mock(AnonymousInteractionRateLimiter.class);
favoriteMvc = ControllerContractSupport.mockMvcOf(
new AppFavoriteController(favoriteService, catalogService));
new AppFavoriteController(
favoriteService,
catalogService,
identityResolver,
rateLimiter));
watchMvc = ControllerContractSupport.mockMvcOf(
new AppWatchController(progressService));
new AppWatchController(progressService, identityResolver, rateLimiter));
SecurityContextHolder.getContext().setAuthentication(
new UsernamePasswordAuthenticationToken(
new LoginUser(USER_ID, "app-user", "APP_USER", List.of()),
@@ -66,9 +80,9 @@ class AppLibraryPagingContractTest {
Drama first = drama(101L, "收藏一");
Drama second = drama(102L, "收藏二");
Drama third = drama(103L, "收藏三");
given(favoriteService.page(USER_ID, 1, 20))
given(favoriteService.page(any(InteractionIdentity.class), eq(1), eq(20)))
.willReturn(new PageResult<>(21L, List.of(first, second)));
given(favoriteService.page(USER_ID, 2, 20))
given(favoriteService.page(any(InteractionIdentity.class), eq(2), eq(20)))
.willReturn(new PageResult<>(21L, List.of(third)));
given(catalogService.toCards(List.of(first, second)))
.willReturn(List.of(card(101L), card(102L)));
@@ -89,8 +103,8 @@ class AppLibraryPagingContractTest {
.andExpect(jsonPath("$.data.list.length()").value(1))
.andExpect(jsonPath("$.data.list[0].id").value("103"));
verify(favoriteService).page(USER_ID, 1, 20);
verify(favoriteService).page(USER_ID, 2, 20);
verify(favoriteService).page(any(InteractionIdentity.class), eq(1), eq(20));
verify(favoriteService).page(any(InteractionIdentity.class), eq(2), eq(20));
}
@Test
@@ -98,9 +112,9 @@ class AppLibraryPagingContractTest {
ProgressRespVO first = progress(12L, 88L, 6, "剧名一", 68);
ProgressRespVO second = progress(13L, 99L, 2, "剧名二", 20);
ProgressRespVO third = progress(14L, 100L, 1, "剧名三", 16);
given(progressService.recentPage(USER_ID, 1, 20))
given(progressService.recentPage(any(InteractionIdentity.class), eq(1), eq(20)))
.willReturn(new PageResult<>(23L, List.of(first, second)));
given(progressService.recentPage(USER_ID, 2, 20))
given(progressService.recentPage(any(InteractionIdentity.class), eq(2), eq(20)))
.willReturn(new PageResult<>(23L, List.of(third)));
watchMvc.perform(get("/app/watch/recent"))
@@ -120,14 +134,51 @@ class AppLibraryPagingContractTest {
.andExpect(jsonPath("$.data.list[0].dramaId").value("14"))
.andExpect(jsonPath("$.data.list[0].dramaTitle").value("剧名三"));
verify(progressService).recentPage(USER_ID, 1, 20);
verify(progressService).recentPage(USER_ID, 2, 20);
verify(progressService).recentPage(any(InteractionIdentity.class), eq(1), eq(20));
verify(progressService).recentPage(any(InteractionIdentity.class), eq(2), eq(20));
}
@Test
void should_remove_favorite_batch_for_signed_in_user() throws Exception {
given(favoriteService.removeBatch(any(InteractionIdentity.class), eq(List.of(7L, 8L))))
.willReturn(2);
favoriteMvc.perform(post("/app/favorites/remove")
.contentType("application/json")
.content("{\"dramaIds\":[\"7\",\"8\"]}"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(0))
.andExpect(jsonPath("$.data").value(2));
verify(favoriteService).removeBatch(any(InteractionIdentity.class), eq(List.of(7L, 8L)));
}
@Test
void should_reject_empty_favorite_batch() throws Exception {
favoriteMvc.perform(post("/app/favorites/remove")
.contentType("application/json")
.content("{\"dramaIds\":[]}"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(400))
.andExpect(jsonPath("$.message").value("dramaIds 不能为空"));
}
@Test
void should_reject_favorite_batch_over_limit() throws Exception {
String dramaIds = String.join(",", Collections.nCopies(201, "\"7\""));
favoriteMvc.perform(post("/app/favorites/remove")
.contentType("application/json")
.content("{\"dramaIds\":[" + dramaIds + "]}"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.code").value(400))
.andExpect(jsonPath("$.message").value("dramaIds 最多 200 条"));
}
@Test
void should_keep_total_and_return_empty_list_when_watch_page_is_out_of_range()
throws Exception {
given(progressService.recentPage(USER_ID, 9, 20))
given(progressService.recentPage(any(InteractionIdentity.class), eq(9), eq(20)))
.willReturn(new PageResult<>(23L, List.of()));
watchMvc.perform(get("/app/watch/recent").param("pageIndex", "9"))
@@ -140,7 +191,7 @@ class AppLibraryPagingContractTest {
@Test
void should_ignore_removed_limit_and_still_return_page_result() throws Exception {
// limit 已从契约移除:即使客户端误传,也应走分页并返回 PageResult。
given(progressService.recentPage(USER_ID, 1, 20))
given(progressService.recentPage(any(InteractionIdentity.class), eq(1), eq(20)))
.willReturn(new PageResult<>(3L, List.of(progress(12L, 88L, 6, "剧名一", 68))));
watchMvc.perform(get("/app/watch/recent").param("limit", "5"))
@@ -150,7 +201,7 @@ class AppLibraryPagingContractTest {
.andExpect(jsonPath("$.data.list.length()").value(1))
.andExpect(jsonPath("$.data.list[0].dramaTitle").value("剧名一"));
verify(progressService).recentPage(USER_ID, 1, 20);
verify(progressService).recentPage(any(InteractionIdentity.class), eq(1), eq(20));
}
private Drama drama(Long id, String title) {
@@ -2,6 +2,8 @@
package com.fruitshorts.app.controller;
import com.fruitshorts.app.service.AppCatalogService;
import com.fruitshorts.app.service.AppEpisodeStatsService;
import com.fruitshorts.app.vo.AppEpisodeStatsVO;
import com.fruitshorts.app.support.ControllerContractSupport;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.EpisodeService;
@@ -37,6 +39,7 @@ class AppPlaybackContractTest {
private PlaybackService playbackService;
private EpisodeService episodeService;
private CommentService commentService;
private AppEpisodeStatsService episodeStatsService;
private MockMvc mockMvc;
@BeforeEach
@@ -44,13 +47,15 @@ class AppPlaybackContractTest {
playbackService = mock(PlaybackService.class);
episodeService = mock(EpisodeService.class);
commentService = mock(CommentService.class);
episodeStatsService = mock(AppEpisodeStatsService.class);
DramaEpisode episode = new DramaEpisode();
episode.setLikeCount(88L);
given(episodeService.getExisting(any())).willReturn(episode);
given(commentService.countByEpisodes(any())).willReturn(Map.of());
mockMvc = ControllerContractSupport.mockMvcOf(
new AppDramaController(mock(AppCatalogService.class), playbackService,
episodeService, commentService));
episodeService, commentService, episodeStatsService,
mock(com.fruitshorts.framework.web.WebUtils.class)));
}
@Test
@@ -102,6 +107,32 @@ class AppPlaybackContractTest {
verify(playbackService).play(isNull(), eq(13L), eq("device-9"), any());
}
@Test
void should_return_episode_stats_without_play_ticket() throws Exception {
AppEpisodeStatsVO stats = new AppEpisodeStatsVO();
stats.setEpisodeId(11L);
stats.setDramaId(7L);
stats.setLikeCount(88L);
stats.setCommentCount(8L);
stats.setFavoriteCount(1800L);
stats.setLiked(true);
stats.setFavorited(false);
given(episodeStatsService.get(11L, null, null)).willReturn(stats);
mockMvc.perform(get("/app/episodes/11/stats"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.data.episodeId").value("11"))
.andExpect(jsonPath("$.data.dramaId").value("7"))
.andExpect(jsonPath("$.data.likeCount").value("88"))
.andExpect(jsonPath("$.data.commentCount").value("8"))
.andExpect(jsonPath("$.data.favoriteCount").value("1800"))
.andExpect(jsonPath("$.data.liked").value(true))
.andExpect(jsonPath("$.data.favorited").value(false));
verify(episodeStatsService).get(11L, null, null);
verify(playbackService, org.mockito.Mockito.never()).play(any(), any(), any(), any());
}
private PlayRespVO ticket(Long episodeId, boolean accessible, String accessType,
String playUrl, String thumbnailsUrl) {
PlayRespVO vo = new PlayRespVO();
@@ -0,0 +1,48 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.framework.redis.RedisService;
import org.junit.jupiter.api.Test;
import org.springframework.test.util.ReflectionTestUtils;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
class AnonymousInteractionRateLimiterTest {
@Test
void should_reject_anonymous_device_after_per_minute_limit() {
RedisService redisService = mock(RedisService.class);
given(redisService.increment(anyString(), any())).willReturn(31L);
AnonymousInteractionRateLimiter limiter =
new AnonymousInteractionRateLimiter(redisService);
ReflectionTestUtils.setField(limiter, "writesPerMinute", 30);
assertThatThrownBy(() -> limiter.check(
new InteractionIdentity(null, "device-a"),
"like"))
.isInstanceOfSatisfying(ServiceException.class, error ->
org.assertj.core.api.Assertions.assertThat(error.getCode())
.isEqualTo(ErrorCode.TOO_MANY_REQUESTS.getCode()));
}
@Test
void should_not_rate_limit_authenticated_user() {
RedisService redisService = mock(RedisService.class);
AnonymousInteractionRateLimiter limiter =
new AnonymousInteractionRateLimiter(redisService);
limiter.check(new InteractionIdentity(9L, ""), "comment");
verify(redisService, never()).increment(anyString(), any());
}
}
@@ -16,18 +16,22 @@ import com.fruitshorts.user.dal.mapper.AppUserMapper;
import com.fruitshorts.user.dal.mapper.AppUserOtpMapper;
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
import com.fruitshorts.user.enums.AppUserStatus;
import com.fruitshorts.user.event.AppInteractionMergeRequested;
import com.fruitshorts.user.service.AppAuthService;
import com.fruitshorts.user.service.AppOtpService;
import com.fruitshorts.user.service.CredentialHasher;
import com.fruitshorts.user.service.GoogleIdTokenVerifier;
import com.fruitshorts.user.service.GoogleUser;
import com.fruitshorts.user.service.IpRegionService;
import com.fruitshorts.user.service.OtpSender;
import com.fruitshorts.user.vo.LoginReqVO;
import com.fruitshorts.user.vo.RegisterReqVO;
import com.fruitshorts.user.vo.TokenRespVO;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.test.util.ReflectionTestUtils;
import java.time.Duration;
@@ -63,6 +67,8 @@ class AppAuthServiceTest {
private PasswordEncoder passwordEncoder;
private ConfigService configService;
private GoogleIdTokenVerifier googleVerifier;
private ApplicationEventPublisher eventPublisher;
private IpRegionService ipRegionService;
private AppAuthService authService;
@BeforeEach
@@ -78,11 +84,15 @@ class AppAuthServiceTest {
passwordEncoder = mock(PasswordEncoder.class);
configService = mock(ConfigService.class);
googleVerifier = mock(GoogleIdTokenVerifier.class);
eventPublisher = mock(ApplicationEventPublisher.class);
ipRegionService = mock(IpRegionService.class);
given(configService.get(anyString(), anyString()))
.willAnswer(invocation -> invocation.getArgument(1));
// 单测里默认解析不出国家,走客户端区域 / 印尼兜底
given(ipRegionService.resolveCountry(any())).willReturn("");
authService = new AppAuthService(userMapper, credentialMapper, sessionMapper, consentMapper,
deviceMapper, otpService, hasher, tokenProvider, passwordEncoder, configService,
googleVerifier);
googleVerifier, eventPublisher, ipRegionService);
}
@Test
@@ -98,6 +108,9 @@ class AppAuthServiceTest {
// 免码注册不等于"已验证":邮箱可达性没验证过,所以先记 0,等日后点验证链接再置 1
assertThat(saved.get().getVerified()).isZero();
verify(otpService, never()).consume(anyString(), anyString(), anyString(), any());
AppInteractionMergeRequested event = capturedMergeEvent();
assertThat(event.userId()).isEqualTo(1001L);
assertThat(event.deviceId()).isEqualTo("device-9");
}
@Test
@@ -251,6 +264,44 @@ class AppAuthServiceTest {
assertThat(resp.getAccessToken()).isEqualTo("access-1");
assertThat(resp.getUser().getUid()).isEqualTo("UAB12CD");
verify(sessionMapper).insert(any(AppUserSession.class));
AppInteractionMergeRequested event = capturedMergeEvent();
assertThat(event.userId()).isEqualTo(1001L);
assertThat(event.deviceId()).isEqualTo("device-9");
}
@Test
void should_sync_country_from_ip_on_login() {
stubLogin(credential("bcrypt-hash"), user(AppUserStatus.NORMAL));
given(passwordEncoder.matches("abcd1234", "bcrypt-hash")).willReturn(true);
given(ipRegionService.resolveCountry(any())).willReturn("TH");
authService.login(emailLogin("abcd1234"), "49.237.8.1");
ArgumentCaptor<AppUser> captor = ArgumentCaptor.forClass(AppUser.class);
verify(userMapper).updateById(captor.capture());
AppUser saved = captor.getValue();
assertThat(saved.getCountryCode()).isEqualTo("TH");
assertThat(saved.getRegionCode()).isEqualTo("TH");
assertThat(saved.getLanguage()).isEqualTo("en-US");
assertThat(saved.getTimezone()).isEqualTo("Asia/Bangkok");
assertThat(saved.getCurrency()).isEqualTo("THB");
}
@Test
void should_keep_user_customized_language_when_syncing_country() {
AppUser user = user(AppUserStatus.NORMAL);
user.setLanguage("zh-CN");
stubLogin(credential("bcrypt-hash"), user);
given(passwordEncoder.matches("abcd1234", "bcrypt-hash")).willReturn(true);
given(ipRegionService.resolveCountry(any())).willReturn("TH");
authService.login(emailLogin("abcd1234"), "49.237.8.1");
ArgumentCaptor<AppUser> captor = ArgumentCaptor.forClass(AppUser.class);
verify(userMapper).updateById(captor.capture());
assertThat(captor.getValue().getCountryCode()).isEqualTo("TH");
// 用户自己改过的语言不被 IP 覆盖
assertThat(captor.getValue().getLanguage()).isEqualTo("zh-CN");
}
@Test
@@ -307,11 +358,14 @@ class AppAuthServiceTest {
});
given(userMapper.selectById(any())).willAnswer(invocation -> inserted.get());
TokenRespVO resp = authService.googleLogin("google-id-token");
TokenRespVO resp = authService.googleLogin("google-id-token", "device-9", "203.0.113.9");
assertThat(resp.getAccessToken()).isEqualTo("access-1");
assertThat(resp.isNewUser()).isTrue();
verify(sessionMapper).insert(any(AppUserSession.class));
AppInteractionMergeRequested event = capturedMergeEvent();
assertThat(event.userId()).isEqualTo(1001L);
assertThat(event.deviceId()).isEqualTo("device-9");
}
@Test
@@ -319,7 +373,7 @@ class AppAuthServiceTest {
given(googleVerifier.verify(anyString()))
.willReturn(new GoogleUser("sub-1", "", "Alice", ""));
assertThatThrownBy(() -> authService.googleLogin("google-id-token"))
assertThatThrownBy(() -> authService.googleLogin("google-id-token", "device-9", "203.0.113.9"))
.isInstanceOf(ServiceException.class)
.hasMessage("Google 账号未返回邮箱,无法登录");
}
@@ -353,6 +407,13 @@ class AppAuthServiceTest {
return saved;
}
private AppInteractionMergeRequested capturedMergeEvent() {
ArgumentCaptor<AppInteractionMergeRequested> event =
ArgumentCaptor.forClass(AppInteractionMergeRequested.class);
verify(eventPublisher).publishEvent(event.capture());
return event.getValue();
}
private void stubLogin(AppUserCredential credential, AppUser user) {
given(credentialMapper.selectOne(any())).willReturn(credential);
given(userMapper.selectById(any())).willReturn(user);
@@ -371,7 +432,7 @@ class AppAuthServiceTest {
AppOtpService realOtpService = new AppOtpService(otpMapper, mock(OtpSender.class), otpHasher);
return new AppAuthService(userMapper, credentialMapper, sessionMapper, consentMapper,
deviceMapper, realOtpService, hasher, tokenProvider, passwordEncoder, configService,
googleVerifier);
googleVerifier, eventPublisher, ipRegionService);
}
private void setRegisterLimits(int perIp, int perDevice) {
@@ -0,0 +1,67 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.app.vo.AppEpisodeStatsVO;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.DramaService;
import com.fruitshorts.content.service.EpisodeService;
import com.fruitshorts.interaction.service.CommentService;
import com.fruitshorts.interaction.service.LikeService;
import com.fruitshorts.playback.service.FavoriteService;
import org.junit.jupiter.api.Test;
import java.util.List;
import java.util.Map;
import java.util.Set;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
class AppEpisodeStatsServiceTest {
@Test
void should_batch_single_episode_stats_without_ticket_side_effects() {
EpisodeService episodeService = mock(EpisodeService.class);
DramaService dramaService = mock(DramaService.class);
CommentService commentService = mock(CommentService.class);
LikeService likeService = mock(LikeService.class);
FavoriteService favoriteService = mock(FavoriteService.class);
DramaEpisode episode = new DramaEpisode();
episode.setId(11L);
episode.setDramaId(7L);
episode.setLikeCount(88L);
Drama drama = new Drama();
drama.setId(7L);
drama.setFavoriteCount(1800L);
given(episodeService.listPublishedByIds(List.of(11L))).willReturn(List.of(episode));
given(dramaService.listPublishedByIds(List.of(7L))).willReturn(List.of(drama));
given(commentService.countByEpisodes(List.of(11L))).willReturn(Map.of(11L, 8L));
given(likeService.likedTargetIds(
100L, "device-1", LikeService.TYPE_EPISODE, List.of(11L)))
.willReturn(Set.of(11L));
given(favoriteService.favoriteDramaIds(100L, "device-1", List.of(7L)))
.willReturn(Set.of());
AppEpisodeStatsService service = new AppEpisodeStatsService(
episodeService, dramaService, commentService, likeService, favoriteService);
AppEpisodeStatsVO result = service.get(11L, 100L, "device-1");
assertThat(result.getEpisodeId()).isEqualTo(11L);
assertThat(result.getDramaId()).isEqualTo(7L);
assertThat(result.getLikeCount()).isEqualTo(88L);
assertThat(result.getCommentCount()).isEqualTo(8L);
assertThat(result.getFavoriteCount()).isEqualTo(1800L);
assertThat(result.getLiked()).isTrue();
assertThat(result.getFavorited()).isFalse();
verify(episodeService).listPublishedByIds(List.of(11L));
verify(commentService).countByEpisodes(List.of(11L));
verify(likeService).likedTargetIds(
100L, "device-1", LikeService.TYPE_EPISODE, List.of(11L));
verify(favoriteService).favoriteDramaIds(100L, "device-1", List.of(7L));
}
}
@@ -0,0 +1,42 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.user.enums.CountryProfile;
import org.junit.jupiter.api.Test;
import static org.assertj.core.api.Assertions.assertThat;
/** 国家解析与映射:IP 库文本 → ISO-2,再据此选语言/时区/币种。 */
class CountryProfileTest {
@Test
void should_map_ip_region_text_to_iso_code() {
assertThat(CountryProfile.fromIpRegion("中国|0|广东省|深圳市|电信")).isEqualTo("CN");
assertThat(CountryProfile.fromIpRegion("中国|0|台湾省|0|中华电信")).isEqualTo("TW");
assertThat(CountryProfile.fromIpRegion("中国|0|香港|0|0")).isEqualTo("HK");
assertThat(CountryProfile.fromIpRegion("中国|0|澳门|0|澳门电讯")).isEqualTo("MO");
assertThat(CountryProfile.fromIpRegion("美国|0|0|0|Level3")).isEqualTo("US");
assertThat(CountryProfile.fromIpRegion("印度尼西亚|0|大雅加达|雅加达|0")).isEqualTo("ID");
}
@Test
void should_return_empty_for_private_ip_and_unknown_country() {
assertThat(CountryProfile.fromIpRegion("0|0|0|内网IP|内网IP")).isEmpty();
assertThat(CountryProfile.fromIpRegion("")).isEmpty();
// 未收录的国家交给上层回退到客户端区域
assertThat(CountryProfile.fromIpRegion("津巴布韦|0|0|0|0")).isEmpty();
}
@Test
void should_normalize_client_region_and_fallback_to_default() {
assertThat(CountryProfile.normalizeClientCode(" id ")).isEqualTo("ID");
assertThat(CountryProfile.normalizeClientCode("us")).isEqualTo("US");
assertThat(CountryProfile.normalizeClientCode("XX")).isEmpty();
assertThat(CountryProfile.normalizeClientCode("")).isEmpty();
assertThat(CountryProfile.of("US").currency()).isEqualTo("USD");
assertThat(CountryProfile.of("US").language()).isEqualTo("en-US");
assertThat(CountryProfile.of("").code()).isEqualTo("ID");
assertThat(CountryProfile.of("").currency()).isEqualTo("IDR");
}
}
@@ -0,0 +1,41 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.interaction.service.CommentService;
import com.fruitshorts.interaction.service.LikeService;
import com.fruitshorts.playback.service.FavoriteService;
import com.fruitshorts.playback.service.WatchProgressService;
import org.junit.jupiter.api.Test;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
class InteractionAccountMergeServiceTest {
@Test
void should_merge_all_device_interactions_and_watch_data() {
LikeService likeService = mock(LikeService.class);
FavoriteService favoriteService = mock(FavoriteService.class);
CommentService commentService = mock(CommentService.class);
WatchProgressService watchService = mock(WatchProgressService.class);
given(likeService.mergeDeviceIntoUser(9L, "device-a"))
.willReturn(new LikeService.MergeResult(1, 0));
given(favoriteService.mergeDeviceIntoUser(9L, "device-a"))
.willReturn(new FavoriteService.MergeResult(1, 0));
given(commentService.mergeDeviceIntoUser(9L, "device-a"))
.willReturn(new CommentService.MergeResult(1, 0));
given(watchService.mergeDeviceIntoUser(9L, "device-a"))
.willReturn(new WatchProgressService.MergeResult(2, 1, 3, 2));
InteractionAccountMergeService service = new InteractionAccountMergeService(
likeService, favoriteService, commentService, watchService);
service.merge(9L, "device-a");
verify(likeService).mergeDeviceIntoUser(9L, "device-a");
verify(favoriteService).mergeDeviceIntoUser(9L, "device-a");
verify(commentService).mergeDeviceIntoUser(9L, "device-a");
verify(watchService).mergeDeviceIntoUser(9L, "device-a");
}
}
@@ -0,0 +1,49 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.user.service.IpRegionService;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import static org.assertj.core.api.Assertions.assertThat;
/**
* IpRegionService 集成测试:真实 xdb 文件 + 真实 resolve。
*
* <p>xdb 在 fruitshorts-module-user 的 resources/ip2region/ip2region.xdb 下(11MB),
* 打包后会进 app-api 的 classpath。这个测试验证 Service 层调 ip2region 库的行为——
* 合法公网 IP → ISO-2 国家码,内网/保留地址/空 → 空串。
*/
class IpRegionServiceTest {
private IpRegionService service;
@BeforeEach
void setUp() {
service = new IpRegionService();
}
@Test
void should_resolve_known_public_ips() {
assertThat(service.resolveCountry("8.8.8.8")).isEqualTo("US");
assertThat(service.resolveCountry("114.114.114.114")).isEqualTo("CN");
}
@Test
void should_return_empty_for_private_loopback_and_reserved() {
assertThat(service.resolveCountry("127.0.0.1")).isEmpty();
assertThat(service.resolveCountry("0.0.0.0")).isEmpty();
assertThat(service.resolveCountry("10.0.0.1")).isEmpty();
assertThat(service.resolveCountry("192.168.1.1")).isEmpty();
assertThat(service.resolveCountry("172.16.0.1")).isEmpty();
assertThat(service.resolveCountry("169.254.1.1")).isEmpty();
}
@Test
void should_return_empty_for_invalid_inputs() {
assertThat(service.resolveCountry("")).isEmpty();
assertThat(service.resolveCountry(null)).isEmpty();
assertThat(service.resolveCountry("not-an-ip")).isEmpty();
assertThat(service.resolveCountry("999.999.999.999")).isEmpty();
}
}
@@ -1,6 +1,8 @@
// Author: fruit.com.cn
package com.fruitshorts.app.service;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.DramaService;
@@ -20,6 +22,7 @@ import org.junit.jupiter.api.Test;
import java.util.List;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.ArgumentMatchers.anyString;
@@ -112,6 +115,30 @@ class PlaybackServiceTest {
verify(ticketMapper, never()).insert(any(PlaybackTicket.class));
}
@Test
void should_reject_locked_episode_ticket_for_anonymous_device() {
given(episodeService.getPublished(12L)).willReturn(episode(12L, 0, 60, "/files/episode/2.mp4"));
assertThatThrownBy(() ->
playbackService.play(null, 12L, "device-9", "203.0.113.8"))
.isInstanceOf(ServiceException.class)
.extracting(error -> ((ServiceException) error).getCode())
.isEqualTo(ErrorCode.UNAUTHORIZED.getCode());
verify(signer, never()).sign(anyString());
}
@Test
void should_reject_locked_episode_ticket_before_no_media_for_anonymous_device() {
given(episodeService.getPublished(17L)).willReturn(episode(17L, 0, 60, ""));
assertThatThrownBy(() ->
playbackService.play(null, 17L, "device-9", "203.0.113.8"))
.isInstanceOf(ServiceException.class)
.extracting(error -> ((ServiceException) error).getCode())
.isEqualTo(ErrorCode.UNAUTHORIZED.getCode());
verify(signer, never()).sign(anyString());
}
@Test
void should_return_entitled_access_when_entitlement_port_confirms_unlock() {
given(episodeService.getPublished(13L)).willReturn(episode(13L, 0, 60, "/files/episode/3.mp4"));
@@ -2,6 +2,7 @@
package com.fruitshorts.app.support;
import com.fruitshorts.framework.log.AccessDeniedRecorder;
import com.fruitshorts.framework.web.WebUtils;
import com.fruitshorts.framework.web.JacksonLongToStringConfig;
import com.fruitshorts.framework.web.GlobalExceptionHandler;
import com.fasterxml.jackson.databind.SerializationFeature;
@@ -30,7 +31,9 @@ public final class ControllerContractSupport {
/** 组装一个只含指定控制器与全局异常处理的 MockMvc。 */
@SuppressWarnings("unchecked")
public static MockMvc mockMvcOf(Object controller) {
AccessDeniedRecorder recorder = new AccessDeniedRecorder(Mockito.mock(ObjectProvider.class));
AccessDeniedRecorder recorder = new AccessDeniedRecorder(
Mockito.mock(ObjectProvider.class),
Mockito.mock(ObjectProvider.class));
return MockMvcBuilders.standaloneSetup(controller)
.setControllerAdvice(new GlobalExceptionHandler(recorder))
// 与 Spring Boot 的 Jackson 配置对齐:时间按 ISO 字符串输出而不是数组,
@@ -0,0 +1,133 @@
// Author: fruit.com.cn
package com.fruitshorts.interaction.service;
import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.content.service.DramaService;
import com.fruitshorts.content.service.EpisodeService;
import com.fruitshorts.interaction.config.InteractionProperties;
import com.fruitshorts.interaction.dal.entity.ContentComment;
import com.fruitshorts.interaction.dal.mapper.ContentAuditRecordMapper;
import com.fruitshorts.interaction.dal.mapper.ContentCommentMapper;
import com.fruitshorts.interaction.dal.mapper.ContentLikeMapper;
import com.fruitshorts.interaction.vo.CommentPublishReqVO;
import com.fruitshorts.interaction.vo.CommentRespVO;
import com.fruitshorts.user.dal.mapper.AppUserMapper;
import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
import java.util.List;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
class AnonymousCommentServiceTest {
private ContentCommentMapper commentMapper;
private SensitiveWordService sensitiveWordService;
private CommentService commentService;
@BeforeAll
static void initTableInfo() {
TableInfoHelper.initTableInfo(
new MapperBuilderAssistant(new MybatisConfiguration(), ""),
ContentComment.class);
}
@BeforeEach
void setUp() {
commentMapper = mock(ContentCommentMapper.class);
sensitiveWordService = mock(SensitiveWordService.class);
AuditRecorder auditRecorder = mock(AuditRecorder.class);
commentService = new CommentService(
commentMapper,
mock(ContentLikeMapper.class),
mock(ContentAuditRecordMapper.class),
mock(AppUserMapper.class),
mock(DramaService.class),
mock(EpisodeService.class),
sensitiveWordService,
auditRecorder,
new InteractionProperties());
}
@Test
void should_publish_anonymous_comment_with_guest_name_and_device_owner() {
given(commentMapper.selectCount(any())).willReturn(0L);
given(commentMapper.insert(any(ContentComment.class))).willReturn(1);
given(sensitiveWordService.check("hello")).willReturn(
SensitiveWordCheck.pass("hello"));
CommentRespVO result = commentService.publish(
new InteractionIdentity(null, "device-a"),
request("hello"),
"127.0.0.1");
ArgumentCaptor<ContentComment> captor =
ArgumentCaptor.forClass(ContentComment.class);
verify(commentMapper).insert(captor.capture());
assertThat(captor.getValue().getUserId()).isNull();
assertThat(captor.getValue().getDeviceId()).isEqualTo("device-a");
assertThat(result.getNickname()).isEqualTo("游客");
assertThat(result.getMine()).isTrue();
}
@Test
void should_reject_duplicate_anonymous_content_within_one_minute() {
given(commentMapper.selectCount(any())).willReturn(0L, 1L);
given(sensitiveWordService.check("hello")).willReturn(
SensitiveWordCheck.pass("hello"));
assertThatThrownBy(() -> commentService.publish(
new InteractionIdentity(null, "device-a"),
request("hello"),
"127.0.0.1"))
.hasMessageContaining("重复");
}
@Test
void should_merge_device_comments_and_remove_duplicate() {
ContentComment duplicate = comment(1L, "same");
ContentComment unique = comment(2L, "new");
given(commentMapper.selectList(any())).willReturn(List.of(duplicate, unique));
given(commentMapper.selectOne(any())).willReturn(comment(3L, "same"), null);
given(commentMapper.physicalDeleteById(1L)).willReturn(1);
CommentService.MergeResult result =
commentService.mergeDeviceIntoUser(9L, "device-a");
assertThat(result.migrated()).isEqualTo(1);
assertThat(result.duplicates()).isEqualTo(1);
verify(commentMapper).physicalDeleteById(1L);
verify(commentMapper).updateById(unique);
assertThat(unique.getUserId()).isEqualTo(9L);
}
private CommentPublishReqVO request(String content) {
CommentPublishReqVO req = new CommentPublishReqVO();
req.setDramaId(7L);
req.setContent(content);
return req;
}
private ContentComment comment(Long id, String content) {
ContentComment comment = new ContentComment();
comment.setId(id);
comment.setDramaId(7L);
comment.setEpisodeId(0L);
comment.setParentId(0L);
comment.setContent(content);
comment.setStatus(CommentService.STATUS_PUBLISHED);
comment.setDeviceId("device-a");
return comment;
}
}
@@ -6,6 +6,7 @@ import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.DramaService;
@@ -19,6 +20,9 @@ import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
import java.util.List;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;
@@ -95,6 +99,44 @@ class LikeServiceEpisodeTest {
verify(likeMapper, never()).insert(any(ContentLike.class));
}
@Test
void should_store_anonymous_episode_like_by_device() {
DramaEpisode published = episode(1L);
given(episodeService.getPublished(11L)).willReturn(published);
given(dramaService.getPublished(5L)).willReturn(new Drama());
given(likeMapper.selectOne(any())).willReturn(null);
given(episodeService.getExisting(11L)).willReturn(published);
likeService.toggle(new InteractionIdentity(null, "device-a"), request(11L));
ArgumentCaptor<ContentLike> captor = ArgumentCaptor.forClass(ContentLike.class);
verify(likeMapper).insert(captor.capture());
assertThat(captor.getValue().getUserId()).isNull();
assertThat(captor.getValue().getDeviceId()).isEqualTo("device-a");
}
@Test
void should_merge_device_likes_and_decrement_duplicate_count_once() {
DramaEpisode published = episode(2L);
ContentLike duplicate = like(1L, LikeService.TYPE_EPISODE, 11L);
ContentLike unique = like(2L, LikeService.TYPE_EPISODE, 12L);
given(likeMapper.selectList(any())).willReturn(List.of(duplicate, unique));
given(likeMapper.selectOne(any())).willReturn(
like(3L, LikeService.TYPE_EPISODE, 11L),
null);
given(likeMapper.physicalDeleteById(1L)).willReturn(1);
given(episodeService.getExisting(11L)).willReturn(published);
given(episodeService.getExisting(12L)).willReturn(published);
LikeService.MergeResult result = likeService.mergeDeviceIntoUser(9L, "device-a");
assertThat(result.migrated()).isEqualTo(1);
assertThat(result.duplicates()).isEqualTo(1);
verify(episodeService).adjustLikeCount(11L, -1);
verify(likeMapper).updateById(unique);
assertThat(unique.getUserId()).isEqualTo(9L);
}
private DramaEpisode episode(Long likeCount) {
DramaEpisode episode = new DramaEpisode();
episode.setId(11L);
@@ -109,4 +151,12 @@ class LikeServiceEpisodeTest {
req.setTargetId(episodeId);
return req;
}
private ContentLike like(Long id, String type, Long targetId) {
ContentLike like = new ContentLike();
like.setId(id);
like.setTargetType(type);
like.setTargetId(targetId);
return like;
}
}
@@ -7,6 +7,7 @@ import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.service.DramaService;
import com.fruitshorts.playback.dal.entity.UserFavorite;
@@ -58,6 +59,38 @@ class FavoriteServiceTest {
verify(dramaService).adjustFavoriteCount(7L, 1);
}
@Test
void should_store_anonymous_favorite_by_device() {
given(favoriteMapper.selectOne(any())).willReturn(null);
given(favoriteMapper.insert(any(UserFavorite.class))).willReturn(1);
favoriteService.toggle(new InteractionIdentity(null, "device-a"), 7L);
ArgumentCaptor<UserFavorite> captor = ArgumentCaptor.forClass(UserFavorite.class);
verify(favoriteMapper).insert(captor.capture());
assertThat(captor.getValue().getUserId()).isNull();
assertThat(captor.getValue().getDeviceId()).isEqualTo("device-a");
verify(dramaService).adjustFavoriteCount(7L, 1);
}
@Test
void should_merge_device_favorites_and_decrement_duplicate_count_once() {
UserFavorite duplicate = favorite(7L);
UserFavorite unique = favorite(8L);
given(favoriteMapper.selectList(any())).willReturn(List.of(duplicate, unique));
given(favoriteMapper.selectOne(any())).willReturn(favorite(9L), null);
given(favoriteMapper.physicalDeleteById(duplicate.getId())).willReturn(1);
FavoriteService.MergeResult result =
favoriteService.mergeDeviceIntoUser(1L, "device-a");
assertThat(result.migrated()).isEqualTo(1);
assertThat(result.duplicates()).isEqualTo(1);
verify(dramaService).adjustFavoriteCount(7L, -1);
verify(favoriteMapper).updateById(unique);
assertThat(unique.getUserId()).isEqualTo(1L);
}
@Test
void should_decrement_favorite_count_when_toggle_removes_relation() {
given(favoriteMapper.selectOne(any())).willReturn(new UserFavorite());
@@ -87,13 +120,28 @@ class FavoriteServiceTest {
given(favoriteMapper.physicalDelete(1L, 8L)).willReturn(0);
given(favoriteMapper.physicalDelete(1L, 9L)).willReturn(1);
favoriteService.removeBatch(1L, Arrays.asList(7L, 8L, 9L, null));
int deleted =
favoriteService.removeBatch(1L, Arrays.asList(7L, 8L, 9L, 7L, null));
assertThat(deleted).isEqualTo(2);
verify(dramaService).adjustFavoriteCount(7L, -1);
verify(dramaService, never()).adjustFavoriteCount(8L, -1);
verify(dramaService).adjustFavoriteCount(9L, -1);
}
@Test
void should_remove_anonymous_batch_by_device_and_return_deleted_count() {
given(favoriteMapper.physicalDeleteDevice("device-a", 7L)).willReturn(1);
given(favoriteMapper.physicalDeleteDevice("device-a", 8L)).willReturn(1);
int deleted = favoriteService.removeBatch(
new InteractionIdentity(null, "device-a"), List.of(7L, 8L));
assertThat(deleted).isEqualTo(2);
verify(dramaService).adjustFavoriteCount(7L, -1);
verify(dramaService).adjustFavoriteCount(8L, -1);
}
@Test
void should_use_database_page_and_keep_favorite_order_when_listing() {
UserFavorite first = favorite(51L);
@@ -124,6 +172,24 @@ class FavoriteServiceTest {
verify(dramaService).listPublishedByIds(List.of(51L, 52L));
}
@Test
void should_page_anonymous_favorites_by_device() {
Page<UserFavorite> databasePage = new Page<>(1, 20, 0);
databasePage.setRecords(List.of());
given(favoriteMapper.selectPage(any(Page.class), any())).willReturn(databasePage);
favoriteService.page(
new InteractionIdentity(null, "device-a"), 1, 20);
@SuppressWarnings("unchecked")
ArgumentCaptor<Wrapper<UserFavorite>> wrapperCaptor =
ArgumentCaptor.forClass(Wrapper.class);
verify(favoriteMapper).selectPage(any(Page.class), wrapperCaptor.capture());
assertThat(wrapperCaptor.getValue().getSqlSegment())
.contains("user_id IS NULL")
.contains("device_id");
}
private UserFavorite favorite(Long dramaId) {
UserFavorite favorite = new UserFavorite();
favorite.setDramaId(dramaId);
@@ -7,27 +7,41 @@ import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.DramaService;
import com.fruitshorts.content.service.EpisodeService;
import com.fruitshorts.playback.dal.entity.WatchProgress;
import com.fruitshorts.playback.dal.entity.WatchHistory;
import com.fruitshorts.playback.dal.mapper.WatchHistoryMapper;
import com.fruitshorts.playback.dal.mapper.WatchProgressMapper;
import com.fruitshorts.playback.vo.ProgressRespVO;
import com.fruitshorts.playback.vo.ProgressSaveReqVO;
import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.apache.ibatis.annotations.Select;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.InOrder;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.support.TransactionTemplate;
import java.time.LocalDateTime;
import java.lang.reflect.Method;
import java.util.List;
import java.util.Map;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.inOrder;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.times;
@@ -43,6 +57,7 @@ class WatchProgressServiceTest {
}
private WatchProgressMapper progressMapper;
private WatchHistoryMapper historyMapper;
private DramaService dramaService;
private EpisodeService episodeService;
private WatchProgressService progressService;
@@ -50,13 +65,21 @@ class WatchProgressServiceTest {
@BeforeEach
void setUp() {
progressMapper = mock(WatchProgressMapper.class);
historyMapper = mock(WatchHistoryMapper.class);
dramaService = mock(DramaService.class);
episodeService = mock(EpisodeService.class);
// 测试统一走缓冲关闭(同步写库)路径:getIfAvailable 返回 null
@SuppressWarnings("unchecked")
ObjectProvider<WatchProgressBuffer> bufferProvider = mock(ObjectProvider.class);
given(bufferProvider.getIfAvailable()).willReturn(null);
progressService = new WatchProgressService(
progressMapper,
mock(WatchHistoryMapper.class),
historyMapper,
dramaService,
episodeService);
episodeService,
bufferProvider,
mock(SqlSessionFactory.class),
new TransactionTemplate(mock(PlatformTransactionManager.class)));
}
@Test
@@ -120,7 +143,7 @@ class WatchProgressServiceTest {
hidden.setId(9L);
hidden.setUserId(7L);
hidden.setDeleted(1);
given(progressMapper.selectAny(7L, 5L)).willReturn(hidden);
given(progressMapper.selectAnyForIdentity(7L, "", 5L)).willReturn(hidden);
given(progressMapper.selectById(9L)).willReturn(hidden);
ProgressSaveReqVO req = new ProgressSaveReqVO();
req.setEpisodeId(88L);
@@ -134,6 +157,302 @@ class WatchProgressServiceTest {
assertThat(hidden.getDeleted()).isZero();
}
@Test
void should_save_and_read_anonymous_progress_by_device() {
DramaEpisode episode = episode(88L, 1);
episode.setDramaId(5L);
given(episodeService.getPublished(88L)).willReturn(episode);
given(dramaService.getPublished(5L)).willReturn(drama(5L, "剧名", 10));
given(progressMapper.selectAnyForIdentity(null, "device-a", 5L))
.willReturn(null);
given(progressMapper.insert(any(WatchProgress.class))).willAnswer(invocation -> {
WatchProgress saved = invocation.getArgument(0);
saved.setId(99L);
return 1;
});
given(progressMapper.selectById(99L)).willAnswer(invocation -> {
WatchProgress saved = new WatchProgress();
saved.setId(99L);
saved.setDramaId(5L);
saved.setEpisodeId(88L);
saved.setPositionMs(12_000L);
saved.setDurationMs(60_000L);
saved.setPlayCount(1);
saved.setDeviceId("device-a");
return saved;
});
ProgressSaveReqVO req = new ProgressSaveReqVO();
req.setEpisodeId(88L);
req.setPositionMs(12_000L);
req.setDurationMs(60_000L);
progressService.save(new InteractionIdentity(null, "device-a"), req);
ArgumentCaptor<WatchProgress> progressCaptor =
ArgumentCaptor.forClass(WatchProgress.class);
verify(progressMapper).insert(progressCaptor.capture());
assertThat(progressCaptor.getValue().getUserId()).isNull();
assertThat(progressCaptor.getValue().getDeviceId()).isEqualTo("device-a");
ArgumentCaptor<WatchHistory> historyCaptor =
ArgumentCaptor.forClass(WatchHistory.class);
verify(historyMapper).insert(historyCaptor.capture());
assertThat(historyCaptor.getValue().getUserId()).isNull();
assertThat(historyCaptor.getValue().getDeviceId()).isEqualTo("device-a");
WatchProgress saved = progressCaptor.getValue();
saved.setUpdatedAt(LocalDateTime.of(2026, 10, 27, 9, 0));
Page<WatchProgress> databasePage = new Page<>(1, 20, 1);
databasePage.setRecords(List.of(saved));
given(progressMapper.selectPage(any(Page.class), any())).willReturn(databasePage);
given(dramaService.listByIds(List.of(5L)))
.willReturn(List.of(drama(5L, "匿名观看", 10)));
given(episodeService.listByIds(List.of(88L)))
.willReturn(List.of(episode));
PageResult<ProgressRespVO> recent = progressService.recentPage(
new InteractionIdentity(null, "device-a"), 1, 20);
assertThat(recent.getList()).hasSize(1);
assertThat(recent.getList().get(0).getEpisodeId()).isEqualTo(88L);
}
@Test
void should_read_anonymous_progress_by_device() {
WatchProgress progress = progress(5L, 88L);
progress.setUserId(null);
progress.setDeviceId("device-a");
given(progressMapper.selectOne(any())).willReturn(progress);
ProgressRespVO result = progressService.ofDrama(
new InteractionIdentity(null, "device-a"), 5L);
assertThat(result.getEpisodeId()).isEqualTo(88L);
@SuppressWarnings("unchecked")
ArgumentCaptor<Wrapper<WatchProgress>> wrapperCaptor =
ArgumentCaptor.forClass(Wrapper.class);
verify(progressMapper).selectOne(wrapperCaptor.capture());
assertThat(wrapperCaptor.getValue().getSqlSegment())
.contains("user_id IS NULL")
.contains("device_id");
}
@Test
void should_read_logged_user_progress() {
WatchProgress progress = progress(5L, 88L);
progress.setUserId(9L);
given(progressMapper.selectOne(any())).willReturn(progress);
ProgressRespVO result = progressService.ofDrama(
new InteractionIdentity(9L, ""), 5L);
assertThat(result.getEpisodeId()).isEqualTo(88L);
@SuppressWarnings("unchecked")
ArgumentCaptor<Wrapper<WatchProgress>> wrapperCaptor =
ArgumentCaptor.forClass(Wrapper.class);
verify(progressMapper).selectOne(wrapperCaptor.capture());
assertThat(wrapperCaptor.getValue().getSqlSegment())
.contains("user_id")
.doesNotContain("user_id IS NULL");
}
@Test
void should_declare_jdbc_type_for_nullable_identity_sql() throws Exception {
Method method = WatchProgressMapper.class.getMethod(
"selectAnyForIdentity", Long.class, String.class, Long.class);
Select select = method.getAnnotation(Select.class);
String sql = String.join("\n", select.value());
assertThat(sql)
.contains("#{userId,jdbcType=BIGINT} IS NOT NULL")
.contains("#{userId,jdbcType=BIGINT} IS NULL")
.doesNotContain("#{userId} ");
}
@Test
void should_merge_watch_data_and_keep_more_advanced_progress() {
DramaEpisode episode = episode(88L, 2);
episode.setDramaId(5L);
given(episodeService.listByIds(List.of(88L))).willReturn(List.of(episode));
WatchProgress deviceProgress = progress(5L, 88L);
deviceProgress.setId(1L);
deviceProgress.setPositionMs(90_000L);
deviceProgress.setPlayCount(3);
given(progressMapper.selectList(any())).willReturn(List.of(deviceProgress));
WatchProgress accountProgress = progress(5L, 88L);
accountProgress.setId(2L);
accountProgress.setUserId(9L);
accountProgress.setPositionMs(60_000L);
accountProgress.setPlayCount(2);
given(progressMapper.selectAnyForIdentity(9L, "", 5L))
.willReturn(accountProgress);
WatchHistory deviceHistory = history(5L, 88L, 90_000L);
deviceHistory.setId(3L);
given(historyMapper.selectList(any())).willReturn(List.of(deviceHistory));
WatchHistory accountHistory = history(5L, 88L, 60_000L);
accountHistory.setId(4L);
accountHistory.setUserId(9L);
given(historyMapper.selectOne(any())).willReturn(accountHistory);
WatchProgressService.MergeResult result =
progressService.mergeDeviceIntoUser(9L, "device-a");
assertThat(result.progressMigrated()).isEqualTo(1);
assertThat(result.progressIgnored()).isZero();
assertThat(result.historyMigrated()).isEqualTo(1);
assertThat(result.historyIgnored()).isZero();
assertThat(accountProgress.getPositionMs()).isEqualTo(90_000L);
assertThat(accountProgress.getPlayCount()).isEqualTo(3);
verify(progressMapper).updateAny(accountProgress);
verify(progressMapper).deleteById(1L);
verify(historyMapper).updateById(accountHistory);
verify(historyMapper).deleteById(3L);
}
@Test
void should_ignore_device_progress_when_account_progress_is_ahead() {
DramaEpisode episode = episode(88L, 2);
episode.setDramaId(5L);
given(episodeService.listByIds(List.of(88L))).willReturn(List.of(episode));
WatchProgress deviceProgress = progress(5L, 88L);
deviceProgress.setId(1L);
deviceProgress.setPositionMs(50_000L);
given(progressMapper.selectList(any())).willReturn(List.of(deviceProgress));
WatchProgress accountProgress = progress(5L, 88L);
accountProgress.setId(2L);
accountProgress.setUserId(9L);
accountProgress.setPositionMs(70_000L);
given(progressMapper.selectAnyForIdentity(9L, "", 5L))
.willReturn(accountProgress);
WatchProgressService.MergeResult result =
progressService.mergeDeviceIntoUser(9L, "device-a");
assertThat(result.progressMigrated()).isZero();
assertThat(result.progressIgnored()).isEqualTo(1);
verify(progressMapper, never()).updateAny(any(WatchProgress.class));
verify(progressMapper).deleteById(1L);
}
@Test
void should_delete_buffer_state_when_removing_or_clearing_history() {
WatchProgressBuffer buffer = mock(WatchProgressBuffer.class);
WatchProgressService service = serviceWithBuffer(buffer);
service.remove(new InteractionIdentity(9L, ""), 5L);
service.clear(new InteractionIdentity(9L, ""));
verify(buffer).remove(any(InteractionIdentity.class), eq(5L));
verify(buffer).clear(any(InteractionIdentity.class));
verify(progressMapper, times(2)).delete(any());
}
@Test
void should_not_bring_back_deleted_drama_from_buffer_in_recent_page() {
WatchProgressBuffer buffer = mock(WatchProgressBuffer.class);
WatchProgressService service = serviceWithBuffer(buffer);
Page<WatchProgress> emptyPage = new Page<>(1, 20, 0);
emptyPage.setRecords(List.of());
given(progressMapper.selectPage(any(Page.class), any())).willReturn(emptyPage);
given(buffer.getAll(any(InteractionIdentity.class))).willReturn(Map.of(
5L, new WatchProgressBuffer.BufferedProgress(
5L, 88L, 1_000L, 2_000L, false, "feed", "", 1_700_000_000_000L)));
// 库里该剧只剩逻辑删除行:缓冲状态不得再把它读回列表
WatchProgress hidden = progress(5L, 88L);
hidden.setDeleted(1);
given(progressMapper.selectAnyForIdentityIn(9L, "", List.of(5L)))
.willReturn(List.of(hidden));
PageResult<ProgressRespVO> result =
service.recentPage(new InteractionIdentity(9L, ""), 1, 20);
assertThat(result.getList()).isEmpty();
assertThat(result.getTotal()).isZero();
}
@Test
void should_drop_buffered_refs_whose_state_is_gone_instead_of_requeue() {
WatchProgressBuffer buffer = mock(WatchProgressBuffer.class);
WatchProgressService service = serviceWithBuffer(buffer);
given(buffer.drain(10)).willReturn(
List.of(new WatchProgressBuffer.IdentityRef("u:9", 9L, "", 5L)));
given(buffer.getAllBatch(List.of("u:9"))).willReturn(Map.of());
int flushed = service.flushBuffered(10);
// 状态已消失(用户删除/清空历史或 TTL 过期):必须丢弃,退回队列会变成永不消失的僵尸条目
assertThat(flushed).isZero();
verify(buffer, never()).requeue(any());
verify(progressMapper, never()).selectBatchForPairs(any());
}
@Test
void should_flush_device_buffer_before_merging_and_clear_it_after() {
WatchProgressBuffer buffer = mock(WatchProgressBuffer.class);
given(buffer.getAll(any(InteractionIdentity.class))).willReturn(Map.of());
WatchProgressService service = serviceWithBuffer(buffer);
given(progressMapper.selectList(any())).willReturn(List.of());
given(historyMapper.selectList(any())).willReturn(List.of());
service.mergeDeviceIntoUser(9L, "device-a");
// 合并前先把缓冲落库,合并完再清掉设备侧缓冲
InOrder order = inOrder(buffer);
order.verify(buffer).getAll(any(InteractionIdentity.class));
order.verify(buffer).clear(any(InteractionIdentity.class));
}
@Test
void should_drop_late_buffered_ref_when_row_is_already_deleted() {
WatchProgressBuffer buffer = mock(WatchProgressBuffer.class);
SqlSessionFactory sessionFactory = mock(SqlSessionFactory.class);
SqlSession session = mock(SqlSession.class);
WatchProgressMapper batchProgress = mock(WatchProgressMapper.class);
WatchHistoryMapper batchHistory = mock(WatchHistoryMapper.class);
given(sessionFactory.openSession(ExecutorType.BATCH, false)).willReturn(session);
given(session.getMapper(WatchProgressMapper.class)).willReturn(batchProgress);
given(session.getMapper(WatchHistoryMapper.class)).willReturn(batchHistory);
WatchProgressService service = serviceWithBuffer(buffer, sessionFactory);
given(buffer.drain(10)).willReturn(
List.of(new WatchProgressBuffer.IdentityRef("u:9", 9L, "", 5L)));
given(buffer.getAllBatch(List.of("u:9"))).willReturn(Map.of("u:9", Map.of(
5L, new WatchProgressBuffer.BufferedProgress(
5L, 88L, 1_000L, 2_000L, false, "feed", "", 1_700_000_000_000L))));
// 库里是逻辑删除行,且删除时刻晚于该条缓冲的上报时刻
WatchProgress hidden = progress(5L, 88L);
hidden.setUserId(9L);
hidden.setDeleted(1);
hidden.setUpdatedAt(LocalDateTime.of(2026, 10, 28, 12, 0));
given(progressMapper.selectBatchForPairs(any())).willReturn(List.of(hidden));
service.flushBuffered(10);
verify(batchProgress, never()).updateRowAny(any());
verify(batchHistory, never()).insert(any(WatchHistory.class));
}
private WatchProgressService serviceWithBuffer(WatchProgressBuffer buffer) {
return serviceWithBuffer(buffer, mock(SqlSessionFactory.class));
}
private WatchProgressService serviceWithBuffer(
WatchProgressBuffer buffer, SqlSessionFactory sessionFactory) {
@SuppressWarnings("unchecked")
ObjectProvider<WatchProgressBuffer> provider = mock(ObjectProvider.class);
given(provider.getIfAvailable()).willReturn(buffer);
return new WatchProgressService(
progressMapper,
historyMapper,
dramaService,
episodeService,
provider,
sessionFactory,
new TransactionTemplate(mock(PlatformTransactionManager.class)));
}
private Drama drama(Long id, String title, int totalEpisodes) {
Drama drama = new Drama();
drama.setId(id);
@@ -149,4 +468,15 @@ class WatchProgressServiceTest {
episode.setEpNo(epNo);
return episode;
}
private WatchHistory history(Long dramaId, Long episodeId, long positionMs) {
WatchHistory history = new WatchHistory();
history.setDramaId(dramaId);
history.setEpisodeId(episodeId);
history.setPositionMs(positionMs);
history.setDurationMs(120_000L);
history.setDeviceId("device-a");
history.setWatchedAt(LocalDateTime.of(2026, 10, 27, 9, 0));
return history;
}
}
@@ -37,7 +37,9 @@ public enum ErrorCode {
*/
CREATOR_NOT_BOUND(200004, "当前账号未绑定创作者主体"),
CREATOR_TARGET_REQUIRED(200005, "请先选择要代看的创作者"),
CREATOR_IMPERSONATE_READONLY(200006, "代看模式下只读,不能修改创作者数据");
CREATOR_IMPERSONATE_READONLY(200006, "代看模式下只读,不能修改创作者数据"),
INTERACTION_IDENTITY_REQUIRED(200007, "缺少登录身份或设备标识,请重新登录");
private final int code;
private final String message;
@@ -0,0 +1,11 @@
// Author: fruit.com.cn
package com.fruitshorts.common.interaction;
/** 非付费互动的归属身份:登录用户优先,否则使用稳定设备号。 */
public record InteractionIdentity(Long userId, String deviceId) {
public boolean authenticated() {
return userId != null;
}
}
@@ -0,0 +1,33 @@
// Author: fruit.com.cn
package com.fruitshorts.framework.interaction;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.common.interaction.InteractionIdentity;
import org.springframework.stereotype.Component;
/** 点赞、收藏、评论共用的身份解析入口。 */
@Component
public class InteractionIdentityResolver {
public InteractionIdentity resolve(Long userId, String headerDeviceId, String bodyDeviceId) {
if (userId != null) {
return new InteractionIdentity(userId, "");
}
String deviceId = normalize(headerDeviceId);
if (deviceId.isEmpty()) {
deviceId = normalize(bodyDeviceId);
}
if (deviceId.isEmpty()) {
throw new ServiceException(
ErrorCode.INTERACTION_IDENTITY_REQUIRED,
ErrorCode.INTERACTION_IDENTITY_REQUIRED.getMessage());
}
return new InteractionIdentity(null, deviceId);
}
private String normalize(String value) {
return value == null ? "" : value.trim();
}
}
@@ -22,6 +22,7 @@ import org.springframework.web.context.request.ServletRequestAttributes;
public class AccessDeniedRecorder {
private final ObjectProvider<OperationLogSink> sinkProvider;
private final ObjectProvider<WebUtils> webUtilsProvider;
public void record(HttpServletRequest request) {
OperationLogSink sink = sinkProvider.getIfAvailable();
@@ -47,10 +48,16 @@ public class AccessDeniedRecorder {
.status("FAIL")
.errorMsg("无权限")
.durationMs(0)
.ip(request == null ? "" : WebUtils.getClientIp(request))
.ip(request == null ? "" : clientIp(request))
.build());
}
/** 优先用 Spring 注入的 WebUtils(尊重可信代理配置),拿不到回退静态方法。 */
private String clientIp(HttpServletRequest request) {
WebUtils webUtils = webUtilsProvider.getIfAvailable();
return webUtils != null ? webUtils.clientIp(request) : WebUtils.getClientIp(request);
}
/** 从当前线程取出请求,供全局异常处理使用。 */
public void recordCurrentRequest() {
if (RequestContextHolder.getRequestAttributes() instanceof ServletRequestAttributes attributes) {
@@ -43,6 +43,7 @@ public class OperationLogAspect {
private final ObjectMapper objectMapper;
private final ObjectProvider<OperationLogSink> sinkProvider;
private final ObjectProvider<WebUtils> webUtilsProvider;
@Around("@annotation(operationLog)")
public Object around(ProceedingJoinPoint point, OperationLog operationLog) throws Throwable {
@@ -86,10 +87,16 @@ public class OperationLogAspect {
.status(error == null ? "SUCCESS" : "FAIL")
.errorMsg(error == null ? "" : truncate(error.getMessage()))
.durationMs(durationMs)
.ip(request == null ? "" : WebUtils.getClientIp(request))
.ip(request == null ? "" : clientIp(request))
.build());
}
/** 优先用 Spring 注入的 WebUtils(尊重可信代理配置),拿不到回退静态方法。 */
private String clientIp(HttpServletRequest request) {
WebUtils webUtils = webUtilsProvider.getIfAvailable();
return webUtils != null ? webUtils.clientIp(request) : WebUtils.getClientIp(request);
}
/** 把方法入参整理成 JSON,文件只记名称和大小。 */
private String body(ProceedingJoinPoint point) {
Object[] args = point.getArgs();
@@ -62,6 +62,8 @@ public final class ImpersonationGuard {
public static String clientIp() {
HttpServletRequest request = currentRequest();
// 这里走静态 getClientIp——等价于 trusted-proxies 为空时的逻辑(只用 remoteAddr,不解析 XFF)。
// 这个类是纯静态工具(无 Spring 注入),且只在 Creator 代看场景用,安全影响低。
return request == null ? "" : com.fruitshorts.framework.web.WebUtils.getClientIp(request);
}
@@ -45,6 +45,7 @@ public class SecurityConfig {
"/app/dramas/**",
"/app/search",
"/app/episodes/*/play",
"/app/episodes/*/stats",
"/app/config/**",
/* 启动配置与公告:客户端登录前就要拿(POST /app/creator/apply 等仍要登录) */
"/app/config",
@@ -80,9 +81,21 @@ public class SecurityConfig {
.sessionManagement(session -> session.sessionCreationPolicy(SessionCreationPolicy.STATELESS))
.authorizeHttpRequests(registry -> registry
.requestMatchers(HttpMethod.OPTIONS, "/**").permitAll()
/* 评论列表允许匿名浏览;发布/删除仍需登录(未列入白名单即需鉴权) */
/* 非付费互动允许设备身份;解锁、钱包与权益接口仍只允许登录用户。 */
.requestMatchers(HttpMethod.POST, "/app/likes/toggle").permitAll()
.requestMatchers(HttpMethod.POST, "/app/favorites/toggle").permitAll()
.requestMatchers(HttpMethod.POST, "/app/favorites/remove").permitAll()
.requestMatchers(HttpMethod.DELETE, "/app/favorites/*").permitAll()
.requestMatchers(HttpMethod.POST, "/app/comments").permitAll()
.requestMatchers(HttpMethod.DELETE, "/app/comments/*").permitAll()
.requestMatchers(HttpMethod.GET, "/app/comments").permitAll()
.requestMatchers(HttpMethod.GET, "/app/comments/count").permitAll()
.requestMatchers(HttpMethod.GET, "/app/favorites").permitAll()
.requestMatchers(HttpMethod.GET, "/app/watch/recent").permitAll()
.requestMatchers(HttpMethod.GET, "/app/watch/progress").permitAll()
.requestMatchers(HttpMethod.POST, "/app/watch/progress").permitAll()
.requestMatchers(HttpMethod.DELETE, "/app/watch/progress/*").permitAll()
.requestMatchers(HttpMethod.DELETE, "/app/watch/progress").permitAll()
.requestMatchers(PUBLIC_PATHS).permitAll()
.anyRequest().authenticated())
.exceptionHandling(handling -> handling
@@ -3,25 +3,146 @@
package com.fruitshorts.framework.web;
import jakarta.servlet.http.HttpServletRequest;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
public final class WebUtils {
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.Arrays;
import java.util.List;
/**
* 取真实客户端 IP。
*
* <p>安全策略:默认信任 {@link HttpServletRequest#getRemoteAddr()} 本身,只有当
* 它属于配置的可信代理 CIDR 时才解析 X-Forwarded-For / X-Real-IP 等头。
* 这样客户端伪造的 XFF 不会生效,只有经 Nginx / K8s ingress 等前置代理转发的请求
* 才能带上可信的 XFF 链。
*
* <p>生产建议:同时开启 Tomcat RemoteIpValve(prod profile 已配置),让
* {@code getRemoteAddr()} 在经过代理后直接就是客户端真实 IP。此时可信代理列表
* 配成空即可,Valve 已经帮我们做了 XFF 解析。
*/
@Slf4j
@Component
public class WebUtils {
private static final String UNKNOWN = "unknown";
private static final String[] PROXY_HEADERS = {"X-Forwarded-For", "X-Real-IP",
"Proxy-Client-IP", "WL-Proxy-Client-IP"};
private WebUtils() {
/** 可信代理 CIDR 列表,逗号分隔;空表示永不解析 XFF,直接用 remoteAddr。 */
@Value("${fruitshorts.web.trusted-proxies:}")
private String trustedProxies = "";
/** 纯静态调用路径(不依赖 Spring),供测试或无 IoC 场景使用。 */
public static String getClientIp(HttpServletRequest request) {
return resolve(request, List.of());
}
/** 取真实客户端 IP,兼容 Nginx 反向代理。 */
public static String getClientIp(HttpServletRequest request) {
String[] headers = {"X-Forwarded-For", "X-Real-IP", "Proxy-Client-IP", "WL-Proxy-Client-IP"};
for (String header : headers) {
String value = request.getHeader(header);
if (StringUtils.hasText(value) && !UNKNOWN.equalsIgnoreCase(value)) {
int comma = value.indexOf(',');
return comma > 0 ? value.substring(0, comma).trim() : value.trim();
/** Spring 注入版:自动按配置解析 XFF。 */
public String clientIp(HttpServletRequest request) {
List<String> cidrs = StringUtils.hasText(trustedProxies)
? Arrays.asList(trustedProxies.split(","))
: List.of();
return resolve(request, cidrs);
}
private static String resolve(HttpServletRequest request, List<String> trustedCidrs) {
String remoteAddr = request.getRemoteAddr();
if (isTrustedProxy(remoteAddr, trustedCidrs)) {
String fromHeader = readProxyHeader(request);
if (StringUtils.hasText(fromHeader)) {
return fromHeader;
}
}
return request.getRemoteAddr();
return remoteAddr;
}
private static String readProxyHeader(HttpServletRequest request) {
for (String header : PROXY_HEADERS) {
String value = request.getHeader(header);
if (StringUtils.hasText(value) && !UNKNOWN.equalsIgnoreCase(value)) {
// XFF 链格式:client, proxy1, proxy2 —— 最左侧是原始客户端 IP
String[] parts = value.split(",");
return parts[0].trim();
}
}
return "";
}
/**
* 判断 remoteAddr 是否属于可信代理。
*
* <p>CIDR 简化实现:只支持单个 IP 或 "/24" 这类前缀掩码,不支持复杂区间。
* 生产反代一般就是 127.0.0.1、10.0.0.0/8 这类,够用。
*/
static boolean isTrustedProxy(String remoteAddr, List<String> trustedCidrs) {
if (!StringUtils.hasText(remoteAddr) || trustedCidrs.isEmpty()) {
return false;
}
try {
InetAddress remote = InetAddress.getByName(remoteAddr.trim());
for (String cidr : trustedCidrs) {
if (cidrMatches(remote, cidr.trim())) {
return true;
}
}
} catch (UnknownHostException ex) {
log.debug("无法解析 remoteAddr {} 为 IP,跳过 XFF 信任判定", remoteAddr);
}
return false;
}
private static boolean cidrMatches(InetAddress target, String cidr) {
if (!cidr.contains("/")) {
// 单 IP 精确匹配
InetAddress trusted;
try {
trusted = InetAddress.getByName(cidr);
} catch (UnknownHostException ex) {
return false;
}
return trusted.equals(target);
}
String[] parts = cidr.split("/", 2);
String netPart = parts[0];
int prefixLen;
try {
prefixLen = Integer.parseInt(parts[1]);
} catch (NumberFormatException ex) {
return false;
}
InetAddress net;
try {
net = InetAddress.getByName(netPart);
} catch (UnknownHostException ex) {
return false;
}
byte[] targetBytes = target.getAddress();
byte[] netBytes = net.getAddress();
if (targetBytes.length != netBytes.length) {
return false;
}
int fullBytes = prefixLen / 8;
int remainingBits = prefixLen % 8;
for (int i = 0; i < fullBytes; i++) {
if (targetBytes[i] != netBytes[i]) {
return false;
}
}
if (remainingBits > 0 && fullBytes < targetBytes.length) {
int mask = (0xFF << (8 - remainingBits)) & 0xFF;
if ((targetBytes[fullBytes] & mask) != (netBytes[fullBytes] & mask)) {
return false;
}
}
return true;
}
/** Spring 直接实例化;保留 package-private 方便单测 new。 */
WebUtils() {
}
}
@@ -158,6 +158,20 @@ public class EpisodeService implements EpisodeTranscodeCompletionPort, EpisodeVi
Wrappers.<DramaEpisode>lambdaQuery().in(DramaEpisode::getId, ids));
}
/** C 端:按 ID 批量取已上线剧集,用于只读统计。 */
public List<DramaEpisode> listPublishedByIds(Collection<Long> ids) {
if (ids == null || ids.isEmpty()) {
return List.of();
}
List<Long> normalized = ids.stream().filter(Objects::nonNull).distinct().toList();
if (normalized.isEmpty()) {
return List.of();
}
return dramaEpisodeMapper.selectList(Wrappers.<DramaEpisode>lambdaQuery()
.in(DramaEpisode::getId, normalized)
.eq(DramaEpisode::getStatus, ContentStatus.PUBLISHED.getCode()));
}
@Transactional(rollbackFor = Exception.class)
public Long create(EpisodeSaveReqVO req) {
dramaService.getExisting(req.getDramaId());
@@ -18,6 +18,7 @@ public class ContentLike {
@TableId(type = IdType.ASSIGN_ID)
private Long id;
private Long userId;
private String deviceId;
/** DRAMA/EPISODE/COMMENT */
private String targetType;
private Long targetId;
@@ -4,6 +4,7 @@ package com.fruitshorts.interaction.dal.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.fruitshorts.interaction.dal.entity.ContentComment;
import org.apache.ibatis.annotations.Delete;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
@@ -12,6 +13,9 @@ import java.util.List;
public interface ContentCommentMapper extends BaseMapper<ContentComment> {
@Delete("DELETE FROM content_comment WHERE id = #{id}")
int physicalDeleteById(@Param("id") Long id);
/**
* 后台举报工单按评论 id 批量读取,包含逻辑删除记录。
*
@@ -20,4 +20,13 @@ public interface ContentLikeMapper extends BaseMapper<ContentLike> {
+ "AND target_id = #{targetId}")
int physicalDelete(@Param("userId") Long userId, @Param("targetType") String targetType,
@Param("targetId") Long targetId);
@Delete("DELETE FROM content_like WHERE user_id IS NULL AND device_id = #{deviceId} "
+ "AND target_type = #{targetType} AND target_id = #{targetId}")
int physicalDeleteDevice(@Param("deviceId") String deviceId,
@Param("targetType") String targetType,
@Param("targetId") Long targetId);
@Delete("DELETE FROM content_like WHERE id = #{id}")
int physicalDeleteById(@Param("id") Long id);
}
@@ -9,6 +9,7 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.common.pojo.CursorResult;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.content.dal.entity.Drama;
@@ -78,6 +79,11 @@ public class CommentService {
@Transactional(rollbackFor = Exception.class)
public CommentRespVO publish(Long userId, CommentPublishReqVO req, String ip, String deviceId) {
return publish(new InteractionIdentity(userId, deviceId), req, ip);
}
@Transactional(rollbackFor = Exception.class)
public CommentRespVO publish(InteractionIdentity identity, CommentPublishReqVO req, String ip) {
dramaService.getPublished(req.getDramaId());
if (req.getEpisodeId() != null) {
episodeService.getPublished(req.getEpisodeId());
@@ -90,7 +96,8 @@ public class CommentService {
throw new ServiceException(ErrorCode.BAD_REQUEST,
"评论最多 " + properties.getCommentMaxLength() + " 字");
}
checkRate(userId);
checkRate(identity);
checkDuplicate(identity, req, text);
SensitiveWordCheck check = sensitiveWordService.check(text);
if (check.blocked()) {
@@ -110,7 +117,7 @@ public class CommentService {
ContentComment comment = new ContentComment();
comment.setDramaId(req.getDramaId());
comment.setEpisodeId(req.getEpisodeId() == null ? 0L : req.getEpisodeId());
comment.setUserId(userId);
comment.setUserId(identity.userId());
comment.setParentId(parent == null ? 0L : parent.getId());
comment.setRootId(resolveRootId(parent));
comment.setContent(check.sanitized());
@@ -121,7 +128,7 @@ public class CommentService {
comment.setReplyCount(0);
comment.setReportCount(0);
comment.setIp(ip == null ? "" : ip);
comment.setDeviceId(deviceId == null ? "" : deviceId);
comment.setDeviceId(identity.authenticated() ? "" : identity.deviceId());
comment.setCountryCode("");
commentMapper.insert(comment);
@@ -134,13 +141,22 @@ public class CommentService {
.eq(ContentComment::getId, parent.getId())
.setSql("reply_count = reply_count + 1"));
}
return toResp(comment, appUserMapper.selectById(userId), false, userId);
return toResp(
comment,
identity.authenticated() ? appUserMapper.selectById(identity.userId()) : null,
false,
identity);
}
/**
* 游标翻页。unsetRootId 表示只看顶层评论;传 rootId 则看该楼层的回复。
*/
public CursorResult<CommentRespVO> page(Long meUserId, CommentQueryReqVO req) {
return page(meUserId, "", req);
}
public CursorResult<CommentRespVO> page(
Long meUserId, String meDeviceId, CommentQueryReqVO req) {
if (req.getDramaId() == null) {
throw new ServiceException(ErrorCode.BAD_REQUEST, "缺少 dramaId");
}
@@ -151,8 +167,13 @@ public class CommentService {
.eq(req.getEpisodeId() != null, ContentComment::getEpisodeId, req.getEpisodeId())
.eq(ContentComment::getRootId, rootFilter)
.lt(req.getCursor() != null, ContentComment::getId, req.getCursor());
if (meUserId == null) {
if (meUserId == null && (meDeviceId == null || meDeviceId.isBlank())) {
wrapper.eq(ContentComment::getStatus, STATUS_PUBLISHED);
} else if (meUserId == null) {
wrapper.and(w -> w.eq(ContentComment::getStatus, STATUS_PUBLISHED)
.or(x -> x.eq(ContentComment::getStatus, STATUS_PENDING)
.isNull(ContentComment::getUserId)
.eq(ContentComment::getDeviceId, meDeviceId)));
} else {
// 已发布 + 自己待审核的
wrapper.and(w -> w.eq(ContentComment::getStatus, STATUS_PUBLISHED)
@@ -166,13 +187,25 @@ public class CommentService {
rows = rows.subList(0, size);
}
Long nextCursor = rows.isEmpty() ? null : rows.get(rows.size() - 1).getId();
return new CursorResult<>(toRespList(rows, meUserId), hasMore ? nextCursor : null, hasMore);
return new CursorResult<>(
toRespList(rows, new InteractionIdentity(meUserId, meDeviceId)),
hasMore ? nextCursor : null,
hasMore);
}
@Transactional(rollbackFor = Exception.class)
public void delete(Long userId, Long commentId) {
delete(new InteractionIdentity(userId, ""), commentId);
}
@Transactional(rollbackFor = Exception.class)
public void delete(InteractionIdentity identity, Long commentId) {
ContentComment comment = getExisting(commentId);
if (!comment.getUserId().equals(userId)) {
boolean mine = identity.authenticated()
? identity.userId().equals(comment.getUserId())
: comment.getUserId() == null
&& identity.deviceId().equals(comment.getDeviceId());
if (!mine) {
throw new ServiceException(ErrorCode.FORBIDDEN, "只能删除自己的评论");
}
softDelete(comment);
@@ -278,7 +311,7 @@ public class CommentService {
req.getKeyword())
.orderByDesc(ContentComment::getId));
List<ContentComment> rows = result.getRecords();
List<CommentRespVO> list = toRespList(rows, null);
List<CommentRespVO> list = toRespList(rows, new InteractionIdentity(null, ""));
fillAdminContentInfo(rows, list);
return new PageResult<>(result.getTotal(), list);
}
@@ -349,15 +382,87 @@ public class CommentService {
commentMapper.updateById(comment);
}
private void checkRate(Long userId) {
Long recent = commentMapper.selectCount(Wrappers.<ContentComment>lambdaQuery()
.eq(ContentComment::getUserId, userId)
.ge(ContentComment::getCreatedAt, LocalDateTime.now().minusMinutes(1)));
private void checkRate(InteractionIdentity identity) {
var query = Wrappers.<ContentComment>lambdaQuery()
.ge(ContentComment::getCreatedAt, LocalDateTime.now().minusMinutes(1));
if (identity.authenticated()) {
query.eq(ContentComment::getUserId, identity.userId());
} else {
query.isNull(ContentComment::getUserId)
.eq(ContentComment::getDeviceId, identity.deviceId());
}
Long recent = commentMapper.selectCount(query);
if (recent != null && recent >= properties.getCommentPerMinute()) {
throw new ServiceException(ErrorCode.TOO_MANY_REQUESTS, "发言过于频繁,请稍后再试");
}
}
private void checkDuplicate(
InteractionIdentity identity, CommentPublishReqVO req, String text) {
var query = Wrappers.<ContentComment>lambdaQuery()
.eq(ContentComment::getDramaId, req.getDramaId())
.eq(ContentComment::getEpisodeId,
req.getEpisodeId() == null ? 0L : req.getEpisodeId())
.eq(ContentComment::getParentId,
req.getParentId() == null ? 0L : req.getParentId())
.eq(ContentComment::getContent, text)
.ne(ContentComment::getStatus, STATUS_DELETED)
.ge(ContentComment::getCreatedAt, LocalDateTime.now().minusMinutes(1));
if (identity.authenticated()) {
query.eq(ContentComment::getUserId, identity.userId());
} else {
query.isNull(ContentComment::getUserId)
.eq(ContentComment::getDeviceId, identity.deviceId());
}
Long count = commentMapper.selectCount(query);
if (count != null && count > 0) {
throw new ServiceException(ErrorCode.DATA_EXISTS, "请勿重复发送相同评论");
}
}
/** 登录时把设备评论归并到账号;重复内容删除设备侧记录。 */
@Transactional(rollbackFor = Exception.class)
public MergeResult mergeDeviceIntoUser(Long userId, String deviceId) {
if (userId == null || !StringUtils.hasText(deviceId)) {
return new MergeResult(0, 0);
}
List<ContentComment> deviceComments = commentMapper.selectList(
Wrappers.<ContentComment>lambdaQuery()
.isNull(ContentComment::getUserId)
.eq(ContentComment::getDeviceId, deviceId));
int migrated = 0;
int duplicates = 0;
for (ContentComment deviceComment : deviceComments) {
ContentComment existing = commentMapper.selectOne(
Wrappers.<ContentComment>lambdaQuery()
.eq(ContentComment::getUserId, userId)
.eq(ContentComment::getDramaId, deviceComment.getDramaId())
.eq(ContentComment::getEpisodeId, deviceComment.getEpisodeId())
.eq(ContentComment::getParentId, deviceComment.getParentId())
.eq(ContentComment::getContent, deviceComment.getContent())
.ne(ContentComment::getStatus, STATUS_DELETED)
.last("limit 1"));
if (existing != null) {
if (commentMapper.physicalDeleteById(deviceComment.getId()) > 0
&& deviceComment.getParentId() != null
&& deviceComment.getParentId() > 0) {
commentMapper.update(null, Wrappers.<ContentComment>lambdaUpdate()
.eq(ContentComment::getId, deviceComment.getParentId())
.setSql("reply_count = GREATEST(reply_count - 1, 0)"));
}
duplicates++;
continue;
}
deviceComment.setUserId(userId);
commentMapper.updateById(deviceComment);
migrated++;
}
return new MergeResult(migrated, duplicates);
}
public record MergeResult(int migrated, int duplicates) {
}
private Long resolveRootId(ContentComment parent) {
if (parent == null) {
return 0L;
@@ -371,27 +476,45 @@ public class CommentService {
return Math.min(value, properties.getPageMaxSize());
}
private List<CommentRespVO> toRespList(List<ContentComment> rows, Long meUserId) {
private List<CommentRespVO> toRespList(
List<ContentComment> rows, InteractionIdentity identity) {
if (rows.isEmpty()) {
return List.of();
}
Map<Long, AppUser> users = new HashMap<>();
List<Long> userIds = rows.stream().map(ContentComment::getUserId).distinct().toList();
for (AppUser user : appUserMapper.selectBatchIds(userIds)) {
users.put(user.getId(), user);
List<Long> userIds = rows.stream()
.map(ContentComment::getUserId)
.filter(Objects::nonNull)
.distinct()
.toList();
if (!userIds.isEmpty()) {
for (AppUser user : appUserMapper.selectBatchIds(userIds)) {
users.put(user.getId(), user);
}
}
Set<Long> likedIds = new HashSet<>();
if (meUserId != null) {
for (ContentLike like : likeMapper.selectList(Wrappers.<ContentLike>lambdaQuery()
.eq(ContentLike::getUserId, meUserId)
if (identity.userId() != null || StringUtils.hasText(identity.deviceId())) {
var query = Wrappers.<ContentLike>lambdaQuery()
.eq(ContentLike::getTargetType, "COMMENT")
.in(ContentLike::getTargetId, rows.stream().map(ContentComment::getId).toList()))) {
.in(ContentLike::getTargetId,
rows.stream().map(ContentComment::getId).toList());
if (identity.authenticated()) {
query.eq(ContentLike::getUserId, identity.userId());
} else {
query.isNull(ContentLike::getUserId)
.eq(ContentLike::getDeviceId, identity.deviceId());
}
for (ContentLike like : likeMapper.selectList(query)) {
likedIds.add(like.getTargetId());
}
}
List<CommentRespVO> list = new ArrayList<>(rows.size());
for (ContentComment row : rows) {
list.add(toResp(row, users.get(row.getUserId()), likedIds.contains(row.getId()), meUserId));
list.add(toResp(
row,
users.get(row.getUserId()),
likedIds.contains(row.getId()),
identity));
}
return list;
}
@@ -429,7 +552,11 @@ public class CommentService {
}
}
private CommentRespVO toResp(ContentComment comment, AppUser author, boolean liked, Long meUserId) {
private CommentRespVO toResp(
ContentComment comment,
AppUser author,
boolean liked,
InteractionIdentity identity) {
CommentRespVO vo = new CommentRespVO();
vo.setId(comment.getId());
vo.setDramaId(comment.getDramaId());
@@ -444,12 +571,20 @@ public class CommentService {
vo.setReplyCount(comment.getReplyCount());
vo.setReportCount(comment.getReportCount());
vo.setCreatedAt(comment.getCreatedAt());
vo.setMine(meUserId != null && meUserId.equals(comment.getUserId()));
boolean mine = identity.authenticated()
? identity.userId().equals(comment.getUserId())
: comment.getUserId() == null
&& identity.deviceId().equals(comment.getDeviceId());
vo.setMine(mine);
vo.setLiked(liked);
if (author != null) {
vo.setUid(author.getUid());
vo.setNickname(author.getNickname());
vo.setAvatarUrl(author.getAvatarUrl());
} else {
vo.setUid("");
vo.setNickname("游客");
vo.setAvatarUrl("");
}
return vo;
}
@@ -5,6 +5,7 @@ package com.fruitshorts.interaction.service;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.DramaService;
import com.fruitshorts.content.service.EpisodeService;
@@ -18,7 +19,11 @@ import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Locale;
import java.util.Set;
/**
* 点赞。关系表是事实来源,计数同步更新在目标行上。
@@ -41,6 +46,11 @@ public class LikeService {
@Transactional(rollbackFor = Exception.class)
public LikeToggleRespVO toggle(Long userId, LikeToggleReqVO req) {
return toggle(new InteractionIdentity(userId, ""), req);
}
@Transactional(rollbackFor = Exception.class)
public LikeToggleRespVO toggle(InteractionIdentity identity, LikeToggleReqVO req) {
String type = req.getTargetType().trim().toUpperCase(Locale.ROOT);
if (!TYPE_DRAMA.equals(type) && !TYPE_EPISODE.equals(type) && !TYPE_COMMENT.equals(type)) {
throw new ServiceException(ErrorCode.BAD_REQUEST, "只支持对短剧、剧集或评论点赞");
@@ -48,22 +58,26 @@ public class LikeService {
Long targetId = req.getTargetId();
validateTarget(type, targetId);
ContentLike existing = likeMapper.selectOne(Wrappers.<ContentLike>lambdaQuery()
.eq(ContentLike::getUserId, userId)
ContentLike existing = likeMapper.selectOne(identityQuery(identity)
.eq(ContentLike::getTargetType, type)
.eq(ContentLike::getTargetId, targetId)
.last("limit 1"));
boolean liked;
if (existing == null) {
ContentLike like = new ContentLike();
like.setUserId(userId);
like.setUserId(identity.userId());
like.setDeviceId(identity.authenticated() ? "" : identity.deviceId());
like.setTargetType(type);
like.setTargetId(targetId);
likeMapper.insert(like);
liked = true;
adjustCount(type, targetId, 1);
} else {
likeMapper.physicalDelete(userId, type, targetId);
if (identity.authenticated()) {
likeMapper.physicalDelete(identity.userId(), type, targetId);
} else {
likeMapper.physicalDeleteDevice(identity.deviceId(), type, targetId);
}
liked = false;
adjustCount(type, targetId, -1);
}
@@ -86,6 +100,83 @@ public class LikeService {
return comment == null || comment.getLikeCount() == null ? 0L : comment.getLikeCount();
}
/** 批量判断用户是否已点赞,一次查询覆盖多个目标。 */
public Set<Long> likedTargetIds(Long userId, String type, Collection<Long> targetIds) {
return likedTargetIds(userId, "", type, targetIds);
}
/** 登录用户优先;匿名状态按设备号查询。 */
public Set<Long> likedTargetIds(Long userId, String deviceId, String type,
Collection<Long> targetIds) {
if (targetIds == null || targetIds.isEmpty()) {
return Set.of();
}
Set<Long> normalized = new HashSet<>();
for (Long targetId : targetIds) {
if (targetId != null) {
normalized.add(targetId);
}
}
if (normalized.isEmpty()) {
return Set.of();
}
if (userId == null && (deviceId == null || deviceId.isBlank())) {
return Set.of();
}
return likeMapper.selectList(identityQuery(new InteractionIdentity(userId, deviceId))
.eq(ContentLike::getTargetType, type)
.in(ContentLike::getTargetId, normalized))
.stream()
.map(ContentLike::getTargetId)
.collect(java.util.stream.Collectors.toSet());
}
/** 登录时把设备点赞归并到账号;重复关系删除并回退目标计数。 */
@Transactional(rollbackFor = Exception.class)
public MergeResult mergeDeviceIntoUser(Long userId, String deviceId) {
if (userId == null || deviceId == null || deviceId.isBlank()) {
return new MergeResult(0, 0);
}
List<ContentLike> deviceLikes = likeMapper.selectList(
Wrappers.<ContentLike>lambdaQuery()
.isNull(ContentLike::getUserId)
.eq(ContentLike::getDeviceId, deviceId));
int migrated = 0;
int duplicates = 0;
for (ContentLike deviceLike : deviceLikes) {
ContentLike existing = likeMapper.selectOne(Wrappers.<ContentLike>lambdaQuery()
.eq(ContentLike::getUserId, userId)
.eq(ContentLike::getTargetType, deviceLike.getTargetType())
.eq(ContentLike::getTargetId, deviceLike.getTargetId())
.last("limit 1"));
if (existing != null) {
if (likeMapper.physicalDeleteById(deviceLike.getId()) > 0) {
adjustCount(deviceLike.getTargetType(), deviceLike.getTargetId(), -1);
}
duplicates++;
continue;
}
deviceLike.setUserId(userId);
likeMapper.updateById(deviceLike);
migrated++;
}
return new MergeResult(migrated, duplicates);
}
private com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper<ContentLike>
identityQuery(InteractionIdentity identity) {
if (identity.authenticated()) {
return Wrappers.<ContentLike>lambdaQuery()
.eq(ContentLike::getUserId, identity.userId());
}
return Wrappers.<ContentLike>lambdaQuery()
.isNull(ContentLike::getUserId)
.eq(ContentLike::getDeviceId, identity.deviceId());
}
public record MergeResult(int migrated, int duplicates) {
}
private void adjustCount(String type, Long targetId, int delta) {
if (TYPE_DRAMA.equals(type)) {
dramaService.adjustLikeCount(targetId, delta);
@@ -20,4 +20,7 @@ public class CommentPublishReqVO {
/** 回复某条评论时传它的 id;顶层评论不传 */
private Long parentId;
/** 兼容旧调用;正常客户端通过 X-Device-Id 下发。 */
private String deviceId;
}
@@ -16,4 +16,7 @@ public class LikeToggleReqVO {
@NotNull(message = "缺少 targetId")
private Long targetId;
/** 兼容旧调用;正常客户端通过 X-Device-Id 下发。 */
private String deviceId;
}
+5
View File
@@ -33,5 +33,10 @@
<groupId>com.fruitshorts</groupId>
<artifactId>fruitshorts-module-user</artifactId>
</dependency>
<!-- 进度上报路径的剧集/短剧进程内缓存(版本由 Spring Boot BOM 管理) -->
<dependency>
<groupId>com.github.ben-manes.caffeine</groupId>
<artifactId>caffeine</artifactId>
</dependency>
</dependencies>
</project>
@@ -20,6 +20,7 @@ public class UserFavorite {
@TableId(type = IdType.ASSIGN_ID)
private Long id;
private Long userId;
private String deviceId;
private Long dramaId;
private Integer sort;
private LocalDateTime createdAt;
@@ -17,4 +17,11 @@ public interface UserFavoriteMapper extends BaseMapper<UserFavorite> {
*/
@Delete("DELETE FROM user_favorite WHERE user_id = #{userId} AND drama_id = #{dramaId}")
int physicalDelete(@Param("userId") Long userId, @Param("dramaId") Long dramaId);
@Delete("DELETE FROM user_favorite WHERE user_id IS NULL AND device_id = #{deviceId} "
+ "AND drama_id = #{dramaId}")
int physicalDeleteDevice(@Param("deviceId") String deviceId, @Param("dramaId") Long dramaId);
@Delete("DELETE FROM user_favorite WHERE id = #{id}")
int physicalDeleteById(@Param("id") Long id);
}
@@ -8,15 +8,25 @@ import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import org.apache.ibatis.annotations.Update;
import java.util.List;
public interface WatchProgressMapper extends BaseMapper<WatchProgress> {
/** 含逻辑删除行。唯一索引不区分 deleted,再看同一部剧必须能找回旧行。 */
/** 按登录用户或匿名设备读取同一部剧的记录,包含逻辑删除行。 */
@Select("""
SELECT * FROM watch_progress
WHERE user_id = #{userId} AND drama_id = #{dramaId}
WHERE (
(#{userId,jdbcType=BIGINT} IS NOT NULL AND user_id = #{userId,jdbcType=BIGINT})
OR
(#{userId,jdbcType=BIGINT} IS NULL AND user_id IS NULL AND device_id = #{deviceId})
)
AND drama_id = #{dramaId}
LIMIT 1
""")
WatchProgress selectAny(@Param("userId") Long userId, @Param("dramaId") Long dramaId);
WatchProgress selectAnyForIdentity(
@Param("userId") Long userId,
@Param("deviceId") String deviceId,
@Param("dramaId") Long dramaId);
/** 恢复并更新。手写 SQL 不带 deleted=0 条件,逻辑删除行也能写回。 */
@Update("""
@@ -32,4 +42,64 @@ public interface WatchProgressMapper extends BaseMapper<WatchProgress> {
WHERE id = #{id}
""")
int updateAny(WatchProgress progress);
/** 固定形状的全字段按 id 更新(绕过逻辑删除过滤)。SQL 不随字段空值变化,
* 缓冲落库时连续调用可被 MyBatis BATCH 执行器合成一个 JDBC batch。 */
@Update("""
UPDATE watch_progress SET
episode_id = #{et.episodeId},
position_ms = #{et.positionMs},
duration_ms = #{et.durationMs},
finished = #{et.finished},
play_count = #{et.playCount},
device_id = #{et.deviceId},
source = #{et.source},
deleted = #{et.deleted}
WHERE id = #{et.id}
""")
int updateRowAny(@Param("et") WatchProgress et);
/** 批量读取同一身份下多部剧的记录(含逻辑删除行),供缓冲落库合并使用。 */
@Select("""
<script>
SELECT * FROM watch_progress
WHERE
<choose>
<when test="userId != null">user_id = #{userId,jdbcType=BIGINT}</when>
<otherwise>user_id IS NULL AND device_id = #{deviceId}</otherwise>
</choose>
AND drama_id IN
<foreach collection="dramaIds" item="d" open="(" separator="," close=")">#{d}</foreach>
</script>
""")
List<WatchProgress> selectAnyForIdentityIn(
@Param("userId") Long userId,
@Param("deviceId") String deviceId,
@Param("dramaIds") List<Long> dramaIds);
/** 落库合并的查询参数:身份(登录用户或匿名设备)+ drama。 */
@lombok.Data
@lombok.NoArgsConstructor
@lombok.AllArgsConstructor
class IdentityDramaPair {
private Long userId;
private String deviceId;
private long dramaId;
}
/** 按多组「身份 + drama」批量读取记录(含逻辑删除行),一组一条索引条件,单次往返。 */
@Select("""
<script>
SELECT * FROM watch_progress WHERE
<foreach collection="pairs" item="p" separator=" OR ">
(
(#{p.userId,jdbcType=BIGINT} IS NOT NULL AND user_id = #{p.userId,jdbcType=BIGINT})
OR
(#{p.userId,jdbcType=BIGINT} IS NULL AND user_id IS NULL AND device_id = #{p.deviceId})
)
AND drama_id = #{p.dramaId}
</foreach>
</script>
""")
List<WatchProgress> selectBatchForPairs(@Param("pairs") List<IdentityDramaPair> pairs);
}
@@ -6,6 +6,7 @@ import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.enums.ContentStatus;
import com.fruitshorts.content.service.DramaService;
@@ -17,9 +18,12 @@ import org.springframework.transaction.annotation.Transactional;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
/** 收藏:切换、列表、批量删除;收藏数在同一事务内同步维护。 */
@Service
@@ -34,21 +38,28 @@ public class FavoriteService {
/** 返回切换后的状态:true 表示已收藏。 */
@Transactional(rollbackFor = Exception.class)
public boolean toggle(Long userId, Long dramaId) {
return toggle(new InteractionIdentity(userId, ""), dramaId);
}
@Transactional(rollbackFor = Exception.class)
public boolean toggle(InteractionIdentity identity, Long dramaId) {
dramaService.getPublished(dramaId);
UserFavorite existing = favoriteMapper.selectOne(Wrappers.<UserFavorite>lambdaQuery()
.eq(UserFavorite::getUserId, userId)
UserFavorite existing = favoriteMapper.selectOne(identityQuery(identity)
.eq(UserFavorite::getDramaId, dramaId)
.last("limit 1"));
if (existing != null) {
// 必须物理删除:唯一索引 (user_id, drama_id) 会挡住"取消后再收藏"
int removed = favoriteMapper.physicalDelete(userId, dramaId);
int removed = identity.authenticated()
? favoriteMapper.physicalDelete(identity.userId(), dramaId)
: favoriteMapper.physicalDeleteDevice(identity.deviceId(), dramaId);
if (removed > 0) {
dramaService.adjustFavoriteCount(dramaId, -1);
}
return false;
}
UserFavorite favorite = new UserFavorite();
favorite.setUserId(userId);
favorite.setUserId(identity.userId());
favorite.setDeviceId(identity.authenticated() ? "" : identity.deviceId());
favorite.setDramaId(dramaId);
favorite.setSort(0);
favoriteMapper.insert(favorite);
@@ -63,16 +74,97 @@ public class FavoriteService {
return count != null && count > 0;
}
/** 批量判断用户是否已收藏,一次查询覆盖多部剧。 */
public Set<Long> favoriteDramaIds(Long userId, Collection<Long> dramaIds) {
return favoriteDramaIds(userId, "", dramaIds);
}
/** 登录用户优先;匿名状态按设备号查询。 */
public Set<Long> favoriteDramaIds(Long userId, String deviceId,
Collection<Long> dramaIds) {
if (dramaIds == null || dramaIds.isEmpty()) {
return Set.of();
}
Set<Long> normalized = new HashSet<>();
for (Long dramaId : dramaIds) {
if (dramaId != null) {
normalized.add(dramaId);
}
}
if (normalized.isEmpty()) {
return Set.of();
}
if (userId == null && (deviceId == null || deviceId.isBlank())) {
return Set.of();
}
return favoriteMapper.selectList(identityQuery(new InteractionIdentity(userId, deviceId))
.in(UserFavorite::getDramaId, normalized))
.stream()
.map(UserFavorite::getDramaId)
.collect(java.util.stream.Collectors.toSet());
}
/** 登录时把设备收藏归并到账号;重复收藏删除并回退剧级计数。 */
@Transactional(rollbackFor = Exception.class)
public MergeResult mergeDeviceIntoUser(Long userId, String deviceId) {
if (userId == null || deviceId == null || deviceId.isBlank()) {
return new MergeResult(0, 0);
}
List<UserFavorite> deviceFavorites = favoriteMapper.selectList(
Wrappers.<UserFavorite>lambdaQuery()
.isNull(UserFavorite::getUserId)
.eq(UserFavorite::getDeviceId, deviceId));
int migrated = 0;
int duplicates = 0;
for (UserFavorite deviceFavorite : deviceFavorites) {
UserFavorite existing = favoriteMapper.selectOne(
Wrappers.<UserFavorite>lambdaQuery()
.eq(UserFavorite::getUserId, userId)
.eq(UserFavorite::getDramaId, deviceFavorite.getDramaId())
.last("limit 1"));
if (existing != null) {
if (favoriteMapper.physicalDeleteById(deviceFavorite.getId()) > 0) {
dramaService.adjustFavoriteCount(deviceFavorite.getDramaId(), -1);
}
duplicates++;
continue;
}
deviceFavorite.setUserId(userId);
favoriteMapper.updateById(deviceFavorite);
migrated++;
}
return new MergeResult(migrated, duplicates);
}
private com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper<UserFavorite>
identityQuery(InteractionIdentity identity) {
if (identity.authenticated()) {
return Wrappers.<UserFavorite>lambdaQuery()
.eq(UserFavorite::getUserId, identity.userId());
}
return Wrappers.<UserFavorite>lambdaQuery()
.isNull(UserFavorite::getUserId)
.eq(UserFavorite::getDeviceId, identity.deviceId());
}
public record MergeResult(int migrated, int duplicates) {
}
/** 分页查询收藏短剧,保持收藏时间与 dramaId 稳定排序。 */
public PageResult<Drama> page(Long userId, int pageIndex, int pageSize) {
return page(new InteractionIdentity(userId, ""), pageIndex, pageSize);
}
/** 分页查询当前登录账号或匿名设备的收藏短剧。 */
public PageResult<Drama> page(
InteractionIdentity identity, int pageIndex, int pageSize) {
int safeIndex = Math.max(pageIndex, 1);
int safeSize = Math.min(Math.max(pageSize, 1), MAX_PAGE_SIZE);
Page<UserFavorite> request = new Page<>(safeIndex, safeSize);
String publishedDramaSql = "SELECT id FROM drama WHERE deleted = 0 AND status = '"
+ ContentStatus.PUBLISHED.getCode() + "'";
IPage<UserFavorite> result = favoriteMapper.selectPage(request,
Wrappers.<UserFavorite>lambdaQuery()
.eq(UserFavorite::getUserId, userId)
identityQuery(identity)
.inSql(UserFavorite::getDramaId, publishedDramaSql)
.orderByDesc(UserFavorite::getCreatedAt)
.orderByDesc(UserFavorite::getDramaId));
@@ -106,19 +198,47 @@ public class FavoriteService {
}
@Transactional(rollbackFor = Exception.class)
public void removeBatch(Long userId, List<Long> dramaIds) {
if (dramaIds == null || dramaIds.isEmpty()) {
return;
public void removeDevice(String deviceId, Long dramaId) {
int removed = favoriteMapper.physicalDeleteDevice(deviceId, dramaId);
if (removed > 0) {
dramaService.adjustFavoriteCount(dramaId, -1);
}
}
@Transactional(rollbackFor = Exception.class)
public int removeBatch(Long userId, List<Long> dramaIds) {
return removeBatchInternal(new InteractionIdentity(userId, ""), dramaIds);
}
/** 登录账号或匿名设备在同一事务内批量取消收藏。 */
@Transactional(rollbackFor = Exception.class)
public int removeBatch(InteractionIdentity identity, List<Long> dramaIds) {
return removeBatchInternal(identity, dramaIds);
}
private int removeBatchInternal(
InteractionIdentity identity, List<Long> dramaIds) {
if (dramaIds == null || dramaIds.isEmpty()) {
return 0;
}
Set<Long> normalized = new HashSet<>();
for (Long dramaId : dramaIds) {
if (dramaId == null) {
continue;
}
int removed = favoriteMapper.physicalDelete(userId, dramaId);
normalized.add(dramaId);
}
int deleted = 0;
for (Long dramaId : normalized) {
int removed = identity.authenticated()
? favoriteMapper.physicalDelete(identity.userId(), dramaId)
: favoriteMapper.physicalDeleteDevice(identity.deviceId(), dramaId);
if (removed > 0) {
dramaService.adjustFavoriteCount(dramaId, -1);
deleted += removed;
}
}
return deleted;
}
/** 查询用户对某部剧最近一次收藏时间。 */
@@ -2,6 +2,8 @@
package com.fruitshorts.playback.service;
import com.fruitshorts.common.exception.ErrorCode;
import com.fruitshorts.common.exception.ServiceException;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.DramaService;
@@ -57,6 +59,10 @@ public class PlaybackService {
boolean free = episode.getIsFree() != null && episode.getIsFree() == 1;
boolean entitled = userId != null && entitlementPort.isUnlocked(userId, drama.getId(), episode.getId());
if (!free && !entitled && userId == null) {
// 锁定集正片必须登录;即使暂时没有片源,也不能把游客引到匿名取票分支。
throw new ServiceException(ErrorCode.UNAUTHORIZED, "登录后才能解锁并观看该集");
}
String source = playableSource(episode);
PlayRespVO vo = new PlayRespVO();
@@ -0,0 +1,355 @@
// Author: fruit.com.cn
package com.fruitshorts.playback.service;
import com.fruitshorts.common.interaction.InteractionIdentity;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.data.redis.core.SessionCallback;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;
import java.time.Duration;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
/**
* 观看进度 Redis 缓冲(write-behind)。
*
* <p>高并发上报的落地形态:POST 只写 Redis(Lua 脚本一次往返完成写状态与入队),由
* {@link WatchProgressFlusher} 定时把缓冲按「账号或设备 + drama」合并后批量 UPSERT
* 进权威表 watch_progress。Redis 不可用时 put 返回 false,调用方回退同步写库。
*
* <p>Redis 结构:
* <ul>
* <li>{@code watch:pg:buf:u:{userId}} / {@code watch:pg:buf:d:{deviceId}}:
* HASH,field=dramaId,value=进度状态,作为读接口的「读己之写」覆盖层,TTL 12h;</li>
* <li>{@code watch:pg:dirty}:LIST,value={identityKey}|{dramaId},待落库队列,
* 定时任务 RPOP 消费;重复条目靠幂等合并消化。</li>
* </ul>
*/
@Slf4j
@Component
@RequiredArgsConstructor
@ConditionalOnProperty(name = "fruitshorts.watch-progress.buffer", havingValue = "true")
public class WatchProgressBuffer {
private static final String BUF_KEY_PREFIX = "watch:pg:buf:";
private static final String DIRTY_KEY = "watch:pg:dirty";
/** 待落库队列长度上限,防止 Redis 在落库停滞时无限膨胀。 */
private static final long MAX_DIRTY = 200_000;
/** 每个身份最多缓冲 30 部剧(续看列表第一页用不完,同时限制 HGETALL 大小)。 */
private static final int MAX_DRAMAS_PER_IDENTITY = 30;
private static final Duration TTL = Duration.ofHours(12);
/**
* 上报热路径的单次往返脚本:写状态、续 TTL、入待落库队列、截断队列一次做完。
* 拆成多条命令时进程若死在「写状态」与「入队」之间,该剧进度只留在 HASH 里、
* 永远不会落库,12 小时后被 TTL 清掉。返回 1 表示该身份的剧数超限,需要补充裁剪。
*/
private static final DefaultRedisScript<Long> PUT_SCRIPT = new DefaultRedisScript<>("""
redis.call('HSET', KEYS[1], ARGV[1], ARGV[2])
redis.call('EXPIRE', KEYS[1], ARGV[3])
redis.call('LPUSH', KEYS[2], ARGV[4])
redis.call('LTRIM', KEYS[2], 0, ARGV[5])
if redis.call('HLEN', KEYS[1]) > tonumber(ARGV[6]) then
return 1
end
return 0
""", Long.class);
private final StringRedisTemplate redis;
/**
* 缓冲中的一条进度状态。ts 为客户端上报时刻(服务端毫秒),用于判断
* 缓冲值是否比数据库行新,以及合并时丢弃乱序旧值。
*/
public record BufferedProgress(
long dramaId,
long episodeId,
long positionMs,
long durationMs,
boolean finished,
String source,
String deviceId,
long ts) {
}
/** dirty 队列弹出的一条待落库引用。 */
public record IdentityRef(String key, Long userId, String deviceId, long dramaId) {
public InteractionIdentity toIdentity() {
return userId != null
? new InteractionIdentity(userId, "")
: new InteractionIdentity(null, deviceId);
}
}
/** 上报进度入缓冲。成功返回 true;Redis 异常返回 false,调用方回退同步写库。 */
public boolean put(InteractionIdentity identity, long dramaId, long episodeId,
long positionMs, long durationMs, boolean finished,
String source, String deviceId) {
try {
String key = identityKey(identity);
String hashKey = BUF_KEY_PREFIX + key;
String field = Long.toString(dramaId);
Long overflow = redis.execute(PUT_SCRIPT,
List.of(hashKey, DIRTY_KEY),
field,
encode(episodeId, positionMs, durationMs, finished, source, deviceId),
Long.toString(TTL.toSeconds()),
key + "|" + field,
Long.toString(MAX_DIRTY - 1),
Integer.toString(MAX_DRAMAS_PER_IDENTITY));
if (overflow != null && overflow > 0) {
// 超出 30 部才做一次 HGETALL + HDEL 裁剪,热路径保持单次往返
trimIdentity(hashKey);
}
return true;
} catch (Exception e) {
log.warn("观看进度写入 Redis 缓冲失败,本次回退同步写库: {}", e.getMessage());
return false;
}
}
/**
* 删除某部剧的缓冲状态,用于用户删除单条观看记录。
*
* <p>缓冲是读接口的权威来源,删库行必须同步删缓冲,否则会被读覆盖层读回;
* 同时缓冲状态消失后,dirty 队列里该条引用在落库时会被丢弃(见
* {@code WatchProgressService#flushRefs}),不会再被写回库。
*/
public boolean remove(InteractionIdentity identity, long dramaId) {
try {
redis.opsForHash().delete(BUF_KEY_PREFIX + identityKey(identity), Long.toString(dramaId));
return true;
} catch (Exception e) {
log.warn("观看进度缓冲删除失败: {}", e.getMessage());
return false;
}
}
/** 清空某身份的全部缓冲状态,用于用户清空观看历史。 */
public boolean clear(InteractionIdentity identity) {
try {
redis.delete(BUF_KEY_PREFIX + identityKey(identity));
return true;
} catch (Exception e) {
log.warn("观看进度缓冲清空失败: {}", e.getMessage());
return false;
}
}
/** 读单个剧的缓冲状态;无缓冲或解码失败返回 null。 */
public BufferedProgress get(InteractionIdentity identity, long dramaId) {
return get(identityKey(identity), dramaId);
}
public BufferedProgress get(String identityKey, long dramaId) {
try {
Object value = redis.opsForHash().get(BUF_KEY_PREFIX + identityKey, Long.toString(dramaId));
return value == null ? null : decode(dramaId, value.toString());
} catch (Exception e) {
log.warn("观看进度读取 Redis 缓冲失败: {}", e.getMessage());
return null;
}
}
/** 读某身份全部缓冲状态(≤30 部剧),读接口覆盖层用。 */
public Map<Long, BufferedProgress> getAll(InteractionIdentity identity) {
return getAll(identityKey(identity));
}
public Map<Long, BufferedProgress> getAll(String identityKey) {
try {
return decodeAll(redis.opsForHash().entries(BUF_KEY_PREFIX + identityKey));
} catch (Exception e) {
log.warn("观看进度读取 Redis 缓冲失败: {}", e.getMessage());
return Map.of();
}
}
/** 批量读取多个身份的缓冲状态(pipeline 单次往返),落库任务用。 */
@SuppressWarnings("unchecked")
public Map<String, Map<Long, BufferedProgress>> getAllBatch(List<String> identityKeys) {
if (identityKeys.isEmpty()) {
return Map.of();
}
try {
List<Object> results = redis.executePipelined(new SessionCallback<>() {
@Override
public Object execute(org.springframework.data.redis.core.RedisOperations operations) {
for (String key : identityKeys) {
operations.opsForHash().entries(BUF_KEY_PREFIX + key);
}
return null;
}
});
Map<String, Map<Long, BufferedProgress>> result = new LinkedHashMap<>();
for (int i = 0; i < identityKeys.size(); i++) {
Object raw = results.get(i);
result.put(identityKeys.get(i),
raw instanceof Map<?, ?> map ? decodeAll((Map<Object, Object>) map) : Map.of());
}
return result;
} catch (Exception e) {
log.warn("观看进度批量读取 Redis 缓冲失败: {}", e.getMessage());
throw new IllegalStateException("Redis 缓冲批量读取失败", e);
}
}
/** 落库任务从 dirty 队头弹出最多 max 条(按 identity+drama 去重,保留顺序)。 */
public List<IdentityRef> drain(int max) {
List<String> raw;
try {
raw = redis.opsForList().leftPop(DIRTY_KEY, Math.max(1, max));
} catch (Exception e) {
log.warn("观看进度 dirty 队列弹出失败: {}", e.getMessage());
return List.of();
}
if (raw == null || raw.isEmpty()) {
return List.of();
}
LinkedHashMap<String, IdentityRef> dedup = new LinkedHashMap<>();
for (String value : raw) {
int split = value.lastIndexOf('|');
if (split <= 0) {
continue;
}
String key = value.substring(0, split);
long dramaId;
try {
dramaId = Long.parseLong(value.substring(split + 1));
} catch (NumberFormatException ignore) {
continue;
}
dedup.putIfAbsent(key + "|" + dramaId, toRef(key, dramaId));
}
return new ArrayList<>(dedup.values());
}
/** 落库失败的条目退回队列,等待下个周期重试。 */
public void requeue(List<IdentityRef> refs) {
try {
List<String> raw = refs.stream()
.map(r -> r.key() + "|" + r.dramaId())
.toList();
redis.opsForList().rightPushAll(DIRTY_KEY, raw);
redis.opsForList().trim(DIRTY_KEY, 0, MAX_DIRTY - 1);
} catch (Exception e) {
log.error("观看进度缓冲条目退回队列失败,这批进度将丢失: {}", e.getMessage());
}
}
/** 身份的 Redis 键:登录用 u:{userId},匿名用 d:{deviceId}。 */
public String identityKey(InteractionIdentity identity) {
return identity.authenticated()
? "u:" + identity.userId()
: "d:" + cleanToken(identity.deviceId());
}
private IdentityRef toRef(String key, long dramaId) {
int split = key.indexOf(':');
if (split <= 0) {
return new IdentityRef(key, null, key, dramaId);
}
String scope = key.substring(0, split);
String id = key.substring(split + 1);
if ("u".equals(scope)) {
try {
return new IdentityRef(key, Long.valueOf(id), "", dramaId);
} catch (NumberFormatException ignore) {
// 落到匿名分支
}
}
return new IdentityRef(key, null, id, dramaId);
}
private String encode(long episodeId, long positionMs, long durationMs,
boolean finished, String source, String deviceId) {
return episodeId + "|" + positionMs + "|" + durationMs + "|" + (finished ? 1 : 0)
+ "|" + cleanToken(source) + "|" + cleanToken(deviceId) + "|" + System.currentTimeMillis();
}
private BufferedProgress decode(long dramaId, String value) {
try {
String[] parts = value.split("\\|", -1);
if (parts.length != 7) {
return null;
}
return new BufferedProgress(
dramaId,
Long.parseLong(parts[0]),
Long.parseLong(parts[1]),
Long.parseLong(parts[2]),
"1".equals(parts[3]),
parts[4],
parts[5],
Long.parseLong(parts[6]));
} catch (NumberFormatException e) {
log.debug("观看进度缓冲值解码失败,忽略: {}", value);
return null;
}
}
private Map<Long, BufferedProgress> decodeAll(Map<Object, Object> raw) {
if (raw == null || raw.isEmpty()) {
return Map.of();
}
Map<Long, BufferedProgress> result = new LinkedHashMap<>();
for (Map.Entry<Object, Object> e : raw.entrySet()) {
long dramaId;
try {
dramaId = Long.parseLong(e.getKey().toString());
} catch (NumberFormatException ignore) {
continue;
}
BufferedProgress state = decode(dramaId, e.getValue().toString());
if (state != null) {
result.put(dramaId, state);
}
}
return result;
}
/** 限制单身份缓冲的剧数:超出时淘汰最旧条目,保证 HGETALL 与读覆盖层开销有界。 */
private void trimIdentity(String hashKey) {
Long size = redis.opsForHash().size(hashKey);
if (size == null || size <= MAX_DRAMAS_PER_IDENTITY) {
return;
}
Map<Object, Object> all = redis.opsForHash().entries(hashKey);
List<Map.Entry<Object, Long>> byTs = new ArrayList<>(all.size());
for (Map.Entry<Object, Object> e : all.entrySet()) {
long ts = 0;
if (e.getValue() != null) {
String[] parts = e.getValue().toString().split("\\|", -1);
try {
ts = Long.parseLong(parts[parts.length - 1]);
} catch (NumberFormatException ignore) {
// ts 取不到当最旧,优先淘汰
}
}
byTs.add(Map.entry(e.getKey(), ts));
}
byTs.sort(Map.Entry.comparingByValue());
int toRemove = Math.max(0, byTs.size() - MAX_DRAMAS_PER_IDENTITY);
if (toRemove > 0) {
redis.opsForHash().delete(hashKey,
byTs.subList(0, toRemove).stream().map(Map.Entry::getKey).toArray());
}
}
/** value 中的 source/deviceId 不允许出现分隔符 | 与换行,统一清洗;长度截到 128。 */
private String cleanToken(String token) {
if (token == null) {
return "";
}
String cleaned = token.replace("|", "").replace("\n", "").replace("\r", "").trim();
return cleaned.length() > 128 ? cleaned.substring(0, 128) : cleaned;
}
}
@@ -0,0 +1,43 @@
// Author: fruit.com.cn
package com.fruitshorts.playback.service;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
/**
* 观看进度缓冲落库任务:定时把 Redis 缓冲里的进度按身份分组合并进 watch_progress。
*
* <p>只在开启缓冲(fruitshorts.watch-progress.buffer=true)的应用内生效,
* app-api 的 prod profile 开启,admin-api 与测试环境不受影响。
* fixedDelay 串行执行,同一时刻只有一个落库批次在跑。
*/
@Slf4j
@Component
@RequiredArgsConstructor
@ConditionalOnProperty(name = "fruitshorts.watch-progress.buffer", havingValue = "true")
public class WatchProgressFlusher {
private final WatchProgressService watchProgressService;
@Value("${fruitshorts.watch-progress.flush-batch-size:10000}")
private int batchSize;
@Scheduled(
fixedDelayString = "${fruitshorts.watch-progress.flush-interval-ms:2000}",
initialDelayString = "${fruitshorts.watch-progress.flush-initial-delay-ms:15000}")
public void flush() {
try {
int flushed = watchProgressService.flushBuffered(batchSize);
if (flushed > 0) {
log.debug("观看进度缓冲落库完成,本批次 {} 条", flushed);
}
} catch (Exception e) {
log.warn("观看进度缓冲落库失败,等待下个周期重试: {}", e.getMessage());
}
}
}
@@ -3,9 +3,11 @@
package com.fruitshorts.playback.service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fruitshorts.common.pojo.PageResult;
import com.fruitshorts.common.interaction.InteractionIdentity;
import com.fruitshorts.content.dal.entity.Drama;
import com.fruitshorts.content.dal.entity.DramaEpisode;
import com.fruitshorts.content.service.ContentLocaleTexts;
@@ -17,28 +19,47 @@ import com.fruitshorts.playback.dal.mapper.WatchHistoryMapper;
import com.fruitshorts.playback.dal.mapper.WatchProgressMapper;
import com.fruitshorts.playback.vo.ProgressRespVO;
import com.fruitshorts.playback.vo.ProgressSaveReqVO;
import com.github.benmanes.caffeine.cache.Cache;
import com.github.benmanes.caffeine.cache.Caffeine;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.support.TransactionTemplate;
import org.springframework.util.StringUtils;
import java.time.Duration;
import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;
/**
* 观看进度。
*
* <p>每次上报都写库在高并发下会成为瓶颈(1 万并发 × 每 10 秒一次 ≈ 1000 写/秒),
* 所以生产形态是「先写 Redis,后台任务按 user+drama 合并后批量 UPSERT」。
* 这里先落地权威表与合并规则,缓存层在接入 Redis 时补齐,接口契约不变。
* <p>生产形态(fruitshorts.watch-progress.buffer=true,prod 已开启):
* POST 只写 Redis 缓冲({@link WatchProgressBuffer})立即返回,
* {@link WatchProgressFlusher} 定时把缓冲按「账号或设备 + drama」合并后
* 批量 UPSERT 进权威表,读接口用缓冲覆盖保证读己之写。
* 缓冲关闭或 Redis 异常时自动回退同步写库,接口契约不变。
*
* <p>合并规则:以客户端上报的 positionMs 为准,但只在「更靠后」时覆盖,
* 避免用户换设备后进度倒退。
*/
@Slf4j
@Service
@RequiredArgsConstructor
public class WatchProgressService {
@@ -50,75 +71,119 @@ public class WatchProgressService {
private final WatchHistoryMapper historyMapper;
private final DramaService dramaService;
private final EpisodeService episodeService;
private final ObjectProvider<WatchProgressBuffer> bufferProvider;
private final SqlSessionFactory sqlSessionFactory;
private final TransactionTemplate transactionTemplate;
/** 已发布剧集/短剧的短 TTL 进程内缓存:上报路径的热点校验,免去每次两条 DB 查询。 */
private final Cache<Long, DramaEpisode> episodeCache = Caffeine.newBuilder()
.maximumSize(50_000)
.expireAfterWrite(Duration.ofMinutes(5))
.build();
private final Cache<Long, Drama> dramaCache = Caffeine.newBuilder()
.maximumSize(50_000)
.expireAfterWrite(Duration.ofMinutes(5))
.build();
@Transactional(rollbackFor = Exception.class)
public ProgressRespVO save(Long userId, ProgressSaveReqVO req) {
DramaEpisode episode = episodeService.getPublished(req.getEpisodeId());
Drama drama = dramaService.getPublished(episode.getDramaId());
return save(new InteractionIdentity(userId, ""), req);
}
public ProgressRespVO save(InteractionIdentity identity, ProgressSaveReqVO req) {
DramaEpisode episode = publishedEpisode(req.getEpisodeId());
Drama drama = publishedDrama(episode.getDramaId());
long position = req.getPositionMs() == null ? 0L : Math.max(0L, req.getPositionMs());
long duration = req.getDurationMs() == null ? 0L : Math.max(0L, req.getDurationMs());
boolean finished = Boolean.TRUE.equals(req.getFinished())
|| (duration > 0 && position + FINISH_THRESHOLD_MS >= duration);
LoadedProgress loaded = loadForSave(userId, drama.getId(), episode, position, duration,
finished, req);
WatchProgress existing = loaded.progress();
if (loaded.created()) {
appendHistory(userId, drama.getId(), episode.getId(), position, duration,
existing.getSource(), existing.getDeviceId());
return toResp(progressMapper.selectById(existing.getId()));
// 高并发形态:写 Redis 缓冲立即返回,落库任务稍后合并进权威表
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
if (buffer != null && buffer.put(identity, drama.getId(), episode.getId(),
position, duration, finished, req.getSource(), req.getDeviceId())) {
ProgressRespVO vo = new ProgressRespVO();
vo.setDramaId(drama.getId());
vo.setEpisodeId(episode.getId());
vo.setPositionMs(position);
vo.setDurationMs(duration);
vo.setFinished(finished);
vo.setUpdatedAt(LocalDateTime.now());
return vo;
}
boolean hidden = existing.getDeleted() != null && existing.getDeleted() != 0;
boolean sameEpisode = existing.getEpisodeId() != null
&& existing.getEpisodeId().equals(episode.getId());
if (hidden && sameEpisode) {
existing.setPlayCount((existing.getPlayCount() == null ? 0 : existing.getPlayCount()) + 1);
appendHistory(userId, drama.getId(), episode.getId(), position, duration,
req.getSource(), existing.getDeviceId());
}
if (!sameEpisode) {
// 换集:直接采纳新集与新的播放位置
existing.setEpisodeId(episode.getId());
existing.setPositionMs(position);
existing.setDurationMs(duration);
existing.setFinished(finished ? 1 : 0);
existing.setPlayCount((existing.getPlayCount() == null ? 0 : existing.getPlayCount()) + 1);
appendHistory(userId, drama.getId(), episode.getId(), position, duration,
req.getSource(), existing.getDeviceId());
} else {
existing.setDurationMs(Math.max(existing.getDurationMs() == null ? 0L : existing.getDurationMs(),
duration));
if (position >= (existing.getPositionMs() == null ? 0L : existing.getPositionMs())) {
existing.setPositionMs(position);
}
if (finished) {
existing.setFinished(1);
}
}
existing.setDeviceId(req.getDeviceId() == null ? "" : req.getDeviceId());
existing.setSource(req.getSource() == null ? "" : req.getSource());
if (hidden) {
existing.setDeleted(0);
progressMapper.updateAny(existing);
} else {
progressMapper.updateById(existing);
}
return toResp(progressMapper.selectById(existing.getId()));
return saveToDb(identity, drama, episode, position, duration, finished, req);
}
private LoadedProgress loadForSave(Long userId, Long dramaId, DramaEpisode episode, long position,
/** 同步写库路径:缓冲关闭或 Redis 异常时的回退,合并规则与缓冲落库一致。 */
private ProgressRespVO saveToDb(InteractionIdentity identity, Drama drama, DramaEpisode episode,
long position, long duration, boolean finished,
ProgressSaveReqVO req) {
return transactionTemplate.execute(status -> {
LoadedProgress loaded = loadForSave(identity, drama.getId(), episode, position, duration,
finished, req);
WatchProgress existing = loaded.progress();
if (loaded.created()) {
appendHistory(identity, drama.getId(), episode.getId(), position, duration,
existing.getSource(), existing.getDeviceId());
return toResp(progressMapper.selectById(existing.getId()));
}
boolean hidden = existing.getDeleted() != null && existing.getDeleted() != 0;
boolean sameEpisode = existing.getEpisodeId() != null
&& existing.getEpisodeId().equals(episode.getId());
if (hidden && sameEpisode) {
existing.setPlayCount((existing.getPlayCount() == null ? 0 : existing.getPlayCount()) + 1);
appendHistory(identity, drama.getId(), episode.getId(), position, duration,
req.getSource(), existing.getDeviceId());
}
if (!sameEpisode) {
// 换集:直接采纳新集与新的播放位置
existing.setEpisodeId(episode.getId());
existing.setPositionMs(position);
existing.setDurationMs(duration);
existing.setFinished(finished ? 1 : 0);
existing.setPlayCount((existing.getPlayCount() == null ? 0 : existing.getPlayCount()) + 1);
appendHistory(identity, drama.getId(), episode.getId(), position, duration,
req.getSource(), existing.getDeviceId());
} else {
existing.setDurationMs(Math.max(existing.getDurationMs() == null ? 0L : existing.getDurationMs(),
duration));
if (position >= (existing.getPositionMs() == null ? 0L : existing.getPositionMs())) {
existing.setPositionMs(position);
}
if (finished) {
existing.setFinished(1);
}
}
existing.setDeviceId(identity.authenticated()
? (req.getDeviceId() == null ? "" : req.getDeviceId())
: identity.deviceId());
existing.setSource(req.getSource() == null ? "" : req.getSource());
if (hidden) {
existing.setDeleted(0);
progressMapper.updateAny(existing);
} else {
progressMapper.updateById(existing);
}
return toResp(progressMapper.selectById(existing.getId()));
});
}
private LoadedProgress loadForSave(InteractionIdentity identity, Long dramaId,
DramaEpisode episode, long position,
long duration, boolean finished, ProgressSaveReqVO req) {
WatchProgress existing = progressMapper.selectAny(userId, dramaId);
WatchProgress existing = progressMapper.selectAnyForIdentity(
identity.userId(), identity.deviceId(), dramaId);
if (existing != null) {
return new LoadedProgress(existing, false);
}
WatchProgress created = newProgress(userId, dramaId, episode, position, duration, finished, req);
WatchProgress created = newProgress(
identity, dramaId, episode, position, duration, finished, req);
try {
progressMapper.insert(created);
return new LoadedProgress(created, true);
} catch (DuplicateKeyException ex) {
WatchProgress raced = progressMapper.selectAny(userId, dramaId);
WatchProgress raced = progressMapper.selectAnyForIdentity(
identity.userId(), identity.deviceId(), dramaId);
if (raced == null) {
throw ex;
}
@@ -126,17 +191,20 @@ public class WatchProgressService {
}
}
private WatchProgress newProgress(Long userId, Long dramaId, DramaEpisode episode, long position,
private WatchProgress newProgress(InteractionIdentity identity, Long dramaId,
DramaEpisode episode, long position,
long duration, boolean finished, ProgressSaveReqVO req) {
WatchProgress progress = new WatchProgress();
progress.setUserId(userId);
progress.setUserId(identity.userId());
progress.setDramaId(dramaId);
progress.setEpisodeId(episode.getId());
progress.setPositionMs(position);
progress.setDurationMs(duration);
progress.setFinished(finished ? 1 : 0);
progress.setPlayCount(1);
progress.setDeviceId(req.getDeviceId() == null ? "" : req.getDeviceId());
progress.setDeviceId(identity.authenticated()
? (req.getDeviceId() == null ? "" : req.getDeviceId())
: identity.deviceId());
progress.setSource(req.getSource() == null ? "" : req.getSource());
return progress;
}
@@ -145,37 +213,309 @@ public class WatchProgressService {
}
public ProgressRespVO ofDrama(Long userId, Long dramaId) {
WatchProgress progress = progressMapper.selectOne(Wrappers.<WatchProgress>lambdaQuery()
.eq(WatchProgress::getUserId, userId)
return ofDrama(new InteractionIdentity(userId, ""), dramaId);
}
public ProgressRespVO ofDrama(InteractionIdentity identity, Long dramaId) {
WatchProgress progress = progressMapper.selectOne(identityQuery(identity)
.eq(WatchProgress::getDramaId, dramaId)
.last("limit 1"));
// 读己之写:缓冲里有比库行更新的状态时以缓冲为准
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
if (buffer != null) {
WatchProgressBuffer.BufferedProgress buffered = buffer.get(identity, dramaId);
if (buffered != null
&& (progress == null
|| buffered.ts() > toMillis(progress.getUpdatedAt()))) {
return toResp(buffered, progress);
}
}
return progress == null ? null : toResp(progress);
}
/** 分页查询最近观看,按更新时间倒序并以 id 兜底保证稳定。 */
public PageResult<ProgressRespVO> recentPage(Long userId, int pageIndex, int pageSize) {
return recentPage(new InteractionIdentity(userId, ""), pageIndex, pageSize);
}
public PageResult<ProgressRespVO> recentPage(
InteractionIdentity identity, int pageIndex, int pageSize) {
int safeIndex = Math.max(pageIndex, 1);
int safeSize = Math.min(Math.max(pageSize, 1), MAX_PAGE_SIZE);
Page<WatchProgress> request = new Page<>(safeIndex, safeSize);
IPage<WatchProgress> result = progressMapper.selectPage(request,
Wrappers.<WatchProgress>lambdaQuery()
.eq(WatchProgress::getUserId, userId)
identityQuery(identity)
.orderByDesc(WatchProgress::getUpdatedAt)
.orderByDesc(WatchProgress::getId));
return new PageResult<>(result.getTotal(), enrichDisplay(result.getRecords()));
// 缓冲的条目都是最近上报的,只会出现在第一页;缓冲关闭时走原逻辑
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
if (buffer == null || safeIndex > 1) {
return new PageResult<>(result.getTotal(), enrichDisplay(result.getRecords()));
}
Map<Long, WatchProgressBuffer.BufferedProgress> states = buffer.getAll(identity);
if (states.isEmpty()) {
return new PageResult<>(result.getTotal(), enrichDisplay(result.getRecords()));
}
List<WatchProgress> merged = new ArrayList<>();
Set<Long> pageDramas = new HashSet<>();
for (WatchProgress row : result.getRecords()) {
pageDramas.add(row.getDramaId());
WatchProgressBuffer.BufferedProgress buffered = row.getDramaId() == null
? null
: states.get(row.getDramaId());
merged.add(overlayRow(row, buffered));
}
long total = result.getTotal();
// 缓冲里有、页里没有的剧补进第一页(刚上报首集、库里还没落的场景)
List<WatchProgressBuffer.BufferedProgress> extra = states.entrySet().stream()
.filter(e -> !pageDramas.contains(e.getKey()))
.map(Map.Entry::getValue)
.sorted(Comparator.comparingLong(WatchProgressBuffer.BufferedProgress::ts).reversed())
.toList();
if (!extra.isEmpty()) {
List<Long> extraDramaIds = extra.stream()
.map(WatchProgressBuffer.BufferedProgress::dramaId)
.toList();
Map<Long, WatchProgress> rowByDrama = progressMapper.selectAnyForIdentityIn(
identity.userId(), identity.deviceId(), extraDramaIds)
.stream()
.collect(Collectors.toMap(WatchProgress::getDramaId, row -> row, (a, b) -> a));
for (WatchProgressBuffer.BufferedProgress buffered : extra) {
WatchProgress existing = rowByDrama.get(buffered.dramaId());
// 用户已删除该剧(库里是逻辑删除行):缓冲状态不能被读覆盖层再读回来
if (existing != null && existing.getDeleted() != null && existing.getDeleted() != 0) {
continue;
}
merged.add(transientRow(buffered));
if (existing == null) {
total++;
}
}
}
merged.sort(Comparator.comparing(WatchProgress::getUpdatedAt,
Comparator.nullsFirst(Comparator.naturalOrder())).reversed());
List<WatchProgress> page = merged.size() > safeSize
? merged.subList(0, safeSize)
: merged;
return new PageResult<>(total, enrichDisplay(page));
}
@Transactional(rollbackFor = Exception.class)
public void remove(Long userId, Long dramaId) {
progressMapper.delete(Wrappers.<WatchProgress>lambdaQuery()
.eq(WatchProgress::getUserId, userId)
remove(new InteractionIdentity(userId, ""), dramaId);
}
@Transactional(rollbackFor = Exception.class)
public void remove(InteractionIdentity identity, Long dramaId) {
// 缓冲是读接口的权威来源:先删缓冲再删库行,否则删掉的记录会被读覆盖层读回、
// 并被下一次落库把行复活。
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
if (buffer != null && dramaId != null) {
buffer.remove(identity, dramaId);
}
progressMapper.delete(identityQuery(identity)
.eq(WatchProgress::getDramaId, dramaId));
}
@Transactional(rollbackFor = Exception.class)
public void clear(Long userId) {
progressMapper.delete(Wrappers.<WatchProgress>lambdaQuery()
.eq(WatchProgress::getUserId, userId));
clear(new InteractionIdentity(userId, ""));
}
@Transactional(rollbackFor = Exception.class)
public void clear(InteractionIdentity identity) {
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
if (buffer != null) {
buffer.clear(identity);
}
progressMapper.delete(identityQuery(identity));
}
/**
* 缓冲落库:由 {@link WatchProgressFlusher} 定时调用。
* 消费 dirty 队列,pipeline 读状态、分块成对查行,先在内存里合并出
* 「新增进度 / 新增流水 / 更新行」三组,再在一个 JDBC batch 会话内
* 按类型连续攒批执行、一次提交(交替提交会退化成逐条往返,吞吐只有约 200 行/秒)。
*
* @return 本批次合并的进度条数
*/
public int flushBuffered(int batchSize) {
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
if (buffer == null) {
return 0;
}
List<WatchProgressBuffer.IdentityRef> refs = buffer.drain(batchSize);
if (refs.isEmpty()) {
return 0;
}
try {
return flushRefs(buffer, refs);
} catch (Exception ex) {
log.warn("观看进度缓冲合并失败,整批退回队列稍后重试: {}", ex.getMessage());
buffer.requeue(refs);
return 0;
}
}
private int flushRefs(WatchProgressBuffer buffer, List<WatchProgressBuffer.IdentityRef> refs) {
// 1) pipeline 批量读缓冲状态,key -> (dramaId -> state)
List<String> identityKeys = refs.stream()
.map(WatchProgressBuffer.IdentityRef::key)
.distinct()
.toList();
Map<String, Map<Long, WatchProgressBuffer.BufferedProgress>> statesByIdentity =
buffer.getAllBatch(identityKeys);
// 2) 组装待合并条目;状态读不到视为异常(ref 必然来自 put),整批退回
List<WatchProgressBuffer.IdentityRef> pendingRefs = new ArrayList<>(refs.size());
List<WatchProgressBuffer.BufferedProgress> pendingStates = new ArrayList<>(refs.size());
List<WatchProgressMapper.IdentityDramaPair> pairs = new ArrayList<>(refs.size());
for (WatchProgressBuffer.IdentityRef ref : refs) {
WatchProgressBuffer.BufferedProgress state =
statesByIdentity.getOrDefault(ref.key(), Map.of()).get(ref.dramaId());
if (state == null) {
continue;
}
pendingRefs.add(ref);
pendingStates.add(state);
pairs.add(new WatchProgressMapper.IdentityDramaPair(
ref.userId(), ref.deviceId(), ref.dramaId()));
}
if (pendingRefs.isEmpty()) {
// 缓冲状态已消失(用户清空/删除历史,或 HASH 12h TTL 到期)。此时 ref 已无
// 可落库的内容,必须丢弃:退回队列会在下个周期被原样弹出、再清空、再退回,
// 这批僵尸条目将永久占住 dirty 队列配额。
log.debug("观看进度缓冲条目已无对应状态,丢弃 {} 条", refs.size());
return 0;
}
// 3) 分块成对查询已有行(每块 100 组 OR 索引条件)
Map<String, WatchProgress> rowByPair = new HashMap<>(pendingRefs.size());
for (int i = 0; i < pairs.size(); i += 100) {
List<WatchProgressMapper.IdentityDramaPair> chunk =
pairs.subList(i, Math.min(i + 100, pairs.size()));
for (WatchProgress row : progressMapper.selectBatchForPairs(chunk)) {
rowByPair.put(rowKey(row.getUserId(), row.getDeviceId(), row.getDramaId()), row);
}
}
// 4) 计算阶段:纯内存合并出「新增进度 / 新增流水 / 更新行」三组
List<WatchProgress> insertProgress = new ArrayList<>(pendingRefs.size());
List<WatchHistory> insertHistory = new ArrayList<>(pendingRefs.size());
List<WatchProgress> updateRows = new ArrayList<>(pendingRefs.size());
for (int i = 0; i < pendingRefs.size(); i++) {
WatchProgress row = rowByPair.get(rowKey(
pendingRefs.get(i).userId(),
pendingRefs.get(i).deviceId(),
pendingStates.get(i).dramaId()));
planOne(pendingRefs.get(i).toIdentity(), pendingStates.get(i), row,
insertProgress, insertHistory, updateRows);
}
// 5) 执行阶段:同一个 batch 会话按类型连续执行,最后一次提交。
// MyBatis BATCH 执行器只合并「连续同形状」的语句,进度/流水交替提交会
// 退化成逐条网络往返(实测只有约 200 行/秒),必须分组攒批。
try (SqlSession session = sqlSessionFactory.openSession(ExecutorType.BATCH, false)) {
WatchProgressMapper batchProgress = session.getMapper(WatchProgressMapper.class);
WatchHistoryMapper batchHistory = session.getMapper(WatchHistoryMapper.class);
for (WatchProgress created : insertProgress) {
batchProgress.insert(created);
}
for (WatchHistory history : insertHistory) {
batchHistory.insert(history);
}
for (WatchProgress row : updateRows) {
batchProgress.updateRowAny(row);
}
session.commit();
}
return pendingRefs.size();
}
/** 计算单条缓冲的合并动作:只改内存对象不触库,产出归入三组待执行列表。 */
private void planOne(InteractionIdentity identity,
WatchProgressBuffer.BufferedProgress state,
WatchProgress row,
List<WatchProgress> insertProgress,
List<WatchHistory> insertHistory,
List<WatchProgress> updateRows) {
if (row == null) {
WatchProgress created = new WatchProgress();
created.setUserId(identity.userId());
created.setDramaId(state.dramaId());
created.setEpisodeId(state.episodeId());
created.setPositionMs(state.positionMs());
created.setDurationMs(state.durationMs());
created.setFinished(state.finished() ? 1 : 0);
created.setPlayCount(1);
created.setDeviceId(identity.authenticated()
? state.deviceId()
: identity.deviceId());
created.setSource(state.source());
insertProgress.add(created);
insertHistory.add(historyOf(identity, state, created.getDeviceId()));
return;
}
boolean hidden = row.getDeleted() != null && row.getDeleted() != 0;
long rowTs = toMillis(row.getUpdatedAt());
if (hidden && rowTs > 0 && state.ts() <= rowTs) {
// 逻辑删除会刷新 updated_at:上报时间早于它的条目是删除前排队的迟到数据,
// 丢弃即可,否则会把已删记录重新写活。
return;
}
boolean sameEpisode = row.getEpisodeId() != null
&& row.getEpisodeId().equals(state.episodeId());
if (hidden && sameEpisode) {
row.setPlayCount((row.getPlayCount() == null ? 0 : row.getPlayCount()) + 1);
insertHistory.add(historyOf(identity, state, row.getDeviceId()));
} else if (!sameEpisode) {
// 换集:直接采纳新集与新的播放位置
row.setEpisodeId(state.episodeId());
row.setPositionMs(state.positionMs());
row.setDurationMs(state.durationMs());
row.setFinished(state.finished() ? 1 : 0);
row.setPlayCount((row.getPlayCount() == null ? 0 : row.getPlayCount()) + 1);
insertHistory.add(historyOf(identity, state, row.getDeviceId()));
} else {
row.setDurationMs(Math.max(row.getDurationMs() == null ? 0L : row.getDurationMs(),
state.durationMs()));
if (state.positionMs() >= (row.getPositionMs() == null ? 0L : row.getPositionMs())) {
row.setPositionMs(state.positionMs());
}
if (state.finished()) {
row.setFinished(1);
}
}
row.setDeviceId(identity.authenticated() ? state.deviceId() : identity.deviceId());
row.setSource(state.source());
if (hidden) {
row.setDeleted(0);
}
updateRows.add(row);
}
/** 组装一条观看流水(首次观剧/换集各记一条)。 */
private WatchHistory historyOf(InteractionIdentity identity,
WatchProgressBuffer.BufferedProgress state,
String deviceId) {
WatchHistory history = new WatchHistory();
history.setUserId(identity.userId());
history.setDramaId(state.dramaId());
history.setEpisodeId(state.episodeId());
history.setPositionMs(state.positionMs());
history.setDurationMs(state.durationMs());
history.setSource(state.source() == null ? "" : state.source());
history.setDeviceId(identity.authenticated()
? (deviceId == null ? "" : deviceId)
: identity.deviceId());
return history;
}
/** 行定位键:登录身份用 userId,匿名用 deviceId。 */
private String rowKey(Long userId, String deviceId, Long dramaId) {
return (userId != null ? "u:" + userId : "d:" + deviceId) + ":" + dramaId;
}
private ProgressRespVO toResp(WatchProgress progress) {
@@ -190,6 +530,68 @@ public class WatchProgressService {
return vo;
}
/** 以缓冲状态为主、库行 playCount 兜底构建续看响应。 */
private ProgressRespVO toResp(WatchProgressBuffer.BufferedProgress buffered, WatchProgress row) {
ProgressRespVO vo = new ProgressRespVO();
vo.setDramaId(buffered.dramaId());
vo.setEpisodeId(buffered.episodeId());
vo.setPositionMs(buffered.positionMs());
vo.setDurationMs(buffered.durationMs());
vo.setFinished(buffered.finished());
vo.setPlayCount(row == null || row.getPlayCount() == null ? 1 : row.getPlayCount());
vo.setUpdatedAt(fromMillis(buffered.ts()));
return vo;
}
/** 库行被缓冲覆盖时返回覆盖后的瞬态行;缓冲不比库行新时原样返回。 */
private WatchProgress overlayRow(WatchProgress row, WatchProgressBuffer.BufferedProgress buffered) {
if (buffered == null || row.getUpdatedAt() == null
|| buffered.ts() <= toMillis(row.getUpdatedAt())) {
return row;
}
WatchProgress overlaid = transientRow(buffered);
overlaid.setPlayCount(row.getPlayCount());
return overlaid;
}
private WatchProgress transientRow(WatchProgressBuffer.BufferedProgress buffered) {
WatchProgress row = new WatchProgress();
row.setDramaId(buffered.dramaId());
row.setEpisodeId(buffered.episodeId());
row.setPositionMs(buffered.positionMs());
row.setDurationMs(buffered.durationMs());
row.setFinished(buffered.finished() ? 1 : 0);
row.setPlayCount(1);
row.setUpdatedAt(fromMillis(buffered.ts()));
return row;
}
private long toMillis(LocalDateTime time) {
return time == null ? 0L : time.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
}
private LocalDateTime fromMillis(long epochMillis) {
return LocalDateTime.ofInstant(Instant.ofEpochMilli(epochMillis), ZoneId.systemDefault());
}
private DramaEpisode publishedEpisode(Long episodeId) {
DramaEpisode episode = episodeCache.getIfPresent(episodeId);
if (episode == null) {
episode = episodeService.getPublished(episodeId);
episodeCache.put(episodeId, episode);
}
return episode;
}
private Drama publishedDrama(Long dramaId) {
Drama drama = dramaCache.getIfPresent(dramaId);
if (drama == null) {
drama = dramaService.getPublished(dramaId);
dramaCache.put(dramaId, drama);
}
return drama;
}
private List<ProgressRespVO> enrichDisplay(List<WatchProgress> records) {
if (records.isEmpty()) {
return List.of();
@@ -248,19 +650,207 @@ public class WatchProgressService {
* 追加一条观看流水。只在「首次观看一部剧」和「切换剧集」时调用,
* 每 10 秒的进度上报不写这里 —— 否则单集会产出几十条噪音。
*/
private void appendHistory(Long userId, Long dramaId, Long episodeId, long positionMs,
private void appendHistory(InteractionIdentity identity, Long dramaId, Long episodeId,
long positionMs,
long durationMs, String source, String deviceId) {
WatchHistory history = new WatchHistory();
history.setUserId(userId);
history.setUserId(identity.userId());
history.setDramaId(dramaId);
history.setEpisodeId(episodeId);
history.setPositionMs(positionMs);
history.setDurationMs(durationMs);
history.setSource(source == null ? "" : source);
history.setDeviceId(deviceId == null ? "" : deviceId);
history.setDeviceId(identity.authenticated()
? (deviceId == null ? "" : deviceId)
: identity.deviceId());
historyMapper.insert(history);
}
/**
* 把某个身份缓冲里尚未落库的进度立即写进权威表。
* 登录合并前必须先调用:合并只扫库表,缓冲里的设备进度会被漏掉,
* 随后定时落库又会生成一条登录用户看不到的匿名孤儿行。
*
* @return 本次合并的条目数
*/
public int flushIdentity(InteractionIdentity identity) {
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
if (buffer == null) {
return 0;
}
Map<Long, WatchProgressBuffer.BufferedProgress> states = buffer.getAll(identity);
if (states.isEmpty()) {
return 0;
}
String key = buffer.identityKey(identity);
List<WatchProgressBuffer.IdentityRef> refs = states.keySet().stream()
.map(dramaId -> new WatchProgressBuffer.IdentityRef(
key, identity.userId(), identity.deviceId(), dramaId))
.toList();
try {
return flushRefs(buffer, refs);
} catch (Exception ex) {
// 预写失败不影响合并本身:dirty 队列里的引用还在,下个周期会再落库
log.warn("观看进度缓冲立即落库失败,跳过预写: {}", ex.getMessage());
return 0;
}
}
/**
* 登录时合并设备进度和观看流水。
*
* <p>同一部剧只保留进度更靠前的账号记录:先比较集号,同集再比较播放位置,
* 最后用更新时间兜底。playCount 只取保留记录,不把两条记录的计数相加。
*/
@Transactional(rollbackFor = Exception.class)
public MergeResult mergeDeviceIntoUser(Long userId, String deviceId) {
if (userId == null || !StringUtils.hasText(deviceId)) {
return new MergeResult(0, 0, 0, 0);
}
InteractionIdentity deviceIdentity = new InteractionIdentity(null, deviceId);
// 缓冲里可能还有没落库的设备进度,先落库再合并,否则这段进度会被漏掉
flushIdentity(deviceIdentity);
List<WatchProgress> deviceProgress = progressMapper.selectList(
Wrappers.<WatchProgress>lambdaQuery()
.isNull(WatchProgress::getUserId)
.eq(WatchProgress::getDeviceId, deviceId));
Map<Long, Integer> episodeNumbers = episodeNumbers(deviceProgress);
int progressMigrated = 0;
int progressIgnored = 0;
for (WatchProgress device : deviceProgress) {
WatchProgress account = progressMapper.selectAnyForIdentity(
userId, "", device.getDramaId());
if (account == null) {
device.setUserId(userId);
progressMapper.updateById(device);
progressMigrated++;
continue;
}
if (progressAhead(device, account, episodeNumbers)) {
copyProgress(device, account);
account.setDeleted(0);
progressMapper.updateAny(account);
progressMigrated++;
} else {
progressIgnored++;
}
progressMapper.deleteById(device.getId());
}
List<WatchHistory> deviceHistory = historyMapper.selectList(
Wrappers.<WatchHistory>lambdaQuery()
.isNull(WatchHistory::getUserId)
.eq(WatchHistory::getDeviceId, deviceId));
int historyMigrated = 0;
int historyIgnored = 0;
for (WatchHistory device : deviceHistory) {
WatchHistory account = historyMapper.selectOne(
Wrappers.<WatchHistory>lambdaQuery()
.eq(WatchHistory::getUserId, userId)
.eq(WatchHistory::getDramaId, device.getDramaId())
.eq(WatchHistory::getEpisodeId, device.getEpisodeId())
.orderByDesc(WatchHistory::getWatchedAt)
.last("limit 1"));
if (account == null) {
device.setUserId(userId);
historyMapper.updateById(device);
historyMigrated++;
continue;
}
if (historyAhead(device, account)) {
account.setPositionMs(device.getPositionMs());
account.setDurationMs(device.getDurationMs());
account.setSource(device.getSource());
account.setWatchedAt(device.getWatchedAt());
historyMapper.updateById(account);
historyMigrated++;
} else {
historyIgnored++;
}
historyMapper.deleteById(device.getId());
}
// 设备侧进度已迁走或淘汰,缓冲留着只会让读覆盖层把已删行读回来
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
if (buffer != null) {
buffer.clear(deviceIdentity);
}
return new MergeResult(
progressMigrated, progressIgnored, historyMigrated, historyIgnored);
}
public record MergeResult(
int progressMigrated,
int progressIgnored,
int historyMigrated,
int historyIgnored) {
}
private Map<Long, Integer> episodeNumbers(List<WatchProgress> progress) {
List<Long> episodeIds = progress.stream()
.map(WatchProgress::getEpisodeId)
.filter(Objects::nonNull)
.distinct()
.toList();
Map<Long, Integer> result = new HashMap<>();
if (episodeIds.isEmpty()) {
return result;
}
for (DramaEpisode episode : episodeService.listByIds(episodeIds)) {
result.put(episode.getId(), episode.getEpNo());
}
return result;
}
private boolean progressAhead(
WatchProgress candidate,
WatchProgress current,
Map<Long, Integer> episodeNumbers) {
Integer candidateNo = episodeNumbers.get(candidate.getEpisodeId());
Integer currentNo = episodeNumbers.get(current.getEpisodeId());
if (candidateNo != null && currentNo != null && !candidateNo.equals(currentNo)) {
return candidateNo > currentNo;
}
long candidatePosition = candidate.getPositionMs() == null ? 0L : candidate.getPositionMs();
long currentPosition = current.getPositionMs() == null ? 0L : current.getPositionMs();
if (candidatePosition != currentPosition) {
return candidatePosition > currentPosition;
}
return candidate.getUpdatedAt() != null
&& (current.getUpdatedAt() == null
|| candidate.getUpdatedAt().isAfter(current.getUpdatedAt()));
}
private void copyProgress(WatchProgress source, WatchProgress target) {
target.setEpisodeId(source.getEpisodeId());
target.setPositionMs(source.getPositionMs());
target.setDurationMs(source.getDurationMs());
target.setFinished(source.getFinished());
target.setPlayCount(source.getPlayCount());
target.setDeviceId(source.getDeviceId());
target.setSource(source.getSource());
}
private boolean historyAhead(WatchHistory candidate, WatchHistory current) {
long candidatePosition = candidate.getPositionMs() == null ? 0L : candidate.getPositionMs();
long currentPosition = current.getPositionMs() == null ? 0L : current.getPositionMs();
if (candidatePosition != currentPosition) {
return candidatePosition > currentPosition;
}
return candidate.getWatchedAt() != null
&& (current.getWatchedAt() == null
|| candidate.getWatchedAt().isAfter(current.getWatchedAt()));
}
private LambdaQueryWrapper<WatchProgress> identityQuery(InteractionIdentity identity) {
if (identity.authenticated()) {
return Wrappers.<WatchProgress>lambdaQuery()
.eq(WatchProgress::getUserId, identity.userId());
}
return Wrappers.<WatchProgress>lambdaQuery()
.isNull(WatchProgress::getUserId)
.eq(WatchProgress::getDeviceId, identity.deviceId());
}
/** 供后续批量合并任务使用:只更新进度字段,避免整行覆盖。 */
public void mergeInto(Long userId, Long dramaId, Long episodeId, long positionMs) {
progressMapper.update(null, Wrappers.<WatchProgress>lambdaUpdate()
@@ -4,6 +4,7 @@ package com.fruitshorts.playback.vo;
import jakarta.validation.constraints.NotEmpty;
import jakarta.validation.constraints.NotNull;
import jakarta.validation.constraints.Size;
import java.util.List;
import lombok.Data;
@@ -11,6 +12,10 @@ import lombok.Data;
@Data
public class FavoriteRemoveReqVO {
@NotEmpty(message = "缺少 dramaIds")
@NotEmpty(message = "dramaIds 不能为空")
@Size(max = 200, message = "dramaIds 最多 200 条")
private List<@NotNull Long> dramaIds;
/** 未登录且未携带 X-Device-Id 时的兜底设备号。 */
private String deviceId;
}
@@ -11,4 +11,7 @@ public class FavoriteToggleReqVO {
@NotNull(message = "缺少 dramaId")
private Long dramaId;
/** 兼容旧调用;正常客户端通过 X-Device-Id 下发。 */
private String deviceId;
}
+5
View File
@@ -25,5 +25,10 @@
<groupId>com.fruitshorts</groupId>
<artifactId>fruitshorts-module-system</artifactId>
</dependency>
<!-- 离线 IP → 国家解析,xdb 数据在 src/main/resources/ip2region 下 -->
<dependency>
<groupId>org.lionsoul</groupId>
<artifactId>ip2region</artifactId>
</dependency>
</dependencies>
</project>
@@ -0,0 +1,32 @@
// Author: fruit.com.cn
package com.fruitshorts.user.dal.entity;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableLogic;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import java.time.LocalDateTime;
/**
* 设备平台:按 device_id 记录客户端是 iOS 还是 Android。
* 与会话不同,它不依赖用户身份,游客请求也会建档。
*/
@Data
@TableName("app_device_profile")
public class AppDeviceProfile {
@TableId(type = IdType.ASSIGN_ID)
private Long id;
private String deviceId;
private String platform;
private LocalDateTime firstSeenAt;
private LocalDateTime lastSeenAt;
private LocalDateTime createdAt;
private LocalDateTime updatedAt;
@TableLogic
private Integer deleted;
}
@@ -0,0 +1,28 @@
// Author: fruit.com.cn
package com.fruitshorts.user.dal.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.fruitshorts.user.dal.entity.AppDeviceProfile;
import org.apache.ibatis.annotations.Insert;
import org.apache.ibatis.annotations.Param;
public interface AppDeviceProfileMapper extends BaseMapper<AppDeviceProfile> {
/**
* 按 device_id 建档或刷新平台。
* 空平台不覆盖已有值:老版本客户端不带平台头,不能把已记录的值抹掉。
*/
@Insert("""
INSERT INTO app_device_profile (id, device_id, platform,
first_seen_at, last_seen_at, created_at, updated_at, deleted)
VALUES (#{p.id}, #{p.deviceId}, #{p.platform},
CURRENT_TIMESTAMP, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP, 0)
ON CONFLICT (device_id) DO UPDATE SET
platform = CASE WHEN EXCLUDED.platform <> '' THEN EXCLUDED.platform
ELSE app_device_profile.platform END,
last_seen_at = CURRENT_TIMESTAMP,
updated_at = CURRENT_TIMESTAMP
""")
int upsert(@Param("p") AppDeviceProfile profile);
}
@@ -4,6 +4,27 @@ package com.fruitshorts.user.dal.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.fruitshorts.user.dal.entity.AppUserSession;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import java.util.List;
public interface AppUserSessionMapper extends BaseMapper<AppUserSession> {
/**
* 最近一次登录会话平台等于指定值的用户。
* 必须取「最近一次」而不是「存在过」:否则同一用户在 Android 登录过、
* 后来又换 iOS,会被两端都命中,筛选结果与实际展示的系统不一致。
*/
@Select("""
SELECT t.user_id
FROM (
SELECT DISTINCT ON (user_id) user_id, platform
FROM app_user_session
WHERE deleted = 0
ORDER BY user_id, last_active_at DESC NULLS LAST, id DESC
) t
WHERE t.platform = #{platform}
""")
List<Long> selectUserIdsByLatestPlatform(@Param("platform") String platform);
}
@@ -0,0 +1,219 @@
// Author: fruit.com.cn
package com.fruitshorts.user.enums;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.StringUtils;
import java.util.HashMap;
import java.util.Locale;
import java.util.Map;
/**
* 国家/地区 → 语言、时区、币种。
*
* <p>App 目前只内置 en / id / zh 三种语言,除印尼、中国及台港澳外的市场统一落到 en-US;
* 时区与币种按国家真实值写入,避免所有用户都变成 IDR。
*
* <p>未收录的国家按 {@link #DEFAULT}(印尼)处理,与历史行为保持一致。
*/
@Slf4j
public enum CountryProfile {
ID("id-ID", "Asia/Jakarta", "IDR"),
MY("en-US", "Asia/Kuala_Lumpur", "MYR"),
SG("en-US", "Asia/Singapore", "SGD"),
TH("en-US", "Asia/Bangkok", "THB"),
VN("en-US", "Asia/Ho_Chi_Minh", "VND"),
PH("en-US", "Asia/Manila", "PHP"),
KH("en-US", "Asia/Phnom_Penh", "KHR"),
MM("en-US", "Asia/Yangon", "MMK"),
LA("en-US", "Asia/Vientiane", "LAK"),
BN("en-US", "Asia/Brunei", "BND"),
CN("zh-CN", "Asia/Shanghai", "CNY"),
TW("zh-TW", "Asia/Taipei", "TWD"),
HK("zh-HK", "Asia/Hong_Kong", "HKD"),
MO("zh-HK", "Asia/Macau", "MOP"),
JP("en-US", "Asia/Tokyo", "JPY"),
KR("en-US", "Asia/Seoul", "KRW"),
IN("en-US", "Asia/Kolkata", "INR"),
PK("en-US", "Asia/Karachi", "PKR"),
BD("en-US", "Asia/Dhaka", "BDT"),
LK("en-US", "Asia/Colombo", "LKR"),
NP("en-US", "Asia/Kathmandu", "NPR"),
US("en-US", "America/New_York", "USD"),
CA("en-US", "America/Toronto", "CAD"),
MX("en-US", "America/Mexico_City", "MXN"),
BR("en-US", "America/Sao_Paulo", "BRL"),
AR("en-US", "America/Argentina/Buenos_Aires", "ARS"),
CL("en-US", "America/Santiago", "CLP"),
CO("en-US", "America/Bogota", "COP"),
PE("en-US", "America/Lima", "PEN"),
VE("en-US", "America/Caracas", "VES"),
EC("en-US", "America/Guayaquil", "USD"),
GT("en-US", "America/Guatemala", "GTQ"),
DO("en-US", "America/Santo_Domingo", "DOP"),
GB("en-US", "Europe/London", "GBP"),
DE("en-US", "Europe/Berlin", "EUR"),
FR("en-US", "Europe/Paris", "EUR"),
ES("en-US", "Europe/Madrid", "EUR"),
IT("en-US", "Europe/Rome", "EUR"),
NL("en-US", "Europe/Amsterdam", "EUR"),
PL("en-US", "Europe/Warsaw", "PLN"),
RU("en-US", "Europe/Moscow", "RUB"),
UA("en-US", "Europe/Kyiv", "UAH"),
TR("en-US", "Europe/Istanbul", "TRY"),
SA("en-US", "Asia/Riyadh", "SAR"),
AE("en-US", "Asia/Dubai", "AED"),
IL("en-US", "Asia/Jerusalem", "ILS"),
EG("en-US", "Africa/Cairo", "EGP"),
NG("en-US", "Africa/Lagos", "NGN"),
ZA("en-US", "Africa/Johannesburg", "ZAR"),
KE("en-US", "Africa/Nairobi", "KES"),
MA("en-US", "Africa/Casablanca", "MAD"),
AU("en-US", "Australia/Sydney", "AUD"),
NZ("en-US", "Pacific/Auckland", "NZD");
public static final CountryProfile DEFAULT = ID;
/** ip2region 返回的中文国家名 → ISO-2,只收录上面配置过的国家。 */
private static final Map<String, String> IP_COUNTRY_NAMES = Map.ofEntries(
Map.entry("印度尼西亚", "ID"),
Map.entry("马来西亚", "MY"),
Map.entry("新加坡", "SG"),
Map.entry("泰国", "TH"),
Map.entry("越南", "VN"),
Map.entry("菲律宾", "PH"),
Map.entry("柬埔寨", "KH"),
Map.entry("缅甸", "MM"),
Map.entry("老挝", "LA"),
Map.entry("文莱", "BN"),
Map.entry("中国", "CN"),
Map.entry("日本", "JP"),
Map.entry("韩国", "KR"),
Map.entry("印度", "IN"),
Map.entry("巴基斯坦", "PK"),
Map.entry("孟加拉国", "BD"),
Map.entry("斯里兰卡", "LK"),
Map.entry("尼泊尔", "NP"),
Map.entry("美国", "US"),
Map.entry("加拿大", "CA"),
Map.entry("墨西哥", "MX"),
Map.entry("巴西", "BR"),
Map.entry("阿根廷", "AR"),
Map.entry("智利", "CL"),
Map.entry("哥伦比亚", "CO"),
Map.entry("秘鲁", "PE"),
Map.entry("委内瑞拉", "VE"),
Map.entry("厄瓜多尔", "EC"),
Map.entry("危地马拉", "GT"),
Map.entry("多米尼加", "DO"),
Map.entry("英国", "GB"),
Map.entry("德国", "DE"),
Map.entry("法国", "FR"),
Map.entry("西班牙", "ES"),
Map.entry("意大利", "IT"),
Map.entry("荷兰", "NL"),
Map.entry("波兰", "PL"),
Map.entry("俄罗斯", "RU"),
Map.entry("乌克兰", "UA"),
Map.entry("土耳其", "TR"),
Map.entry("沙特阿拉伯", "SA"),
Map.entry("阿联酋", "AE"),
Map.entry("以色列", "IL"),
Map.entry("埃及", "EG"),
Map.entry("尼日利亚", "NG"),
Map.entry("南非", "ZA"),
Map.entry("肯尼亚", "KE"),
Map.entry("摩洛哥", "MA"),
Map.entry("澳大利亚", "AU"),
Map.entry("新西兰", "NZ"));
private static final Map<String, CountryProfile> BY_CODE = new HashMap<>();
static {
for (CountryProfile profile : values()) {
BY_CODE.put(profile.name(), profile);
}
}
private final String language;
private final String timezone;
private final String currency;
CountryProfile(String language, String timezone, String currency) {
this.language = language;
this.timezone = timezone;
this.currency = currency;
}
public String code() {
return name();
}
public String language() {
return language;
}
public String timezone() {
return timezone;
}
public String currency() {
return currency;
}
/** 未收录的国家统一回退到印尼。 */
public static CountryProfile of(String countryCode) {
CountryProfile profile = BY_CODE.get(normalize(countryCode));
return profile == null ? DEFAULT : profile;
}
/** 客户端上报的区域:只接受两位字母且已收录的国家,其余返回空串。 */
public static String normalizeClientCode(String raw) {
String code = normalize(raw);
return BY_CODE.containsKey(code) ? code : "";
}
/**
* 把 ip2region 的 `国家|区域|省份|城市|运营商` 转成 ISO-2。
* 台港澳在库里的国家字段都是「中国」,需要看省份区分。
*/
public static String fromIpRegion(String region) {
if (!StringUtils.hasText(region)) {
return "";
}
String[] parts = region.split("\\|", -1);
String country = parts[0].trim();
if (!StringUtils.hasText(country) || "0".equals(country)) {
return "";
}
if ("中国".equals(country)) {
String province = parts.length > 2 ? parts[2].trim() : "";
if (province.contains("台湾")) {
return "TW";
}
if (province.contains("香港")) {
return "HK";
}
if (province.contains("澳门")) {
return "MO";
}
return "CN";
}
String result = IP_COUNTRY_NAMES.getOrDefault(country, "");
if (!StringUtils.hasText(result) && StringUtils.hasText(country)) {
// ip2region 库里有但我们没收录的国家,debug 日志方便排查
log.debug("ip2region 返回未收录的国家 '{}',回退到默认值(印尼)", country);
}
return result;
}
private static String normalize(String raw) {
if (!StringUtils.hasText(raw)) {
return "";
}
String code = raw.trim().toUpperCase(Locale.ROOT);
return code.length() == 2 ? code : "";
}
}
@@ -0,0 +1,7 @@
// Author: fruit.com.cn
package com.fruitshorts.user.event;
/** 登录/注册成功后,在同一事务内触发设备互动合并。 */
public record AppInteractionMergeRequested(Long userId, String deviceId) {
}
@@ -18,6 +18,7 @@ import com.fruitshorts.user.dal.mapper.AppUserDeviceMapper;
import com.fruitshorts.user.dal.mapper.AppUserMapper;
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
import com.fruitshorts.user.enums.AppUserStatus;
import com.fruitshorts.user.enums.CountryProfile;
import com.fruitshorts.user.enums.CredentialType;
import com.fruitshorts.user.enums.OtpScene;
import com.fruitshorts.user.vo.AppUserRespVO;
@@ -25,8 +26,10 @@ import com.fruitshorts.user.vo.LoginReqVO;
import com.fruitshorts.user.vo.LogoutReqVO;
import com.fruitshorts.user.vo.RegisterReqVO;
import com.fruitshorts.user.vo.TokenRespVO;
import com.fruitshorts.user.event.AppInteractionMergeRequested;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.stereotype.Service;
@@ -63,6 +66,8 @@ public class AppAuthService {
private final PasswordEncoder passwordEncoder;
private final ConfigService configService;
private final GoogleIdTokenVerifier googleVerifier;
private final ApplicationEventPublisher eventPublisher;
private final IpRegionService ipRegionService;
/** 后台 sys_config 缺失或非法时使用;小于等于 0 表示不限制。 */
@Value("${fruitshorts.auth.register-max-per-ip-per-day:5}")
@@ -130,6 +135,7 @@ public class AppAuthService {
TokenRespVO resp = issueTokens(user, credential, req.getDeviceId(), req.getDeviceModel(),
req.getOsVersion(), req.getAppVersion(), req.getPlatform(), clientIp);
resp.setNewUser(true);
requestInteractionMerge(user.getId(), req.getDeviceId());
return resp;
}
@@ -174,11 +180,14 @@ public class AppAuthService {
user.setLastLoginAt(LocalDateTime.now());
user.setLastLoginIp(AppIdentifier.nullSafe(clientIp));
syncCountryByIp(user, clientIp);
appUserMapper.updateById(user);
touchDevice(user.getId(), req.getDeviceId(), req.getPlatform(), req.getDeviceModel());
return issueTokens(user, credential, req.getDeviceId(), req.getDeviceModel(), req.getOsVersion(),
req.getAppVersion(), req.getPlatform(), clientIp);
TokenRespVO resp = issueTokens(user, credential, req.getDeviceId(), req.getDeviceModel(),
req.getOsVersion(), req.getAppVersion(), req.getPlatform(), clientIp);
requestInteractionMerge(user.getId(), req.getDeviceId());
return resp;
}
/**
@@ -188,7 +197,7 @@ public class AppAuthService {
* 再落一条 GOOGLE 凭证用于下次直接登录;已有邮箱账号时只补 GOOGLE 凭证,不重复建号。
*/
@Transactional(rollbackFor = Exception.class)
public TokenRespVO googleLogin(String idToken) {
public TokenRespVO googleLogin(String idToken, String deviceId, String clientIp) {
GoogleUser googleUser = googleVerifier.verify(idToken);
String email = requireEmail(googleUser);
AppUserCredential googleCredential = findCredential(CredentialType.GOOGLE.name(), email);
@@ -197,7 +206,7 @@ public class AppAuthService {
if (googleCredential == null) {
AppUserCredential emailCredential = findCredential(CredentialType.EMAIL.name(), email);
if (emailCredential == null) {
AppUser user = createGoogleUser(googleUser);
AppUser user = createGoogleUser(googleUser, clientIp);
insertCredential(user.getId(), CredentialType.EMAIL.name(), email, "");
userId = user.getId();
newUser = true;
@@ -213,10 +222,13 @@ public class AppAuthService {
googleCredential.setLastUsedAt(LocalDateTime.now());
credentialMapper.updateById(googleCredential);
user.setLastLoginAt(LocalDateTime.now());
user.setLastLoginIp("");
user.setLastLoginIp(AppIdentifier.nullSafe(clientIp));
syncCountryByIp(user, clientIp);
appUserMapper.updateById(user);
TokenRespVO resp = issueTokens(user, googleCredential, null, null, null, null, null, null);
touchDevice(user.getId(), deviceId, null, null);
TokenRespVO resp = issueTokens(user, googleCredential, deviceId, null, null, null, null, clientIp);
resp.setNewUser(newUser);
requestInteractionMerge(user.getId(), deviceId);
return resp;
}
@@ -230,7 +242,7 @@ public class AppAuthService {
}
/** 创建 Google 用户:邮箱已验证,昵称与头像直接取 Google 返回。 */
private AppUser createGoogleUser(GoogleUser googleUser) {
private AppUser createGoogleUser(GoogleUser googleUser, String clientIp) {
AppUser user = new AppUser();
user.setUid(hasher.newUid());
String nickname = StringUtils.hasText(googleUser.name())
@@ -238,11 +250,13 @@ public class AppAuthService {
user.setNickname(nickname);
user.setAvatarUrl(AppIdentifier.nullSafe(googleUser.picture()));
user.setGender("");
user.setCountryCode("ID");
user.setRegionCode("ID");
user.setLanguage("id-ID");
user.setTimezone("Asia/Jakarta");
user.setCurrency("IDR");
String country = resolveCountry(clientIp, "");
CountryProfile profile = CountryProfile.of(country);
user.setCountryCode(country);
user.setRegionCode(country);
user.setLanguage(profile.language());
user.setTimezone(profile.timezone());
user.setCurrency(profile.currency());
user.setStatus(AppUserStatus.NORMAL);
user.setRegisterSource("APP");
user.setRegisterChannel("");
@@ -436,6 +450,13 @@ public class AppAuthService {
return resp;
}
private void requestInteractionMerge(Long userId, String deviceId) {
if (StringUtils.hasText(deviceId)) {
eventPublisher.publishEvent(
new AppInteractionMergeRequested(userId, deviceId.trim()));
}
}
private static final String REGISTER_IP_LIMIT_KEY = "app.register.max-per-ip-per-day";
private static final String REGISTER_DEVICE_LIMIT_KEY = "app.register.max-per-device-per-day";
@@ -512,12 +533,13 @@ public class AppAuthService {
: "User" + user.getUid().substring(1, 7));
user.setAvatarUrl("");
user.setGender("");
String country = StringUtils.hasText(req.getCountryCode()) ? req.getCountryCode() : "ID";
String country = resolveCountry(clientIp, req.getCountryCode());
CountryProfile profile = CountryProfile.of(country);
user.setCountryCode(country);
user.setRegionCode(country);
user.setLanguage("id-ID");
user.setTimezone("Asia/Jakarta");
user.setCurrency("IDR");
user.setLanguage(profile.language());
user.setTimezone(profile.timezone());
user.setCurrency(profile.currency());
user.setStatus(AppUserStatus.NORMAL);
user.setRegisterSource("APP");
user.setRegisterChannel("");
@@ -534,6 +556,49 @@ public class AppAuthService {
return user;
}
/**
* 国家判定优先级:服务端 IP 解析 → 客户端上报的设备区域 → 印尼(历史默认)。
* IP 在本地开发/内网环境解析不出来,这时才用客户端区域。
*/
private String resolveCountry(String clientIp, String clientCountryCode) {
String byIp = ipRegionService.resolveCountry(clientIp);
if (StringUtils.hasText(byIp)) {
return byIp;
}
String byClient = CountryProfile.normalizeClientCode(clientCountryCode);
return StringUtils.hasText(byClient) ? byClient : CountryProfile.DEFAULT.code();
}
/**
* 登录时按 IP 修正国家,解决历史账号被统一写成 ID 的问题(老用户下次登录即自愈)。
*
* <p>regionCode / 语言 / 时区 / 币种 只在它们仍是「旧国家映射值」时才跟着改,
* 用户自己在资料页改过就保留,不被 IP 覆盖。countryCode 始终用 IP 结果覆盖
* (它是国家判断的权威来源,不涉及用户偏好)。
*/
private void syncCountryByIp(AppUser user, String clientIp) {
String byIp = ipRegionService.resolveCountry(clientIp);
if (!StringUtils.hasText(byIp) || byIp.equalsIgnoreCase(user.getCountryCode())) {
return;
}
CountryProfile before = CountryProfile.of(user.getCountryCode());
CountryProfile after = CountryProfile.of(byIp);
user.setCountryCode(byIp);
// regionCode 与 countryCode 永远同时写入(注册时就是这样),所以保护条件和语言一致
if (!StringUtils.hasText(user.getRegionCode()) || user.getRegionCode().equalsIgnoreCase(before.code())) {
user.setRegionCode(byIp);
}
if (!StringUtils.hasText(user.getLanguage()) || user.getLanguage().equals(before.language())) {
user.setLanguage(after.language());
}
if (!StringUtils.hasText(user.getTimezone()) || user.getTimezone().equals(before.timezone())) {
user.setTimezone(after.timezone());
}
if (!StringUtils.hasText(user.getCurrency()) || user.getCurrency().equals(before.currency())) {
user.setCurrency(after.currency());
}
}
private void recordConsent(Long userId, String scene, String version, String clientIp, String deviceId) {
AppUserConsent consent = new AppUserConsent();
consent.setUserId(userId);
@@ -2,6 +2,7 @@
package com.fruitshorts.user.service;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
@@ -30,6 +31,7 @@ import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
/**
@@ -52,33 +54,36 @@ public class AppUserAdminService {
req.getPageIndex() == null || req.getPageIndex() < 1 ? 1 : req.getPageIndex(),
req.getPageSize() == null || req.getPageSize() < 1 ? 10 : req.getPageSize());
IPage<AppUser> result;
List<Long> platformUserIds = null;
if (StringUtils.hasText(req.getPlatform())) {
// 与列表展示保持一致:取每个用户最近一次会话的平台
platformUserIds = sessionMapper.selectUserIdsByLatestPlatform(
req.getPlatform().trim().toUpperCase(Locale.ROOT));
if (platformUserIds.isEmpty()) {
return new PageResult<>(0L, List.of());
}
}
LambdaQueryWrapper<AppUser> wrapper = Wrappers.lambdaQuery();
if (StringUtils.hasText(req.getKeyword())) {
// uid / 昵称 直接查;手机号与邮箱要先在凭证表命中 userId
List<Long> matchedUserIds = userIdsByCredential(req.getKeyword());
result = appUserMapper.selectPage(page, Wrappers.<AppUser>lambdaQuery()
.and(wrapper -> {
wrapper.like(AppUser::getUid, req.getKeyword())
.or().like(AppUser::getNickname, req.getKeyword());
if (!matchedUserIds.isEmpty()) {
wrapper.or().in(AppUser::getId, matchedUserIds);
}
})
.eq(StringUtils.hasText(req.getStatus()), AppUser::getStatus, req.getStatus())
.eq(StringUtils.hasText(req.getCountryCode()), AppUser::getCountryCode,
req.getCountryCode())
.eq(StringUtils.hasText(req.getRegisterSource()), AppUser::getRegisterSource,
req.getRegisterSource())
.orderByDesc(AppUser::getId));
} else {
result = appUserMapper.selectPage(page, Wrappers.<AppUser>lambdaQuery()
.eq(StringUtils.hasText(req.getStatus()), AppUser::getStatus, req.getStatus())
.eq(StringUtils.hasText(req.getCountryCode()), AppUser::getCountryCode,
req.getCountryCode())
.eq(StringUtils.hasText(req.getRegisterSource()), AppUser::getRegisterSource,
req.getRegisterSource())
.orderByDesc(AppUser::getId));
wrapper.and(nested -> {
nested.like(AppUser::getUid, req.getKeyword())
.or().like(AppUser::getNickname, req.getKeyword());
if (!matchedUserIds.isEmpty()) {
nested.or().in(AppUser::getId, matchedUserIds);
}
});
}
wrapper.eq(StringUtils.hasText(req.getStatus()), AppUser::getStatus, req.getStatus())
.eq(StringUtils.hasText(req.getCountryCode()), AppUser::getCountryCode,
req.getCountryCode())
.eq(StringUtils.hasText(req.getRegisterSource()), AppUser::getRegisterSource,
req.getRegisterSource())
.in(platformUserIds != null, AppUser::getId, platformUserIds)
.orderByDesc(AppUser::getId);
IPage<AppUser> result = appUserMapper.selectPage(page, wrapper);
List<AppUserAdminRespVO> list = toRespList(result.getRecords());
fillLatestSystem(list);
return new PageResult<>(result.getTotal(), list);
@@ -164,7 +169,7 @@ public class AppUserAdminService {
return vo;
}
/** 用最近一次未吊销会话上的系统信息,供后台用户列表展示。 */
/** 用最近一次会话上的系统信息,供后台用户列表展示;排序需与平台筛选一致。 */
private void fillLatestSystem(List<AppUserAdminRespVO> list) {
if (list.isEmpty()) {
return;
@@ -180,7 +185,6 @@ public class AppUserAdminService {
}
List<AppUserSession> sessions = sessionMapper.selectList(Wrappers.<AppUserSession>lambdaQuery()
.in(AppUserSession::getUserId, ids)
.isNull(AppUserSession::getRevokedAt)
.orderByDesc(AppUserSession::getLastActiveAt)
.orderByDesc(AppUserSession::getId));
Map<Long, AppUserSession> latest = new HashMap<>();
@@ -0,0 +1,51 @@
// Author: fruit.com.cn
package com.fruitshorts.user.service;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.fruitshorts.user.dal.entity.AppDeviceProfile;
import com.fruitshorts.user.dal.mapper.AppDeviceProfileMapper;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import java.util.Locale;
/** 设备平台写入:按 device_id 建档,匿名请求也记录。 */
@Slf4j
@Service
@RequiredArgsConstructor
public class DeviceProfileService {
private static final int DEVICE_ID_MAX = 128;
private static final int PLATFORM_MAX = 16;
private final AppDeviceProfileMapper mapper;
/** 记录或刷新设备平台;平台为空时直接跳过,避免产生无意义记录。 */
public void touch(String deviceId, String platform) {
String cleanPlatform = sanitize(platform, PLATFORM_MAX).toUpperCase(Locale.ROOT);
if (!StringUtils.hasText(deviceId) || cleanPlatform.isEmpty()) {
return;
}
AppDeviceProfile profile = new AppDeviceProfile();
profile.setId(IdWorker.getId());
profile.setDeviceId(sanitize(deviceId, DEVICE_ID_MAX));
profile.setPlatform(cleanPlatform);
try {
mapper.upsert(profile);
} catch (RuntimeException ex) {
// 平台只用于后台展示,写失败不能影响正常业务请求
log.warn("device-platform upsert failed deviceId={} reason={}", profile.getDeviceId(), ex.getMessage());
}
}
private String sanitize(String value, int maxLength) {
if (!StringUtils.hasText(value)) {
return "";
}
String trimmed = value.trim().replaceAll("[\\p{Cntrl}]", "");
return trimmed.length() <= maxLength ? trimmed : trimmed.substring(0, maxLength);
}
}
@@ -0,0 +1,60 @@
// Author: fruit.com.cn
package com.fruitshorts.user.service;
import com.fruitshorts.user.enums.CountryProfile;
import lombok.extern.slf4j.Slf4j;
import org.lionsoul.ip2region.xdb.Searcher;
import org.springframework.core.io.ClassPathResource;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import java.io.InputStream;
/**
* 把登录/注册请求的客户端 IP 解析成国家码。
*
* <p>用离线 ip2region(xdb 随包发布),不依赖外部接口;内网、保留地址与未收录国家返回空串,
* 由调用方回退到客户端上报的设备区域。数据加载失败也不阻断启动,只是 IP 解析不生效。
*/
@Slf4j
@Service
public class IpRegionService {
private static final String DB_PATH = "ip2region/ip2region.xdb";
private volatile Searcher searcher;
private volatile boolean loaded;
/** 返回 ISO-2 国家码;解析不出来返回空串。 */
public String resolveCountry(String ip) {
Searcher current = searcher();
if (current == null || !StringUtils.hasText(ip)) {
return "";
}
try {
return CountryProfile.fromIpRegion(current.search(ip.trim()));
} catch (Exception ex) {
// 非法 IP(本地开发、测试桩)不当作错误,交给客户端区域兜底
return "";
}
}
private Searcher searcher() {
if (loaded) {
return searcher;
}
synchronized (this) {
if (!loaded) {
try (InputStream in = new ClassPathResource(DB_PATH).getInputStream()) {
searcher = Searcher.newWithBuffer(in.readAllBytes());
} catch (Exception ex) {
log.warn("ip2region 数据加载失败,IP 国家解析不可用: {}", ex.getMessage());
} finally {
loaded = true;
}
}
}
return searcher;
}
}
@@ -13,6 +13,8 @@ public class AppUserQueryReqVO {
private String status;
private String countryCode;
private String registerSource;
/** 客户端平台筛选:ANDROID / IOS / WEB,留空不限制;按最近一次会话的平台判定 */
private String platform;
private Integer pageIndex = 1;
private Integer pageSize = 10;
}
@@ -25,6 +25,7 @@ public class RegisterReqVO {
private String password;
private String nickname;
/** 客户端上报的设备区域(ISO-2),仅在服务端 IP 解析失败时作为兜底 */
private String countryCode;
private String deviceId;
private String deviceModel;
Loaded 100 of 118 files, more files were not shown because too many files have changed in this diff. Show more