Remove deprecated API usage in KuduTableProperties

trinodb · May 18, 2021 · 6d0d774 · 6d0d774
1 parent 68ea403
commit 6d0d774
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plugin/trino-kudu/src/main/java/io/trino/plugin/kudu/properties/KuduTableProperties.java b/plugin/trino-kudu/src/main/java/io/trino/plugin/kudu/properties/KuduTableProperties.java
@@ -330,7 +330,7 @@ public static Map<String, Object> toMap(KuduTable table)
     private static List<RangePartition> getRangePartitionList(KuduTable table, long deadline)
     {
         List<RangePartition> rangePartitions = new ArrayList<>();
-        if (!table.getPartitionSchema().getRangeSchema().getColumns().isEmpty()) {
+        if (!table.getPartitionSchema().getRangeSchema().getColumnIds().isEmpty()) {
             try {
                 for (LocatedTablet tablet : table.getTabletsLocations(deadline)) {
                     Partition partition = tablet.getPartition();
@@ -365,7 +365,7 @@ private static RangeBoundValue buildRangePartitionBound(KuduTable table, byte[]
             Schema schema = table.getSchema();
             PartitionSchema partitionSchema = table.getPartitionSchema();
             PartitionSchema.RangeSchema rangeSchema = partitionSchema.getRangeSchema();
-            List<Integer> rangeColumns = rangeSchema.getColumns();
+            List<Integer> rangeColumns = rangeSchema.getColumnIds();
 
             int numColumns = rangeColumns.size();
 
@@ -427,7 +427,7 @@ public static PartitionDesign getPartitionDesign(KuduTable table)
                 }).collect(toImmutableList());
         partitionDesign.setHash(hashPartitions);
 
-        List<Integer> rangeColumns = partitionSchema.getRangeSchema().getColumns();
+        List<Integer> rangeColumns = partitionSchema.getRangeSchema().getColumnIds();
         if (!rangeColumns.isEmpty()) {
             RangePartitionDefinition definition = new RangePartitionDefinition();
             definition.setColumns(rangeColumns.stream()