修改mybatis版本启动报错,swagger注解问题

This commit is contained in:
wh
2026-04-12 00:15:59 +08:00
parent c3308e069d
commit a489e2b204
25 changed files with 510 additions and 64 deletions

2
.gitignore vendored
View File

@@ -15,6 +15,8 @@ target/
*.iml *.iml
*.ipr *.ipr
*.iws *.iws
.agents/
logs/
# ========================================== # ==========================================
# 3. 项目特定工具目录 (根据你的文件列表) # 3. 项目特定工具目录 (根据你的文件列表)

23
pom.xml
View File

@@ -1,7 +1,6 @@
<?xml version="1.0" encoding="UTF-8"?> <?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" <project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0
https://maven.apache.org/xsd/maven-4.0.0.xsd"> https://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>
@@ -78,15 +77,21 @@
<!-- MyBatis Plus --> <!-- MyBatis Plus -->
<dependency> <dependency>
<groupId>com.baomidou</groupId> <groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId> <artifactId>mybatis-plus-spring-boot3-starter</artifactId>
<version>3.5.9</version> <version>3.5.10</version>
</dependency> </dependency>
<!-- MyBatis Plus JSqlParser (required for TenantLineInnerInterceptor in 3.5.7+) --> <!-- MyBatis Plus JSqlParser (required for TenantLineInnerInterceptor in 3.5.7+) -->
<dependency> <dependency>
<groupId>com.baomidou</groupId> <groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-jsqlparser</artifactId> <artifactId>mybatis-plus-jsqlparser</artifactId>
<version>3.5.9</version> <version>3.5.10</version>
</dependency>
<dependency>
<groupId>org.springdoc</groupId>
<artifactId>springdoc-openapi-starter-webmvc-ui</artifactId>
<version>2.5.0</version>
</dependency> </dependency>
<!-- Apache Shiro --> <!-- Apache Shiro -->
@@ -169,7 +174,9 @@
<execution> <execution>
<id>copy-dependencies</id> <id>copy-dependencies</id>
<phase>package</phase> <phase>package</phase>
<goals><goal>copy-dependencies</goal></goals> <goals>
<goal>copy-dependencies</goal>
</goals>
<configuration> <configuration>
<outputDirectory>${project.build.directory}/libs</outputDirectory> <outputDirectory>${project.build.directory}/libs</outputDirectory>
<includeScope>runtime</includeScope> <includeScope>runtime</includeScope>
@@ -186,7 +193,9 @@
<execution> <execution>
<id>create-distribution</id> <id>create-distribution</id>
<phase>package</phase> <phase>package</phase>
<goals><goal>single</goal></goals> <goals>
<goal>single</goal>
</goals>
<configuration> <configuration>
<descriptors> <descriptors>
<descriptor>src/main/assembly/distribution.xml</descriptor> <descriptor>src/main/assembly/distribution.xml</descriptor>

View File

