presto源码学习2——RESTFul服务

presto源码学习2——RESTFul服务

presto集群中的数据传输、节点通信、计算调度和监控等都是通过接口间的互相调用来实现的。其中，主要的的查询相关的接口有Statement、Query、Stage、Task这几类，都位于presto-main模块中

Statement接口

接口概览

Statement接口在QueuedStatementResource和ExecutingStatementResource类中声明

QueuedStatementResource中有以下几个操作接口

• POST /v1/statement：所有查询的入口，客户端直接调用的接口
• GET /v1/statement/queued/{queryId}/{token}：提交queryId代表的一个查询
• DELETE /v1/statement/queued/{queryId}/{token}：取消等待查询队列中的某个查询

ExecutingStatementResource类中是对正在执行的查询的操作

• GET /v1/statement/executing/{queryId}/{token}：执行queryId代表的查询，并将结果返回
• DELETE /v1/statement/executing/{queryId}/{token}：取消正在执行的查询

Statement语句的处理流程

1、statement会以参数的形式发送至/v1/statement接口，客户端连接信息（user、catalog、schema等）都会携带在请求的header中。

@POST
    @Path("/v1/statement")
    @Produces(APPLICATION_JSON)
    public Response postStatement(
            String statement,
            @HeaderParam(X_FORWARDED_PROTO) String xForwardedProto,
            @Context HttpServletRequest servletRequest,
            @Context UriInfo uriInfo)
    {
   
        if (isNullOrEmpty(statement)) {
   
            throw badRequest(BAD_REQUEST, "SQL statement is empty");
        }

        SessionContext sessionContext = new HttpRequestSessionContext(servletRequest);
        Query query = new