79 lines
2.7 KiB
Java
79 lines
2.7 KiB
Java
package com.cloudhandson.vpdbackoffice.service;
|
|
|
|
import com.fasterxml.jackson.databind.JsonNode;
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
import com.fasterxml.jackson.databind.node.TextNode;
|
|
import java.sql.CallableStatement;
|
|
import java.sql.Clob;
|
|
import java.sql.PreparedStatement;
|
|
import java.sql.ResultSet;
|
|
import org.springframework.jdbc.core.ConnectionCallback;
|
|
import org.springframework.jdbc.core.JdbcTemplate;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
/** JDBC implementation shared with the standalone HMM MCP facade. */
|
|
@Service
|
|
public class JdbcHmmAiAgentToolRunner implements HmmAiAgentToolRunner {
|
|
|
|
private final JdbcTemplate jdbcTemplate;
|
|
private final ObjectMapper objectMapper;
|
|
|
|
public JdbcHmmAiAgentToolRunner(JdbcTemplate jdbcTemplate, ObjectMapper objectMapper) {
|
|
this.jdbcTemplate = jdbcTemplate;
|
|
this.objectMapper = objectMapper;
|
|
}
|
|
|
|
@Override
|
|
public JsonNode run(
|
|
String toolName,
|
|
com.fasterxml.jackson.databind.node.ObjectNode input,
|
|
String bearerToken
|
|
) {
|
|
try {
|
|
String request = objectMapper.writeValueAsString(input);
|
|
String response = jdbcTemplate.execute((ConnectionCallback<String>) connection -> {
|
|
boolean contextSet = false;
|
|
try {
|
|
try (CallableStatement statement = connection.prepareCall(
|
|
"BEGIN ADMIN.HMM_ACCESS_CTX_PKG.SET_USER_BY_BEARER(?); END;")) {
|
|
statement.setString(1, bearerToken);
|
|
statement.execute();
|
|
contextSet = true;
|
|
}
|
|
try (PreparedStatement statement = connection.prepareStatement("""
|
|
SELECT DBMS_CLOUD_AI_AGENT.RUN_TOOL(?, TO_CLOB(?))
|
|
FROM dual
|
|
""")) {
|
|
statement.setString(1, toolName);
|
|
statement.setString(2, request);
|
|
try (ResultSet resultSet = statement.executeQuery()) {
|
|
if (!resultSet.next()) {
|
|
return "";
|
|
}
|
|
Clob clob = resultSet.getClob(1);
|
|
return clob == null ? "" : clob.getSubString(1, (int) clob.length());
|
|
}
|
|
}
|
|
} finally {
|
|
if (contextSet) {
|
|
try (CallableStatement statement = connection.prepareCall(
|
|
"BEGIN ADMIN.HMM_ACCESS_CTX_PKG.CLEAR_USER; END;")) {
|
|
statement.execute();
|
|
}
|
|
}
|
|
}
|
|
});
|
|
if (response == null || response.isBlank()) {
|
|
return objectMapper.createObjectNode();
|
|
}
|
|
try {
|
|
return objectMapper.readTree(response);
|
|
} catch (Exception ignored) {
|
|
return TextNode.valueOf(response);
|
|
}
|
|
} catch (Exception exception) {
|
|
throw new AppException("HMM AI Agent Tool 실행에 실패했습니다: " + exception.getMessage());
|
|
}
|
|
}
|
|
}
|