Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.trino.plugin.hive;
import com.google.common.collect.ImmutableList;
import io.trino.plugin.hive.acid.AcidOperation;
import io.trino.plugin.hive.acid.AcidTransaction;
import io.trino.plugin.hive.authentication.HiveIdentity;
import io.trino.plugin.hive.metastore.Database;
import io.trino.plugin.hive.metastore.HiveMetastore;
import io.trino.plugin.hive.metastore.HivePrincipal;
import io.trino.plugin.hive.metastore.HivePrivilegeInfo;
import io.trino.plugin.hive.metastore.Partition;
import io.trino.plugin.hive.metastore.PartitionWithStatistics;
import io.trino.plugin.hive.metastore.PrincipalPrivileges;
import io.trino.plugin.hive.metastore.Table;
import io.trino.spi.connector.SchemaTableName;
import io.trino.spi.connector.TableNotFoundException;
import io.trino.spi.predicate.TupleDomain;
import io.trino.spi.security.RoleGrant;
import io.trino.spi.statistics.ColumnStatisticType;
import io.trino.spi.type.Type;
import org.apache.hadoop.hive.metastore.api.DataOperationType;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.OptionalLong;
import java.util.Set;
import java.util.function.Function;
import static com.google.common.collect.ImmutableList.toImmutableList;
import static com.google.common.collect.ImmutableMap.toImmutableMap;
import static com.google.common.collect.Maps.immutableEntry;
import static io.trino.plugin.hive.HivePartitionManager.extractPartitionValues;
import static java.util.Objects.requireNonNull;
public class HiveMetastoreClosure
{
private final HiveMetastore delegate;
public HiveMetastoreClosure(HiveMetastore delegate)
{
this.delegate = requireNonNull(delegate, "delegate is null");
}
public Optional getDatabase(String databaseName)
{
return delegate.getDatabase(databaseName);
}
public List getAllDatabases()
{
return delegate.getAllDatabases();
}
private Table getExistingTable(HiveIdentity identity, String databaseName, String tableName)
{
return delegate.getTable(identity, databaseName, tableName)
.orElseThrow(() -> new TableNotFoundException(new SchemaTableName(databaseName, tableName)));
}
public Optional
getTable(HiveIdentity identity, String databaseName, String tableName)
{
return delegate.getTable(identity, databaseName, tableName);
}
public Set getSupportedColumnStatistics(Type type)
{
return delegate.getSupportedColumnStatistics(type);
}
public PartitionStatistics getTableStatistics(HiveIdentity identity, String databaseName, String tableName)
{
return delegate.getTableStatistics(identity, getExistingTable(identity, databaseName, tableName));
}
public Map getPartitionStatistics(HiveIdentity identity, String databaseName, String tableName, Set partitionNames)
{
Table table = getExistingTable(identity, databaseName, tableName);
List partitions = getExistingPartitionsByNames(identity, table, ImmutableList.copyOf(partitionNames));
return delegate.getPartitionStatistics(identity, table, partitions);
}
public void updateTableStatistics(HiveIdentity identity, String databaseName, String tableName, AcidTransaction transaction, Function update)
{
delegate.updateTableStatistics(identity, databaseName, tableName, transaction, update);
}
public void updatePartitionStatistics(HiveIdentity identity, String databaseName, String tableName, String partitionName, Function update)
{
Table table = getExistingTable(identity, databaseName, tableName);
delegate.updatePartitionStatistics(identity, table, partitionName, update);
}
public List getAllTables(String databaseName)
{
return delegate.getAllTables(databaseName);
}
public List getTablesWithParameter(String databaseName, String parameterKey, String parameterValue)
{
return delegate.getTablesWithParameter(databaseName, parameterKey, parameterValue);
}
public List getAllViews(String databaseName)
{
return delegate.getAllViews(databaseName);
}
public void createDatabase(HiveIdentity identity, Database database)
{
delegate.createDatabase(identity, database);
}
public void dropDatabase(HiveIdentity identity, String databaseName)
{
delegate.dropDatabase(identity, databaseName);
}
public void renameDatabase(HiveIdentity identity, String databaseName, String newDatabaseName)
{
delegate.renameDatabase(identity, databaseName, newDatabaseName);
}
public void setDatabaseOwner(HiveIdentity identity, String databaseName, HivePrincipal principal)
{
delegate.setDatabaseOwner(identity, databaseName, principal);
}
public void setTableOwner(HiveIdentity identity, String databaseName, String tableName, HivePrincipal principal)
{
delegate.setTableOwner(identity, databaseName, tableName, principal);
}
public void createTable(HiveIdentity identity, Table table, PrincipalPrivileges principalPrivileges)
{
delegate.createTable(identity, table, principalPrivileges);
}
public void dropTable(HiveIdentity identity, String databaseName, String tableName, boolean deleteData)
{
delegate.dropTable(identity, databaseName, tableName, deleteData);
}
public void replaceTable(HiveIdentity identity, String databaseName, String tableName, Table newTable, PrincipalPrivileges principalPrivileges)
{
delegate.replaceTable(identity, databaseName, tableName, newTable, principalPrivileges);
}
public void renameTable(HiveIdentity identity, String databaseName, String tableName, String newDatabaseName, String newTableName)
{
delegate.renameTable(identity, databaseName, tableName, newDatabaseName, newTableName);
}
public void commentTable(HiveIdentity identity, String databaseName, String tableName, Optional comment)
{
delegate.commentTable(identity, databaseName, tableName, comment);
}
public void commentColumn(HiveIdentity identity, String databaseName, String tableName, String columnName, Optional comment)
{
delegate.commentColumn(identity, databaseName, tableName, columnName, comment);
}
public void addColumn(HiveIdentity identity, String databaseName, String tableName, String columnName, HiveType columnType, String columnComment)
{
delegate.addColumn(identity, databaseName, tableName, columnName, columnType, columnComment);
}
public void renameColumn(HiveIdentity identity, String databaseName, String tableName, String oldColumnName, String newColumnName)
{
delegate.renameColumn(identity, databaseName, tableName, oldColumnName, newColumnName);
}
public void dropColumn(HiveIdentity identity, String databaseName, String tableName, String columnName)
{
delegate.dropColumn(identity, databaseName, tableName, columnName);
}
public Optional getPartition(HiveIdentity identity, String databaseName, String tableName, List partitionValues)
{
return delegate.getTable(identity, databaseName, tableName)
.flatMap(table -> delegate.getPartition(identity, table, partitionValues));
}
public Optional> getPartitionNamesByFilter(
HiveIdentity identity,
String databaseName,
String tableName,
List columnNames,
TupleDomain partitionKeysFilter)
{
return delegate.getPartitionNamesByFilter(identity, databaseName, tableName, columnNames, partitionKeysFilter);
}
private List getExistingPartitionsByNames(HiveIdentity identity, Table table, List partitionNames)
{
Map partitions = delegate.getPartitionsByNames(identity, table, partitionNames).entrySet().stream()
.map(entry -> immutableEntry(entry.getKey(), entry.getValue().orElseThrow(() ->
new PartitionNotFoundException(table.getSchemaTableName(), extractPartitionValues(entry.getKey())))))
.collect(toImmutableMap(Map.Entry::getKey, Map.Entry::getValue));
return partitionNames.stream()
.map(partitions::get)
.collect(toImmutableList());
}
public Map> getPartitionsByNames(HiveIdentity identity, String databaseName, String tableName, List partitionNames)
{
return delegate.getTable(identity, databaseName, tableName)
.map(table -> delegate.getPartitionsByNames(identity, table, partitionNames))
.orElseGet(() -> partitionNames.stream()
.collect(toImmutableMap(name -> name, name -> Optional.empty())));
}
public void addPartitions(HiveIdentity identity, String databaseName, String tableName, List partitions)
{
delegate.addPartitions(identity, databaseName, tableName, partitions);
}
public void dropPartition(HiveIdentity identity, String databaseName, String tableName, List parts, boolean deleteData)
{
delegate.dropPartition(identity, databaseName, tableName, parts, deleteData);
}
public void alterPartition(HiveIdentity identity, String databaseName, String tableName, PartitionWithStatistics partition)
{
delegate.alterPartition(identity, databaseName, tableName, partition);
}
public void createRole(String role, String grantor)
{
delegate.createRole(role, grantor);
}
public void dropRole(String role)
{
delegate.dropRole(role);
}
public Set listRoles()
{
return delegate.listRoles();
}
public void grantRoles(Set roles, Set grantees, boolean adminOption, HivePrincipal grantor)
{
delegate.grantRoles(roles, grantees, adminOption, grantor);
}
public void revokeRoles(Set roles, Set grantees, boolean adminOption, HivePrincipal grantor)
{
delegate.revokeRoles(roles, grantees, adminOption, grantor);
}
public Set listGrantedPrincipals(String role)
{
return delegate.listGrantedPrincipals(role);
}
public Set listRoleGrants(HivePrincipal principal)
{
return delegate.listRoleGrants(principal);
}
public void grantTablePrivileges(String databaseName, String tableName, String tableOwner, HivePrincipal grantee, Set privileges)
{
delegate.grantTablePrivileges(databaseName, tableName, tableOwner, grantee, privileges);
}
public void revokeTablePrivileges(String databaseName, String tableName, String tableOwner, HivePrincipal grantee, Set privileges)
{
delegate.revokeTablePrivileges(databaseName, tableName, tableOwner, grantee, privileges);
}
public Set listTablePrivileges(String databaseName, String tableName, String tableOwner, Optional principal)
{
return delegate.listTablePrivileges(databaseName, tableName, tableOwner, principal);
}
public boolean isImpersonationEnabled()
{
return delegate.isImpersonationEnabled();
}
public long openTransaction(HiveIdentity identity)
{
return delegate.openTransaction(identity);
}
public void commitTransaction(HiveIdentity identity, long transactionId)
{
delegate.commitTransaction(identity, transactionId);
}
public void sendTransactionHeartbeat(HiveIdentity identity, long transactionId)
{
delegate.sendTransactionHeartbeat(identity, transactionId);
}
public void acquireSharedReadLock(HiveIdentity identity, String queryId, long transactionId, List fullTables, List partitions)
{
delegate.acquireSharedReadLock(identity, queryId, transactionId, fullTables, partitions);
}
public String getValidWriteIds(HiveIdentity identity, List tables, long currentTransactionId)
{
return delegate.getValidWriteIds(identity, tables, currentTransactionId);
}
public Optional getConfigValue(String name)
{
return delegate.getConfigValue(name);
}
public long allocateWriteId(HiveIdentity identity, String dbName, String tableName, long transactionId)
{
return delegate.allocateWriteId(identity, dbName, tableName, transactionId);
}
public void acquireTableWriteLock(HiveIdentity identity, String queryId, long transactionId, String dbName, String tableName, DataOperationType operation, boolean isPartitioned)
{
delegate.acquireTableWriteLock(identity, queryId, transactionId, dbName, tableName, operation, isPartitioned);
}
public void updateTableWriteId(HiveIdentity identity, String dbName, String tableName, long transactionId, long writeId, OptionalLong rowCountChange)
{
delegate.updateTableWriteId(identity, dbName, tableName, transactionId, writeId, rowCountChange);
}
public void alterPartitions(HiveIdentity identity, String dbName, String tableName, List partitions, long writeId)
{
delegate.alterPartitions(identity, dbName, tableName, partitions, writeId);
}
public void addDynamicPartitions(HiveIdentity identity, String dbName, String tableName, List partitionNames, long transactionId, long writeId, AcidOperation operation)
{
delegate.addDynamicPartitions(identity, dbName, tableName, partitionNames, transactionId, writeId, operation);
}
public void alterTransactionalTable(HiveIdentity identity, Table table, long transactionId, long writeId, PrincipalPrivileges principalPrivileges)
{
delegate.alterTransactionalTable(identity, table, transactionId, writeId, principalPrivileges);
}
}
