Ejemplo n.º 1
0
 static ClientSession processSessionParameterChange(
     Object parsedStatement, ClientSession session) {
   if (parsedStatement instanceof Use) {
     Use use = (Use) parsedStatement;
     return ClientSession.withCatalogAndSchema(
         session, use.getCatalog().orElse(session.getCatalog()), use.getSchema());
   }
   return session;
 }
Ejemplo n.º 2
0
    private static Request buildQueryRequest(ClientSession session, String query) {
      Request.Builder builder =
          preparePost()
              .setUri(uriBuilderFrom(session.getServer()).replacePath("/v1/execute").build())
              .setBodyGenerator(createStaticBodyGenerator(query, Charsets.UTF_8));

      if (session.getUser() != null) {
        builder.setHeader(PrestoHeaders.PRESTO_USER, session.getUser());
      }
      if (session.getSource() != null) {
        builder.setHeader(PrestoHeaders.PRESTO_SOURCE, session.getSource());
      }
      if (session.getCatalog() != null) {
        builder.setHeader(PrestoHeaders.PRESTO_CATALOG, session.getCatalog());
      }
      if (session.getSchema() != null) {
        builder.setHeader(PrestoHeaders.PRESTO_SCHEMA, session.getSchema());
      }
      builder.setHeader(PrestoHeaders.PRESTO_SCHEMA, session.getTimeZoneId());
      builder.setHeader(USER_AGENT, USER_AGENT_VALUE);

      return builder.build();
    }