@@ -1,3 +1,4 @@
package com.label; package com.label;
import org.springframework.boot.SpringApplication; import org.springframework.boot.SpringApplication;
@@ -11,11 +12,13 @@ import org.springframework.boot.autoconfigure.SpringBootApplication;
* Spring Boot 3. 的 jakarta.servlet 命名空间冲突。 认证/ 授权逻辑改由 * Spring Boot 3. 的 jakarta.servlet 命名空间冲突。 认证/ 授权逻辑改由
* TokenFilterOncePerRequestFilter+ ShiroConfig 手动装配。 * TokenFilterOncePerRequestFilter+ ShiroConfig 手动装配。
*/ */
@SpringBootApplication(excludeName = {
"org.apache.shiro.spring.config.web.autoconfigure.ShiroWebAutoConfiguration", // (excludeName = {
"org.apache.shiro.spring.config.web.autoconfigure.ShiroWebFilterConfiguration",
"org.apache.shiro.spring.config.web.autoconfigure.ShiroWebMvcAutoConfiguration" }) // "org.apache.shiro.spring.config.web.autoconfigure.ShiroWebAutoConfiguration",
// "org.apache.shiro.spring.config.web.autoconfigure.ShiroWebFilterConfiguration",
// "org.apache.shiro.spring.config.web.autoconfigure.ShiroWebMvcAutoConfiguration" })
@SpringBootApplication
public class LabelBackendApplication { public class LabelBackendApplication {
public static void main(String[] args) { public static void main(String[] args) {

View File

@@ -23,9 +23,7 @@ public class AiServiceClient {
@PostConstruct @PostConstruct
public void init() { public void init() {
restClient = RestClient.builder() restClient = RestClient.builder().baseUrl(baseUrl).build();
.baseUrl(baseUrl)
.build();
} }
// DTO classes // DTO classes
@@ -85,65 +83,34 @@ public class AiServiceClient {
// The 8 endpoints: // The 8 endpoints:
public ExtractionResponse extractText(ExtractionRequest request) { public ExtractionResponse extractText(ExtractionRequest request) {
return restClient.post() return restClient.post().uri("/extract/text").body(request).retrieve().body(ExtractionResponse.class);
.uri("/extract/text")
.body(request)
.retrieve()
.body(ExtractionResponse.class);
} }
public ExtractionResponse extractImage(ExtractionRequest request) { public ExtractionResponse extractImage(ExtractionRequest request) {
return restClient.post() return restClient.post().uri("/extract/image").body(request).retrieve().body(ExtractionResponse.class);
.uri("/extract/image")
.body(request)
.retrieve()
.body(ExtractionResponse.class);
} }
public void extractFrames(VideoProcessRequest request) { public void extractFrames(VideoProcessRequest request) {
restClient.post() restClient.post().uri("/video/extract-frames").body(request).retrieve().toBodilessEntity();
.uri("/video/extract-frames")
.body(request)
.retrieve()
.toBodilessEntity();
} }
public void videoToText(VideoProcessRequest request) { public void videoToText(VideoProcessRequest request) {
restClient.post() restClient.post().uri("/video/to-text").body(request).retrieve().toBodilessEntity();
.uri("/video/to-text")
.body(request)
.retrieve()
.toBodilessEntity();
} }
public QaGenResponse genTextQa(ExtractionRequest request) { public QaGenResponse genTextQa(ExtractionRequest request) {
return restClient.post() return restClient.post().uri("/qa/gen-text").body(request).retrieve().body(QaGenResponse.class);
.uri("/qa/gen-text")
.body(request)
.retrieve()
.body(QaGenResponse.class);
} }
public QaGenResponse genImageQa(ExtractionRequest request) { public QaGenResponse genImageQa(ExtractionRequest request) {
return restClient.post() return restClient.post().uri("/qa/gen-image").body(request).retrieve().body(QaGenResponse.class);
.uri("/qa/gen-image")
.body(request)
.retrieve()
.body(QaGenResponse.class);
} }
public FinetuneResponse startFinetune(FinetuneRequest request) { public FinetuneResponse startFinetune(FinetuneRequest request) {
return restClient.post() return restClient.post().uri("/finetune/start").body(request).retrieve().body(FinetuneResponse.class);
.uri("/finetune/start")
.body(request)
.retrieve()
.body(FinetuneResponse.class);
} }
public FinetuneStatusResponse getFinetuneStatus(String jobId) { public FinetuneStatusResponse getFinetuneStatus(String jobId) {
return restClient.get() return restClient.get().uri("/finetune/status/{jobId}", jobId).retrieve().body(FinetuneStatusResponse.class);
.uri("/finetune/status/{jobId}", jobId)
.retrieve()
.body(FinetuneStatusResponse.class);
} }
} }

View File

@@ -13,6 +13,7 @@ import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.shiro.SecurityUtils; import org.apache.shiro.SecurityUtils;
import org.apache.shiro.util.ThreadContext; import org.apache.shiro.util.ThreadContext;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.MediaType; import org.springframework.http.MediaType;
import org.springframework.web.filter.OncePerRequestFilter; import org.springframework.web.filter.OncePerRequestFilter;
@@ -38,6 +39,24 @@ public class TokenFilter extends OncePerRequestFilter {
private final RedisService redisService; private final RedisService redisService;
private final ObjectMapper objectMapper; private final ObjectMapper objectMapper;
@Value("${shiro.auth.enabled:true}")
private boolean authEnabled;
@Value("${shiro.auth.mock-company-id:1}")
private Long mockCompanyId;
@Value("${shiro.auth.mock-user-id:1}")
private Long mockUserId;
@Value("${shiro.auth.mock-role:ADMIN}")
private String mockRole;
@Value("${shiro.auth.mock-username:mock}")
private String mockUsername;
@Value("${token.ttl-seconds:7200}")
private long tokenTtlSeconds;
/** /**
* 公开端点跳过过滤:非 /api/ 前缀路径,以及登录接口本身。 * 公开端点跳过过滤:非 /api/ 前缀路径,以及登录接口本身。
*/ */
@@ -46,13 +65,25 @@ public class TokenFilter extends OncePerRequestFilter {
String path = request.getServletPath(); String path = request.getServletPath();
return !path.startsWith("/api/") return !path.startsWith("/api/")
|| path.equals("/api/auth/login") || path.equals("/api/auth/login")
|| path.equals("/api/video/callback"); // AI 服务内部回调,不走用户 Token 认证 || path.equals("/api/video/callback")
|| path.startsWith("/swagger-ui")
|| path.startsWith("/v3/api-docs"); // AI 服务内部回调,不走用户 Token 认证
} }
@Override @Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response,
FilterChain filterChain) throws ServletException, IOException { FilterChain filterChain) throws ServletException, IOException {
try { try {
if (!authEnabled) {
TokenPrincipal principal = new TokenPrincipal(
mockUserId, mockRole, mockCompanyId, mockUsername, "mock-token");
CompanyContext.set(mockCompanyId);
SecurityUtils.getSubject().login(new BearerToken("mock-token", principal));
request.setAttribute("__token_principal__", principal);
filterChain.doFilter(request, response);
return;
}
String authHeader = request.getHeader("Authorization"); String authHeader = request.getHeader("Authorization");
if (authHeader == null || !authHeader.startsWith("Bearer ")) { if (authHeader == null || !authHeader.startsWith("Bearer ")) {
writeUnauthorized(response, "缺少或无效的认证令牌"); writeUnauthorized(response, "缺少或无效的认证令牌");
@@ -79,6 +110,8 @@ public class TokenFilter extends OncePerRequestFilter {
TokenPrincipal principal = new TokenPrincipal(userId, role, companyId, username, token); TokenPrincipal principal = new TokenPrincipal(userId, role, companyId, username, token);
SecurityUtils.getSubject().login(new BearerToken(token, principal)); SecurityUtils.getSubject().login(new BearerToken(token, principal));
request.setAttribute("__token_principal__", principal); request.setAttribute("__token_principal__", principal);
redisService.expire(RedisKeyManager.tokenKey(token), tokenTtlSeconds);
redisService.expire(RedisKeyManager.userSessionsKey(userId), tokenTtlSeconds);
filterChain.doFilter(request, response); filterChain.doFilter(request, response);
} catch (Exception e) { } catch (Exception e) {

View File

@@ -3,6 +3,8 @@ package com.label.module.annotation.controller;
import com.label.common.result.Result; import com.label.common.result.Result;
import com.label.common.shiro.TokenPrincipal; import com.label.common.shiro.TokenPrincipal;
import com.label.module.annotation.service.ExtractionService; import com.label.module.annotation.service.ExtractionService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresRoles; import org.apache.shiro.authz.annotation.RequiresRoles;
@@ -13,6 +15,7 @@ import java.util.Map;
/** /**
* 提取阶段标注工作台接口5 个端点)。 * 提取阶段标注工作台接口5 个端点)。
*/ */
@Tag(name = "提取标注", description = "提取阶段的查看、编辑、提交和审批")
@RestController @RestController
@RequestMapping("/api/extraction") @RequestMapping("/api/extraction")
@RequiredArgsConstructor @RequiredArgsConstructor
@@ -21,6 +24,7 @@ public class ExtractionController {
private final ExtractionService extractionService; private final ExtractionService extractionService;
/** GET /api/extraction/{taskId} — 获取当前标注结果 */ /** GET /api/extraction/{taskId} — 获取当前标注结果 */
@Operation(summary = "获取提取标注结果")
@GetMapping("/{taskId}") @GetMapping("/{taskId}")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Map<String, Object>> getResult(@PathVariable Long taskId, public Result<Map<String, Object>> getResult(@PathVariable Long taskId,
@@ -29,6 +33,7 @@ public class ExtractionController {
} }
/** PUT /api/extraction/{taskId} — 更新标注结果(整体覆盖) */ /** PUT /api/extraction/{taskId} — 更新标注结果(整体覆盖) */
@Operation(summary = "更新提取标注结果")
@PutMapping("/{taskId}") @PutMapping("/{taskId}")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Void> updateResult(@PathVariable Long taskId, public Result<Void> updateResult(@PathVariable Long taskId,
@@ -39,6 +44,7 @@ public class ExtractionController {
} }
/** POST /api/extraction/{taskId}/submit — 提交标注结果 */ /** POST /api/extraction/{taskId}/submit — 提交标注结果 */
@Operation(summary = "提交提取标注结果")
@PostMapping("/{taskId}/submit") @PostMapping("/{taskId}/submit")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Void> submit(@PathVariable Long taskId, public Result<Void> submit(@PathVariable Long taskId,
@@ -48,6 +54,7 @@ public class ExtractionController {
} }
/** POST /api/extraction/{taskId}/approve — 审批通过REVIEWER */ /** POST /api/extraction/{taskId}/approve — 审批通过REVIEWER */
@Operation(summary = "审批通过提取结果")
@PostMapping("/{taskId}/approve") @PostMapping("/{taskId}/approve")
@RequiresRoles("REVIEWER") @RequiresRoles("REVIEWER")
public Result<Void> approve(@PathVariable Long taskId, public Result<Void> approve(@PathVariable Long taskId,
@@ -57,6 +64,7 @@ public class ExtractionController {
} }
/** POST /api/extraction/{taskId}/reject — 驳回REVIEWER */ /** POST /api/extraction/{taskId}/reject — 驳回REVIEWER */
@Operation(summary = "驳回提取结果")
@PostMapping("/{taskId}/reject") @PostMapping("/{taskId}/reject")
@RequiresRoles("REVIEWER") @RequiresRoles("REVIEWER")
public Result<Void> reject(@PathVariable Long taskId, public Result<Void> reject(@PathVariable Long taskId,

View File

@@ -3,6 +3,8 @@ package com.label.module.annotation.controller;
import com.label.common.result.Result; import com.label.common.result.Result;
import com.label.common.shiro.TokenPrincipal; import com.label.common.shiro.TokenPrincipal;
import com.label.module.annotation.service.QaService; import com.label.module.annotation.service.QaService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresRoles; import org.apache.shiro.authz.annotation.RequiresRoles;
@@ -13,6 +15,7 @@ import java.util.Map;
/** /**
* 问答生成阶段标注工作台接口5 个端点)。 * 问答生成阶段标注工作台接口5 个端点)。
*/ */
@Tag(name = "问答生成", description = "问答生成阶段的查看、编辑、提交和审批")
@RestController @RestController
@RequestMapping("/api/qa") @RequestMapping("/api/qa")
@RequiredArgsConstructor @RequiredArgsConstructor
@@ -21,6 +24,7 @@ public class QaController {
private final QaService qaService; private final QaService qaService;
/** GET /api/qa/{taskId} — 获取候选问答对 */ /** GET /api/qa/{taskId} — 获取候选问答对 */
@Operation(summary = "获取候选问答对")
@GetMapping("/{taskId}") @GetMapping("/{taskId}")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Map<String, Object>> getResult(@PathVariable Long taskId, public Result<Map<String, Object>> getResult(@PathVariable Long taskId,
@@ -29,6 +33,7 @@ public class QaController {
} }
/** PUT /api/qa/{taskId} — 整体覆盖问答对 */ /** PUT /api/qa/{taskId} — 整体覆盖问答对 */
@Operation(summary = "更新候选问答对")
@PutMapping("/{taskId}") @PutMapping("/{taskId}")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Void> updateResult(@PathVariable Long taskId, public Result<Void> updateResult(@PathVariable Long taskId,
@@ -39,6 +44,7 @@ public class QaController {
} }
/** POST /api/qa/{taskId}/submit — 提交问答对 */ /** POST /api/qa/{taskId}/submit — 提交问答对 */
@Operation(summary = "提交问答对")
@PostMapping("/{taskId}/submit") @PostMapping("/{taskId}/submit")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Void> submit(@PathVariable Long taskId, public Result<Void> submit(@PathVariable Long taskId,
@@ -48,6 +54,7 @@ public class QaController {
} }
/** POST /api/qa/{taskId}/approve — 审批通过REVIEWER */ /** POST /api/qa/{taskId}/approve — 审批通过REVIEWER */
@Operation(summary = "审批通过问答对")
@PostMapping("/{taskId}/approve") @PostMapping("/{taskId}/approve")
@RequiresRoles("REVIEWER") @RequiresRoles("REVIEWER")
public Result<Void> approve(@PathVariable Long taskId, public Result<Void> approve(@PathVariable Long taskId,
@@ -57,6 +64,7 @@ public class QaController {
} }
/** POST /api/qa/{taskId}/reject — 驳回REVIEWER */ /** POST /api/qa/{taskId}/reject — 驳回REVIEWER */
@Operation(summary = "驳回答案对")
@PostMapping("/{taskId}/reject") @PostMapping("/{taskId}/reject")
@RequiresRoles("REVIEWER") @RequiresRoles("REVIEWER")
public Result<Void> reject(@PathVariable Long taskId, public Result<Void> reject(@PathVariable Long taskId,

View File

@@ -4,6 +4,8 @@ import com.label.common.result.Result;
import com.label.common.shiro.TokenPrincipal; import com.label.common.shiro.TokenPrincipal;
import com.label.module.config.entity.SysConfig; import com.label.module.config.entity.SysConfig;
import com.label.module.config.service.SysConfigService; import com.label.module.config.service.SysConfigService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresRoles; import org.apache.shiro.authz.annotation.RequiresRoles;
@@ -18,6 +20,7 @@ import java.util.Map;
* GET /api/config — 查询当前公司所有可见配置(公司专属 + 全局默认合并) * GET /api/config — 查询当前公司所有可见配置(公司专属 + 全局默认合并)
* PUT /api/config/{key} — 更新/创建公司专属配置UPSERT * PUT /api/config/{key} — 更新/创建公司专属配置UPSERT
*/ */
@Tag(name = "系统配置", description = "全局和公司级系统配置管理")
@RestController @RestController
@RequiredArgsConstructor @RequiredArgsConstructor
public class SysConfigController { public class SysConfigController {
@@ -31,6 +34,7 @@ public class SysConfigController {
* - "COMPANY":当前公司专属配置(优先生效) * - "COMPANY":当前公司专属配置(优先生效)
* - "GLOBAL":全局默认配置(公司未覆盖时生效) * - "GLOBAL":全局默认配置(公司未覆盖时生效)
*/ */
@Operation(summary = "查询合并后的系统配置")
@GetMapping("/api/config") @GetMapping("/api/config")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<List<Map<String, Object>>> listConfig(HttpServletRequest request) { public Result<List<Map<String, Object>>> listConfig(HttpServletRequest request) {
@@ -43,6 +47,7 @@ public class SysConfigController {
* *
* Body: { "value": "...", "description": "..." } * Body: { "value": "...", "description": "..." }
*/ */
@Operation(summary = "更新或创建公司专属配置")
@PutMapping("/api/config/{key}") @PutMapping("/api/config/{key}")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<SysConfig> updateConfig(@PathVariable String key, public Result<SysConfig> updateConfig(@PathVariable String key,

View File

@@ -7,6 +7,8 @@ import com.label.module.annotation.entity.TrainingDataset;
import com.label.module.export.entity.ExportBatch; import com.label.module.export.entity.ExportBatch;
import com.label.module.export.service.ExportService; import com.label.module.export.service.ExportService;
import com.label.module.export.service.FinetuneService; import com.label.module.export.service.FinetuneService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresRoles; import org.apache.shiro.authz.annotation.RequiresRoles;
@@ -19,6 +21,7 @@ import java.util.Map;
/** /**
* 训练数据导出与微调接口5 个端点,全部 ADMIN 权限)。 * 训练数据导出与微调接口5 个端点,全部 ADMIN 权限)。
*/ */
@Tag(name = "导出管理", description = "训练样本查询、导出批次和微调任务")
@RestController @RestController
@RequiredArgsConstructor @RequiredArgsConstructor
public class ExportController { public class ExportController {
@@ -27,6 +30,7 @@ public class ExportController {
private final FinetuneService finetuneService; private final FinetuneService finetuneService;
/** GET /api/training/samples — 分页查询已审批可导出样本 */ /** GET /api/training/samples — 分页查询已审批可导出样本 */
@Operation(summary = "分页查询可导出训练样本")
@GetMapping("/api/training/samples") @GetMapping("/api/training/samples")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<PageResult<TrainingDataset>> listSamples( public Result<PageResult<TrainingDataset>> listSamples(
@@ -39,6 +43,7 @@ public class ExportController {
} }
/** POST /api/export/batch — 创建导出批次 */ /** POST /api/export/batch — 创建导出批次 */
@Operation(summary = "创建导出批次")
@PostMapping("/api/export/batch") @PostMapping("/api/export/batch")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
@ResponseStatus(HttpStatus.CREATED) @ResponseStatus(HttpStatus.CREATED)
@@ -53,6 +58,7 @@ public class ExportController {
} }
/** POST /api/export/{batchId}/finetune — 提交微调任务 */ /** POST /api/export/{batchId}/finetune — 提交微调任务 */
@Operation(summary = "提交微调任务")
@PostMapping("/api/export/{batchId}/finetune") @PostMapping("/api/export/{batchId}/finetune")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<Map<String, Object>> triggerFinetune(@PathVariable Long batchId, public Result<Map<String, Object>> triggerFinetune(@PathVariable Long batchId,
@@ -61,6 +67,7 @@ public class ExportController {
} }
/** GET /api/export/{batchId}/status — 查询微调状态 */ /** GET /api/export/{batchId}/status — 查询微调状态 */
@Operation(summary = "查询微调状态")
@GetMapping("/api/export/{batchId}/status") @GetMapping("/api/export/{batchId}/status")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<Map<String, Object>> getFinetuneStatus(@PathVariable Long batchId, public Result<Map<String, Object>> getFinetuneStatus(@PathVariable Long batchId,
@@ -69,6 +76,7 @@ public class ExportController {
} }
/** GET /api/export/list — 分页查询导出批次列表 */ /** GET /api/export/list — 分页查询导出批次列表 */
@Operation(summary = "分页查询导出批次")
@GetMapping("/api/export/list") @GetMapping("/api/export/list")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<PageResult<ExportBatch>> listBatches( public Result<PageResult<ExportBatch>> listBatches(

View File

@@ -5,6 +5,8 @@ import com.label.common.result.Result;
import com.label.common.shiro.TokenPrincipal; import com.label.common.shiro.TokenPrincipal;
import com.label.module.source.dto.SourceResponse; import com.label.module.source.dto.SourceResponse;
import com.label.module.source.service.SourceService; import com.label.module.source.service.SourceService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresRoles; import org.apache.shiro.authz.annotation.RequiresRoles;
@@ -19,6 +21,7 @@ import org.springframework.web.multipart.MultipartFile;
* - 上传 / 列表 / 详情UPLOADER 及以上角色(含 ANNOTATOR、REVIEWER、ADMIN * - 上传 / 列表 / 详情UPLOADER 及以上角色(含 ANNOTATOR、REVIEWER、ADMIN
* - 删除:仅 ADMIN * - 删除:仅 ADMIN
*/ */
@Tag(name = "资料管理", description = "原始资料上传、查询和删除")
@RestController @RestController
@RequestMapping("/api/source") @RequestMapping("/api/source")
@RequiredArgsConstructor @RequiredArgsConstructor
@@ -30,6 +33,7 @@ public class SourceController {
* 上传文件multipart/form-data * 上传文件multipart/form-data
* 返回 201 Created + 资料摘要。 * 返回 201 Created + 资料摘要。
*/ */
@Operation(summary = "上传原始资料")
@PostMapping("/upload") @PostMapping("/upload")
@RequiresRoles("UPLOADER") @RequiresRoles("UPLOADER")
@ResponseStatus(HttpStatus.CREATED) @ResponseStatus(HttpStatus.CREATED)
@@ -45,6 +49,7 @@ public class SourceController {
* 分页查询资料列表。 * 分页查询资料列表。
* UPLOADER 只见自己的资料ADMIN 见全公司资料。 * UPLOADER 只见自己的资料ADMIN 见全公司资料。
*/ */
@Operation(summary = "分页查询资料列表")
@GetMapping("/list") @GetMapping("/list")
@RequiresRoles("UPLOADER") @RequiresRoles("UPLOADER")
public Result<PageResult<SourceResponse>> list( public Result<PageResult<SourceResponse>> list(
@@ -60,6 +65,7 @@ public class SourceController {
/** /**
* 查询资料详情(含 15 分钟预签名下载链接)。 * 查询资料详情(含 15 分钟预签名下载链接)。
*/ */
@Operation(summary = "查询资料详情")
@GetMapping("/{id}") @GetMapping("/{id}")
@RequiresRoles("UPLOADER") @RequiresRoles("UPLOADER")
public Result<SourceResponse> findById(@PathVariable Long id) { public Result<SourceResponse> findById(@PathVariable Long id) {
@@ -70,6 +76,7 @@ public class SourceController {
* 删除资料(仅 PENDING 状态可删)。 * 删除资料(仅 PENDING 状态可删)。
* 同步删除 RustFS 文件及 DB 记录。 * 同步删除 RustFS 文件及 DB 记录。
*/ */
@Operation(summary = "删除资料")
@DeleteMapping("/{id}") @DeleteMapping("/{id}")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<Void> delete(@PathVariable Long id, HttpServletRequest request) { public Result<Void> delete(@PathVariable Long id, HttpServletRequest request) {

View File

@@ -1,5 +1,6 @@
package com.label.module.source.dto; package com.label.module.source.dto;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Builder; import lombok.Builder;
import lombok.Data; import lombok.Data;
@@ -11,17 +12,27 @@ import java.time.LocalDateTime;
*/ */
@Data @Data
@Builder @Builder
@Schema(description = "原始资料响应")
public class SourceResponse { public class SourceResponse {
@Schema(description = "资料主键")
private Long id; private Long id;
@Schema(description = "文件名")
private String fileName; private String fileName;
@Schema(description = "资料类型", example = "TEXT")
private String dataType; private String dataType;
@Schema(description = "文件大小(字节)")
private Long fileSize; private Long fileSize;
@Schema(description = "资料状态", example = "PENDING")
private String status; private String status;
/** 上传用户 ID列表端点返回 */ /** 上传用户 ID列表端点返回 */
@Schema(description = "上传用户 ID")
private Long uploaderId; private Long uploaderId;
/** 15 分钟预签名下载链接(详情端点返回) */ /** 15 分钟预签名下载链接(详情端点返回) */
@Schema(description = "预签名下载链接")
private String presignedUrl; private String presignedUrl;
/** 父资料 ID视频帧 / 文本片段;详情端点返回) */ /** 父资料 ID视频帧 / 文本片段;详情端点返回) */
@Schema(description = "父资料 ID")
private Long parentSourceId; private Long parentSourceId;
@Schema(description = "创建时间")
private LocalDateTime createdAt; private LocalDateTime createdAt;
} }

View File

@@ -6,6 +6,8 @@ import com.label.common.shiro.TokenPrincipal;
import com.label.module.task.dto.TaskResponse; import com.label.module.task.dto.TaskResponse;
import com.label.module.task.service.TaskClaimService; import com.label.module.task.service.TaskClaimService;
import com.label.module.task.service.TaskService; import com.label.module.task.service.TaskService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresRoles; import org.apache.shiro.authz.annotation.RequiresRoles;
@@ -16,6 +18,7 @@ import java.util.Map;
/** /**
* 任务管理接口10 个端点)。 * 任务管理接口10 个端点)。
*/ */
@Tag(name = "任务管理", description = "任务池、我的任务、审批队列和管理操作")
@RestController @RestController
@RequestMapping("/api/tasks") @RequestMapping("/api/tasks")
@RequiredArgsConstructor @RequiredArgsConstructor
@@ -25,6 +28,7 @@ public class TaskController {
private final TaskClaimService taskClaimService; private final TaskClaimService taskClaimService;
/** GET /api/tasks/pool — 查询可领取任务池(角色感知) */ /** GET /api/tasks/pool — 查询可领取任务池(角色感知) */
@Operation(summary = "查询可领取任务池")
@GetMapping("/pool") @GetMapping("/pool")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<PageResult<TaskResponse>> getPool( public Result<PageResult<TaskResponse>> getPool(
@@ -35,6 +39,7 @@ public class TaskController {
} }
/** GET /api/tasks/mine — 查询我的任务 */ /** GET /api/tasks/mine — 查询我的任务 */
@Operation(summary = "查询我的任务")
@GetMapping("/mine") @GetMapping("/mine")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<PageResult<TaskResponse>> getMine( public Result<PageResult<TaskResponse>> getMine(
@@ -46,6 +51,7 @@ public class TaskController {
} }
/** GET /api/tasks/pending-review — 待审批队列REVIEWER 专属) */ /** GET /api/tasks/pending-review — 待审批队列REVIEWER 专属) */
@Operation(summary = "查询待审批任务")
@GetMapping("/pending-review") @GetMapping("/pending-review")
@RequiresRoles("REVIEWER") @RequiresRoles("REVIEWER")
public Result<PageResult<TaskResponse>> getPendingReview( public Result<PageResult<TaskResponse>> getPendingReview(
@@ -56,6 +62,7 @@ public class TaskController {
} }
/** GET /api/tasks — 查询全部任务ADMIN */ /** GET /api/tasks — 查询全部任务ADMIN */
@Operation(summary = "管理员查询全部任务")
@GetMapping @GetMapping
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<PageResult<TaskResponse>> getAll( public Result<PageResult<TaskResponse>> getAll(
@@ -67,6 +74,7 @@ public class TaskController {
} }
/** POST /api/tasks — 创建任务ADMIN */ /** POST /api/tasks — 创建任务ADMIN */
@Operation(summary = "管理员创建任务")
@PostMapping @PostMapping
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<TaskResponse> createTask(@RequestBody Map<String, Object> body, public Result<TaskResponse> createTask(@RequestBody Map<String, Object> body,
@@ -79,6 +87,7 @@ public class TaskController {
} }
/** GET /api/tasks/{id} — 查询任务详情 */ /** GET /api/tasks/{id} — 查询任务详情 */
@Operation(summary = "查询任务详情")
@GetMapping("/{id}") @GetMapping("/{id}")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<TaskResponse> getById(@PathVariable Long id) { public Result<TaskResponse> getById(@PathVariable Long id) {
@@ -86,6 +95,7 @@ public class TaskController {
} }
/** POST /api/tasks/{id}/claim — 领取任务 */ /** POST /api/tasks/{id}/claim — 领取任务 */
@Operation(summary = "领取任务")
@PostMapping("/{id}/claim") @PostMapping("/{id}/claim")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Void> claim(@PathVariable Long id, HttpServletRequest request) { public Result<Void> claim(@PathVariable Long id, HttpServletRequest request) {
@@ -94,6 +104,7 @@ public class TaskController {
} }
/** POST /api/tasks/{id}/unclaim — 放弃任务 */ /** POST /api/tasks/{id}/unclaim — 放弃任务 */
@Operation(summary = "放弃任务")
@PostMapping("/{id}/unclaim") @PostMapping("/{id}/unclaim")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Void> unclaim(@PathVariable Long id, HttpServletRequest request) { public Result<Void> unclaim(@PathVariable Long id, HttpServletRequest request) {
@@ -102,6 +113,7 @@ public class TaskController {
} }
/** POST /api/tasks/{id}/reclaim — 重领被驳回的任务 */ /** POST /api/tasks/{id}/reclaim — 重领被驳回的任务 */
@Operation(summary = "重领被驳回的任务")
@PostMapping("/{id}/reclaim") @PostMapping("/{id}/reclaim")
@RequiresRoles("ANNOTATOR") @RequiresRoles("ANNOTATOR")
public Result<Void> reclaim(@PathVariable Long id, HttpServletRequest request) { public Result<Void> reclaim(@PathVariable Long id, HttpServletRequest request) {
@@ -110,6 +122,7 @@ public class TaskController {
} }
/** PUT /api/tasks/{id}/reassign — ADMIN 强制指派 */ /** PUT /api/tasks/{id}/reassign — ADMIN 强制指派 */
@Operation(summary = "管理员强制指派任务")
@PutMapping("/{id}/reassign") @PutMapping("/{id}/reassign")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<Void> reassign(@PathVariable Long id, public Result<Void> reassign(@PathVariable Long id,

View File

@@ -1,5 +1,6 @@
package com.label.module.task.dto; package com.label.module.task.dto;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Builder; import lombok.Builder;
import lombok.Data; import lombok.Data;
@@ -10,17 +11,28 @@ import java.time.LocalDateTime;
*/ */
@Data @Data
@Builder @Builder
@Schema(description = "标注任务响应")
public class TaskResponse { public class TaskResponse {
@Schema(description = "任务主键")
private Long id; private Long id;
@Schema(description = "关联资料 ID")
private Long sourceId; private Long sourceId;
/** 任务类型(对应 taskType 字段EXTRACTION / QA_GENERATION */ /** 任务类型(对应 taskType 字段EXTRACTION / QA_GENERATION */
@Schema(description = "任务类型", example = "EXTRACTION")
private String taskType; private String taskType;
@Schema(description = "任务状态", example = "UNCLAIMED")
private String status; private String status;
@Schema(description = "领取人用户 ID")
private Long claimedBy; private Long claimedBy;
@Schema(description = "领取时间")
private LocalDateTime claimedAt; private LocalDateTime claimedAt;
@Schema(description = "提交时间")
private LocalDateTime submittedAt; private LocalDateTime submittedAt;
@Schema(description = "完成时间")
private LocalDateTime completedAt; private LocalDateTime completedAt;
/** 驳回原因REJECTED 状态时非空) */ /** 驳回原因REJECTED 状态时非空) */
@Schema(description = "驳回原因")
private String rejectReason; private String rejectReason;
@Schema(description = "创建时间")
private LocalDateTime createdAt; private LocalDateTime createdAt;
} }

View File

@@ -6,6 +6,8 @@ import com.label.module.user.dto.LoginRequest;
import com.label.module.user.dto.LoginResponse; import com.label.module.user.dto.LoginResponse;
import com.label.module.user.dto.UserInfoResponse; import com.label.module.user.dto.UserInfoResponse;
import com.label.module.user.service.AuthService; import com.label.module.user.service.AuthService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
@@ -18,6 +20,7 @@ import org.springframework.web.bind.annotation.*;
* - POST /api/auth/logout → 需要有效 TokenTokenFilter 校验) * - POST /api/auth/logout → 需要有效 TokenTokenFilter 校验)
* - GET /api/auth/me → 需要有效 TokenTokenFilter 校验) * - GET /api/auth/me → 需要有效 TokenTokenFilter 校验)
*/ */
@Tag(name = "认证管理", description = "登录、退出和当前用户信息")
@RestController @RestController
@RequestMapping("/api/auth") @RequestMapping("/api/auth")
@RequiredArgsConstructor @RequiredArgsConstructor
@@ -28,6 +31,7 @@ public class AuthController {
/** /**
* 登录接口(匿名,无需 Token * 登录接口(匿名,无需 Token
*/ */
@Operation(summary = "用户登录,返回 Bearer Token")
@PostMapping("/login") @PostMapping("/login")
public Result<LoginResponse> login(@RequestBody LoginRequest request) { public Result<LoginResponse> login(@RequestBody LoginRequest request) {
return Result.success(authService.login(request)); return Result.success(authService.login(request));
@@ -36,6 +40,7 @@ public class AuthController {
/** /**
* 退出登录,立即删除 Redis Token。 * 退出登录,立即删除 Redis Token。
*/ */
@Operation(summary = "退出登录并立即失效当前 Token")
@PostMapping("/logout") @PostMapping("/logout")
public Result<Void> logout(HttpServletRequest request) { public Result<Void> logout(HttpServletRequest request) {
String token = extractToken(request); String token = extractToken(request);
@@ -47,6 +52,7 @@ public class AuthController {
* 获取当前登录用户信息。 * 获取当前登录用户信息。
* TokenPrincipal 由 TokenFilter 写入请求属性 "__token_principal__"。 * TokenPrincipal 由 TokenFilter 写入请求属性 "__token_principal__"。
*/ */
@Operation(summary = "获取当前登录用户信息")
@GetMapping("/me") @GetMapping("/me")
public Result<UserInfoResponse> me(HttpServletRequest request) { public Result<UserInfoResponse> me(HttpServletRequest request) {
TokenPrincipal principal = (TokenPrincipal) request.getAttribute("__token_principal__"); TokenPrincipal principal = (TokenPrincipal) request.getAttribute("__token_principal__");

View File

@@ -5,6 +5,8 @@ import com.label.common.result.Result;
import com.label.common.shiro.TokenPrincipal; import com.label.common.shiro.TokenPrincipal;
import com.label.module.user.entity.SysUser; import com.label.module.user.entity.SysUser;
import com.label.module.user.service.UserService; import com.label.module.user.service.UserService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresRoles; import org.apache.shiro.authz.annotation.RequiresRoles;
@@ -15,6 +17,7 @@ import java.util.Map;
/** /**
* 用户管理接口5 个端点,全部 ADMIN 权限)。 * 用户管理接口5 个端点,全部 ADMIN 权限)。
*/ */
@Tag(name = "用户管理", description = "管理员维护公司用户")
@RestController @RestController
@RequestMapping("/api/users") @RequestMapping("/api/users")
@RequiredArgsConstructor @RequiredArgsConstructor
@@ -23,6 +26,7 @@ public class UserController {
private final UserService userService; private final UserService userService;
/** GET /api/users — 分页查询用户列表 */ /** GET /api/users — 分页查询用户列表 */
@Operation(summary = "分页查询用户列表")
@GetMapping @GetMapping
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<PageResult<SysUser>> listUsers( public Result<PageResult<SysUser>> listUsers(
@@ -33,6 +37,7 @@ public class UserController {
} }
/** POST /api/users — 创建用户 */ /** POST /api/users — 创建用户 */
@Operation(summary = "创建用户")
@PostMapping @PostMapping
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<SysUser> createUser(@RequestBody Map<String, String> body, public Result<SysUser> createUser(@RequestBody Map<String, String> body,
@@ -46,6 +51,7 @@ public class UserController {
} }
/** PUT /api/users/{id} — 更新用户基本信息 */ /** PUT /api/users/{id} — 更新用户基本信息 */
@Operation(summary = "更新用户基本信息")
@PutMapping("/{id}") @PutMapping("/{id}")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<SysUser> updateUser(@PathVariable Long id, public Result<SysUser> updateUser(@PathVariable Long id,
@@ -59,6 +65,7 @@ public class UserController {
} }
/** PUT /api/users/{id}/status — 变更用户状态 */ /** PUT /api/users/{id}/status — 变更用户状态 */
@Operation(summary = "变更用户状态")
@PutMapping("/{id}/status") @PutMapping("/{id}/status")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<Void> updateStatus(@PathVariable Long id, public Result<Void> updateStatus(@PathVariable Long id,
@@ -69,6 +76,7 @@ public class UserController {
} }
/** PUT /api/users/{id}/role — 变更用户角色 */ /** PUT /api/users/{id}/role — 变更用户角色 */
@Operation(summary = "变更用户角色")
@PutMapping("/{id}/role") @PutMapping("/{id}/role")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<Void> updateRole(@PathVariable Long id, public Result<Void> updateRole(@PathVariable Long id,

View File

@@ -1,16 +1,21 @@
package com.label.module.user.dto; package com.label.module.user.dto;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data; import lombok.Data;
/** /**
* 登录请求体。 * 登录请求体。
*/ */
@Data @Data
@Schema(description = "登录请求")
public class LoginRequest { public class LoginRequest {
/** 公司代码(英文简写),用于确定租户 */ /** 公司代码(英文简写),用于确定租户 */
@Schema(description = "公司代码(英文简写)", example = "DEMO")
private String companyCode; private String companyCode;
/** 登录用户名 */ /** 登录用户名 */
@Schema(description = "登录用户名", example = "admin")
private String username; private String username;
/** 明文密码(传输层应使用 HTTPS 保护) */ /** 明文密码(传输层应使用 HTTPS 保护) */
@Schema(description = "明文密码", example = "admin123")
private String password; private String password;
} }

View File

@@ -1,5 +1,6 @@
package com.label.module.user.dto; package com.label.module.user.dto;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor; import lombok.AllArgsConstructor;
import lombok.Data; import lombok.Data;
@@ -8,15 +9,21 @@ import lombok.Data;
*/ */
@Data @Data
@AllArgsConstructor @AllArgsConstructor
@Schema(description = "登录响应")
public class LoginResponse { public class LoginResponse {
/** Bearer TokenUUID v4后续请求放入 Authorization 头 */ /** Bearer TokenUUID v4后续请求放入 Authorization 头 */
@Schema(description = "Bearer Token", example = "550e8400-e29b-41d4-a716-446655440000")
private String token; private String token;
/** 用户主键 */ /** 用户主键 */
@Schema(description = "用户主键")
private Long userId; private Long userId;
/** 登录用户名 */ /** 登录用户名 */
@Schema(description = "登录用户名")
private String username; private String username;
/** 角色UPLOADER / ANNOTATOR / REVIEWER / ADMIN */ /** 角色UPLOADER / ANNOTATOR / REVIEWER / ADMIN */
@Schema(description = "角色", example = "ADMIN")
private String role; private String role;
/** Token 有效期(秒) */ /** Token 有效期(秒) */
@Schema(description = "Token 有效期(秒)", example = "7200")
private Long expiresIn; private Long expiresIn;
} }

View File

@@ -1,5 +1,6 @@
package com.label.module.user.dto; package com.label.module.user.dto;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor; import lombok.AllArgsConstructor;
import lombok.Data; import lombok.Data;
@@ -8,11 +9,18 @@ import lombok.Data;
*/ */
@Data @Data
@AllArgsConstructor @AllArgsConstructor
@Schema(description = "当前登录用户信息")
public class UserInfoResponse { public class UserInfoResponse {
@Schema(description = "用户主键")
private Long id; private Long id;
@Schema(description = "用户名")
private String username; private String username;
@Schema(description = "真实姓名")
private String realName; private String realName;
@Schema(description = "角色", example = "ADMIN")
private String role; private String role;
@Schema(description = "所属公司 ID")
private Long companyId; private Long companyId;
@Schema(description = "所属公司名称")
private String companyName; private String companyName;
} }

View File

@@ -4,6 +4,8 @@ import com.label.common.result.Result;
import com.label.common.shiro.TokenPrincipal; import com.label.common.shiro.TokenPrincipal;
import com.label.module.video.entity.VideoProcessJob; import com.label.module.video.entity.VideoProcessJob;
import com.label.module.video.service.VideoProcessService; import com.label.module.video.service.VideoProcessService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
@@ -21,6 +23,7 @@ import java.util.Map;
* POST /api/video/jobs/{jobId}/reset — 重置失败任务ADMIN * POST /api/video/jobs/{jobId}/reset — 重置失败任务ADMIN
* POST /api/video/callback — AI 回调接口(无需认证,已在 TokenFilter 中排除) * POST /api/video/callback — AI 回调接口(无需认证,已在 TokenFilter 中排除)
*/ */
@Tag(name = "视频处理", description = "视频处理任务创建、查询、重置和回调")
@Slf4j @Slf4j
@RestController @RestController
@RequiredArgsConstructor @RequiredArgsConstructor
@@ -32,6 +35,7 @@ public class VideoController {
private String callbackSecret; private String callbackSecret;
/** POST /api/video/process — 触发视频处理任务 */ /** POST /api/video/process — 触发视频处理任务 */
@Operation(summary = "触发视频处理任务")
@PostMapping("/api/video/process") @PostMapping("/api/video/process")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<VideoProcessJob> createJob(@RequestBody Map<String, Object> body, public Result<VideoProcessJob> createJob(@RequestBody Map<String, Object> body,
@@ -51,6 +55,7 @@ public class VideoController {
} }
/** GET /api/video/jobs/{jobId} — 查询视频处理任务 */ /** GET /api/video/jobs/{jobId} — 查询视频处理任务 */
@Operation(summary = "查询视频处理任务状态")
@GetMapping("/api/video/jobs/{jobId}") @GetMapping("/api/video/jobs/{jobId}")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<VideoProcessJob> getJob(@PathVariable Long jobId, public Result<VideoProcessJob> getJob(@PathVariable Long jobId,
@@ -59,6 +64,7 @@ public class VideoController {
} }
/** POST /api/video/jobs/{jobId}/reset — 管理员重置失败任务 */ /** POST /api/video/jobs/{jobId}/reset — 管理员重置失败任务 */
@Operation(summary = "重置失败的视频处理任务")
@PostMapping("/api/video/jobs/{jobId}/reset") @PostMapping("/api/video/jobs/{jobId}/reset")
@RequiresRoles("ADMIN") @RequiresRoles("ADMIN")
public Result<VideoProcessJob> resetJob(@PathVariable Long jobId, public Result<VideoProcessJob> resetJob(@PathVariable Long jobId,
@@ -76,6 +82,7 @@ public class VideoController {
* { "jobId": 123, "status": "SUCCESS", "outputPath": "processed/123/frames.zip" } * { "jobId": 123, "status": "SUCCESS", "outputPath": "processed/123/frames.zip" }
* { "jobId": 123, "status": "FAILED", "errorMessage": "ffmpeg error: ..." } * { "jobId": 123, "status": "FAILED", "errorMessage": "ffmpeg error: ..." }
*/ */
@Operation(summary = "接收 AI 服务视频处理回调")
@PostMapping("/api/video/callback") @PostMapping("/api/video/callback")
public Result<Void> handleCallback(@RequestBody Map<String, Object> body, public Result<Void> handleCallback(@RequestBody Map<String, Object> body,
HttpServletRequest request) { HttpServletRequest request) {

View File

@@ -2,6 +2,8 @@ server:
port: 8080 port: 8080
spring: spring:
application:
name: label-backend
datasource: datasource:
url: ${SPRING_DATASOURCE_URL:jdbc:postgresql://localhost:5432/label_db} url: ${SPRING_DATASOURCE_URL:jdbc:postgresql://localhost:5432/label_db}
username: ${SPRING_DATASOURCE_USERNAME:label} username: ${SPRING_DATASOURCE_USERNAME:label}
@@ -33,6 +35,14 @@ spring:
pathmatch: pathmatch:
matching-strategy: ant_path_matcher # Shiro 与 Spring Boot 3 兼容性需要 matching-strategy: ant_path_matcher # Shiro 与 Spring Boot 3 兼容性需要
springdoc:
api-docs:
enabled: true
path: /v3/api-docs
swagger-ui:
enabled: true
path: /swagger-ui.html
mybatis-plus: mybatis-plus:
mapper-locations: classpath*:mapper/**/*.xml mapper-locations: classpath*:mapper/**/*.xml
type-aliases-package: com.label.module type-aliases-package: com.label.module
@@ -53,6 +63,14 @@ ai-service:
base-url: ${AI_SERVICE_BASE_URL:http://localhost:8000} base-url: ${AI_SERVICE_BASE_URL:http://localhost:8000}
timeout: 30000 # milliseconds timeout: 30000 # milliseconds
shiro:
auth:
enabled: true
mock-company-id: 1
mock-user-id: 1
mock-role: ADMIN
mock-username: mock
token: token:
ttl-seconds: 7200 # Token 默认有效期(秒),与 sys_config token_ttl_seconds 保持一致 ttl-seconds: 7200 # Token 默认有效期(秒),与 sys_config token_ttl_seconds 保持一致
@@ -61,6 +79,6 @@ video:
logging: logging:
level: level:
com.label: DEBUG com.label: INFO
org.apache.shiro: INFO org.apache.shiro: INFO
com.baomidou.mybatisplus: INFO com.baomidou.mybatisplus: INFO

View File

@@ -1,7 +1,12 @@
package com.label; package com.label;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest; import org.springframework.boot.test.context.SpringBootTest;
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE) @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)
class LabelBackendApplicationTests { class LabelBackendApplicationTests {
@Test
void contextLoads() {
}
} }

View File

@@ -0,0 +1,55 @@
package com.label.unit;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.boot.env.YamlPropertySourceLoader;
import org.springframework.core.env.PropertySource;
import org.springframework.core.io.ClassPathResource;
import java.nio.charset.StandardCharsets;
import static org.assertj.core.api.Assertions.assertThat;
@DisplayName("应用配置单元测试")
class ApplicationConfigTest {
@Test
@DisplayName("application.yml 提供 Swagger 和 shiro.auth 测试开关配置")
void applicationYaml_containsSwaggerAndShiroAuthToggle() throws Exception {
PropertySource<?> source = new YamlPropertySourceLoader()
.load("application", new ClassPathResource("application.yml"))
.get(0);
assertThat(source.getProperty("springdoc.api-docs.enabled")).isEqualTo(true);
assertThat(source.getProperty("springdoc.api-docs.path")).isEqualTo("/v3/api-docs");
assertThat(source.getProperty("springdoc.swagger-ui.enabled")).isEqualTo(true);
assertThat(source.getProperty("springdoc.swagger-ui.path")).isEqualTo("/swagger-ui.html");
assertThat(source.getProperty("shiro.auth.enabled")).isEqualTo(true);
assertThat(source.getProperty("shiro.auth.mock-company-id")).isEqualTo(1);
assertThat(source.getProperty("shiro.auth.mock-user-id")).isEqualTo(1);
assertThat(source.getProperty("shiro.auth.mock-role")).isEqualTo("ADMIN");
assertThat(source.getProperty("logging.level.com.label")).isEqualTo("INFO");
}
@Test
@DisplayName("application.yml 默认值不指向公网服务或携带真实默认密码")
void applicationYaml_doesNotShipPublicInfrastructureDefaults() throws Exception {
String yaml = new ClassPathResource("application.yml")
.getContentAsString(StandardCharsets.UTF_8);
assertThat(yaml).doesNotContain("39.107.112.174");
assertThat(yaml).doesNotContain("postgres!Pw");
assertThat(yaml).doesNotContain("jsti@2024");
}
@Test
@DisplayName("logback.xml 启用 60 MB 滚动文件日志")
void logback_enablesRollingFileAppender() throws Exception {
String xml = new ClassPathResource("logback.xml")
.getContentAsString(StandardCharsets.UTF_8);
assertThat(xml).contains("<maxFileSize>60MB</maxFileSize>");
assertThat(xml).contains("<appender-ref ref=\"FILE\"/>");
assertThat(xml).doesNotContain("<!-- <appender-ref ref=\"FILE\"/> -->");
}
}

View File

@@ -0,0 +1,98 @@
package com.label.unit;
import com.label.module.annotation.controller.ExtractionController;
import com.label.module.annotation.controller.QaController;
import com.label.module.config.controller.SysConfigController;
import com.label.module.export.controller.ExportController;
import com.label.module.source.controller.SourceController;
import com.label.module.source.dto.SourceResponse;
import com.label.module.task.controller.TaskController;
import com.label.module.task.dto.TaskResponse;
import com.label.module.user.controller.AuthController;
import com.label.module.user.controller.UserController;
import com.label.module.user.dto.LoginRequest;
import com.label.module.user.dto.LoginResponse;
import com.label.module.user.dto.UserInfoResponse;
import com.label.module.video.controller.VideoController;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.Arrays;
import java.util.List;
import static org.assertj.core.api.Assertions.assertThat;
@DisplayName("OpenAPI 注解覆盖测试")
class OpenApiAnnotationTest {
private static final List<Class<?>> CONTROLLERS = List.of(
AuthController.class,
UserController.class,
SourceController.class,
TaskController.class,
ExtractionController.class,
QaController.class,
ExportController.class,
SysConfigController.class,
VideoController.class
);
private static final List<Class<?>> DTOS = List.of(
LoginRequest.class,
LoginResponse.class,
UserInfoResponse.class,
TaskResponse.class,
SourceResponse.class
);
@Test
@DisplayName("所有 REST Controller 都声明 @Tag")
void allControllersHaveTag() {
assertThat(CONTROLLERS)
.allSatisfy(controller ->
assertThat(controller.getAnnotation(Tag.class))
.as(controller.getSimpleName() + " should have @Tag")
.isNotNull());
}
@Test
@DisplayName("所有 REST endpoint 方法都声明 @Operation")
void allEndpointMethodsHaveOperation() {
for (Class<?> controller : CONTROLLERS) {
Arrays.stream(controller.getDeclaredMethods())
.filter(method -> !Modifier.isPrivate(method.getModifiers()))
.filter(OpenApiAnnotationTest::isEndpointMethod)
.forEach(method -> assertThat(method.getAnnotation(Operation.class))
.as(controller.getSimpleName() + "." + method.getName() + " should have @Operation")
.isNotNull());
}
}
@Test
@DisplayName("核心 DTO 都声明 @Schema")
void coreDtosHaveSchema() {
assertThat(DTOS)
.allSatisfy(dto ->
assertThat(dto.getAnnotation(Schema.class))
.as(dto.getSimpleName() + " should have @Schema")
.isNotNull());
}
private static boolean isEndpointMethod(Method method) {
return method.isAnnotationPresent(GetMapping.class)
|| method.isAnnotationPresent(PostMapping.class)
|| method.isAnnotationPresent(PutMapping.class)
|| method.isAnnotationPresent(DeleteMapping.class)
|| method.isAnnotationPresent(RequestMapping.class);
}
}

View File

@@ -0,0 +1,143 @@
package com.label.unit;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.label.common.context.CompanyContext;
import com.label.common.redis.RedisKeyManager;
import com.label.common.redis.RedisService;
import com.label.common.shiro.BearerToken;
import com.label.common.shiro.TokenFilter;
import com.label.common.shiro.TokenPrincipal;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.AuthenticationInfo;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authc.SimpleAuthenticationInfo;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.mgt.DefaultSecurityManager;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.util.ThreadContext;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.test.util.ReflectionTestUtils;
import jakarta.servlet.FilterChain;
import jakarta.servlet.ServletRequest;
import jakarta.servlet.ServletResponse;
import java.util.Map;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.*;
@DisplayName("TokenFilter 单元测试")
class TokenFilterTest {
private RedisService redisService;
private TestableTokenFilter filter;
@BeforeEach
void setUp() {
redisService = mock(RedisService.class);
filter = new TestableTokenFilter(redisService, new ObjectMapper());
SecurityUtils.setSecurityManager(new DefaultSecurityManager(new BearerTokenRealm()));
}
@AfterEach
void tearDown() {
CompanyContext.clear();
ThreadContext.remove();
}
@Test
@DisplayName("有效 Token 请求会刷新 token TTL实现滑动过期")
void validToken_refreshesTokenTtl() throws Exception {
ReflectionTestUtils.setField(filter, "authEnabled", true);
ReflectionTestUtils.setField(filter, "tokenTtlSeconds", 7200L);
String token = "valid-token";
when(redisService.hGetAll(RedisKeyManager.tokenKey(token))).thenReturn(Map.of(
"userId", "10",
"role", "ADMIN",
"companyId", "20",
"username", "admin"
));
MockHttpServletRequest request = new MockHttpServletRequest("GET", "/api/tasks");
request.addHeader("Authorization", "Bearer " + token);
MockHttpServletResponse response = new MockHttpServletResponse();
RecordingChain chain = new RecordingChain();
filter.invoke(request, response, chain);
assertThat(response.getStatus()).isEqualTo(200);
assertThat(chain.principal).isInstanceOf(TokenPrincipal.class);
verify(redisService).expire(RedisKeyManager.tokenKey(token), 7200L);
}
@Test
@DisplayName("shiro.auth.enabled=false 时注入 mock Principal 并跳过 Redis 校验")
void authDisabled_injectsMockPrincipalWithoutRedisLookup() throws Exception {
ReflectionTestUtils.setField(filter, "authEnabled", false);
ReflectionTestUtils.setField(filter, "mockCompanyId", 3L);
ReflectionTestUtils.setField(filter, "mockUserId", 4L);
ReflectionTestUtils.setField(filter, "mockRole", "ADMIN");
ReflectionTestUtils.setField(filter, "mockUsername", "mock-admin");
MockHttpServletRequest request = new MockHttpServletRequest("GET", "/api/tasks");
MockHttpServletResponse response = new MockHttpServletResponse();
RecordingChain chain = new RecordingChain();
filter.invoke(request, response, chain);
assertThat(response.getStatus()).isEqualTo(200);
TokenPrincipal principal = chain.principal;
assertThat(principal.getCompanyId()).isEqualTo(3L);
assertThat(principal.getUserId()).isEqualTo(4L);
assertThat(principal.getRole()).isEqualTo("ADMIN");
assertThat(principal.getUsername()).isEqualTo("mock-admin");
verify(redisService, never()).hGetAll(anyString());
}
private static final class BearerTokenRealm extends AuthorizingRealm {
@Override
public boolean supports(AuthenticationToken token) {
return token instanceof BearerToken;
}
@Override
protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) {
return new SimpleAuthenticationInfo(token.getPrincipal(), token.getCredentials(), getName());
}
@Override
protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {
TokenPrincipal principal = (TokenPrincipal) principals.getPrimaryPrincipal();
SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
info.addRole(principal.getRole());
return info;
}
}
private static final class RecordingChain implements FilterChain {
private TokenPrincipal principal;
@Override
public void doFilter(ServletRequest request, ServletResponse response) {
principal = (TokenPrincipal) request.getAttribute("__token_principal__");
}
}
private static final class TestableTokenFilter extends TokenFilter {
private TestableTokenFilter(RedisService redisService, ObjectMapper objectMapper) {
super(redisService, objectMapper);
}
private void invoke(MockHttpServletRequest request, MockHttpServletResponse response, FilterChain chain)
throws Exception {
super.doFilterInternal(request, response, chain);
}
}
}