Trino LDAP认证场景下REST API文档及Java/Python连接示例咨询

关于REST API官方文档的说明 #

Trino提供了完整的对外REST API能力，覆盖认证、查询提交、状态轮询、结果获取、查询终止全链路操作，所有端点的路径规则、请求要求、响应结构都随版本同步收录在官方随发行版发布的文档中，完全可以支撑直接基于API做二次开发。
针对LDAP认证场景有几个核心注意点：

• Trino没有单独的登录接口，LDAP认证逻辑嵌入在每一次API请求的校验环节：请求时携带HTTP Basic认证头，填入LDAP的用户名、密码即可通过校验，服务端不会返回独立的会话令牌，后续所有请求持续携带合法的Basic认证头即可
• 提交查询的核心端点为POST /v1/statement，请求体直接传入待执行的SQL纯文本，除认证头外还需要携带X-Trino-User请求头指定访问用户，可按需附加X-Trino-Catalog、X-Trino-Schema等请求头指定默认的目录、schema
• 查询提交成功后，响应体中会直接返回nextUri字段，后续轮询查询状态、拉取分页结果、判断查询结束都直接访问对应返回的URI即可，不需要自行拼接接口路径

LDAP模式下连接代码参考 #

以下是直接调用原生REST API的实现片段，如果是生产环境使用，优先推荐直接用官方提供的JDBC/Python客户端，只需要配置LDAP相关连接参数即可，不需要自行实现全量的API交互逻辑，容错性更好。

Python 实现 #

基于requests库实现，核心是在所有请求中附加LDAP账号对应的Basic认证信息：

import requests
from requests.auth import HTTPBasicAuth

# 替换为实际的Trino协调节点地址
TRINO_COORDINATOR = "https://<your-trino-coordinator-host>:8443"
# 替换为实际的LDAP账号密码
LDAP_USERNAME = "<your-ldap-account>"
LDAP_PASSWORD = "<your-ldap-password>"

common_headers = {
    "X-Trino-User": LDAP_USERNAME,
    "X-Trino-Catalog": "hive", # 替换为实际使用的catalog
    "X-Trino-Schema": "default", # 替换为实际使用的schema
    "Content-Type": "text/plain"
}
common_auth = HTTPBasicAuth(LDAP_USERNAME, LDAP_PASSWORD)

# 提交查询
sql_text = "SELECT * FROM your_table LIMIT 10"
submit_resp = requests.post(
    url=f"{TRINO_COORDINATOR}/v1/statement",
    data=sql_text.encode("utf-8"),
    headers=common_headers,
    auth=common_auth,
    verify=False # 生产环境请替换为实际CA证书路径，不要关闭证书校验
)
submit_resp.raise_for_status()
current_resp_data = submit_resp.json()

# 轮询拉取结果
all_result = []
while True:
    if "error" in current_resp_data:
        raise RuntimeError(f"查询执行失败: {current_resp_data['error']['message']}")
    if "data" in current_resp_data:
        all_result.extend(current_resp_data["data"])
    if "nextUri" not in current_resp_data:
        break
    next_resp = requests.get(
        url=current_resp_data["nextUri"],
        auth=common_auth,
        verify=False
    )
    next_resp.raise_for_status()
    current_resp_data = next_resp.json()

print(f"查询返回结果共{len(all_result)}条:", all_result)

Java 实现 #

基于JDK原生HttpClient实现，核心是手动构造HTTP Basic认证头随请求发送：

import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
import java.util.Base64;

public class TrinoLdapApiClient {
    public static void main(String[] args) throws Exception {
        // 替换为实际配置
        String coordinatorAddr = "https://<your-trino-coordinator-host>:8443";
        String ldapUser = "<your-ldap-account>";
        String ldapPwd = "<your-ldap-password>";

        // 构造Basic认证头
        String authValue = "Basic " + Base64.getEncoder().encodeToString(
                (ldapUser + ":" + ldapPwd).getBytes(StandardCharsets.UTF_8)
        );

        // 构造查询提交请求
        String sql = "SELECT * FROM your_table LIMIT 10";
        HttpRequest submitRequest = HttpRequest.newBuilder()
                .uri(URI.create(coordinatorAddr + "/v1/statement"))
                .header("Authorization", authValue)
                .header("X-Trino-User", ldapUser)
                .header("X-Trino-Catalog", "hive")
                .header("X-Trino-Schema", "default")
                .header("Content-Type", "text/plain")
                .POST(HttpRequest.BodyPublishers.ofString(sql))
                .build();

        HttpClient httpClient = HttpClient.newHttpClient();
        HttpResponse<String> submitResp = httpClient.send(submitRequest, HttpResponse.BodyHandlers.ofString());
        
        if (submitResp.statusCode() != 200) {
            throw new RuntimeException("查询提交失败，响应码：" + submitResp.statusCode() + "，响应内容：" + submitResp.body());
        }
        // 后续可引入JSON解析包处理响应，按照nextUri轮询拉取结果的逻辑和Python版本一致
        System.out.println("查询提交成功，初始响应：" + submitResp.body());
    }
}

注意：如果集群配置了HTTPS双向认证、自定义认证过滤规则，需要在请求中额外附加对应要求的头信息或证书配置。

内容的提问来源于stack exchange，提问作者Rock_17