AX-HUB/ax-hub-mcp_tool_new
2
0
Fork 0
forked from kimhyungsik/ax_hub_mcp_tool
Code Pull Requests Activity

fix: Extract traceId validation from headers via payload meta

Browse Source
This commit is contained in:
jade
2026-07-04 02:01:31 +09:00
parent f5c3145a27
commit f9287a8c5f
2 changed files with 27 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/main/java/io/shinhanlife/axhub/biz/mcp/gateway/controller/McpRouterController.java
View File

@@ -113,8 +113,21 @@ public class McpRouterController {
// 1. [신규] 표준 MCP 파이프라인 호출 (가장 중요!)
@Operation(summary = "MCP 파이프라인 호출", description = "MCP 표준 파이프라인(ExecuteService)을 통해 레거시 툴을 호출합니다.")
@PostMapping("/tools/call")
public ResponseEntity<Object> callTool(@RequestBody Map<String, Object> payload, @RequestAttribute(value = "tenantId", required = false) String tenantId) {
log.info("[MCP 표준] ExecuteService 파이프라인을 통한 툴 호출 시작 (Tenant: {})", tenantId);
public ResponseEntity<Object> callTool(
@RequestHeader(value = "X-Trace-Id", required = false) String traceId,
@RequestHeader(value = "X-Agent-Id", required = false) String agentId,
@RequestHeader(value = "X-User-Prompt", required = false) String userPrompt,
@RequestBody Map<String, Object> payload,
@RequestAttribute(value = "tenantId", required = false) String tenantId) {
// 메타데이터 주입
java.util.Map<String, Object> meta = new java.util.HashMap<>();
meta.put("traceId", traceId != null ? traceId : java.util.UUID.randomUUID().toString());
meta.put("agentId", agentId != null ? agentId : "UNKNOWN");
meta.put("userPrompt", userPrompt != null ? userPrompt : "");
payload.put("meta", meta);
log.info("[MCP 표준] ExecuteService 파이프라인을 통한 툴 호출 시작 (Tenant: {}, Trace: {})", tenantId, meta.get("traceId"));
if (payload != null) {
try {
log.info(" [Gateway] AI Agent 요청 파라미터: {}", objectMapper.writeValueAsString(payload));

19
src/main/java/io/shinhanlife/axhub/biz/mcp/gateway/router/RequestValidator.java
View File

@@ -40,23 +40,28 @@ public class RequestValidator {
throw new IllegalArgumentException("params.arguments 객체가 누락되었습니다.");
}
// 2. 필수 파라미터 누락 및 빈 값 검증 (arguments 내에서 검증)
// 2. 필수 파라미터 누락 및 빈 값 검증 (meta 내에서 검증)
Map<String, Object> meta = (Map<String, Object>) payload.get("meta");
if (meta == null) {
throw new IllegalArgumentException("메타데이터 객체(meta)가 누락되었습니다.");
}
for (String field : REQUIRED_FIELDS) {
if (!arguments.containsKey(field)) {
log.error(" [Validator] 검증 실패: 필수 키 누락 [{}]", field);
if (!meta.containsKey(field)) {
log.error(" [Validator] 검증 실패: 필수 메타데이터 누락 [{}]", field);
throw new IllegalArgumentException("필수 파라미터가 누락되었습니다: " + field);
}
Object value = arguments.get(field);
Object value = meta.get(field);
if (value == null || value.toString().trim().isEmpty()) {
log.error(" [Validator] 검증 실패: 필수 키의 값이 비어 있음 [{}]", field);
log.error(" [Validator] 검증 실패: 필수 메타데이터의 값이 비어 있음 [{}]", field);
throw new IllegalArgumentException("필수 파라미터의 값이 비어있을 수 없습니다: " + field);
}
}
// 3. 비즈니스 로직에 따른 추가 데이터 길이 또는 타입 검증 (예: 프롬프트 길이)
String traceId = arguments.get("traceId").toString();
String userPrompt = arguments.get("userPrompt").toString();
String traceId = meta.get("traceId").toString();
String userPrompt = meta.get("userPrompt").toString();
if (userPrompt.length() > 2000) {
log.warn(" [Validator] 프롬프트 길이 초과 (Trace ID: {})", traceId);