105 lines
3.9 KiB
Java
105 lines
3.9 KiB
Java
package com.yupi.springbootinit.controller;
|
||
|
||
import com.yupi.springbootinit.Interceptor.FeatureTicketInterceptor;
|
||
import com.yupi.springbootinit.annotation.RequireFeatureTicket;
|
||
import com.yupi.springbootinit.common.BaseResponse;
|
||
import com.yupi.springbootinit.common.ResultUtils;
|
||
import com.yupi.springbootinit.model.dto.ticket.FeatureTicketPayload;
|
||
import io.swagger.annotations.Api;
|
||
import io.swagger.annotations.ApiOperation;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
import org.springframework.web.bind.annotation.PostMapping;
|
||
import org.springframework.web.bind.annotation.RequestBody;
|
||
import org.springframework.web.bind.annotation.RequestMapping;
|
||
import org.springframework.web.bind.annotation.RestController;
|
||
|
||
import javax.servlet.http.HttpServletRequest;
|
||
import java.util.HashMap;
|
||
import java.util.Map;
|
||
|
||
/**
|
||
* 任务执行控制器（示例）
|
||
* 展示如何使用 @RequireFeatureTicket 注解保护接口
|
||
*
|
||
* @author ziin
|
||
*/
|
||
@RestController
|
||
@RequestMapping("/v1")
|
||
@Api(tags = "任务执行（示例）")
|
||
public class TaskExecuteController {
|
||
|
||
private static final Logger log = LoggerFactory.getLogger(TaskExecuteController.class);
|
||
|
||
/**
|
||
* 执行任务接口（受 Feature Ticket 保护）
|
||
*
|
||
* 调用此接口需要：
|
||
* 1. 请求头 X-Feature-Ticket: 有效的 Ticket
|
||
* 2. 请求头 X-Machine-Id: 与 Ticket 中一致的设备 ID
|
||
*/
|
||
@PostMapping("/execute-task")
|
||
@RequireFeatureTicket(featureCode = "CRAWL")
|
||
@ApiOperation(value = "执行任务", notes = "需要 Feature Ticket 校验，功能代码: CRAWL")
|
||
public BaseResponse<Map<String, Object>> executeTask(
|
||
@RequestBody Map<String, Object> taskParams,
|
||
HttpServletRequest request) {
|
||
|
||
// 从请求属性中获取已校验的 Ticket 载荷
|
||
FeatureTicketPayload payload = (FeatureTicketPayload) request.getAttribute(
|
||
FeatureTicketInterceptor.TICKET_PAYLOAD_ATTR);
|
||
|
||
log.info("执行任务，用户: {}, 租户: {}, 设备: {}",
|
||
payload.getUserId(), payload.getTenantId(), payload.getMachineId());
|
||
|
||
// 业务逻辑处理
|
||
Map<String, Object> result = new HashMap<>();
|
||
result.put("success", true);
|
||
result.put("taskId", System.currentTimeMillis());
|
||
result.put("tenantId", payload.getTenantId());
|
||
result.put("userId", payload.getUserId());
|
||
result.put("message", "任务已提交执行");
|
||
|
||
return ResultUtils.success(result);
|
||
}
|
||
|
||
/**
|
||
* 另一个受保护的接口示例（不限定功能代码）
|
||
*
|
||
* 只校验 Ticket 有效性和设备 ID，不校验功能代码
|
||
*/
|
||
@PostMapping("/query-status")
|
||
@RequireFeatureTicket
|
||
@ApiOperation(value = "查询状态", notes = "需要 Feature Ticket 校验，不限定功能代码")
|
||
public BaseResponse<Map<String, Object>> queryStatus(HttpServletRequest request) {
|
||
|
||
FeatureTicketPayload payload = (FeatureTicketPayload) request.getAttribute(
|
||
FeatureTicketInterceptor.TICKET_PAYLOAD_ATTR);
|
||
|
||
Map<String, Object> result = new HashMap<>();
|
||
result.put("status", "running");
|
||
result.put("featureCode", payload.getFeatureCode());
|
||
result.put("tenantId", payload.getTenantId());
|
||
|
||
return ResultUtils.success(result);
|
||
}
|
||
|
||
/**
|
||
* 自定义请求头名称的示例
|
||
*/
|
||
@PostMapping("/custom-header-task")
|
||
@RequireFeatureTicket(
|
||
featureCode = "AI_CHAT",
|
||
ticketHeader = "Authorization-Ticket",
|
||
machineIdHeader = "Device-Id"
|
||
)
|
||
@ApiOperation(value = "自定义请求头任务", notes = "使用自定义请求头名称")
|
||
public BaseResponse<String> customHeaderTask(HttpServletRequest request) {
|
||
|
||
FeatureTicketPayload payload = (FeatureTicketPayload) request.getAttribute(
|
||
FeatureTicketInterceptor.TICKET_PAYLOAD_ATTR);
|
||
|
||
return ResultUtils.success("任务执行成功，用户: " + payload.getUserId());
|
||
}
|
||
}
|