Package io.trino.plugin.pinot.client

Class PinotClient

java.lang.Object

io.trino.plugin.pinot.client.PinotClient

public class PinotClient
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	PinotClient.BrokerResultRow
static class	PinotClient.BrokersForTable
static class	PinotClient.GetTables
static class	PinotClient.QueryRequest
static class	PinotClient.ResultsIterator
static class	PinotClient.TimeBoundary

Constructor Summary

Constructors

Constructor

Description

PinotClient(PinotConfig config, PinotHostMapper pinotHostMapper, io.airlift.http.client.HttpClient httpClient, ExecutorService executor, io.airlift.json.JsonCodec<PinotClient.GetTables> tablesJsonCodec, io.airlift.json.JsonCodec<PinotClient.BrokersForTable> brokersForTableJsonCodec, io.airlift.json.JsonCodec<PinotClient.TimeBoundary> timeBoundaryJsonCodec, io.airlift.json.JsonCodec<org.apache.pinot.common.response.broker.BrokerResponseNative> brokerResponseCodec, PinotControllerAuthenticationProvider controllerAuthenticationProvider, PinotBrokerAuthenticationProvider brokerAuthenticationProvider)

Method Summary

Modifier and Type	Method	Description
static void	addJsonBinders(io.airlift.json.JsonCodecBinder jsonCodecBinder)
Iterator<PinotClient.BrokerResultRow>	createResultIterator(ConnectorSession session, PinotQueryInfo query, List<PinotColumnHandle> columnHandles)	columnIndices: column name -> column index from column handles indiceToGroupByFunction<Int,String> (groupByFunctions): aggregationIndex -> groupByFunctionName(columnName) groupByFunctions is for values groupByColumnNames: from aggregationResult.groupByResult.groupByColumnNames() aggregationResults[GroupByColumns, GroupByResult] GroupByColumns: String[] // column names, i.e.
protected <T> T	doHttpActionWithHeadersJson(io.airlift.http.client.Request.Builder requestBuilder, Optional<String> requestBody, io.airlift.json.JsonCodec<T> codec, com.google.common.collect.Multimap<String,String> additionalHeaders)
static <T> T	doWithRetries(int retries, Function<Integer,T> caller)
static PinotClient.ResultsIterator	fromResultTable(org.apache.pinot.common.response.broker.BrokerResponseNative brokerResponse, List<PinotColumnHandle> columnHandles, int groupByClauses)
List<String>	getAllBrokersForTable(String table)
protected com.google.common.collect.Multimap<String,String>	getAllTables()
String	getBrokerHost(String table)
static <K, V> V	getFromCache(com.google.common.cache.LoadingCache<K,V> cache, K key)
String	getPinotTableNameFromTrinoTableName(String trinoTableName)
String	getPinotTableNameFromTrinoTableNameIfExists(String trinoTableName)
List<String>	getPinotTableNames()
Map<String,Map<String,List<String>>>	getRoutingTableForTable(String tableName)
org.apache.pinot.spi.data.Schema	getTableSchema(String table)
PinotClient.TimeBoundary	getTimeBoundaryForTable(String table)

Methods inherited from class java.lang.Object

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

Constructor Details

PinotClient

@Inject
public PinotClient(PinotConfig config,
 PinotHostMapper pinotHostMapper,
 io.airlift.http.client.HttpClient httpClient,
 ExecutorService executor,
 io.airlift.json.JsonCodec<PinotClient.GetTables> tablesJsonCodec,
 io.airlift.json.JsonCodec<PinotClient.BrokersForTable> brokersForTableJsonCodec,
 io.airlift.json.JsonCodec<PinotClient.TimeBoundary> timeBoundaryJsonCodec,
 io.airlift.json.JsonCodec<org.apache.pinot.common.response.broker.BrokerResponseNative> brokerResponseCodec,
 PinotControllerAuthenticationProvider controllerAuthenticationProvider,
 PinotBrokerAuthenticationProvider brokerAuthenticationProvider)

Method Details

addJsonBinders

public static void addJsonBinders(io.airlift.json.JsonCodecBinder jsonCodecBinder)

doHttpActionWithHeadersJson

protected <T> T doHttpActionWithHeadersJson(io.airlift.http.client.Request.Builder requestBuilder,
 Optional<String> requestBody,
 io.airlift.json.JsonCodec<T> codec,
 com.google.common.collect.Multimap<String,String> additionalHeaders)

getAllTables

protected com.google.common.collect.Multimap<String,String> getAllTables()

getTableSchema

public org.apache.pinot.spi.data.Schema getTableSchema(String table)
                                                throws Exception

Throws:

Exception

getPinotTableNames

public List<String> getPinotTableNames()

getFromCache

public static <K,
V> V getFromCache(com.google.common.cache.LoadingCache<K,V> cache,
 K key)

getPinotTableNameFromTrinoTableNameIfExists

public String getPinotTableNameFromTrinoTableNameIfExists(String trinoTableName)

getPinotTableNameFromTrinoTableName

public String getPinotTableNameFromTrinoTableName(String trinoTableName)

getAllBrokersForTable

public List<String> getAllBrokersForTable(String table)

getBrokerHost

public String getBrokerHost(String table)

getRoutingTableForTable

public Map<String,Map<String,List<String>>> getRoutingTableForTable(String tableName)

getTimeBoundaryForTable

public PinotClient.TimeBoundary getTimeBoundaryForTable(String table)

createResultIterator

public Iterator<PinotClient.BrokerResultRow> createResultIterator(ConnectorSession session,
 PinotQueryInfo query,
 List<PinotColumnHandle> columnHandles)

columnIndices: column name -> column index from column handles indiceToGroupByFunction<Int,String> (groupByFunctions): aggregationIndex -> groupByFunctionName(columnName) groupByFunctions is for values groupByColumnNames: from aggregationResult.groupByResult.groupByColumnNames() aggregationResults[GroupByColumns, GroupByResult] GroupByColumns: String[] // column names, i.e. group by foo, bar, baz GroupByResult: Row[] Row: {group: String[] // values of groupBy columns, value: aggregationResult}

Results: aggregationResults.get(0..aggregationResults.size()) Result: function, value means columnName -> columnValue

fromResultTable

public static PinotClient.ResultsIterator fromResultTable(org.apache.pinot.common.response.broker.BrokerResponseNative brokerResponse,
 List<PinotColumnHandle> columnHandles,
 int groupByClauses)

doWithRetries

public static <T> T doWithRetries(int retries,
 Function<Integer,T> caller)