Compare commits
27
Commits
7baaeddf26
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
aa5954ef6a | ||
|
|
963bf5c0da | ||
|
|
a71d2701d1 | ||
|
|
025fbe73f0 | ||
|
|
0d161f318f | ||
|
|
2eac1fe31d | ||
|
|
09b0f89e8d | ||
|
|
8fb1f52777 | ||
|
|
44b7abcd0f | ||
|
|
181b338780 | ||
|
|
657190d5a8 | ||
|
|
3e2d6a06d9 | ||
|
|
fc9e429187 | ||
|
|
5d139783c9 | ||
|
|
ddd2406f32 | ||
|
|
272aa62434 | ||
|
|
a41ab18636 | ||
|
|
8999ff395d | ||
|
|
229f1d0198 | ||
|
|
2dadca6288 | ||
|
|
2cf8f9a1c3 | ||
|
|
d9a1670a72 | ||
|
|
5ff4626b33 | ||
|
|
6bbde0b100 | ||
|
|
884324b8fd | ||
|
|
6abb09e5c2 | ||
|
|
8b37e23951 |
No files matched your search
@@ -19,3 +19,7 @@ data/uploads/*
|
|||||||
|
|
||||||
# 部署用的真实环境变量(含密钥),只提交 .env.example
|
# 部署用的真实环境变量(含密钥),只提交 .env.example
|
||||||
.env
|
.env
|
||||||
|
dump.rdb
|
||||||
|
|
||||||
|
# 看门狗飞书告警 Webhook,只存在于部署机
|
||||||
|
deploy/feishu-webhook.txt
|
||||||
+15
-1
@@ -15,6 +15,20 @@
|
|||||||
FROM maven:3.9-eclipse-temurin-21 AS build
|
FROM maven:3.9-eclipse-temurin-21 AS build
|
||||||
WORKDIR /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 再拉依赖:依赖层可复用,改业务代码不会重新下载依赖
|
# 先只拷各模块 pom 再拉依赖:依赖层可复用,改业务代码不会重新下载依赖
|
||||||
COPY pom.xml .
|
COPY pom.xml .
|
||||||
COPY fruitshorts-common/pom.xml fruitshorts-common/
|
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-admin-api/pom.xml fruitshorts-admin-api/
|
||||||
COPY fruitshorts-app-api/pom.xml fruitshorts-app-api/
|
COPY fruitshorts-app-api/pom.xml fruitshorts-app-api/
|
||||||
COPY fruitshorts-transcode-worker/pom.xml fruitshorts-transcode-worker/
|
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 . .
|
COPY . .
|
||||||
ARG MODULE=fruitshorts-app-api
|
ARG MODULE=fruitshorts-app-api
|
||||||
|
|||||||
@@ -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>
|
||||||
@@ -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
@@ -71,8 +71,10 @@ services:
|
|||||||
env_file: .env
|
env_file: .env
|
||||||
environment:
|
environment:
|
||||||
SERVER_PORT: "8081"
|
SERVER_PORT: "8081"
|
||||||
# 默认连编排里的 db;要连外部库就在 .env 里覆盖 DB_URL
|
# 默认连编排里的 db;要连外部库就在 .env 里覆盖 DB_URL。
|
||||||
DB_URL: ${DB_URL:-jdbc:postgresql://db:5432/${DB_NAME:-fruitshorts}}
|
# 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;容器内 127.0.0.1 是容器自身,必须指向编排里的 redis
|
||||||
REDIS_HOST: ${REDIS_HOST:-redis}
|
REDIS_HOST: ${REDIS_HOST:-redis}
|
||||||
REDIS_PORT: ${REDIS_PORT:-6379}
|
REDIS_PORT: ${REDIS_PORT:-6379}
|
||||||
@@ -197,9 +199,40 @@ services:
|
|||||||
GF_SECURITY_ADMIN_USER: ${GRAFANA_ADMIN_USER:-admin}
|
GF_SECURITY_ADMIN_USER: ${GRAFANA_ADMIN_USER:-admin}
|
||||||
GF_SECURITY_ADMIN_PASSWORD: ${GRAFANA_ADMIN_PASSWORD:?Grafana 管理员密码必填,见 .env.example}
|
GF_SECURITY_ADMIN_PASSWORD: ${GRAFANA_ADMIN_PASSWORD:?Grafana 管理员密码必填,见 .env.example}
|
||||||
GF_USERS_ALLOW_SIGN_UP: "false"
|
GF_USERS_ALLOW_SIGN_UP: "false"
|
||||||
|
# 新用户默认界面语言:简体中文
|
||||||
|
GF_USERS_LANGUAGE: "zh-Hans"
|
||||||
TZ: Asia/Shanghai
|
TZ: Asia/Shanghai
|
||||||
depends_on:
|
depends_on:
|
||||||
- prometheus
|
- 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:
|
blackbox:
|
||||||
@@ -227,3 +260,4 @@ volumes:
|
|||||||
redis-data:
|
redis-data:
|
||||||
prometheus-data:
|
prometheus-data:
|
||||||
grafana-data:
|
grafana-data:
|
||||||
|
loki-data:
|
||||||
+6
-4
@@ -115,6 +115,8 @@ POST /app/auth/register
|
|||||||
| 密码强度 | 8–64 位,且**同时包含字母与数字**(不强制特殊字符:低端机输入成本高,拦机器注册靠频控) |
|
| 密码强度 | 8–64 位,且**同时包含字母与数字**(不强制特殊字符:低端机输入成本高,拦机器注册靠频控) |
|
||||||
| 注册频控 | 同 IP / 同设备 24h 默认上限分别为 5 / 3 个,超出返回 `429`(`TOO_MANY_REQUESTS`) |
|
| 注册频控 | 同 IP / 同设备 24h 默认上限分别为 5 / 3 个,超出返回 `429`(`TOO_MANY_REQUESTS`) |
|
||||||
| 邮箱验证状态 | 免码注册时 `app_user_credential.verified = 0`(保留"待验证"),**不阻塞登录**;只有**验证码登录**会把 `verified` 置 1,密码登录不改变该状态 |
|
| 邮箱验证状态 | 免码注册时 `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 计数,不新建计数表)。
|
频控口径落库在 `app_user.register_ip` / `app_user.register_device_id` 上(按 24h 计数,不新建计数表)。
|
||||||
|
|
||||||
@@ -863,13 +865,13 @@ 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}` | `app-user:list` | 详情 |
|
||||||
| GET | `/admin/app-users/{id}/ban-records` | `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/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` | 收藏只读分页,含游客 |
|
| GET | `/admin/favorites/page` | `favorite:list` | 收藏只读分页,含游客;支持 platform 筛选(ANDROID/IOS,按**最近一次会话**) |
|
||||||
| GET | `/admin/likes/page` | `like:list` | 点赞只读分页,含游客及多态目标 |
|
| GET | `/admin/likes/page` | `like:list` | 点赞只读分页,含游客及多态目标;支持 platform 筛选(ANDROID/IOS,按**最近一次会话**) |
|
||||||
| PUT | `/admin/app-users/{id}/status` | `app-user:update` | 改状态:normal/limited/banned |
|
| PUT | `/admin/app-users/{id}/status` | `app-user:update` | 改状态:normal/limited/banned |
|
||||||
|
|
||||||
状态处置会**立即吊销该用户全部会话**并写入封禁记录,避免"封了还能用"。
|
状态处置会**立即吊销该用户全部会话**并写入封禁记录,避免"封了还能用"。
|
||||||
|
|||||||
+28
@@ -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));
|
||||||
|
}
|
||||||
|
}
|
||||||
+2
-1
@@ -32,12 +32,13 @@ import java.util.List;
|
|||||||
public class AuthController {
|
public class AuthController {
|
||||||
|
|
||||||
private final AuthService authService;
|
private final AuthService authService;
|
||||||
|
private final WebUtils webUtils;
|
||||||
|
|
||||||
@Operation(summary = "管理端登录")
|
@Operation(summary = "管理端登录")
|
||||||
@PostMapping("/login")
|
@PostMapping("/login")
|
||||||
@OperationLog(module = "认证", action = "登录")
|
@OperationLog(module = "认证", action = "登录")
|
||||||
public Result<LoginRespVO> login(@Valid @RequestBody LoginReqVO req, HttpServletRequest request) {
|
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 的动态路由一致。 */
|
/** 当前用户的菜单路由树,结构与前端 element-plus-admin 的动态路由一致。 */
|
||||||
|
|||||||
+52
-1
@@ -3,7 +3,9 @@
|
|||||||
package com.fruitshorts.admin.service;
|
package com.fruitshorts.admin.service;
|
||||||
|
|
||||||
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
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.entity.AppUserSession;
|
||||||
|
import com.fruitshorts.user.dal.mapper.AppDeviceProfileMapper;
|
||||||
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
|
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
|
||||||
import lombok.RequiredArgsConstructor;
|
import lombok.RequiredArgsConstructor;
|
||||||
import org.springframework.stereotype.Service;
|
import org.springframework.stereotype.Service;
|
||||||
@@ -13,16 +15,19 @@ import java.util.ArrayList;
|
|||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Locale;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 按用户最近会话解析系统。同一设备优先,否则用仍有效的最近一次登录。
|
* 按设备画像解析系统,缺画像时回退到用户最近会话。
|
||||||
|
* 设备画像不依赖登录态,游客行为也能显示系统。
|
||||||
*/
|
*/
|
||||||
@Service
|
@Service
|
||||||
@RequiredArgsConstructor
|
@RequiredArgsConstructor
|
||||||
public class AppSessionSystemService {
|
public class AppSessionSystemService {
|
||||||
|
|
||||||
private final AppUserSessionMapper sessionMapper;
|
private final AppUserSessionMapper sessionMapper;
|
||||||
|
private final AppDeviceProfileMapper profileMapper;
|
||||||
|
|
||||||
/** 按用户分组,组内按最近活跃倒序。 */
|
/** 按用户分组,组内按最近活跃倒序。 */
|
||||||
public Map<Long, List<AppUserSession>> groupByUser(Collection<Long> userIds) {
|
public Map<Long, List<AppUserSession>> groupByUser(Collection<Long> userIds) {
|
||||||
@@ -31,6 +36,9 @@ public class AppSessionSystemService {
|
|||||||
return grouped;
|
return grouped;
|
||||||
}
|
}
|
||||||
List<AppUserSession> sessions = sessionMapper.selectList(Wrappers.<AppUserSession>lambdaQuery()
|
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)
|
.in(AppUserSession::getUserId, userIds)
|
||||||
.orderByDesc(AppUserSession::getLastActiveAt)
|
.orderByDesc(AppUserSession::getLastActiveAt)
|
||||||
.orderByDesc(AppUserSession::getId));
|
.orderByDesc(AppUserSession::getId));
|
||||||
@@ -40,6 +48,49 @@ public class AppSessionSystemService {
|
|||||||
return grouped;
|
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) {
|
public AppUserSession match(List<AppUserSession> sessions, String deviceId) {
|
||||||
if (sessions == null || sessions.isEmpty()) {
|
if (sessions == null || sessions.isEmpty()) {
|
||||||
|
|||||||
+45
@@ -9,13 +9,19 @@ import com.fruitshorts.common.exception.ErrorCode;
|
|||||||
import com.fruitshorts.common.exception.ServiceException;
|
import com.fruitshorts.common.exception.ServiceException;
|
||||||
import com.fruitshorts.content.dal.entity.Drama;
|
import com.fruitshorts.content.dal.entity.Drama;
|
||||||
import com.fruitshorts.content.dal.mapper.DramaMapper;
|
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.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.AppUserMapper;
|
||||||
|
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
|
||||||
import lombok.RequiredArgsConstructor;
|
import lombok.RequiredArgsConstructor;
|
||||||
import org.springframework.stereotype.Component;
|
import org.springframework.stereotype.Component;
|
||||||
import org.springframework.util.StringUtils;
|
import org.springframework.util.StringUtils;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Locale;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Objects;
|
import java.util.Objects;
|
||||||
import java.util.function.Function;
|
import java.util.function.Function;
|
||||||
@@ -35,6 +41,9 @@ public class EngagementAdminLookup {
|
|||||||
private static final int LOOKUP_PROBE = MAX_LOOKUP_SIZE + 1;
|
private static final int LOOKUP_PROBE = MAX_LOOKUP_SIZE + 1;
|
||||||
private final AppUserMapper userMapper;
|
private final AppUserMapper userMapper;
|
||||||
private final DramaMapper dramaMapper;
|
private final DramaMapper dramaMapper;
|
||||||
|
private final AppDeviceProfileMapper deviceProfileMapper;
|
||||||
|
private final AppUserSessionMapper sessionMapper;
|
||||||
|
private final AppSessionSystemService sessionSystem;
|
||||||
|
|
||||||
/** 规范化页码和页大小。 */
|
/** 规范化页码和页大小。 */
|
||||||
public <T> Page<T> page(EngagementQueryReqVO req) {
|
public <T> Page<T> page(EngagementQueryReqVO req) {
|
||||||
@@ -96,6 +105,42 @@ public class EngagementAdminLookup {
|
|||||||
.collect(Collectors.toMap(Drama::getId, Function.identity()));
|
.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) {
|
private List<Long> keys(Collection<Long> ids) {
|
||||||
return ids.stream().filter(Objects::nonNull).distinct().toList();
|
return ids.stream().filter(Objects::nonNull).distinct().toList();
|
||||||
}
|
}
|
||||||
|
|||||||
+38
-4
@@ -10,7 +10,9 @@ import com.fruitshorts.content.dal.entity.Drama;
|
|||||||
import com.fruitshorts.content.enums.ContentStatus;
|
import com.fruitshorts.content.enums.ContentStatus;
|
||||||
import com.fruitshorts.playback.dal.entity.UserFavorite;
|
import com.fruitshorts.playback.dal.entity.UserFavorite;
|
||||||
import com.fruitshorts.playback.dal.mapper.UserFavoriteMapper;
|
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.AppUser;
|
||||||
|
import com.fruitshorts.user.dal.entity.AppUserSession;
|
||||||
import lombok.RequiredArgsConstructor;
|
import lombok.RequiredArgsConstructor;
|
||||||
import org.springframework.stereotype.Service;
|
import org.springframework.stereotype.Service;
|
||||||
import org.springframework.util.StringUtils;
|
import org.springframework.util.StringUtils;
|
||||||
@@ -34,6 +36,10 @@ public class FavoriteAdminService {
|
|||||||
if (lookup.noMatch(dramaIds)) {
|
if (lookup.noMatch(dramaIds)) {
|
||||||
return PageResult.empty();
|
return PageResult.empty();
|
||||||
}
|
}
|
||||||
|
var scope = lookup.platformScope(req.getPlatform());
|
||||||
|
if (scope != null && scope.empty()) {
|
||||||
|
return PageResult.empty();
|
||||||
|
}
|
||||||
var query = Wrappers.<UserFavorite>lambdaQuery()
|
var query = Wrappers.<UserFavorite>lambdaQuery()
|
||||||
.eq(req.getUserId() != null, UserFavorite::getUserId, req.getUserId())
|
.eq(req.getUserId() != null, UserFavorite::getUserId, req.getUserId())
|
||||||
.in(userIds != null, UserFavorite::getUserId, userIds)
|
.in(userIds != null, UserFavorite::getUserId, userIds)
|
||||||
@@ -41,8 +47,22 @@ public class FavoriteAdminService {
|
|||||||
.eq(req.getDramaId() != null, UserFavorite::getDramaId, req.getDramaId())
|
.eq(req.getDramaId() != null, UserFavorite::getDramaId, req.getDramaId())
|
||||||
.in(dramaIds != null, UserFavorite::getDramaId, dramaIds)
|
.in(dramaIds != null, UserFavorite::getDramaId, dramaIds)
|
||||||
.ge(req.getStartTime() != null, UserFavorite::getCreatedAt, req.getStartTime())
|
.ge(req.getStartTime() != null, UserFavorite::getCreatedAt, req.getStartTime())
|
||||||
.le(req.getEndTime() != null, UserFavorite::getCreatedAt, req.getEndTime())
|
.le(req.getEndTime() != null, UserFavorite::getCreatedAt, req.getEndTime());
|
||||||
.orderByDesc(UserFavorite::getCreatedAt, UserFavorite::getId);
|
// 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);
|
var result = favoriteMapper.selectPage(lookup.page(req), query);
|
||||||
return new PageResult<>(result.getTotal(), assemble(result.getRecords()));
|
return new PageResult<>(result.getTotal(), assemble(result.getRecords()));
|
||||||
}
|
}
|
||||||
@@ -53,17 +73,31 @@ public class FavoriteAdminService {
|
|||||||
}
|
}
|
||||||
Map<Long, AppUser> users = lookup.users(rows.stream().map(UserFavorite::getUserId).toList());
|
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<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()),
|
return rows.stream().map(row -> toVo(row, row.getUserId() == null ? null : users.get(row.getUserId()),
|
||||||
row.getDramaId() == null ? null : dramas.get(row.getDramaId())))
|
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();
|
.toList();
|
||||||
}
|
}
|
||||||
|
|
||||||
private AdminFavoriteVO toVo(UserFavorite row, AppUser user, Drama drama) {
|
private AdminFavoriteVO toVo(UserFavorite row, AppUser user, Drama drama,
|
||||||
|
AppDeviceProfile profile, AppUserSession session) {
|
||||||
AdminFavoriteVO vo = new AdminFavoriteVO();
|
AdminFavoriteVO vo = new AdminFavoriteVO();
|
||||||
vo.setId(row.getId());
|
vo.setId(row.getId());
|
||||||
vo.setUserId(row.getUserId());
|
vo.setUserId(row.getUserId());
|
||||||
vo.setDeviceId(row.getDeviceId());
|
vo.setDeviceId(row.getDeviceId());
|
||||||
vo.setAnonymous(row.getUserId() == null);
|
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.setDramaId(row.getDramaId());
|
||||||
vo.setCreatedAt(row.getCreatedAt());
|
vo.setCreatedAt(row.getCreatedAt());
|
||||||
if (user != null) {
|
if (user != null) {
|
||||||
|
|||||||
+43
-4
@@ -14,7 +14,9 @@ import com.fruitshorts.interaction.dal.entity.ContentComment;
|
|||||||
import com.fruitshorts.interaction.dal.entity.ContentLike;
|
import com.fruitshorts.interaction.dal.entity.ContentLike;
|
||||||
import com.fruitshorts.interaction.dal.mapper.ContentCommentMapper;
|
import com.fruitshorts.interaction.dal.mapper.ContentCommentMapper;
|
||||||
import com.fruitshorts.interaction.dal.mapper.ContentLikeMapper;
|
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.AppUser;
|
||||||
|
import com.fruitshorts.user.dal.entity.AppUserSession;
|
||||||
import lombok.RequiredArgsConstructor;
|
import lombok.RequiredArgsConstructor;
|
||||||
import org.springframework.stereotype.Service;
|
import org.springframework.stereotype.Service;
|
||||||
import org.springframework.util.StringUtils;
|
import org.springframework.util.StringUtils;
|
||||||
@@ -67,6 +69,23 @@ public class LikeAdminService {
|
|||||||
return PageResult.empty();
|
return PageResult.empty();
|
||||||
}
|
}
|
||||||
var query = buildQuery(req, users);
|
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) {
|
if (req.getDramaId() != null) {
|
||||||
query.and(q -> q.apply(DRAMA_MATCH, req.getDramaId()));
|
query.and(q -> q.apply(DRAMA_MATCH, req.getDramaId()));
|
||||||
}
|
}
|
||||||
@@ -80,6 +99,8 @@ public class LikeAdminService {
|
|||||||
if (req.getEpisodeId() != null) {
|
if (req.getEpisodeId() != null) {
|
||||||
query.and(q -> q.apply(EPISODE_MATCH, req.getEpisodeId()));
|
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);
|
var result = likeMapper.selectPage(lookup.page(req), query);
|
||||||
return new PageResult<>(result.getTotal(), assemble(result.getRecords()));
|
return new PageResult<>(result.getTotal(), assemble(result.getRecords()));
|
||||||
}
|
}
|
||||||
@@ -92,8 +113,7 @@ public class LikeAdminService {
|
|||||||
.eq(StringUtils.hasText(req.getTargetType()), ContentLike::getTargetType, req.getTargetType())
|
.eq(StringUtils.hasText(req.getTargetType()), ContentLike::getTargetType, req.getTargetType())
|
||||||
.eq(req.getTargetId() != null, ContentLike::getTargetId, req.getTargetId())
|
.eq(req.getTargetId() != null, ContentLike::getTargetId, req.getTargetId())
|
||||||
.ge(req.getStartTime() != null, ContentLike::getCreatedAt, req.getStartTime())
|
.ge(req.getStartTime() != null, ContentLike::getCreatedAt, req.getStartTime())
|
||||||
.le(req.getEndTime() != null, ContentLike::getCreatedAt, req.getEndTime())
|
.le(req.getEndTime() != null, ContentLike::getCreatedAt, req.getEndTime());
|
||||||
.orderByDesc(ContentLike::getCreatedAt, ContentLike::getId);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private List<AdminLikeVO> assemble(List<ContentLike> rows) {
|
private List<AdminLikeVO> assemble(List<ContentLike> rows) {
|
||||||
@@ -110,7 +130,13 @@ public class LikeAdminService {
|
|||||||
comments.values().stream().map(ContentComment::getDramaId).filter(Objects::nonNull).forEach(dramaIds::add);
|
comments.values().stream().map(ContentComment::getDramaId).filter(Objects::nonNull).forEach(dramaIds::add);
|
||||||
episodes.values().stream().map(DramaEpisode::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<Long, Drama> dramas = lookup.dramas(dramaIds);
|
||||||
return rows.stream().map(row -> toVo(row, users, comments, episodes, dramas)).toList();
|
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) {
|
private List<Long> targetIds(List<ContentLike> rows, String type) {
|
||||||
@@ -130,12 +156,25 @@ public class LikeAdminService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private AdminLikeVO toVo(ContentLike row, Map<Long, AppUser> users, Map<Long, ContentComment> comments,
|
private AdminLikeVO toVo(ContentLike row, Map<Long, AppUser> users, Map<Long, ContentComment> comments,
|
||||||
Map<Long, DramaEpisode> episodes, Map<Long, Drama> dramas) {
|
Map<Long, DramaEpisode> episodes, Map<Long, Drama> dramas,
|
||||||
|
Map<String, AppDeviceProfile> profiles,
|
||||||
|
Map<Long, AppUserSession> sessions) {
|
||||||
AdminLikeVO vo = new AdminLikeVO();
|
AdminLikeVO vo = new AdminLikeVO();
|
||||||
vo.setId(row.getId());
|
vo.setId(row.getId());
|
||||||
vo.setUserId(row.getUserId());
|
vo.setUserId(row.getUserId());
|
||||||
vo.setDeviceId(row.getDeviceId());
|
vo.setDeviceId(row.getDeviceId());
|
||||||
vo.setAnonymous(row.getUserId() == null);
|
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.setTargetType(row.getTargetType());
|
||||||
vo.setTargetId(row.getTargetId());
|
vo.setTargetId(row.getTargetId());
|
||||||
vo.setCreatedAt(row.getCreatedAt());
|
vo.setCreatedAt(row.getCreatedAt());
|
||||||
|
|||||||
+76
-15
@@ -8,6 +8,8 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|||||||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||||
import com.fruitshorts.admin.vo.AdminWatchHistoryVO;
|
import com.fruitshorts.admin.vo.AdminWatchHistoryVO;
|
||||||
import com.fruitshorts.admin.vo.WatchHistoryQueryReqVO;
|
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.common.pojo.PageResult;
|
||||||
import com.fruitshorts.content.dal.entity.Drama;
|
import com.fruitshorts.content.dal.entity.Drama;
|
||||||
import com.fruitshorts.content.dal.entity.DramaEpisode;
|
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.content.enums.ContentStatus;
|
||||||
import com.fruitshorts.playback.dal.entity.WatchHistory;
|
import com.fruitshorts.playback.dal.entity.WatchHistory;
|
||||||
import com.fruitshorts.playback.dal.mapper.WatchHistoryMapper;
|
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.AppUser;
|
||||||
import com.fruitshorts.user.dal.entity.AppUserSession;
|
import com.fruitshorts.user.dal.entity.AppUserSession;
|
||||||
import com.fruitshorts.user.dal.mapper.AppUserMapper;
|
import com.fruitshorts.user.dal.mapper.AppUserMapper;
|
||||||
@@ -27,6 +30,7 @@ import java.util.ArrayList;
|
|||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
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 DEFAULT_PAGE_SIZE = 10;
|
||||||
private static final int MAX_PAGE_SIZE = 100;
|
private static final int MAX_PAGE_SIZE = 100;
|
||||||
private static final long FINISHED_TOLERANCE_MS = 1000L;
|
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 WatchHistoryMapper watchHistoryMapper;
|
||||||
private final AppUserMapper appUserMapper;
|
private final AppUserMapper appUserMapper;
|
||||||
@@ -65,14 +71,19 @@ public class WatchHistoryAdminService {
|
|||||||
if (matchedDramaIds != null && matchedDramaIds.isEmpty()) {
|
if (matchedDramaIds != null && matchedDramaIds.isEmpty()) {
|
||||||
return emptyPage();
|
return emptyPage();
|
||||||
}
|
}
|
||||||
|
var scope = sessionSystem.platformScope(req.getPlatform());
|
||||||
|
if (scope != null && scope.empty()) {
|
||||||
|
return emptyPage();
|
||||||
|
}
|
||||||
IPage<WatchHistory> result = watchHistoryMapper.selectPage(new Page<>(pageIndex, pageSize),
|
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()));
|
return new PageResult<>(result.getTotal(), toVoList(result.getRecords()));
|
||||||
}
|
}
|
||||||
|
|
||||||
private LambdaQueryWrapper<WatchHistory> buildQuery(WatchHistoryQueryReqVO req,
|
private LambdaQueryWrapper<WatchHistory> buildQuery(WatchHistoryQueryReqVO req,
|
||||||
List<Long> matchedUserIds,
|
List<Long> matchedUserIds,
|
||||||
List<Long> matchedDramaIds) {
|
List<Long> matchedDramaIds,
|
||||||
|
AppSessionSystemService.PlatformScope scope) {
|
||||||
LambdaQueryWrapper<WatchHistory> wrapper = Wrappers.<WatchHistory>lambdaQuery()
|
LambdaQueryWrapper<WatchHistory> wrapper = Wrappers.<WatchHistory>lambdaQuery()
|
||||||
.eq(req.getUserId() != null, WatchHistory::getUserId, req.getUserId())
|
.eq(req.getUserId() != null, WatchHistory::getUserId, req.getUserId())
|
||||||
.in(matchedUserIds != null, WatchHistory::getUserId, matchedUserIds)
|
.in(matchedUserIds != null, WatchHistory::getUserId, matchedUserIds)
|
||||||
@@ -82,6 +93,19 @@ public class WatchHistoryAdminService {
|
|||||||
.eq(StringUtils.hasText(req.getSource()), WatchHistory::getSource, req.getSource())
|
.eq(StringUtils.hasText(req.getSource()), WatchHistory::getSource, req.getSource())
|
||||||
.ge(req.getStartTime() != null, WatchHistory::getWatchedAt, req.getStartTime())
|
.ge(req.getStartTime() != null, WatchHistory::getWatchedAt, req.getStartTime())
|
||||||
.le(req.getEndTime() != null, WatchHistory::getWatchedAt, req.getEndTime());
|
.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());
|
applyFinished(wrapper, req.getFinished());
|
||||||
return wrapper.orderByDesc(WatchHistory::getWatchedAt)
|
return wrapper.orderByDesc(WatchHistory::getWatchedAt)
|
||||||
.orderByDesc(WatchHistory::getId);
|
.orderByDesc(WatchHistory::getId);
|
||||||
@@ -100,24 +124,37 @@ public class WatchHistoryAdminService {
|
|||||||
|
|
||||||
private List<Long> searchUserIds(String keyword) {
|
private List<Long> searchUserIds(String keyword) {
|
||||||
String value = keyword.trim();
|
String value = keyword.trim();
|
||||||
return appUserMapper.selectList(Wrappers.<AppUser>lambdaQuery()
|
List<Long> ids = appUserMapper.selectList(Wrappers.<AppUser>lambdaQuery()
|
||||||
.select(AppUser::getId)
|
.select(AppUser::getId)
|
||||||
.and(query -> query.like(AppUser::getUid, value)
|
.and(query -> query.like(AppUser::getUid, value)
|
||||||
.or()
|
.or()
|
||||||
.like(AppUser::getNickname, value)))
|
.like(AppUser::getNickname, value))
|
||||||
|
.last("LIMIT " + LOOKUP_PROBE))
|
||||||
.stream()
|
.stream()
|
||||||
.map(AppUser::getId)
|
.map(AppUser::getId)
|
||||||
.toList();
|
.toList();
|
||||||
|
return ensureLookupSize(ids, "uid / 昵称");
|
||||||
}
|
}
|
||||||
|
|
||||||
private List<Long> searchDramaIds(String dramaTitle) {
|
private List<Long> searchDramaIds(String dramaTitle) {
|
||||||
String value = dramaTitle.trim();
|
String value = dramaTitle.trim();
|
||||||
return dramaMapper.selectList(Wrappers.<Drama>lambdaQuery()
|
List<Long> ids = dramaMapper.selectList(Wrappers.<Drama>lambdaQuery()
|
||||||
.select(Drama::getId)
|
.select(Drama::getId)
|
||||||
.like(Drama::getTitle, value))
|
.like(Drama::getTitle, value)
|
||||||
|
.last("LIMIT " + LOOKUP_PROBE))
|
||||||
.stream()
|
.stream()
|
||||||
.map(Drama::getId)
|
.map(Drama::getId)
|
||||||
.toList();
|
.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) {
|
private List<AdminWatchHistoryVO> toVoList(List<WatchHistory> rows) {
|
||||||
@@ -129,19 +166,28 @@ public class WatchHistoryAdminService {
|
|||||||
Map<Long, DramaEpisode> episodes = loadEpisodes(rows);
|
Map<Long, DramaEpisode> episodes = loadEpisodes(rows);
|
||||||
Map<Long, List<AppUserSession>> sessions = sessionSystem.groupByUser(
|
Map<Long, List<AppUserSession>> sessions = sessionSystem.groupByUser(
|
||||||
rows.stream().map(WatchHistory::getUserId).distinct().toList());
|
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());
|
List<AdminWatchHistoryVO> list = new ArrayList<>(rows.size());
|
||||||
for (WatchHistory row : rows) {
|
for (WatchHistory row : rows) {
|
||||||
list.add(toVo(row, users.get(row.getUserId()), dramas.get(row.getDramaId()),
|
list.add(toVo(row, users.get(row.getUserId()), dramas.get(row.getDramaId()),
|
||||||
episodes.get(row.getEpisodeId()),
|
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;
|
return list;
|
||||||
}
|
}
|
||||||
|
|
||||||
private Map<Long, AppUser> loadUsers(List<WatchHistory> rows) {
|
private Map<Long, AppUser> loadUsers(List<WatchHistory> rows) {
|
||||||
Map<Long, AppUser> result = new HashMap<>();
|
Map<Long, AppUser> result = new HashMap<>();
|
||||||
List<Long> ids = rows.stream().map(WatchHistory::getUserId).distinct().toList();
|
List<Long> ids = rows.stream().map(WatchHistory::getUserId).filter(Objects::nonNull).distinct().toList();
|
||||||
for (AppUser user : appUserMapper.selectBatchIds(ids)) {
|
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);
|
result.put(user.getId(), user);
|
||||||
}
|
}
|
||||||
return result;
|
return result;
|
||||||
@@ -149,8 +195,13 @@ public class WatchHistoryAdminService {
|
|||||||
|
|
||||||
private Map<Long, Drama> loadDramas(List<WatchHistory> rows) {
|
private Map<Long, Drama> loadDramas(List<WatchHistory> rows) {
|
||||||
Map<Long, Drama> result = new HashMap<>();
|
Map<Long, Drama> result = new HashMap<>();
|
||||||
List<Long> ids = rows.stream().map(WatchHistory::getDramaId).distinct().toList();
|
List<Long> ids = rows.stream().map(WatchHistory::getDramaId).filter(Objects::nonNull).distinct().toList();
|
||||||
for (Drama drama : dramaMapper.selectBatchIds(ids)) {
|
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);
|
result.put(drama.getId(), drama);
|
||||||
}
|
}
|
||||||
return result;
|
return result;
|
||||||
@@ -158,15 +209,22 @@ public class WatchHistoryAdminService {
|
|||||||
|
|
||||||
private Map<Long, DramaEpisode> loadEpisodes(List<WatchHistory> rows) {
|
private Map<Long, DramaEpisode> loadEpisodes(List<WatchHistory> rows) {
|
||||||
Map<Long, DramaEpisode> result = new HashMap<>();
|
Map<Long, DramaEpisode> result = new HashMap<>();
|
||||||
List<Long> ids = rows.stream().map(WatchHistory::getEpisodeId).distinct().toList();
|
List<Long> ids = rows.stream().map(WatchHistory::getEpisodeId).filter(Objects::nonNull).distinct().toList();
|
||||||
for (DramaEpisode episode : dramaEpisodeMapper.selectBatchIds(ids)) {
|
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);
|
result.put(episode.getId(), episode);
|
||||||
}
|
}
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
private AdminWatchHistoryVO toVo(WatchHistory row, AppUser user, Drama drama,
|
private AdminWatchHistoryVO toVo(WatchHistory row, AppUser user, Drama drama,
|
||||||
DramaEpisode episode, AppUserSession session) {
|
DramaEpisode episode, AppUserSession session,
|
||||||
|
AppDeviceProfile profile) {
|
||||||
AdminWatchHistoryVO vo = new AdminWatchHistoryVO();
|
AdminWatchHistoryVO vo = new AdminWatchHistoryVO();
|
||||||
vo.setId(row.getId());
|
vo.setId(row.getId());
|
||||||
vo.setUserId(row.getUserId());
|
vo.setUserId(row.getUserId());
|
||||||
@@ -177,7 +235,10 @@ public class WatchHistoryAdminService {
|
|||||||
vo.setSource(row.getSource());
|
vo.setSource(row.getSource());
|
||||||
vo.setDeviceId(row.getDeviceId());
|
vo.setDeviceId(row.getDeviceId());
|
||||||
vo.setWatchedAt(row.getWatchedAt());
|
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.setPlatform(session.getPlatform());
|
||||||
vo.setOsVersion(session.getOsVersion());
|
vo.setOsVersion(session.getOsVersion());
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -13,6 +13,8 @@ public class AdminFavoriteVO {
|
|||||||
private String uid;
|
private String uid;
|
||||||
private String nickname;
|
private String nickname;
|
||||||
private String deviceId;
|
private String deviceId;
|
||||||
|
/** 客户端平台:ANDROID / IOS / WEB,来自设备平台记录,游客同样有值。 */
|
||||||
|
private String platform;
|
||||||
private Boolean anonymous;
|
private Boolean anonymous;
|
||||||
private Long dramaId;
|
private Long dramaId;
|
||||||
private String dramaTitle;
|
private String dramaTitle;
|
||||||
|
|||||||
@@ -13,6 +13,8 @@ public class AdminLikeVO {
|
|||||||
private String uid;
|
private String uid;
|
||||||
private String nickname;
|
private String nickname;
|
||||||
private String deviceId;
|
private String deviceId;
|
||||||
|
/** 客户端平台:ANDROID / IOS / WEB,来自设备平台记录,游客同样有值。 */
|
||||||
|
private String platform;
|
||||||
private Boolean anonymous;
|
private Boolean anonymous;
|
||||||
private Long dramaId;
|
private Long dramaId;
|
||||||
private String dramaTitle;
|
private String dramaTitle;
|
||||||
|
|||||||
@@ -16,6 +16,8 @@ public class EngagementQueryReqVO {
|
|||||||
private String keyword;
|
private String keyword;
|
||||||
private Long userId;
|
private Long userId;
|
||||||
private String deviceId;
|
private String deviceId;
|
||||||
|
/** 客户端平台筛选:ANDROID / IOS / WEB,留空不限制。 */
|
||||||
|
private String platform;
|
||||||
private String dramaTitle;
|
private String dramaTitle;
|
||||||
private Long dramaId;
|
private Long dramaId;
|
||||||
@DateTimeFormat(pattern = TIME_PATTERN)
|
@DateTimeFormat(pattern = TIME_PATTERN)
|
||||||
|
|||||||
+2
@@ -20,6 +20,8 @@ public class WatchHistoryQueryReqVO {
|
|||||||
private String dramaTitle;
|
private String dramaTitle;
|
||||||
private Long dramaId;
|
private Long dramaId;
|
||||||
private Long episodeId;
|
private Long episodeId;
|
||||||
|
/** 客户端平台筛选:ANDROID / IOS / WEB,留空不限制。 */
|
||||||
|
private String platform;
|
||||||
private String source;
|
private String source;
|
||||||
private Boolean finished;
|
private Boolean finished;
|
||||||
|
|
||||||
|
|||||||
@@ -14,6 +14,12 @@ spring:
|
|||||||
url: ${DB_URL:jdbc:postgresql://127.0.0.1:5432/fruitshorts}
|
url: ${DB_URL:jdbc:postgresql://127.0.0.1:5432/fruitshorts}
|
||||||
username: ${DB_USER:fruitshorts}
|
username: ${DB_USER:fruitshorts}
|
||||||
password: ${DB_PASSWORD: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:
|
jackson:
|
||||||
date-format: yyyy-MM-dd HH:mm:ss
|
date-format: yyyy-MM-dd HH:mm:ss
|
||||||
time-zone: Asia/Shanghai
|
time-zone: Asia/Shanghai
|
||||||
@@ -40,6 +46,11 @@ mybatis-plus:
|
|||||||
datacenter-id: 1
|
datacenter-id: 1
|
||||||
|
|
||||||
fruitshorts:
|
fruitshorts:
|
||||||
|
web:
|
||||||
|
# 可信代理 CIDR 列表(逗号分隔)。默认空 = 永不解析 XFF,直接用 request.getRemoteAddr()。
|
||||||
|
# 配了反代(Nginx/K8s ingress)时填上其 IP 段,才会从 XFF 取最右侧可信一跳。
|
||||||
|
# 生产若开启了 Tomcat RemoteIpValve(prod profile 已配),保持空即可。
|
||||||
|
trusted-proxies: ${FRUITSHORTS_TRUSTED_PROXIES:}
|
||||||
media:
|
media:
|
||||||
# 媒资目录:容器里指向挂载卷;本地默认 ./data/uploads
|
# 媒资目录:容器里指向挂载卷;本地默认 ./data/uploads
|
||||||
root-dir: ${MEDIA_ROOT_DIR:./data/uploads}
|
root-dir: ${MEDIA_ROOT_DIR:./data/uploads}
|
||||||
|
|||||||
+1
-1
@@ -13,7 +13,7 @@ import static org.assertj.core.api.Assertions.assertThat;
|
|||||||
/** 观看记录匹配登录系统。 */
|
/** 观看记录匹配登录系统。 */
|
||||||
class AppSessionSystemServiceTest {
|
class AppSessionSystemServiceTest {
|
||||||
|
|
||||||
private final AppSessionSystemService service = new AppSessionSystemService(null);
|
private final AppSessionSystemService service = new AppSessionSystemService(null, null);
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
void should_use_same_device_session_when_watch_history_has_device() {
|
void should_use_same_device_session_when_watch_history_has_device() {
|
||||||
|
|||||||
+4
-3
@@ -199,9 +199,10 @@ class WatchHistoryAdminServiceTest {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private void stubBatchData() {
|
private void stubBatchData() {
|
||||||
given(appUserMapper.selectBatchIds(any())).willReturn(List.of(user(7L, "Alice")));
|
// 装配阶段按需列查询,而不是 selectBatchIds,桩法要跟着改
|
||||||
given(dramaMapper.selectBatchIds(any())).willReturn(List.of(drama(5L, "Signal")));
|
given(appUserMapper.selectList(any())).willReturn(List.of(user(7L, "Alice")));
|
||||||
given(dramaEpisodeMapper.selectBatchIds(any())).willReturn(List.of(episode(11L)));
|
given(dramaMapper.selectList(any())).willReturn(List.of(drama(5L, "Signal")));
|
||||||
|
given(dramaEpisodeMapper.selectList(any())).willReturn(List.of(episode(11L)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@SuppressWarnings("unchecked")
|
@SuppressWarnings("unchecked")
|
||||||
|
|||||||
+3
-1
@@ -30,7 +30,9 @@ public final class ControllerContractSupport {
|
|||||||
/** 组装一个只含指定控制器与全局异常处理的 MockMvc。 */
|
/** 组装一个只含指定控制器与全局异常处理的 MockMvc。 */
|
||||||
@SuppressWarnings("unchecked")
|
@SuppressWarnings("unchecked")
|
||||||
public static MockMvc mockMvcOf(Object controller) {
|
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)
|
return MockMvcBuilders.standaloneSetup(controller)
|
||||||
.setControllerAdvice(new GlobalExceptionHandler(recorder))
|
.setControllerAdvice(new GlobalExceptionHandler(recorder))
|
||||||
// 与 Spring Boot 的 Jackson 配置对齐:时间按 ISO 字符串输出而不是数组,
|
// 与 Spring Boot 的 Jackson 配置对齐:时间按 ISO 字符串输出而不是数组,
|
||||||
|
|||||||
+6
-4
@@ -35,13 +35,14 @@ public class AppAuthController {
|
|||||||
|
|
||||||
private final AppOtpService otpService;
|
private final AppOtpService otpService;
|
||||||
private final AppAuthService authService;
|
private final AppAuthService authService;
|
||||||
|
private final WebUtils webUtils;
|
||||||
|
|
||||||
/** 发送验证码。频控在服务端,客户端不需要自己限流。 */
|
/** 发送验证码。频控在服务端,客户端不需要自己限流。 */
|
||||||
@Operation(summary = "发送验证码(频控在服务端)")
|
@Operation(summary = "发送验证码(频控在服务端)")
|
||||||
@RateLimit(limit = 20, windowSeconds = 60, key = "otp")
|
@RateLimit(limit = 20, windowSeconds = 60, key = "otp")
|
||||||
@PostMapping("/otp")
|
@PostMapping("/otp")
|
||||||
public Result<Void> sendOtp(@Valid @RequestBody OtpSendReqVO req, HttpServletRequest request) {
|
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();
|
return Result.ok();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -50,7 +51,7 @@ public class AppAuthController {
|
|||||||
public Result<TokenRespVO> register(@Valid @RequestBody RegisterReqVO req,
|
public Result<TokenRespVO> register(@Valid @RequestBody RegisterReqVO req,
|
||||||
HttpServletRequest request) {
|
HttpServletRequest request) {
|
||||||
applyHeaderDeviceId(req, request);
|
applyHeaderDeviceId(req, request);
|
||||||
return Result.ok(authService.register(req, WebUtils.getClientIp(request)));
|
return Result.ok(authService.register(req, webUtils.clientIp(request)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Operation(summary = "登录(验证码或密码)")
|
@Operation(summary = "登录(验证码或密码)")
|
||||||
@@ -59,7 +60,7 @@ public class AppAuthController {
|
|||||||
public Result<TokenRespVO> login(@Valid @RequestBody LoginReqVO req,
|
public Result<TokenRespVO> login(@Valid @RequestBody LoginReqVO req,
|
||||||
HttpServletRequest request) {
|
HttpServletRequest request) {
|
||||||
applyHeaderDeviceId(req, request);
|
applyHeaderDeviceId(req, request);
|
||||||
return Result.ok(authService.login(req, WebUtils.getClientIp(request)));
|
return Result.ok(authService.login(req, webUtils.clientIp(request)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Operation(summary = "Google 登录(未配置 clientId 时返回明确错误)")
|
@Operation(summary = "Google 登录(未配置 clientId 时返回明确错误)")
|
||||||
@@ -69,7 +70,8 @@ public class AppAuthController {
|
|||||||
String headerDeviceId = request.getHeader("X-Device-Id");
|
String headerDeviceId = request.getHeader("X-Device-Id");
|
||||||
String deviceId = StringUtils.hasText(headerDeviceId)
|
String deviceId = StringUtils.hasText(headerDeviceId)
|
||||||
? headerDeviceId : req.getDeviceId();
|
? headerDeviceId : req.getDeviceId();
|
||||||
return Result.ok(authService.googleLogin(req.getIdToken(), deviceId));
|
return Result.ok(authService.googleLogin(req.getIdToken(), deviceId,
|
||||||
|
webUtils.clientIp(request)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Operation(summary = "刷新令牌(旧 refresh token 立即失效)")
|
@Operation(summary = "刷新令牌(旧 refresh token 立即失效)")
|
||||||
|
|||||||
+2
-1
@@ -48,6 +48,7 @@ public class AppCommentController {
|
|||||||
private final EpisodeService episodeService;
|
private final EpisodeService episodeService;
|
||||||
private final InteractionIdentityResolver identityResolver;
|
private final InteractionIdentityResolver identityResolver;
|
||||||
private final AnonymousInteractionRateLimiter anonymousRateLimiter;
|
private final AnonymousInteractionRateLimiter anonymousRateLimiter;
|
||||||
|
private final WebUtils webUtils;
|
||||||
|
|
||||||
@Operation(summary = "发布评论")
|
@Operation(summary = "发布评论")
|
||||||
@PostMapping
|
@PostMapping
|
||||||
@@ -59,7 +60,7 @@ public class AppCommentController {
|
|||||||
req.getDeviceId());
|
req.getDeviceId());
|
||||||
anonymousRateLimiter.check(identity, "comment");
|
anonymousRateLimiter.check(identity, "comment");
|
||||||
return Result.ok(commentService.publish(
|
return Result.ok(commentService.publish(
|
||||||
identity, req, WebUtils.getClientIp(request)));
|
identity, req, webUtils.clientIp(request)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Operation(summary = "评论列表(游标翻页,可匿名浏览)")
|
@Operation(summary = "评论列表(游标翻页,可匿名浏览)")
|
||||||
|
|||||||
+2
-1
@@ -37,6 +37,7 @@ public class AppDramaController {
|
|||||||
private final EpisodeService episodeService;
|
private final EpisodeService episodeService;
|
||||||
private final CommentService commentService;
|
private final CommentService commentService;
|
||||||
private final AppEpisodeStatsService appEpisodeStatsService;
|
private final AppEpisodeStatsService appEpisodeStatsService;
|
||||||
|
private final WebUtils webUtils;
|
||||||
|
|
||||||
/** 详情:包含剧集列表,客户端进详情页一次请求即可渲染 */
|
/** 详情:包含剧集列表,客户端进详情页一次请求即可渲染 */
|
||||||
@Operation(summary = "短剧详情(含剧集列表,进详情页一次请求即可渲染)")
|
@Operation(summary = "短剧详情(含剧集列表,进详情页一次请求即可渲染)")
|
||||||
@@ -56,7 +57,7 @@ public class AppDramaController {
|
|||||||
SecurityUtils.getUserIdOrNull(),
|
SecurityUtils.getUserIdOrNull(),
|
||||||
id,
|
id,
|
||||||
request.getHeader("X-Device-Id"),
|
request.getHeader("X-Device-Id"),
|
||||||
WebUtils.getClientIp(request));
|
webUtils.clientIp(request));
|
||||||
DramaEpisode episode = episodeService.getExisting(id);
|
DramaEpisode episode = episodeService.getExisting(id);
|
||||||
vo.setLikeCount(episode.getLikeCount() == null ? 0L : episode.getLikeCount());
|
vo.setLikeCount(episode.getLikeCount() == null ? 0L : episode.getLikeCount());
|
||||||
vo.setCommentCount(commentService.countByEpisodes(List.of(id)).getOrDefault(id, 0L));
|
vo.setCommentCount(commentService.countByEpisodes(List.of(id)).getOrDefault(id, 0L));
|
||||||
|
|||||||
+2
-1
@@ -25,11 +25,12 @@ import org.springframework.web.bind.annotation.RestController;
|
|||||||
public class AppEventController {
|
public class AppEventController {
|
||||||
|
|
||||||
private final AppEventService eventService;
|
private final AppEventService eventService;
|
||||||
|
private final WebUtils webUtils;
|
||||||
|
|
||||||
@Operation(summary = "埋点事件批量上报")
|
@Operation(summary = "埋点事件批量上报")
|
||||||
@PostMapping("/events")
|
@PostMapping("/events")
|
||||||
public Result<Void> report(@Valid @RequestBody AppEventReqVO req, HttpServletRequest request) {
|
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();
|
return Result.ok();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -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:
|
cors:
|
||||||
# 生产只允许显式配置的来源;没配就等于空,浏览器侧会被拦住(比"*"安全)
|
# 生产只允许显式配置的来源;没配就等于空,浏览器侧会被拦住(比"*"安全)
|
||||||
allowed-origins: ${CORS_ALLOWED_ORIGINS:}
|
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:
|
springdoc:
|
||||||
# 生产只留机器可读契约,关掉 UI(doc.html 不对外开放)
|
# 生产只留机器可读契约,关掉 UI(doc.html 不对外开放)
|
||||||
|
|||||||
@@ -11,13 +11,22 @@ spring:
|
|||||||
datasource:
|
datasource:
|
||||||
driver-class-name: org.postgresql.Driver
|
driver-class-name: org.postgresql.Driver
|
||||||
# 地址必须外置:不同环境(本机 / 测试 / 生产)连的库不一样,
|
# 地址必须外置:不同环境(本机 / 测试 / 生产)连的库不一样,
|
||||||
# 生产请用 DB_URL 注入(prod profile 会校验它非空)
|
# 生产请用 DB_URL 注入(prod profile 会校验它非空)。
|
||||||
url: ${DB_URL:jdbc:postgresql://127.0.0.1:5432/fruitshorts}
|
# reWriteBatchedInserts:pgjdbc 把批内多条 INSERT 重写成多值 INSERT,
|
||||||
|
# 观看进度缓冲落库依赖它才能跑满批量吞吐。
|
||||||
|
url: ${DB_URL:jdbc:postgresql://127.0.0.1:5432/fruitshorts?reWriteBatchedInserts=true}
|
||||||
username: ${DB_USER:fruitshorts}
|
username: ${DB_USER:fruitshorts}
|
||||||
password: ${DB_PASSWORD:fruitshorts}
|
password: ${DB_PASSWORD:fruitshorts}
|
||||||
jackson:
|
jackson:
|
||||||
date-format: yyyy-MM-dd HH:mm:ss
|
date-format: yyyy-MM-dd HH:mm:ss
|
||||||
time-zone: Asia/Shanghai
|
time-zone: Asia/Shanghai
|
||||||
|
task:
|
||||||
|
scheduling:
|
||||||
|
# 默认调度池只有 1 条线程:观看进度落库(每 2s)会和其他定时任务互相排队,
|
||||||
|
# 落库一旦拖慢就会连带推迟对时延敏感的任务,必须给池子多几条线程。
|
||||||
|
pool:
|
||||||
|
size: 4
|
||||||
|
thread-name-prefix: fruitshorts-sched-
|
||||||
data:
|
data:
|
||||||
redis:
|
redis:
|
||||||
host: ${REDIS_HOST:127.0.0.1}
|
host: ${REDIS_HOST:127.0.0.1}
|
||||||
@@ -47,6 +56,11 @@ springdoc:
|
|||||||
path: /v3/api-docs
|
path: /v3/api-docs
|
||||||
|
|
||||||
fruitshorts:
|
fruitshorts:
|
||||||
|
web:
|
||||||
|
# 可信代理 CIDR 列表(逗号分隔)。默认空 = 永不解析 XFF,直接用 request.getRemoteAddr()。
|
||||||
|
# 配了反代(Nginx/K8s ingress)时填上其 IP 段,才会从 XFF 取最右侧可信一跳。
|
||||||
|
# 生产若开启了 Tomcat RemoteIpValve(prod profile 已配),保持空即可。
|
||||||
|
trusted-proxies: ${FRUITSHORTS_TRUSTED_PROXIES:}
|
||||||
auth:
|
auth:
|
||||||
# 仅作 sys_config 缺失/非法时的兜底;运行时优先读取后台「系统配置」
|
# 仅作 sys_config 缺失/非法时的兜底;运行时优先读取后台「系统配置」
|
||||||
register-max-per-ip-per-day: ${REGISTER_MAX_PER_IP_PER_DAY:5}
|
register-max-per-ip-per-day: ${REGISTER_MAX_PER_IP_PER_DAY:5}
|
||||||
|
|||||||
+4
-1
@@ -4,6 +4,7 @@ package com.fruitshorts.app.controller;
|
|||||||
import com.fruitshorts.app.support.ControllerContractSupport;
|
import com.fruitshorts.app.support.ControllerContractSupport;
|
||||||
import com.fruitshorts.common.exception.ErrorCode;
|
import com.fruitshorts.common.exception.ErrorCode;
|
||||||
import com.fruitshorts.common.exception.ServiceException;
|
import com.fruitshorts.common.exception.ServiceException;
|
||||||
|
import com.fruitshorts.framework.web.WebUtils;
|
||||||
import com.fruitshorts.user.service.AppAuthService;
|
import com.fruitshorts.user.service.AppAuthService;
|
||||||
import com.fruitshorts.user.service.AppOtpService;
|
import com.fruitshorts.user.service.AppOtpService;
|
||||||
import com.fruitshorts.user.vo.TokenRespVO;
|
import com.fruitshorts.user.vo.TokenRespVO;
|
||||||
@@ -32,13 +33,15 @@ class AppAuthControllerContractTest {
|
|||||||
|
|
||||||
private AppOtpService otpService;
|
private AppOtpService otpService;
|
||||||
private AppAuthService authService;
|
private AppAuthService authService;
|
||||||
|
private WebUtils webUtils;
|
||||||
private MockMvc mockMvc;
|
private MockMvc mockMvc;
|
||||||
|
|
||||||
@BeforeEach
|
@BeforeEach
|
||||||
void setUp() {
|
void setUp() {
|
||||||
otpService = mock(AppOtpService.class);
|
otpService = mock(AppOtpService.class);
|
||||||
authService = mock(AppAuthService.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
|
@Test
|
||||||
|
|||||||
+2
-1
@@ -51,7 +51,8 @@ class AppCatalogControllerContractTest {
|
|||||||
dramaMvc = ControllerContractSupport.mockMvcOf(
|
dramaMvc = ControllerContractSupport.mockMvcOf(
|
||||||
new AppDramaController(catalogService, mock(PlaybackService.class),
|
new AppDramaController(catalogService, mock(PlaybackService.class),
|
||||||
mock(EpisodeService.class), mock(CommentService.class),
|
mock(EpisodeService.class), mock(CommentService.class),
|
||||||
mock(AppEpisodeStatsService.class)));
|
mock(AppEpisodeStatsService.class),
|
||||||
|
mock(com.fruitshorts.framework.web.WebUtils.class)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
|
|||||||
+2
-1
@@ -38,7 +38,8 @@ class AppCommentControllerContractTest {
|
|||||||
commentService,
|
commentService,
|
||||||
episodeService,
|
episodeService,
|
||||||
mock(InteractionIdentityResolver.class),
|
mock(InteractionIdentityResolver.class),
|
||||||
mock(AnonymousInteractionRateLimiter.class)));
|
mock(AnonymousInteractionRateLimiter.class),
|
||||||
|
mock(com.fruitshorts.framework.web.WebUtils.class)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
|
|||||||
+5
-3
@@ -234,7 +234,8 @@ class AppCommentCountAnonymousSecurityTest {
|
|||||||
commentService,
|
commentService,
|
||||||
episodeService,
|
episodeService,
|
||||||
identityResolver,
|
identityResolver,
|
||||||
rateLimiter);
|
rateLimiter,
|
||||||
|
mock(com.fruitshorts.framework.web.WebUtils.class));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Bean
|
@Bean
|
||||||
@@ -303,7 +304,8 @@ class AppCommentCountAnonymousSecurityTest {
|
|||||||
mock(PlaybackService.class),
|
mock(PlaybackService.class),
|
||||||
episodeService,
|
episodeService,
|
||||||
commentService,
|
commentService,
|
||||||
appEpisodeStatsService);
|
appEpisodeStatsService,
|
||||||
|
mock(com.fruitshorts.framework.web.WebUtils.class));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Bean
|
@Bean
|
||||||
@@ -340,7 +342,7 @@ class AppCommentCountAnonymousSecurityTest {
|
|||||||
RestAuthEntryPoint restAuthEntryPoint(ObjectMapper objectMapper) {
|
RestAuthEntryPoint restAuthEntryPoint(ObjectMapper objectMapper) {
|
||||||
return new RestAuthEntryPoint(
|
return new RestAuthEntryPoint(
|
||||||
objectMapper,
|
objectMapper,
|
||||||
new AccessDeniedRecorder(mock(ObjectProvider.class)));
|
new AccessDeniedRecorder(mock(ObjectProvider.class), mock(ObjectProvider.class)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Bean
|
@Bean
|
||||||
|
|||||||
+2
-1
@@ -54,7 +54,8 @@ class AppPlaybackContractTest {
|
|||||||
given(commentService.countByEpisodes(any())).willReturn(Map.of());
|
given(commentService.countByEpisodes(any())).willReturn(Map.of());
|
||||||
mockMvc = ControllerContractSupport.mockMvcOf(
|
mockMvc = ControllerContractSupport.mockMvcOf(
|
||||||
new AppDramaController(mock(AppCatalogService.class), playbackService,
|
new AppDramaController(mock(AppCatalogService.class), playbackService,
|
||||||
episodeService, commentService, episodeStatsService));
|
episodeService, commentService, episodeStatsService,
|
||||||
|
mock(com.fruitshorts.framework.web.WebUtils.class)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
|
|||||||
+44
-4
@@ -22,6 +22,7 @@ import com.fruitshorts.user.service.AppOtpService;
|
|||||||
import com.fruitshorts.user.service.CredentialHasher;
|
import com.fruitshorts.user.service.CredentialHasher;
|
||||||
import com.fruitshorts.user.service.GoogleIdTokenVerifier;
|
import com.fruitshorts.user.service.GoogleIdTokenVerifier;
|
||||||
import com.fruitshorts.user.service.GoogleUser;
|
import com.fruitshorts.user.service.GoogleUser;
|
||||||
|
import com.fruitshorts.user.service.IpRegionService;
|
||||||
import com.fruitshorts.user.service.OtpSender;
|
import com.fruitshorts.user.service.OtpSender;
|
||||||
import com.fruitshorts.user.vo.LoginReqVO;
|
import com.fruitshorts.user.vo.LoginReqVO;
|
||||||
import com.fruitshorts.user.vo.RegisterReqVO;
|
import com.fruitshorts.user.vo.RegisterReqVO;
|
||||||
@@ -67,6 +68,7 @@ class AppAuthServiceTest {
|
|||||||
private ConfigService configService;
|
private ConfigService configService;
|
||||||
private GoogleIdTokenVerifier googleVerifier;
|
private GoogleIdTokenVerifier googleVerifier;
|
||||||
private ApplicationEventPublisher eventPublisher;
|
private ApplicationEventPublisher eventPublisher;
|
||||||
|
private IpRegionService ipRegionService;
|
||||||
private AppAuthService authService;
|
private AppAuthService authService;
|
||||||
|
|
||||||
@BeforeEach
|
@BeforeEach
|
||||||
@@ -83,11 +85,14 @@ class AppAuthServiceTest {
|
|||||||
configService = mock(ConfigService.class);
|
configService = mock(ConfigService.class);
|
||||||
googleVerifier = mock(GoogleIdTokenVerifier.class);
|
googleVerifier = mock(GoogleIdTokenVerifier.class);
|
||||||
eventPublisher = mock(ApplicationEventPublisher.class);
|
eventPublisher = mock(ApplicationEventPublisher.class);
|
||||||
|
ipRegionService = mock(IpRegionService.class);
|
||||||
given(configService.get(anyString(), anyString()))
|
given(configService.get(anyString(), anyString()))
|
||||||
.willAnswer(invocation -> invocation.getArgument(1));
|
.willAnswer(invocation -> invocation.getArgument(1));
|
||||||
|
// 单测里默认解析不出国家,走客户端区域 / 印尼兜底
|
||||||
|
given(ipRegionService.resolveCountry(any())).willReturn("");
|
||||||
authService = new AppAuthService(userMapper, credentialMapper, sessionMapper, consentMapper,
|
authService = new AppAuthService(userMapper, credentialMapper, sessionMapper, consentMapper,
|
||||||
deviceMapper, otpService, hasher, tokenProvider, passwordEncoder, configService,
|
deviceMapper, otpService, hasher, tokenProvider, passwordEncoder, configService,
|
||||||
googleVerifier, eventPublisher);
|
googleVerifier, eventPublisher, ipRegionService);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
@@ -264,6 +269,41 @@ class AppAuthServiceTest {
|
|||||||
assertThat(event.deviceId()).isEqualTo("device-9");
|
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
|
@Test
|
||||||
void should_reject_login_when_password_does_not_match() {
|
void should_reject_login_when_password_does_not_match() {
|
||||||
stubLogin(credential("bcrypt-hash"), user(AppUserStatus.NORMAL));
|
stubLogin(credential("bcrypt-hash"), user(AppUserStatus.NORMAL));
|
||||||
@@ -318,7 +358,7 @@ class AppAuthServiceTest {
|
|||||||
});
|
});
|
||||||
given(userMapper.selectById(any())).willAnswer(invocation -> inserted.get());
|
given(userMapper.selectById(any())).willAnswer(invocation -> inserted.get());
|
||||||
|
|
||||||
TokenRespVO resp = authService.googleLogin("google-id-token", "device-9");
|
TokenRespVO resp = authService.googleLogin("google-id-token", "device-9", "203.0.113.9");
|
||||||
|
|
||||||
assertThat(resp.getAccessToken()).isEqualTo("access-1");
|
assertThat(resp.getAccessToken()).isEqualTo("access-1");
|
||||||
assertThat(resp.isNewUser()).isTrue();
|
assertThat(resp.isNewUser()).isTrue();
|
||||||
@@ -333,7 +373,7 @@ class AppAuthServiceTest {
|
|||||||
given(googleVerifier.verify(anyString()))
|
given(googleVerifier.verify(anyString()))
|
||||||
.willReturn(new GoogleUser("sub-1", "", "Alice", ""));
|
.willReturn(new GoogleUser("sub-1", "", "Alice", ""));
|
||||||
|
|
||||||
assertThatThrownBy(() -> authService.googleLogin("google-id-token", "device-9"))
|
assertThatThrownBy(() -> authService.googleLogin("google-id-token", "device-9", "203.0.113.9"))
|
||||||
.isInstanceOf(ServiceException.class)
|
.isInstanceOf(ServiceException.class)
|
||||||
.hasMessage("Google 账号未返回邮箱,无法登录");
|
.hasMessage("Google 账号未返回邮箱,无法登录");
|
||||||
}
|
}
|
||||||
@@ -392,7 +432,7 @@ class AppAuthServiceTest {
|
|||||||
AppOtpService realOtpService = new AppOtpService(otpMapper, mock(OtpSender.class), otpHasher);
|
AppOtpService realOtpService = new AppOtpService(otpMapper, mock(OtpSender.class), otpHasher);
|
||||||
return new AppAuthService(userMapper, credentialMapper, sessionMapper, consentMapper,
|
return new AppAuthService(userMapper, credentialMapper, sessionMapper, consentMapper,
|
||||||
deviceMapper, realOtpService, hasher, tokenProvider, passwordEncoder, configService,
|
deviceMapper, realOtpService, hasher, tokenProvider, passwordEncoder, configService,
|
||||||
googleVerifier, eventPublisher);
|
googleVerifier, eventPublisher, ipRegionService);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void setRegisterLimits(int perIp, int perDevice) {
|
private void setRegisterLimits(int perIp, int perDevice) {
|
||||||
|
|||||||
@@ -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");
|
||||||
|
}
|
||||||
|
}
|
||||||
+49
@@ -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();
|
||||||
|
}
|
||||||
|
}
|
||||||
+4
-1
@@ -2,6 +2,7 @@
|
|||||||
package com.fruitshorts.app.support;
|
package com.fruitshorts.app.support;
|
||||||
|
|
||||||
import com.fruitshorts.framework.log.AccessDeniedRecorder;
|
import com.fruitshorts.framework.log.AccessDeniedRecorder;
|
||||||
|
import com.fruitshorts.framework.web.WebUtils;
|
||||||
import com.fruitshorts.framework.web.JacksonLongToStringConfig;
|
import com.fruitshorts.framework.web.JacksonLongToStringConfig;
|
||||||
import com.fruitshorts.framework.web.GlobalExceptionHandler;
|
import com.fruitshorts.framework.web.GlobalExceptionHandler;
|
||||||
import com.fasterxml.jackson.databind.SerializationFeature;
|
import com.fasterxml.jackson.databind.SerializationFeature;
|
||||||
@@ -30,7 +31,9 @@ public final class ControllerContractSupport {
|
|||||||
/** 组装一个只含指定控制器与全局异常处理的 MockMvc。 */
|
/** 组装一个只含指定控制器与全局异常处理的 MockMvc。 */
|
||||||
@SuppressWarnings("unchecked")
|
@SuppressWarnings("unchecked")
|
||||||
public static MockMvc mockMvcOf(Object controller) {
|
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)
|
return MockMvcBuilders.standaloneSetup(controller)
|
||||||
.setControllerAdvice(new GlobalExceptionHandler(recorder))
|
.setControllerAdvice(new GlobalExceptionHandler(recorder))
|
||||||
// 与 Spring Boot 的 Jackson 配置对齐:时间按 ISO 字符串输出而不是数组,
|
// 与 Spring Boot 的 Jackson 配置对齐:时间按 ISO 字符串输出而不是数组,
|
||||||
|
|||||||
+135
-1
@@ -20,18 +20,28 @@ import com.fruitshorts.playback.vo.ProgressRespVO;
|
|||||||
import com.fruitshorts.playback.vo.ProgressSaveReqVO;
|
import com.fruitshorts.playback.vo.ProgressSaveReqVO;
|
||||||
import org.apache.ibatis.builder.MapperBuilderAssistant;
|
import org.apache.ibatis.builder.MapperBuilderAssistant;
|
||||||
import org.apache.ibatis.annotations.Select;
|
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.BeforeAll;
|
||||||
import org.junit.jupiter.api.BeforeEach;
|
import org.junit.jupiter.api.BeforeEach;
|
||||||
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.Test;
|
||||||
import org.mockito.ArgumentCaptor;
|
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.time.LocalDateTime;
|
||||||
import java.lang.reflect.Method;
|
import java.lang.reflect.Method;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
import static org.assertj.core.api.Assertions.assertThat;
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.ArgumentMatchers.any;
|
||||||
|
import static org.mockito.ArgumentMatchers.eq;
|
||||||
import static org.mockito.BDDMockito.given;
|
import static org.mockito.BDDMockito.given;
|
||||||
|
import static org.mockito.Mockito.inOrder;
|
||||||
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.mock;
|
||||||
import static org.mockito.Mockito.never;
|
import static org.mockito.Mockito.never;
|
||||||
import static org.mockito.Mockito.times;
|
import static org.mockito.Mockito.times;
|
||||||
@@ -58,11 +68,18 @@ class WatchProgressServiceTest {
|
|||||||
historyMapper = mock(WatchHistoryMapper.class);
|
historyMapper = mock(WatchHistoryMapper.class);
|
||||||
dramaService = mock(DramaService.class);
|
dramaService = mock(DramaService.class);
|
||||||
episodeService = mock(EpisodeService.class);
|
episodeService = mock(EpisodeService.class);
|
||||||
|
// 测试统一走缓冲关闭(同步写库)路径:getIfAvailable 返回 null
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
ObjectProvider<WatchProgressBuffer> bufferProvider = mock(ObjectProvider.class);
|
||||||
|
given(bufferProvider.getIfAvailable()).willReturn(null);
|
||||||
progressService = new WatchProgressService(
|
progressService = new WatchProgressService(
|
||||||
progressMapper,
|
progressMapper,
|
||||||
historyMapper,
|
historyMapper,
|
||||||
dramaService,
|
dramaService,
|
||||||
episodeService);
|
episodeService,
|
||||||
|
bufferProvider,
|
||||||
|
mock(SqlSessionFactory.class),
|
||||||
|
new TransactionTemplate(mock(PlatformTransactionManager.class)));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
@@ -319,6 +336,123 @@ class WatchProgressServiceTest {
|
|||||||
verify(progressMapper).deleteById(1L);
|
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) {
|
private Drama drama(Long id, String title, int totalEpisodes) {
|
||||||
Drama drama = new Drama();
|
Drama drama = new Drama();
|
||||||
drama.setId(id);
|
drama.setId(id);
|
||||||
|
|||||||
+8
-1
@@ -22,6 +22,7 @@ import org.springframework.web.context.request.ServletRequestAttributes;
|
|||||||
public class AccessDeniedRecorder {
|
public class AccessDeniedRecorder {
|
||||||
|
|
||||||
private final ObjectProvider<OperationLogSink> sinkProvider;
|
private final ObjectProvider<OperationLogSink> sinkProvider;
|
||||||
|
private final ObjectProvider<WebUtils> webUtilsProvider;
|
||||||
|
|
||||||
public void record(HttpServletRequest request) {
|
public void record(HttpServletRequest request) {
|
||||||
OperationLogSink sink = sinkProvider.getIfAvailable();
|
OperationLogSink sink = sinkProvider.getIfAvailable();
|
||||||
@@ -47,10 +48,16 @@ public class AccessDeniedRecorder {
|
|||||||
.status("FAIL")
|
.status("FAIL")
|
||||||
.errorMsg("无权限")
|
.errorMsg("无权限")
|
||||||
.durationMs(0)
|
.durationMs(0)
|
||||||
.ip(request == null ? "" : WebUtils.getClientIp(request))
|
.ip(request == null ? "" : clientIp(request))
|
||||||
.build());
|
.build());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** 优先用 Spring 注入的 WebUtils(尊重可信代理配置),拿不到回退静态方法。 */
|
||||||
|
private String clientIp(HttpServletRequest request) {
|
||||||
|
WebUtils webUtils = webUtilsProvider.getIfAvailable();
|
||||||
|
return webUtils != null ? webUtils.clientIp(request) : WebUtils.getClientIp(request);
|
||||||
|
}
|
||||||
|
|
||||||
/** 从当前线程取出请求,供全局异常处理使用。 */
|
/** 从当前线程取出请求,供全局异常处理使用。 */
|
||||||
public void recordCurrentRequest() {
|
public void recordCurrentRequest() {
|
||||||
if (RequestContextHolder.getRequestAttributes() instanceof ServletRequestAttributes attributes) {
|
if (RequestContextHolder.getRequestAttributes() instanceof ServletRequestAttributes attributes) {
|
||||||
|
|||||||
+8
-1
@@ -43,6 +43,7 @@ public class OperationLogAspect {
|
|||||||
|
|
||||||
private final ObjectMapper objectMapper;
|
private final ObjectMapper objectMapper;
|
||||||
private final ObjectProvider<OperationLogSink> sinkProvider;
|
private final ObjectProvider<OperationLogSink> sinkProvider;
|
||||||
|
private final ObjectProvider<WebUtils> webUtilsProvider;
|
||||||
|
|
||||||
@Around("@annotation(operationLog)")
|
@Around("@annotation(operationLog)")
|
||||||
public Object around(ProceedingJoinPoint point, OperationLog operationLog) throws Throwable {
|
public Object around(ProceedingJoinPoint point, OperationLog operationLog) throws Throwable {
|
||||||
@@ -86,10 +87,16 @@ public class OperationLogAspect {
|
|||||||
.status(error == null ? "SUCCESS" : "FAIL")
|
.status(error == null ? "SUCCESS" : "FAIL")
|
||||||
.errorMsg(error == null ? "" : truncate(error.getMessage()))
|
.errorMsg(error == null ? "" : truncate(error.getMessage()))
|
||||||
.durationMs(durationMs)
|
.durationMs(durationMs)
|
||||||
.ip(request == null ? "" : WebUtils.getClientIp(request))
|
.ip(request == null ? "" : clientIp(request))
|
||||||
.build());
|
.build());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** 优先用 Spring 注入的 WebUtils(尊重可信代理配置),拿不到回退静态方法。 */
|
||||||
|
private String clientIp(HttpServletRequest request) {
|
||||||
|
WebUtils webUtils = webUtilsProvider.getIfAvailable();
|
||||||
|
return webUtils != null ? webUtils.clientIp(request) : WebUtils.getClientIp(request);
|
||||||
|
}
|
||||||
|
|
||||||
/** 把方法入参整理成 JSON,文件只记名称和大小。 */
|
/** 把方法入参整理成 JSON,文件只记名称和大小。 */
|
||||||
private String body(ProceedingJoinPoint point) {
|
private String body(ProceedingJoinPoint point) {
|
||||||
Object[] args = point.getArgs();
|
Object[] args = point.getArgs();
|
||||||
|
|||||||
+2
@@ -62,6 +62,8 @@ public final class ImpersonationGuard {
|
|||||||
|
|
||||||
public static String clientIp() {
|
public static String clientIp() {
|
||||||
HttpServletRequest request = currentRequest();
|
HttpServletRequest request = currentRequest();
|
||||||
|
// 这里走静态 getClientIp——等价于 trusted-proxies 为空时的逻辑(只用 remoteAddr,不解析 XFF)。
|
||||||
|
// 这个类是纯静态工具(无 Spring 注入),且只在 Creator 代看场景用,安全影响低。
|
||||||
return request == null ? "" : com.fruitshorts.framework.web.WebUtils.getClientIp(request);
|
return request == null ? "" : com.fruitshorts.framework.web.WebUtils.getClientIp(request);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -3,25 +3,146 @@
|
|||||||
package com.fruitshorts.framework.web;
|
package com.fruitshorts.framework.web;
|
||||||
|
|
||||||
import jakarta.servlet.http.HttpServletRequest;
|
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;
|
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 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 反向代理。 */
|
/** Spring 注入版:自动按配置解析 XFF。 */
|
||||||
public static String getClientIp(HttpServletRequest request) {
|
public String clientIp(HttpServletRequest request) {
|
||||||
String[] headers = {"X-Forwarded-For", "X-Real-IP", "Proxy-Client-IP", "WL-Proxy-Client-IP"};
|
List<String> cidrs = StringUtils.hasText(trustedProxies)
|
||||||
for (String header : headers) {
|
? Arrays.asList(trustedProxies.split(","))
|
||||||
String value = request.getHeader(header);
|
: List.of();
|
||||||
if (StringUtils.hasText(value) && !UNKNOWN.equalsIgnoreCase(value)) {
|
return resolve(request, cidrs);
|
||||||
int comma = value.indexOf(',');
|
}
|
||||||
return comma > 0 ? value.substring(0, comma).trim() : value.trim();
|
|
||||||
|
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() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -33,5 +33,10 @@
|
|||||||
<groupId>com.fruitshorts</groupId>
|
<groupId>com.fruitshorts</groupId>
|
||||||
<artifactId>fruitshorts-module-user</artifactId>
|
<artifactId>fruitshorts-module-user</artifactId>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
<!-- 进度上报路径的剧集/短剧进程内缓存(版本由 Spring Boot BOM 管理) -->
|
||||||
|
<dependency>
|
||||||
|
<groupId>com.github.ben-manes.caffeine</groupId>
|
||||||
|
<artifactId>caffeine</artifactId>
|
||||||
|
</dependency>
|
||||||
</dependencies>
|
</dependencies>
|
||||||
</project>
|
</project>
|
||||||
+62
@@ -8,6 +8,8 @@ import org.apache.ibatis.annotations.Param;
|
|||||||
import org.apache.ibatis.annotations.Select;
|
import org.apache.ibatis.annotations.Select;
|
||||||
import org.apache.ibatis.annotations.Update;
|
import org.apache.ibatis.annotations.Update;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
public interface WatchProgressMapper extends BaseMapper<WatchProgress> {
|
public interface WatchProgressMapper extends BaseMapper<WatchProgress> {
|
||||||
|
|
||||||
/** 按登录用户或匿名设备读取同一部剧的记录,包含逻辑删除行。 */
|
/** 按登录用户或匿名设备读取同一部剧的记录,包含逻辑删除行。 */
|
||||||
@@ -40,4 +42,64 @@ public interface WatchProgressMapper extends BaseMapper<WatchProgress> {
|
|||||||
WHERE id = #{id}
|
WHERE id = #{id}
|
||||||
""")
|
""")
|
||||||
int updateAny(WatchProgress progress);
|
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);
|
||||||
}
|
}
|
||||||
+355
@@ -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;
|
||||||
|
}
|
||||||
|
}
|
||||||
+43
@@ -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());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
+458
-50
@@ -19,29 +19,47 @@ import com.fruitshorts.playback.dal.mapper.WatchHistoryMapper;
|
|||||||
import com.fruitshorts.playback.dal.mapper.WatchProgressMapper;
|
import com.fruitshorts.playback.dal.mapper.WatchProgressMapper;
|
||||||
import com.fruitshorts.playback.vo.ProgressRespVO;
|
import com.fruitshorts.playback.vo.ProgressRespVO;
|
||||||
import com.fruitshorts.playback.vo.ProgressSaveReqVO;
|
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.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.dao.DuplicateKeyException;
|
||||||
import org.springframework.stereotype.Service;
|
import org.springframework.stereotype.Service;
|
||||||
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.transaction.annotation.Transactional;
|
||||||
|
import org.springframework.transaction.support.TransactionTemplate;
|
||||||
import org.springframework.util.StringUtils;
|
import org.springframework.util.StringUtils;
|
||||||
|
|
||||||
|
import java.time.Duration;
|
||||||
|
import java.time.Instant;
|
||||||
import java.time.LocalDateTime;
|
import java.time.LocalDateTime;
|
||||||
|
import java.time.ZoneId;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
|
import java.util.Comparator;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
|
import java.util.HashSet;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Objects;
|
import java.util.Objects;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 观看进度。
|
* 观看进度。
|
||||||
*
|
*
|
||||||
* <p>每次上报都写库在高并发下会成为瓶颈(1 万并发 × 每 10 秒一次 ≈ 1000 写/秒),
|
* <p>生产形态(fruitshorts.watch-progress.buffer=true,prod 已开启):
|
||||||
* 所以生产形态是「先写 Redis,后台任务按登录账号或设备 + drama 合并后批量 UPSERT」。
|
* POST 只写 Redis 缓冲({@link WatchProgressBuffer})立即返回,
|
||||||
* 这里先落地权威表与合并规则,缓存层在接入 Redis 时补齐,接口契约不变。
|
* {@link WatchProgressFlusher} 定时把缓冲按「账号或设备 + drama」合并后
|
||||||
|
* 批量 UPSERT 进权威表,读接口用缓冲覆盖保证读己之写。
|
||||||
|
* 缓冲关闭或 Redis 异常时自动回退同步写库,接口契约不变。
|
||||||
*
|
*
|
||||||
* <p>合并规则:以客户端上报的 positionMs 为准,但只在「更靠后」时覆盖,
|
* <p>合并规则:以客户端上报的 positionMs 为准,但只在「更靠后」时覆盖,
|
||||||
* 避免用户换设备后进度倒退。
|
* 避免用户换设备后进度倒退。
|
||||||
*/
|
*/
|
||||||
|
@Slf4j
|
||||||
@Service
|
@Service
|
||||||
@RequiredArgsConstructor
|
@RequiredArgsConstructor
|
||||||
public class WatchProgressService {
|
public class WatchProgressService {
|
||||||
@@ -53,68 +71,101 @@ public class WatchProgressService {
|
|||||||
private final WatchHistoryMapper historyMapper;
|
private final WatchHistoryMapper historyMapper;
|
||||||
private final DramaService dramaService;
|
private final DramaService dramaService;
|
||||||
private final EpisodeService episodeService;
|
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) {
|
public ProgressRespVO save(Long userId, ProgressSaveReqVO req) {
|
||||||
return save(new InteractionIdentity(userId, ""), req);
|
return save(new InteractionIdentity(userId, ""), req);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Transactional(rollbackFor = Exception.class)
|
|
||||||
public ProgressRespVO save(InteractionIdentity identity, ProgressSaveReqVO req) {
|
public ProgressRespVO save(InteractionIdentity identity, ProgressSaveReqVO req) {
|
||||||
DramaEpisode episode = episodeService.getPublished(req.getEpisodeId());
|
DramaEpisode episode = publishedEpisode(req.getEpisodeId());
|
||||||
Drama drama = dramaService.getPublished(episode.getDramaId());
|
Drama drama = publishedDrama(episode.getDramaId());
|
||||||
long position = req.getPositionMs() == null ? 0L : Math.max(0L, req.getPositionMs());
|
long position = req.getPositionMs() == null ? 0L : Math.max(0L, req.getPositionMs());
|
||||||
long duration = req.getDurationMs() == null ? 0L : Math.max(0L, req.getDurationMs());
|
long duration = req.getDurationMs() == null ? 0L : Math.max(0L, req.getDurationMs());
|
||||||
boolean finished = Boolean.TRUE.equals(req.getFinished())
|
boolean finished = Boolean.TRUE.equals(req.getFinished())
|
||||||
|| (duration > 0 && position + FINISH_THRESHOLD_MS >= duration);
|
|| (duration > 0 && position + FINISH_THRESHOLD_MS >= duration);
|
||||||
|
|
||||||
LoadedProgress loaded = loadForSave(identity, drama.getId(), episode, position, duration,
|
// 高并发形态:写 Redis 缓冲立即返回,落库任务稍后合并进权威表
|
||||||
finished, req);
|
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
|
||||||
WatchProgress existing = loaded.progress();
|
if (buffer != null && buffer.put(identity, drama.getId(), episode.getId(),
|
||||||
if (loaded.created()) {
|
position, duration, finished, req.getSource(), req.getDeviceId())) {
|
||||||
appendHistory(identity, drama.getId(), episode.getId(), position, duration,
|
ProgressRespVO vo = new ProgressRespVO();
|
||||||
existing.getSource(), existing.getDeviceId());
|
vo.setDramaId(drama.getId());
|
||||||
return toResp(progressMapper.selectById(existing.getId()));
|
vo.setEpisodeId(episode.getId());
|
||||||
|
vo.setPositionMs(position);
|
||||||
|
vo.setDurationMs(duration);
|
||||||
|
vo.setFinished(finished);
|
||||||
|
vo.setUpdatedAt(LocalDateTime.now());
|
||||||
|
return vo;
|
||||||
}
|
}
|
||||||
|
return saveToDb(identity, drama, episode, position, duration, finished, req);
|
||||||
|
}
|
||||||
|
|
||||||
boolean hidden = existing.getDeleted() != null && existing.getDeleted() != 0;
|
/** 同步写库路径:缓冲关闭或 Redis 异常时的回退,合并规则与缓冲落库一致。 */
|
||||||
boolean sameEpisode = existing.getEpisodeId() != null
|
private ProgressRespVO saveToDb(InteractionIdentity identity, Drama drama, DramaEpisode episode,
|
||||||
&& existing.getEpisodeId().equals(episode.getId());
|
long position, long duration, boolean finished,
|
||||||
if (hidden && sameEpisode) {
|
ProgressSaveReqVO req) {
|
||||||
existing.setPlayCount((existing.getPlayCount() == null ? 0 : existing.getPlayCount()) + 1);
|
return transactionTemplate.execute(status -> {
|
||||||
appendHistory(identity, drama.getId(), episode.getId(), position, duration,
|
LoadedProgress loaded = loadForSave(identity, drama.getId(), episode, position, duration,
|
||||||
req.getSource(), existing.getDeviceId());
|
finished, req);
|
||||||
}
|
WatchProgress existing = loaded.progress();
|
||||||
if (!sameEpisode) {
|
if (loaded.created()) {
|
||||||
// 换集:直接采纳新集与新的播放位置
|
appendHistory(identity, drama.getId(), episode.getId(), position, duration,
|
||||||
existing.setEpisodeId(episode.getId());
|
existing.getSource(), existing.getDeviceId());
|
||||||
existing.setPositionMs(position);
|
return toResp(progressMapper.selectById(existing.getId()));
|
||||||
existing.setDurationMs(duration);
|
}
|
||||||
existing.setFinished(finished ? 1 : 0);
|
|
||||||
existing.setPlayCount((existing.getPlayCount() == null ? 0 : existing.getPlayCount()) + 1);
|
boolean hidden = existing.getDeleted() != null && existing.getDeleted() != 0;
|
||||||
appendHistory(identity, drama.getId(), episode.getId(), position, duration,
|
boolean sameEpisode = existing.getEpisodeId() != null
|
||||||
req.getSource(), existing.getDeviceId());
|
&& existing.getEpisodeId().equals(episode.getId());
|
||||||
} else {
|
if (hidden && sameEpisode) {
|
||||||
existing.setDurationMs(Math.max(existing.getDurationMs() == null ? 0L : existing.getDurationMs(),
|
existing.setPlayCount((existing.getPlayCount() == null ? 0 : existing.getPlayCount()) + 1);
|
||||||
duration));
|
appendHistory(identity, drama.getId(), episode.getId(), position, duration,
|
||||||
if (position >= (existing.getPositionMs() == null ? 0L : existing.getPositionMs())) {
|
req.getSource(), existing.getDeviceId());
|
||||||
|
}
|
||||||
|
if (!sameEpisode) {
|
||||||
|
// 换集:直接采纳新集与新的播放位置
|
||||||
|
existing.setEpisodeId(episode.getId());
|
||||||
existing.setPositionMs(position);
|
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);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
if (finished) {
|
existing.setDeviceId(identity.authenticated()
|
||||||
existing.setFinished(1);
|
? (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()));
|
||||||
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,
|
private LoadedProgress loadForSave(InteractionIdentity identity, Long dramaId,
|
||||||
@@ -169,6 +220,16 @@ public class WatchProgressService {
|
|||||||
WatchProgress progress = progressMapper.selectOne(identityQuery(identity)
|
WatchProgress progress = progressMapper.selectOne(identityQuery(identity)
|
||||||
.eq(WatchProgress::getDramaId, dramaId)
|
.eq(WatchProgress::getDramaId, dramaId)
|
||||||
.last("limit 1"));
|
.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);
|
return progress == null ? null : toResp(progress);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -186,7 +247,59 @@ public class WatchProgressService {
|
|||||||
identityQuery(identity)
|
identityQuery(identity)
|
||||||
.orderByDesc(WatchProgress::getUpdatedAt)
|
.orderByDesc(WatchProgress::getUpdatedAt)
|
||||||
.orderByDesc(WatchProgress::getId));
|
.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)
|
@Transactional(rollbackFor = Exception.class)
|
||||||
@@ -196,6 +309,12 @@ public class WatchProgressService {
|
|||||||
|
|
||||||
@Transactional(rollbackFor = Exception.class)
|
@Transactional(rollbackFor = Exception.class)
|
||||||
public void remove(InteractionIdentity identity, Long dramaId) {
|
public void remove(InteractionIdentity identity, Long dramaId) {
|
||||||
|
// 缓冲是读接口的权威来源:先删缓冲再删库行,否则删掉的记录会被读覆盖层读回、
|
||||||
|
// 并被下一次落库把行复活。
|
||||||
|
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
|
||||||
|
if (buffer != null && dramaId != null) {
|
||||||
|
buffer.remove(identity, dramaId);
|
||||||
|
}
|
||||||
progressMapper.delete(identityQuery(identity)
|
progressMapper.delete(identityQuery(identity)
|
||||||
.eq(WatchProgress::getDramaId, dramaId));
|
.eq(WatchProgress::getDramaId, dramaId));
|
||||||
}
|
}
|
||||||
@@ -207,9 +326,198 @@ public class WatchProgressService {
|
|||||||
|
|
||||||
@Transactional(rollbackFor = Exception.class)
|
@Transactional(rollbackFor = Exception.class)
|
||||||
public void clear(InteractionIdentity identity) {
|
public void clear(InteractionIdentity identity) {
|
||||||
|
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
|
||||||
|
if (buffer != null) {
|
||||||
|
buffer.clear(identity);
|
||||||
|
}
|
||||||
progressMapper.delete(identityQuery(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) {
|
private ProgressRespVO toResp(WatchProgress progress) {
|
||||||
ProgressRespVO vo = new ProgressRespVO();
|
ProgressRespVO vo = new ProgressRespVO();
|
||||||
vo.setDramaId(progress.getDramaId());
|
vo.setDramaId(progress.getDramaId());
|
||||||
@@ -222,6 +530,68 @@ public class WatchProgressService {
|
|||||||
return vo;
|
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) {
|
private List<ProgressRespVO> enrichDisplay(List<WatchProgress> records) {
|
||||||
if (records.isEmpty()) {
|
if (records.isEmpty()) {
|
||||||
return List.of();
|
return List.of();
|
||||||
@@ -296,6 +666,36 @@ public class WatchProgressService {
|
|||||||
historyMapper.insert(history);
|
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;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 登录时合并设备进度和观看流水。
|
* 登录时合并设备进度和观看流水。
|
||||||
*
|
*
|
||||||
@@ -307,6 +707,9 @@ public class WatchProgressService {
|
|||||||
if (userId == null || !StringUtils.hasText(deviceId)) {
|
if (userId == null || !StringUtils.hasText(deviceId)) {
|
||||||
return new MergeResult(0, 0, 0, 0);
|
return new MergeResult(0, 0, 0, 0);
|
||||||
}
|
}
|
||||||
|
InteractionIdentity deviceIdentity = new InteractionIdentity(null, deviceId);
|
||||||
|
// 缓冲里可能还有没落库的设备进度,先落库再合并,否则这段进度会被漏掉
|
||||||
|
flushIdentity(deviceIdentity);
|
||||||
List<WatchProgress> deviceProgress = progressMapper.selectList(
|
List<WatchProgress> deviceProgress = progressMapper.selectList(
|
||||||
Wrappers.<WatchProgress>lambdaQuery()
|
Wrappers.<WatchProgress>lambdaQuery()
|
||||||
.isNull(WatchProgress::getUserId)
|
.isNull(WatchProgress::getUserId)
|
||||||
@@ -366,6 +769,11 @@ public class WatchProgressService {
|
|||||||
}
|
}
|
||||||
historyMapper.deleteById(device.getId());
|
historyMapper.deleteById(device.getId());
|
||||||
}
|
}
|
||||||
|
// 设备侧进度已迁走或淘汰,缓冲留着只会让读覆盖层把已删行读回来
|
||||||
|
WatchProgressBuffer buffer = bufferProvider.getIfAvailable();
|
||||||
|
if (buffer != null) {
|
||||||
|
buffer.clear(deviceIdentity);
|
||||||
|
}
|
||||||
return new MergeResult(
|
return new MergeResult(
|
||||||
progressMigrated, progressIgnored, historyMigrated, historyIgnored);
|
progressMigrated, progressIgnored, historyMigrated, historyIgnored);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -25,5 +25,10 @@
|
|||||||
<groupId>com.fruitshorts</groupId>
|
<groupId>com.fruitshorts</groupId>
|
||||||
<artifactId>fruitshorts-module-system</artifactId>
|
<artifactId>fruitshorts-module-system</artifactId>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
<!-- 离线 IP → 国家解析,xdb 数据在 src/main/resources/ip2region 下 -->
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.lionsoul</groupId>
|
||||||
|
<artifactId>ip2region</artifactId>
|
||||||
|
</dependency>
|
||||||
</dependencies>
|
</dependencies>
|
||||||
</project>
|
</project>
|
||||||
+32
@@ -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;
|
||||||
|
}
|
||||||
+28
@@ -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);
|
||||||
|
}
|
||||||
+21
@@ -4,6 +4,27 @@ package com.fruitshorts.user.dal.mapper;
|
|||||||
|
|
||||||
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||||
import com.fruitshorts.user.dal.entity.AppUserSession;
|
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> {
|
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 : "";
|
||||||
|
}
|
||||||
|
}
|
||||||
+64
-14
@@ -18,6 +18,7 @@ import com.fruitshorts.user.dal.mapper.AppUserDeviceMapper;
|
|||||||
import com.fruitshorts.user.dal.mapper.AppUserMapper;
|
import com.fruitshorts.user.dal.mapper.AppUserMapper;
|
||||||
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
|
import com.fruitshorts.user.dal.mapper.AppUserSessionMapper;
|
||||||
import com.fruitshorts.user.enums.AppUserStatus;
|
import com.fruitshorts.user.enums.AppUserStatus;
|
||||||
|
import com.fruitshorts.user.enums.CountryProfile;
|
||||||
import com.fruitshorts.user.enums.CredentialType;
|
import com.fruitshorts.user.enums.CredentialType;
|
||||||
import com.fruitshorts.user.enums.OtpScene;
|
import com.fruitshorts.user.enums.OtpScene;
|
||||||
import com.fruitshorts.user.vo.AppUserRespVO;
|
import com.fruitshorts.user.vo.AppUserRespVO;
|
||||||
@@ -66,6 +67,7 @@ public class AppAuthService {
|
|||||||
private final ConfigService configService;
|
private final ConfigService configService;
|
||||||
private final GoogleIdTokenVerifier googleVerifier;
|
private final GoogleIdTokenVerifier googleVerifier;
|
||||||
private final ApplicationEventPublisher eventPublisher;
|
private final ApplicationEventPublisher eventPublisher;
|
||||||
|
private final IpRegionService ipRegionService;
|
||||||
|
|
||||||
/** 后台 sys_config 缺失或非法时使用;小于等于 0 表示不限制。 */
|
/** 后台 sys_config 缺失或非法时使用;小于等于 0 表示不限制。 */
|
||||||
@Value("${fruitshorts.auth.register-max-per-ip-per-day:5}")
|
@Value("${fruitshorts.auth.register-max-per-ip-per-day:5}")
|
||||||
@@ -178,6 +180,7 @@ public class AppAuthService {
|
|||||||
|
|
||||||
user.setLastLoginAt(LocalDateTime.now());
|
user.setLastLoginAt(LocalDateTime.now());
|
||||||
user.setLastLoginIp(AppIdentifier.nullSafe(clientIp));
|
user.setLastLoginIp(AppIdentifier.nullSafe(clientIp));
|
||||||
|
syncCountryByIp(user, clientIp);
|
||||||
appUserMapper.updateById(user);
|
appUserMapper.updateById(user);
|
||||||
touchDevice(user.getId(), req.getDeviceId(), req.getPlatform(), req.getDeviceModel());
|
touchDevice(user.getId(), req.getDeviceId(), req.getPlatform(), req.getDeviceModel());
|
||||||
|
|
||||||
@@ -194,7 +197,7 @@ public class AppAuthService {
|
|||||||
* 再落一条 GOOGLE 凭证用于下次直接登录;已有邮箱账号时只补 GOOGLE 凭证,不重复建号。
|
* 再落一条 GOOGLE 凭证用于下次直接登录;已有邮箱账号时只补 GOOGLE 凭证,不重复建号。
|
||||||
*/
|
*/
|
||||||
@Transactional(rollbackFor = Exception.class)
|
@Transactional(rollbackFor = Exception.class)
|
||||||
public TokenRespVO googleLogin(String idToken, String deviceId) {
|
public TokenRespVO googleLogin(String idToken, String deviceId, String clientIp) {
|
||||||
GoogleUser googleUser = googleVerifier.verify(idToken);
|
GoogleUser googleUser = googleVerifier.verify(idToken);
|
||||||
String email = requireEmail(googleUser);
|
String email = requireEmail(googleUser);
|
||||||
AppUserCredential googleCredential = findCredential(CredentialType.GOOGLE.name(), email);
|
AppUserCredential googleCredential = findCredential(CredentialType.GOOGLE.name(), email);
|
||||||
@@ -203,7 +206,7 @@ public class AppAuthService {
|
|||||||
if (googleCredential == null) {
|
if (googleCredential == null) {
|
||||||
AppUserCredential emailCredential = findCredential(CredentialType.EMAIL.name(), email);
|
AppUserCredential emailCredential = findCredential(CredentialType.EMAIL.name(), email);
|
||||||
if (emailCredential == null) {
|
if (emailCredential == null) {
|
||||||
AppUser user = createGoogleUser(googleUser);
|
AppUser user = createGoogleUser(googleUser, clientIp);
|
||||||
insertCredential(user.getId(), CredentialType.EMAIL.name(), email, "");
|
insertCredential(user.getId(), CredentialType.EMAIL.name(), email, "");
|
||||||
userId = user.getId();
|
userId = user.getId();
|
||||||
newUser = true;
|
newUser = true;
|
||||||
@@ -219,10 +222,11 @@ public class AppAuthService {
|
|||||||
googleCredential.setLastUsedAt(LocalDateTime.now());
|
googleCredential.setLastUsedAt(LocalDateTime.now());
|
||||||
credentialMapper.updateById(googleCredential);
|
credentialMapper.updateById(googleCredential);
|
||||||
user.setLastLoginAt(LocalDateTime.now());
|
user.setLastLoginAt(LocalDateTime.now());
|
||||||
user.setLastLoginIp("");
|
user.setLastLoginIp(AppIdentifier.nullSafe(clientIp));
|
||||||
|
syncCountryByIp(user, clientIp);
|
||||||
appUserMapper.updateById(user);
|
appUserMapper.updateById(user);
|
||||||
touchDevice(user.getId(), deviceId, null, null);
|
touchDevice(user.getId(), deviceId, null, null);
|
||||||
TokenRespVO resp = issueTokens(user, googleCredential, deviceId, null, null, null, null, null);
|
TokenRespVO resp = issueTokens(user, googleCredential, deviceId, null, null, null, null, clientIp);
|
||||||
resp.setNewUser(newUser);
|
resp.setNewUser(newUser);
|
||||||
requestInteractionMerge(user.getId(), deviceId);
|
requestInteractionMerge(user.getId(), deviceId);
|
||||||
return resp;
|
return resp;
|
||||||
@@ -238,7 +242,7 @@ public class AppAuthService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/** 创建 Google 用户:邮箱已验证,昵称与头像直接取 Google 返回。 */
|
/** 创建 Google 用户:邮箱已验证,昵称与头像直接取 Google 返回。 */
|
||||||
private AppUser createGoogleUser(GoogleUser googleUser) {
|
private AppUser createGoogleUser(GoogleUser googleUser, String clientIp) {
|
||||||
AppUser user = new AppUser();
|
AppUser user = new AppUser();
|
||||||
user.setUid(hasher.newUid());
|
user.setUid(hasher.newUid());
|
||||||
String nickname = StringUtils.hasText(googleUser.name())
|
String nickname = StringUtils.hasText(googleUser.name())
|
||||||
@@ -246,11 +250,13 @@ public class AppAuthService {
|
|||||||
user.setNickname(nickname);
|
user.setNickname(nickname);
|
||||||
user.setAvatarUrl(AppIdentifier.nullSafe(googleUser.picture()));
|
user.setAvatarUrl(AppIdentifier.nullSafe(googleUser.picture()));
|
||||||
user.setGender("");
|
user.setGender("");
|
||||||
user.setCountryCode("ID");
|
String country = resolveCountry(clientIp, "");
|
||||||
user.setRegionCode("ID");
|
CountryProfile profile = CountryProfile.of(country);
|
||||||
user.setLanguage("id-ID");
|
user.setCountryCode(country);
|
||||||
user.setTimezone("Asia/Jakarta");
|
user.setRegionCode(country);
|
||||||
user.setCurrency("IDR");
|
user.setLanguage(profile.language());
|
||||||
|
user.setTimezone(profile.timezone());
|
||||||
|
user.setCurrency(profile.currency());
|
||||||
user.setStatus(AppUserStatus.NORMAL);
|
user.setStatus(AppUserStatus.NORMAL);
|
||||||
user.setRegisterSource("APP");
|
user.setRegisterSource("APP");
|
||||||
user.setRegisterChannel("");
|
user.setRegisterChannel("");
|
||||||
@@ -527,12 +533,13 @@ public class AppAuthService {
|
|||||||
: "User" + user.getUid().substring(1, 7));
|
: "User" + user.getUid().substring(1, 7));
|
||||||
user.setAvatarUrl("");
|
user.setAvatarUrl("");
|
||||||
user.setGender("");
|
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.setCountryCode(country);
|
||||||
user.setRegionCode(country);
|
user.setRegionCode(country);
|
||||||
user.setLanguage("id-ID");
|
user.setLanguage(profile.language());
|
||||||
user.setTimezone("Asia/Jakarta");
|
user.setTimezone(profile.timezone());
|
||||||
user.setCurrency("IDR");
|
user.setCurrency(profile.currency());
|
||||||
user.setStatus(AppUserStatus.NORMAL);
|
user.setStatus(AppUserStatus.NORMAL);
|
||||||
user.setRegisterSource("APP");
|
user.setRegisterSource("APP");
|
||||||
user.setRegisterChannel("");
|
user.setRegisterChannel("");
|
||||||
@@ -549,6 +556,49 @@ public class AppAuthService {
|
|||||||
return user;
|
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) {
|
private void recordConsent(Long userId, String scene, String version, String clientIp, String deviceId) {
|
||||||
AppUserConsent consent = new AppUserConsent();
|
AppUserConsent consent = new AppUserConsent();
|
||||||
consent.setUserId(userId);
|
consent.setUserId(userId);
|
||||||
|
|||||||
+29
-25
@@ -2,6 +2,7 @@
|
|||||||
|
|
||||||
package com.fruitshorts.user.service;
|
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.metadata.IPage;
|
||||||
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
||||||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||||
@@ -30,6 +31,7 @@ import java.time.LocalDateTime;
|
|||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Locale;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -52,33 +54,36 @@ public class AppUserAdminService {
|
|||||||
req.getPageIndex() == null || req.getPageIndex() < 1 ? 1 : req.getPageIndex(),
|
req.getPageIndex() == null || req.getPageIndex() < 1 ? 1 : req.getPageIndex(),
|
||||||
req.getPageSize() == null || req.getPageSize() < 1 ? 10 : req.getPageSize());
|
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())) {
|
if (StringUtils.hasText(req.getKeyword())) {
|
||||||
// uid / 昵称 直接查;手机号与邮箱要先在凭证表命中 userId
|
// uid / 昵称 直接查;手机号与邮箱要先在凭证表命中 userId
|
||||||
List<Long> matchedUserIds = userIdsByCredential(req.getKeyword());
|
List<Long> matchedUserIds = userIdsByCredential(req.getKeyword());
|
||||||
result = appUserMapper.selectPage(page, Wrappers.<AppUser>lambdaQuery()
|
wrapper.and(nested -> {
|
||||||
.and(wrapper -> {
|
nested.like(AppUser::getUid, req.getKeyword())
|
||||||
wrapper.like(AppUser::getUid, req.getKeyword())
|
.or().like(AppUser::getNickname, req.getKeyword());
|
||||||
.or().like(AppUser::getNickname, req.getKeyword());
|
if (!matchedUserIds.isEmpty()) {
|
||||||
if (!matchedUserIds.isEmpty()) {
|
nested.or().in(AppUser::getId, matchedUserIds);
|
||||||
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.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());
|
List<AppUserAdminRespVO> list = toRespList(result.getRecords());
|
||||||
fillLatestSystem(list);
|
fillLatestSystem(list);
|
||||||
return new PageResult<>(result.getTotal(), list);
|
return new PageResult<>(result.getTotal(), list);
|
||||||
@@ -164,7 +169,7 @@ public class AppUserAdminService {
|
|||||||
return vo;
|
return vo;
|
||||||
}
|
}
|
||||||
|
|
||||||
/** 用最近一次未吊销会话上的系统信息,供后台用户列表展示。 */
|
/** 用最近一次会话上的系统信息,供后台用户列表展示;排序需与平台筛选一致。 */
|
||||||
private void fillLatestSystem(List<AppUserAdminRespVO> list) {
|
private void fillLatestSystem(List<AppUserAdminRespVO> list) {
|
||||||
if (list.isEmpty()) {
|
if (list.isEmpty()) {
|
||||||
return;
|
return;
|
||||||
@@ -180,7 +185,6 @@ public class AppUserAdminService {
|
|||||||
}
|
}
|
||||||
List<AppUserSession> sessions = sessionMapper.selectList(Wrappers.<AppUserSession>lambdaQuery()
|
List<AppUserSession> sessions = sessionMapper.selectList(Wrappers.<AppUserSession>lambdaQuery()
|
||||||
.in(AppUserSession::getUserId, ids)
|
.in(AppUserSession::getUserId, ids)
|
||||||
.isNull(AppUserSession::getRevokedAt)
|
|
||||||
.orderByDesc(AppUserSession::getLastActiveAt)
|
.orderByDesc(AppUserSession::getLastActiveAt)
|
||||||
.orderByDesc(AppUserSession::getId));
|
.orderByDesc(AppUserSession::getId));
|
||||||
Map<Long, AppUserSession> latest = new HashMap<>();
|
Map<Long, AppUserSession> latest = new HashMap<>();
|
||||||
|
|||||||
+51
@@ -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);
|
||||||
|
}
|
||||||
|
}
|
||||||
+60
@@ -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 status;
|
||||||
private String countryCode;
|
private String countryCode;
|
||||||
private String registerSource;
|
private String registerSource;
|
||||||
|
/** 客户端平台筛选:ANDROID / IOS / WEB,留空不限制;按最近一次会话的平台判定 */
|
||||||
|
private String platform;
|
||||||
private Integer pageIndex = 1;
|
private Integer pageIndex = 1;
|
||||||
private Integer pageSize = 10;
|
private Integer pageSize = 10;
|
||||||
}
|
}
|
||||||
@@ -25,6 +25,7 @@ public class RegisterReqVO {
|
|||||||
private String password;
|
private String password;
|
||||||
|
|
||||||
private String nickname;
|
private String nickname;
|
||||||
|
/** 客户端上报的设备区域(ISO-2),仅在服务端 IP 解析失败时作为兜底 */
|
||||||
private String countryCode;
|
private String countryCode;
|
||||||
private String deviceId;
|
private String deviceId;
|
||||||
private String deviceModel;
|
private String deviceModel;
|
||||||
|
|||||||
@@ -0,0 +1,12 @@
|
|||||||
|
# Author: fruit.com.cn
|
||||||
|
# ip2region 离线库数据版本追踪
|
||||||
|
# 每次替换 ip2region.xdb 时更新此文件
|
||||||
|
|
||||||
|
version: 2.7.0
|
||||||
|
updated: 2026-10-21
|
||||||
|
source: org.lionsoul:ip2region:2.7.0
|
||||||
|
xdb-path: fruitshorts-module-user/src/main/resources/ip2region/ip2region.xdb
|
||||||
|
|
||||||
|
# 覆盖范围:国家 + 省份 + 城市 + 运营商(IPv4)
|
||||||
|
# 生产请留意:xdb 随 jar 打包约 11MB,admin-api 不解析 IP 但也会带上这份数据。
|
||||||
|
# 后续可考虑用 Spring Profile + 条件装配,让 admin-api 不带 ip2region。
|
||||||
Binary file not shown.
@@ -0,0 +1,14 @@
|
|||||||
|
# Author: fruit.com.cn
|
||||||
|
# Grafana 数据源:Loki 日志
|
||||||
|
apiVersion: 1
|
||||||
|
|
||||||
|
datasources:
|
||||||
|
- name: Loki
|
||||||
|
uid: loki
|
||||||
|
type: loki
|
||||||
|
access: proxy
|
||||||
|
url: http://loki:3100
|
||||||
|
isDefault: false
|
||||||
|
editable: false
|
||||||
|
jsonData:
|
||||||
|
maxLines: 1000
|
||||||
@@ -0,0 +1,55 @@
|
|||||||
|
# Author: fruit.com.cn
|
||||||
|
# Loki 单机配置:文件系统存储,7 天保留,供 Grafana 查询容器日志。
|
||||||
|
auth_enabled: false
|
||||||
|
|
||||||
|
server:
|
||||||
|
http_listen_port: 3100
|
||||||
|
grpc_listen_port: 9096
|
||||||
|
|
||||||
|
common:
|
||||||
|
instance_addr: 127.0.0.1
|
||||||
|
path_prefix: /loki
|
||||||
|
storage:
|
||||||
|
filesystem:
|
||||||
|
chunks_directory: /loki/chunks
|
||||||
|
rules_directory: /loki/rules
|
||||||
|
replication_factor: 1
|
||||||
|
ring:
|
||||||
|
kvstore:
|
||||||
|
store: inmemory
|
||||||
|
|
||||||
|
query_range:
|
||||||
|
results_cache:
|
||||||
|
cache:
|
||||||
|
embedded_cache:
|
||||||
|
enabled: true
|
||||||
|
max_size_mb: 100
|
||||||
|
|
||||||
|
schema_config:
|
||||||
|
configs:
|
||||||
|
- from: 2024-01-01
|
||||||
|
store: tsdb
|
||||||
|
object_store: filesystem
|
||||||
|
schema: v13
|
||||||
|
index:
|
||||||
|
prefix: index_
|
||||||
|
period: 24h
|
||||||
|
|
||||||
|
# 7 天保留,避免磁盘无限增长
|
||||||
|
limits_config:
|
||||||
|
retention_period: 168h
|
||||||
|
max_query_series: 1000
|
||||||
|
# 允许采集历史日志(容器启动时的旧日志也能入库),靠 retention_period 控制总量
|
||||||
|
reject_old_samples: false
|
||||||
|
ingestion_rate_mb: 20
|
||||||
|
ingestion_burst_size_mb: 40
|
||||||
|
per_stream_rate_limit: 10MB
|
||||||
|
per_stream_rate_limit_burst: 40MB
|
||||||
|
|
||||||
|
compactor:
|
||||||
|
working_directory: /loki/compactor
|
||||||
|
compaction_interval: 10m
|
||||||
|
retention_enabled: true
|
||||||
|
retention_delete_delay: 2h
|
||||||
|
retention_delete_worker_count: 150
|
||||||
|
delete_request_store: filesystem
|
||||||
@@ -44,6 +44,7 @@
|
|||||||
<mybatis-plus.version>3.5.7</mybatis-plus.version>
|
<mybatis-plus.version>3.5.7</mybatis-plus.version>
|
||||||
<jjwt.version>0.12.5</jjwt.version>
|
<jjwt.version>0.12.5</jjwt.version>
|
||||||
<springdoc.version>2.5.0</springdoc.version>
|
<springdoc.version>2.5.0</springdoc.version>
|
||||||
|
<ip2region.version>2.7.0</ip2region.version>
|
||||||
<!-- Spring Boot 3.2 默认的 1.18.32 在 JDK 23+ 上会编译失败,这里提升到支持新 JDK 的版本 -->
|
<!-- Spring Boot 3.2 默认的 1.18.32 在 JDK 23+ 上会编译失败,这里提升到支持新 JDK 的版本 -->
|
||||||
<lombok.version>1.18.48</lombok.version>
|
<lombok.version>1.18.48</lombok.version>
|
||||||
</properties>
|
</properties>
|
||||||
@@ -130,6 +131,12 @@
|
|||||||
<artifactId>springdoc-openapi-starter-webmvc-ui</artifactId>
|
<artifactId>springdoc-openapi-starter-webmvc-ui</artifactId>
|
||||||
<version>${springdoc.version}</version>
|
<version>${springdoc.version}</version>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.lionsoul</groupId>
|
||||||
|
<artifactId>ip2region</artifactId>
|
||||||
|
<version>${ip2region.version}</version>
|
||||||
|
</dependency>
|
||||||
</dependencies>
|
</dependencies>
|
||||||
</dependencyManagement>
|
</dependencyManagement>
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,34 @@
|
|||||||
|
# Author: fruit.com.cn
|
||||||
|
# Promtail:通过 Docker 服务发现自动采集所有容器的 stdout/stderr 日志,推送到 Loki。
|
||||||
|
server:
|
||||||
|
http_listen_port: 9080
|
||||||
|
grpc_listen_port: 0
|
||||||
|
|
||||||
|
positions:
|
||||||
|
filename: /tmp/positions.yaml
|
||||||
|
|
||||||
|
clients:
|
||||||
|
- url: http://loki:3100/loki/api/v1/push
|
||||||
|
batchwait: 1s
|
||||||
|
batchsize: 1048576
|
||||||
|
|
||||||
|
scrape_configs:
|
||||||
|
- job_name: docker-containers
|
||||||
|
docker_sd_configs:
|
||||||
|
- host: unix:///var/run/docker.sock
|
||||||
|
refresh_interval: 5s
|
||||||
|
relabel_configs:
|
||||||
|
# 容器名(去掉开头的 /)
|
||||||
|
- source_labels: ['__meta_docker_container_name']
|
||||||
|
regex: '/(.*)'
|
||||||
|
target_label: container
|
||||||
|
# 日志流:stdout / stderr
|
||||||
|
- source_labels: ['__meta_docker_container_log_stream']
|
||||||
|
target_label: stream
|
||||||
|
# 容器 ID 前 12 位
|
||||||
|
- source_labels: ['__meta_docker_container_id']
|
||||||
|
target_label: container_id
|
||||||
|
regex: '(.{12}).*'
|
||||||
|
# 镜像名
|
||||||
|
- source_labels: ['__meta_docker_container_image']
|
||||||
|
target_label: image
|
||||||
@@ -0,0 +1,113 @@
|
|||||||
|
-- Author: fruit.com.cn
|
||||||
|
|
||||||
|
-- FruitShorts 短剧服务 · 印尼语(id-ID)敏感词库增量补全(PostgreSQL 18)
|
||||||
|
--
|
||||||
|
-- 续接基线:sensitive_word 已有 id 9000001~9000053(zh-CN/en-US/id-ID 三语初版)。
|
||||||
|
-- 本文件只补 id-ID 缺失/不足的分类,id 从 9000054 续接,避免与现有雪花 ID 冲突。
|
||||||
|
--
|
||||||
|
-- 分类(category):POLITICS 政治 / PORN 色情 / ABUSE 辱骂 / AD 广告导流 / RELIGION 宗教
|
||||||
|
-- 级别(level):
|
||||||
|
-- BLOCK 直接拒绝发布
|
||||||
|
-- REVIEW 命中转人工审核(歧义词、可能误伤的业务词放这里)
|
||||||
|
-- REPLACE 替换后放行(轻度脏话,用 replace_text 打码)
|
||||||
|
--
|
||||||
|
-- 幂等:唯一索引 uk_sensitive_word(language, word),重复执行靠 ON CONFLICT DO NOTHING 跳过。
|
||||||
|
-- 说明:词库内容需印尼运营/合规方复核后启用,本文件仅提供开发联调可用的初始基线。
|
||||||
|
|
||||||
|
-- ============ 辱骂 / 脏话(ABUSE)· 补全 ============
|
||||||
|
INSERT INTO sensitive_word (id, word, category, level, replace_text, language, status) VALUES
|
||||||
|
(9000054, 'babi', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000055, 'brengsek', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000056, 'kampret', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000057, 'keparat', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000058, 'sundal', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000059, 'lonte', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000060, 'pelacur', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000061, 'jancok', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000062, 'asu', 'ABUSE', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000063, 'sialan', 'ABUSE', 'REPLACE', '***', 'id-ID', 1),
|
||||||
|
(9000064, 'tolol', 'ABUSE', 'REPLACE', '***', 'id-ID', 1),
|
||||||
|
(9000065, 'goblok', 'ABUSE', 'REPLACE', '***', 'id-ID', 1),
|
||||||
|
(9000066, 'bodoh', 'ABUSE', 'REPLACE', '***', 'id-ID', 1),
|
||||||
|
(9000067, 'bego', 'ABUSE', 'REPLACE', '***', 'id-ID', 1),
|
||||||
|
(9000068, 'kirik', 'ABUSE', 'REPLACE', '***', 'id-ID', 1),
|
||||||
|
(9000069, 'tai', 'ABUSE', 'REPLACE', '***', 'id-ID', 1),
|
||||||
|
(9000070, 'bejat', 'ABUSE', 'REPLACE', '***', 'id-ID', 1)
|
||||||
|
ON CONFLICT (language, word) DO NOTHING;
|
||||||
|
|
||||||
|
-- ============ 色情(PORN)· 少量补全 ============
|
||||||
|
INSERT INTO sensitive_word (id, word, category, level, replace_text, language, status) VALUES
|
||||||
|
(9000071, 'ngewe', 'PORN', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000072, 'toket', 'PORN', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000073, 'onani', 'PORN', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000074, 'perkosa', 'PORN', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000075, 'pemerkosaan', 'PORN', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000076, 'jembut', 'PORN', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000077, 'pepek', 'PORN', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000078, 'selingkuh', 'PORN', 'REVIEW', '*', 'id-ID', 1),
|
||||||
|
(9000079, 'gairah', 'PORN', 'REVIEW', '*', 'id-ID', 1)
|
||||||
|
ON CONFLICT (language, word) DO NOTHING;
|
||||||
|
|
||||||
|
-- ============ 政治(POLITICS)· 新增(基线缺失) ============
|
||||||
|
INSERT INTO sensitive_word (id, word, category, level, replace_text, language, status) VALUES
|
||||||
|
(9000080, 'PKI', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000081, 'komunis', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000082, 'Papua Merdeka', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000083, 'OPM', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000084, 'GAM', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000085, 'separatis', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000086, 'separatisme', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000087, 'teroris', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000088, 'terorisme', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000089, 'ISIS', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000090, 'jihadis', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000091, 'radikalisme', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000092, 'kudeta', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000093, 'makar', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000094, 'pemberontak', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000095, 'pemberontakan', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000096, 'kerusuhan', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000097, 'penjarahan', 'POLITICS', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000098, 'radikal', 'POLITICS', 'REVIEW', '*', 'id-ID', 1),
|
||||||
|
(9000099, 'jihad', 'POLITICS', 'REVIEW', '*', 'id-ID', 1),
|
||||||
|
(9000100, 'anti-pemerintah', 'POLITICS', 'REVIEW', '*', 'id-ID', 1),
|
||||||
|
(9000101, 'bubarkan', 'POLITICS', 'REVIEW', '*', 'id-ID', 1)
|
||||||
|
ON CONFLICT (language, word) DO NOTHING;
|
||||||
|
|
||||||
|
-- ============ 宗教(RELIGION)· 补全 ============
|
||||||
|
INSERT INTO sensitive_word (id, word, category, level, replace_text, language, status) VALUES
|
||||||
|
(9000102, 'penistaan agama', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000103, 'menghina agama', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000104, 'menghina Nabi', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000105, 'menghina Allah', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000106, 'menghina Kristen','RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000107, 'menghina Katolik','RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000108, 'menghina Hindu', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000109, 'menghina Buddha', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000110, 'kafir', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000111, 'murtad', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000112, 'sara', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000113, 'provokasi agama', 'RELIGION', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000114, 'konflik agama', 'RELIGION', 'REVIEW', '*', 'id-ID', 1)
|
||||||
|
ON CONFLICT (language, word) DO NOTHING;
|
||||||
|
|
||||||
|
-- ============ 广告导流 / 非法(AD)· 补全 ============
|
||||||
|
INSERT INTO sensitive_word (id, word, category, level, replace_text, language, status) VALUES
|
||||||
|
(9000115, 'judi online', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000116, 'slot online', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000117, 'casino', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000118, 'poker online', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000119, 'situs judi', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000120, 'deposit slot', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000121, 'link alternatif', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000122, 'pinjol', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000123, 'rentenir', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000124, 'scam', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000125, 'penipuan', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000126, 'investasi bodong', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000127, 'arisan berantai', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000128, 'money game', 'AD', 'BLOCK', '*', 'id-ID', 1),
|
||||||
|
(9000129, 'pinjaman online', 'AD', 'REVIEW', '*', 'id-ID', 1),
|
||||||
|
(9000130, 'hack', 'AD', 'REVIEW', '*', 'id-ID', 1),
|
||||||
|
(9000131, 'jual akun', 'AD', 'REVIEW', '*', 'id-ID', 1)
|
||||||
|
ON CONFLICT (language, word) DO NOTHING;
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
-- Author: fruit.com.cn
|
||||||
|
-- K109 清理历史遗留的「按整部剧点赞」数据。
|
||||||
|
-- 产品口径:收藏按剧,点赞与评论按集。客户端点赞入口只作用于当前剧集
|
||||||
|
-- (targetType = EPISODE),以下 DRAMA 记录均为 2026-10-21 15:49 之前的旧模型遗留。
|
||||||
|
BEGIN;
|
||||||
|
|
||||||
|
UPDATE content_like
|
||||||
|
SET deleted = 1
|
||||||
|
WHERE target_type = 'DRAMA' AND deleted = 0;
|
||||||
|
|
||||||
|
-- 剧级点赞数由这些行累加而来,随软删一并按实际剩余行数校准,避免详情页展示虚高数字。
|
||||||
|
UPDATE drama d
|
||||||
|
SET like_count = sub.cnt
|
||||||
|
FROM (
|
||||||
|
SELECT d2.id, COUNT(l.id) AS cnt
|
||||||
|
FROM drama d2
|
||||||
|
LEFT JOIN content_like l
|
||||||
|
ON l.target_type = 'DRAMA' AND l.target_id = d2.id AND l.deleted = 0
|
||||||
|
GROUP BY d2.id
|
||||||
|
) sub
|
||||||
|
WHERE d.id = sub.id AND d.like_count IS DISTINCT FROM sub.cnt;
|
||||||
|
|
||||||
|
COMMIT;
|
||||||
@@ -0,0 +1,30 @@
|
|||||||
|
-- Author: fruit.com.cn
|
||||||
|
-- K110 校准互动计数字段。
|
||||||
|
-- 历史压测数据把部分计数直接改成 100000 量级,与真实行数不符;
|
||||||
|
-- 这里按实际有效互动行重算剧集点赞数与剧收藏数,可重复执行。
|
||||||
|
BEGIN;
|
||||||
|
|
||||||
|
-- 剧集点赞数 = 该集未删除的 EPISODE 点赞行数
|
||||||
|
UPDATE drama_episode e
|
||||||
|
SET like_count = sub.cnt
|
||||||
|
FROM (
|
||||||
|
SELECT e2.id, COUNT(l.id) AS cnt
|
||||||
|
FROM drama_episode e2
|
||||||
|
LEFT JOIN content_like l
|
||||||
|
ON l.target_type = 'EPISODE' AND l.target_id = e2.id AND l.deleted = 0
|
||||||
|
GROUP BY e2.id
|
||||||
|
) sub
|
||||||
|
WHERE e.id = sub.id AND e.like_count IS DISTINCT FROM sub.cnt;
|
||||||
|
|
||||||
|
-- 剧收藏数 = 该剧未删除的收藏行数
|
||||||
|
UPDATE drama d
|
||||||
|
SET favorite_count = sub.cnt
|
||||||
|
FROM (
|
||||||
|
SELECT d2.id, COUNT(f.id) AS cnt
|
||||||
|
FROM drama d2
|
||||||
|
LEFT JOIN user_favorite f ON f.drama_id = d2.id AND f.deleted = 0
|
||||||
|
GROUP BY d2.id
|
||||||
|
) sub
|
||||||
|
WHERE d.id = sub.id AND d.favorite_count IS DISTINCT FROM sub.cnt;
|
||||||
|
|
||||||
|
COMMIT;
|
||||||
@@ -0,0 +1,28 @@
|
|||||||
|
-- Author: fruit.com.cn
|
||||||
|
-- K111 设备平台:记录每个 device_id 是 iOS 还是 Android。
|
||||||
|
-- 会话/设备表都依赖已登录账号,游客行为拿不到平台;这里按 device_id 独立建档,
|
||||||
|
-- 只存平台,系统版本与机型不上报。
|
||||||
|
BEGIN;
|
||||||
|
|
||||||
|
DROP TABLE IF EXISTS app_device_profile;
|
||||||
|
|
||||||
|
CREATE TABLE app_device_profile (
|
||||||
|
id BIGINT NOT NULL,
|
||||||
|
device_id VARCHAR(128) NOT NULL,
|
||||||
|
platform VARCHAR(16) NOT NULL DEFAULT '',
|
||||||
|
first_seen_at TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
last_seen_at TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
created_at TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
updated_at TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
deleted SMALLINT NOT NULL DEFAULT 0,
|
||||||
|
PRIMARY KEY (id)
|
||||||
|
);
|
||||||
|
|
||||||
|
-- 不带 deleted 谓词:写入用的 ON CONFLICT (device_id) 需要能推断到唯一索引;
|
||||||
|
-- 设备平台不做软删,重复建档走更新。
|
||||||
|
CREATE UNIQUE INDEX uk_app_device_profile_device
|
||||||
|
ON app_device_profile (device_id);
|
||||||
|
|
||||||
|
COMMENT ON TABLE app_device_profile IS '设备平台:device_id 维度区分 iOS/Android,匿名请求同样记录';
|
||||||
|
|
||||||
|
COMMIT;
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
-- Author: fruit.com.cn
|
||||||
|
-- K112 观看记录默认列表的排序索引。
|
||||||
|
-- 后台按 watched_at DESC, id DESC 倒序翻页,原有索引都以 user_id/drama_id/device_id 开头,
|
||||||
|
-- 不带筛选条件时只能全表扫描再排序;补一条时间索引让翻页直接走索引。
|
||||||
|
-- watch_history 是分区表,在父表建索引会自动下发到各分区。
|
||||||
|
BEGIN;
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_watch_history_watched_at
|
||||||
|
ON watch_history (watched_at DESC, id DESC);
|
||||||
|
|
||||||
|
COMMIT;
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
-- Author: fruit.com.cn
|
||||||
|
-- K113 app_user_session 按「最近一次」平台筛选用户的索引。
|
||||||
|
--
|
||||||
|
-- selectUserIdsByLatestPlatform 用的是 DISTINCT ON (user_id) ... ORDER BY user_id, last_active_at DESC NULLS LAST, id DESC。
|
||||||
|
-- 原表只有 (user_id, revoked_at) 索引,DISTINCT ON + ORDER BY 退化成全表 Sort + Seq Scan,
|
||||||
|
-- 500 行还行,10 万行就扛不住了。补一个匹配 ORDER BY 的部分索引:deleted = 0 覆盖逻辑删除。
|
||||||
|
--
|
||||||
|
-- 注意:CREATE INDEX CONCURRENTLY 不能在事务块里执行,这里直接跑,不加 BEGIN/COMMIT。
|
||||||
|
CREATE INDEX CONCURRENTLY IF NOT EXISTS idx_session_user_latest
|
||||||
|
ON app_user_session (user_id, last_active_at DESC NULLS LAST, id DESC)
|
||||||
|
WHERE deleted = 0;
|
||||||
@@ -0,0 +1,7 @@
|
|||||||
|
-- 缩略图时间轴唯一约束改为部分索引:
|
||||||
|
-- media_thumbnail.deleted 为 @TableLogic 逻辑删除,软删行仍占用 episode_id 会导致
|
||||||
|
-- ThumbnailRecordService.upsert 的 delete+insert 触发 uk_media_thumbnail_episode 冲突。
|
||||||
|
-- 唯一性只约束未删除行(deleted = 0)。
|
||||||
|
DROP INDEX IF EXISTS uk_media_thumbnail_episode;
|
||||||
|
CREATE UNIQUE INDEX IF NOT EXISTS uk_media_thumbnail_episode
|
||||||
|
ON media_thumbnail (episode_id) WHERE deleted = 0;
|
||||||
@@ -806,7 +806,7 @@ CREATE TABLE IF NOT EXISTS media_thumbnail (
|
|||||||
deleted SMALLINT NOT NULL DEFAULT 0
|
deleted SMALLINT NOT NULL DEFAULT 0
|
||||||
);
|
);
|
||||||
CREATE UNIQUE INDEX IF NOT EXISTS uk_media_thumbnail_episode
|
CREATE UNIQUE INDEX IF NOT EXISTS uk_media_thumbnail_episode
|
||||||
ON media_thumbnail (episode_id);
|
ON media_thumbnail (episode_id) WHERE deleted = 0;
|
||||||
CREATE INDEX IF NOT EXISTS idx_media_thumbnail_drama
|
CREATE INDEX IF NOT EXISTS idx_media_thumbnail_drama
|
||||||
ON media_thumbnail (drama_id);
|
ON media_thumbnail (drama_id);
|
||||||
COMMENT ON TABLE media_thumbnail IS '单集缩略图雪碧图与 VTT 时间轴';
|
COMMENT ON TABLE media_thumbnail IS '单集缩略图雪碧图与 VTT 时间轴';
|
||||||
|
|||||||
Reference in new issue
Block a user