Examples with CatalogException org.apache.flink.table.catalog.exceptions.CatalogException used on opensource projects

Example 6 with CatalogException

use of org.apache.flink.table.catalog.exceptions.CatalogException in project flink by apache.

the class HiveCatalog method createDatabase.

@Override
public void createDatabase(String databaseName, CatalogDatabase database, boolean ignoreIfExists) throws DatabaseAlreadyExistException, CatalogException {
    checkArgument(!isNullOrWhitespaceOnly(databaseName), "databaseName cannot be null or empty");
    checkNotNull(database, "database cannot be null");
    Map<String, String> properties = database.getProperties();
    String dbLocationUri = properties.remove(SqlCreateHiveDatabase.DATABASE_LOCATION_URI);
    Database hiveDatabase = new Database(databaseName, database.getComment(), dbLocationUri, properties);
    try {
        client.createDatabase(hiveDatabase);
    } catch (AlreadyExistsException e) {
        if (!ignoreIfExists) {
            throw new DatabaseAlreadyExistException(getName(), hiveDatabase.getName());
        }
    } catch (TException e) {
        throw new CatalogException(String.format("Failed to create database %s", hiveDatabase.getName()), e);
    }
}

Example 7 with CatalogException

use of org.apache.flink.table.catalog.exceptions.CatalogException in project flink by apache.

the class HiveCatalog method listPartitions.

@Override
public List<CatalogPartitionSpec> listPartitions(ObjectPath tablePath, CatalogPartitionSpec partitionSpec) throws TableNotExistException, TableNotPartitionedException, PartitionSpecInvalidException, CatalogException {
    checkNotNull(tablePath, "Table path cannot be null");
    checkNotNull(partitionSpec, "CatalogPartitionSpec cannot be null");
    Table hiveTable = getHiveTable(tablePath);
    ensurePartitionedTable(tablePath, hiveTable);
    checkValidPartitionSpec(partitionSpec, getFieldNames(hiveTable.getPartitionKeys()), tablePath);
    try {
        // partition spec can be partial
        List<String> partialVals = HiveReflectionUtils.getPvals(hiveShim, hiveTable.getPartitionKeys(), partitionSpec.getPartitionSpec());
        return client.listPartitionNames(tablePath.getDatabaseName(), tablePath.getObjectName(), partialVals, (short) -1).stream().map(HiveCatalog::createPartitionSpec).collect(Collectors.toList());
    } catch (TException e) {
        throw new CatalogException(String.format("Failed to list partitions of table %s", tablePath), e);
    }
}

Example 8 with CatalogException

use of org.apache.flink.table.catalog.exceptions.CatalogException in project flink by apache.

the class HiveShimV100 method listBuiltInFunctions.

@Override
public Set<String> listBuiltInFunctions() {
    try {
        Method method = FunctionRegistry.class.getDeclaredMethod("getFunctionNames", boolean.class);
        method.setAccessible(true);
        // don't search HMS cause we're only interested in built-in functions
        Set<String> names = (Set<String>) method.invoke(null, false);
        return names.stream().filter(n -> getBuiltInFunctionInfo(n).isPresent()).collect(Collectors.toSet());
    } catch (Exception ex) {
        throw new CatalogException("Failed to invoke FunctionRegistry.getFunctionNames()", ex);
    }
}

Example 9 with CatalogException

use of org.apache.flink.table.catalog.exceptions.CatalogException in project flink by apache.

the class HiveShimV100 method alterPartition.

@Override
public void alterPartition(IMetaStoreClient client, String databaseName, String tableName, Partition partition) throws InvalidOperationException, MetaException, TException {
    String errorMsg = "Failed to alter partition for table %s in database %s";
    try {
        Method method = client.getClass().getMethod("alter_partition", String.class, String.class, Partition.class);
        method.invoke(client, databaseName, tableName, partition);
    } catch (InvocationTargetException ite) {
        Throwable targetEx = ite.getTargetException();
        if (targetEx instanceof TException) {
            throw (TException) targetEx;
        } else {
            throw new CatalogException(String.format(errorMsg, tableName, databaseName), targetEx);
        }
    } catch (NoSuchMethodException | IllegalAccessException e) {
        throw new CatalogException(String.format(errorMsg, tableName, databaseName), e);
    }
}

Example 10 with CatalogException

use of org.apache.flink.table.catalog.exceptions.CatalogException in project flink by apache.

the class HiveCatalog method alterPartitionColumnStatistics.

@Override
public void alterPartitionColumnStatistics(ObjectPath tablePath, CatalogPartitionSpec partitionSpec, CatalogColumnStatistics columnStatistics, boolean ignoreIfNotExists) throws PartitionNotExistException, CatalogException {
    try {
        Partition hivePartition = getHivePartition(tablePath, partitionSpec);
        Table hiveTable = getHiveTable(tablePath);
        String partName = getEscapedPartitionName(tablePath, partitionSpec, hiveTable);
        client.updatePartitionColumnStatistics(HiveStatsUtil.createPartitionColumnStats(hivePartition, partName, columnStatistics.getColumnStatisticsData(), hiveVersion));
    } catch (TableNotExistException | PartitionSpecInvalidException e) {
        if (!ignoreIfNotExists) {
            throw new PartitionNotExistException(getName(), tablePath, partitionSpec, e);
        }
    } catch (TException e) {
        throw new CatalogException(String.format("Failed to alter table column stats of table %s 's partition %s", tablePath.getFullName(), String.valueOf(partitionSpec)), e);
    }
}