org.apache.hadoop.hive.metastore.api.ThriftHiveMetastore Java Examples

@Test
public void testWithThriftHiveMetastoreClient()
        throws Exception
{
    List<String> messages = new ArrayList<>();
    // LoggingInvocationHandler is used e.g. with ThriftHiveMetastore.Iface. Since the logging is reflection-based,
    // we test it with this interface as well.
    ThriftHiveMetastore.Iface proxy = newProxy(ThriftHiveMetastore.Iface.class, new LoggingInvocationHandler(
            dummyThriftHiveMetastoreClient(),
            new AirliftParameterNamesProvider(ThriftHiveMetastore.Iface.class, ThriftHiveMetastore.Client.class),
            messages::add));
    proxy.get_table("some_database", "some_table_name");
    assertThat(messages)
            .hasSize(1)
            .element(0).matches(message -> message.matches("\\QInvocation of get_table(dbname='some_database', tbl_name='some_table_name') succeeded in\\E " + DURATION_PATTERN));
}
 

public ThriftHiveMetastoreClient(TTransport transport, String hostname)
{
    this.transport = requireNonNull(transport, "transport is null");
    ThriftHiveMetastore.Client client = new ThriftHiveMetastore.Client(new TBinaryProtocol(transport));
    if (log.isDebugEnabled()) {
        this.client = newProxy(ThriftHiveMetastore.Iface.class, new LoggingInvocationHandler(client, PARAMETER_NAMES_PROVIDER, log::debug));
    }
    else {
        this.client = client;
    }
    this.hostname = requireNonNull(hostname, "hostname is null");
}
 

private static ThriftHiveMetastore.Iface dummyThriftHiveMetastoreClient()
{
    return (ThriftHiveMetastore.Iface) newProxyInstance(
            TestLoggingInvocationHandlerWithHiveMetastore.class.getClassLoader(),
            new Class<?>[] {ThriftHiveMetastore.Iface.class},
            (proxy, method, args) -> null);
}
 

private ThriftHiveMetastore.Client getClient() {
  final URI metastoreUri = URI.create(MetastoreConf.getAsString(conf, THRIFT_URIS));
  final int socketTimeOut = (int) MetastoreConf.getTimeVar(conf, CLIENT_SOCKET_TIMEOUT, TimeUnit.MILLISECONDS);
  TTransport transport = new TSocket(metastoreUri.getHost(), metastoreUri.getPort(), socketTimeOut);
  try {
    transport.open();
  } catch (TTransportException e) {
    throw new RuntimeException("failed to open socket for " + metastoreUri + " with timeoutMillis " + socketTimeOut);
  }
  return new ThriftHiveMetastore.Client(new TBinaryProtocol(transport));
}
 

private CloseableThriftHiveMetastoreIface newInstance(
    ThriftHiveMetastore.Client delegate,
    HiveThriftMetaStoreIfaceCompatibility compatibility) {
  ClassLoader classLoader = CloseableThriftHiveMetastoreIface.class.getClassLoader();
  Class<?>[] interfaces = new Class<?>[] { CloseableThriftHiveMetastoreIface.class };
  ThriftMetaStoreClientInvocationHandler handler = new ThriftMetaStoreClientInvocationHandler(delegate,
      compatibility);
  return (CloseableThriftHiveMetastoreIface) Proxy.newProxyInstance(classLoader, interfaces, handler);
}
 

/**
 * {@inheritDoc}
 */
@Override
public TProcessor getProcessor() {
    return new ThriftHiveMetastore.Processor<>(
        new CatalogThriftHiveMetastore(config, hiveConverters, metacatV1, partitionV1, catalogName, registry)
    );
}
 

protected HiveTableOperations(Configuration conf, ThriftHiveMetastore.Client metaStoreClient, String database, String table) {
  super(conf);
  this.metaStoreClient = metaStoreClient;
  this.database = database;
  this.tableName = table;
}
 

@Override
public ThriftHiveMetastore.Iface getClient() {
  return metaStoreMapping.getClient();
}
 

@Override
public ThriftHiveMetastore.Iface getClient() {
  return client;
}
 

protected ThriftHiveMetastore.Iface getClient() {
  return client;
}
 

ThriftMetaStoreClientInvocationHandler(
    ThriftHiveMetastore.Client delegate,
    HiveThriftMetaStoreIfaceCompatibility compatibility) {
  this.delegate = delegate;
  this.compatibility = compatibility;
}
 

public CloseableThriftHiveMetastoreIface newInstance(ThriftHiveMetastore.Client delegate) {
  HiveThriftMetaStoreIfaceCompatibility compatibility = new HiveThriftMetaStoreIfaceCompatibility1xx(delegate);
  return newInstance(delegate, compatibility);
}
 

public HiveThriftMetaStoreIfaceCompatibility1xx(ThriftHiveMetastore.Client client) {
  this.client = client;
}
 

private ThriftHiveMetastore.Iface getPrimaryClient() throws TException {
  return databaseMappingService.primaryDatabaseMapping().getClient();
}
 

public static ThriftHiveMetastore.Iface getClient(HiveMetaStoreClient client){
  return client.client;
}
 

ThriftHiveMetastore.Iface getClient();