org.apache.calcite.adapter.druid

Class DruidConnectionImpl

java.lang.Object

org.apache.calcite.adapter.druid.DruidConnectionImpl

All Implemented Interfaces:

DruidConnection

class DruidConnectionImpl
extends Object
implements DruidConnection

Implementation of DruidConnection.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
private static class	DruidConnectionImpl.BlockingQueueEnumerator<E> An Enumerator that gets its rows from a BlockingQueue.
private static class	DruidConnectionImpl.JsonAggregator Element of the "aggregators" collection in the result of a "segmentMetadata" call, populated by Jackson.
private static class	DruidConnectionImpl.JsonColumn Element of the "columns" collection in the result of a "segmentMetadata" call, populated by Jackson.
private static class	DruidConnectionImpl.JsonSegmentMetadata Result of a "segmentMetadata" call, populated by Jackson.
(package private) static class	DruidConnectionImpl.Page Progress through a large fetch.
private static interface	DruidConnectionImpl.RunnableQueueSink A Sink that is also Runnable.

Field Summary

Fields

Modifier and Type	Field and Description
private String	coordinatorUrl
private static String	DEFAULT_RESPONSE_TIMESTAMP_COLUMN
private static Set<String>	SUPPORTED_TYPES
private String	url
private static SimpleDateFormat	UTC_TIMESTAMP_FORMAT

Constructor Summary

Constructors

Constructor and Description
DruidConnectionImpl(String url, String coordinatorUrl)

Method Summary

Modifier and Type	Method and Description
Enumerable<Row>	enumerable(QueryType queryType, String request, List<String> fieldNames, ExecutorService service) Executes a request and returns the resulting rows as an Enumerable, running the parser in a thread provided by service.
private void	expect(com.fasterxml.jackson.core.JsonParser parser, com.fasterxml.jackson.core.JsonToken token)
private void	expectObjectField(com.fasterxml.jackson.core.JsonParser parser, String name)
private void	expectScalarField(com.fasterxml.jackson.core.JsonParser parser, String name)
private Long	extractTimestampField(com.fasterxml.jackson.core.JsonParser parser)
private boolean	isSupportedType(String type)
(package private) void	metadata(String dataSourceName, String timestampColumnName, List<org.joda.time.Interval> intervals, Map<String,SqlTypeName> fieldBuilder, Set<String> metricNameBuilder) Reads segment metadata, and populates a list of columns and metrics.
private void	parse(QueryType queryType, InputStream in, Sink sink, List<String> fieldNames, List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes, DruidConnectionImpl.Page page) Parses the output of a query, sending the results to a Sink.
private void	parseField(List<String> fieldNames, List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes, int posTimestampField, Row.RowBuilder rowBuilder, com.fasterxml.jackson.core.JsonParser parser)
private void	parseFields(List<String> fieldNames, List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes, int posTimestampField, Row.RowBuilder rowBuilder, com.fasterxml.jackson.core.JsonParser parser)
private void	parseFields(List<String> fieldNames, List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes, Row.RowBuilder rowBuilder, com.fasterxml.jackson.core.JsonParser parser)
void	request(QueryType queryType, String data, Sink sink, List<String> fieldNames, List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes, DruidConnectionImpl.Page page) Executes a query request.
(package private) Set<String>	tableNames() Reads data source names from Druid.
private InputStream	traceResponse(InputStream in)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

url

private final String url

coordinatorUrl

private final String coordinatorUrl

DEFAULT_RESPONSE_TIMESTAMP_COLUMN

private static final String DEFAULT_RESPONSE_TIMESTAMP_COLUMN

See Also:

Constant Field Values

UTC_TIMESTAMP_FORMAT

private static final SimpleDateFormat UTC_TIMESTAMP_FORMAT

SUPPORTED_TYPES

private static final Set<String> SUPPORTED_TYPES

Constructor Detail

DruidConnectionImpl

DruidConnectionImpl(String url,
                    String coordinatorUrl)

Method Detail

request

public void request(QueryType queryType,
                    String data,
                    Sink sink,
                    List<String> fieldNames,
                    List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes,
                    DruidConnectionImpl.Page page)

Executes a query request.

Parameters:

queryType - Query type

data - Data to post

sink - Sink to which to send the parsed rows

fieldNames - Names of fields

fieldTypes - Types of fields (never null, but elements may be null)

page - Page definition (in/out)

parse

private void parse(QueryType queryType,
                   InputStream in,
                   Sink sink,
                   List<String> fieldNames,
                   List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes,
                   DruidConnectionImpl.Page page)

Parses the output of a query, sending the results to a Sink.

parseFields

private void parseFields(List<String> fieldNames,
                         List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes,
                         Row.RowBuilder rowBuilder,
                         com.fasterxml.jackson.core.JsonParser parser)
                  throws IOException

Throws:

IOException

parseFields

private void parseFields(List<String> fieldNames,
                         List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes,
                         int posTimestampField,
                         Row.RowBuilder rowBuilder,
                         com.fasterxml.jackson.core.JsonParser parser)
                  throws IOException

Throws:

IOException

parseField

private void parseField(List<String> fieldNames,
                        List<org.apache.calcite.avatica.ColumnMetaData.Rep> fieldTypes,
                        int posTimestampField,
                        Row.RowBuilder rowBuilder,
                        com.fasterxml.jackson.core.JsonParser parser)
                 throws IOException

Throws:

IOException

expect

private void expect(com.fasterxml.jackson.core.JsonParser parser,
                    com.fasterxml.jackson.core.JsonToken token)
             throws IOException

Throws:

IOException

expectScalarField

private void expectScalarField(com.fasterxml.jackson.core.JsonParser parser,
                               String name)
                        throws IOException

Throws:

IOException

expectObjectField

private void expectObjectField(com.fasterxml.jackson.core.JsonParser parser,
                               String name)
                        throws IOException

Throws:

IOException

extractTimestampField

private Long extractTimestampField(com.fasterxml.jackson.core.JsonParser parser)
                            throws IOException

Throws:

IOException

enumerable

public Enumerable<Row> enumerable(QueryType queryType,
                                  String request,
                                  List<String> fieldNames,
                                  ExecutorService service)
                           throws IOException

Executes a request and returns the resulting rows as an Enumerable, running the parser in a thread provided by service.

Throws:

IOException

isSupportedType

private boolean isSupportedType(String type)

metadata

void metadata(String dataSourceName,
              String timestampColumnName,
              List<org.joda.time.Interval> intervals,
              Map<String,SqlTypeName> fieldBuilder,
              Set<String> metricNameBuilder)

Reads segment metadata, and populates a list of columns and metrics.

tableNames

Set<String> tableNames()

Reads data source names from Druid.

traceResponse

private InputStream traceResponse(InputStream in)