SourabhBadhya commented on code in PR #5076: URL: https://github.com/apache/hive/pull/5076#discussion_r1505606288
########## ql/src/java/org/apache/hadoop/hive/ql/plan/ConditionalResolverMergeFiles.java: ########## @@ -125,6 +133,22 @@ public ListBucketingCtx getLbCtx() { public void setLbCtx(ListBucketingCtx lbCtx) { this.lbCtx = lbCtx; } + + public void setCustomStorageHandlerProps(Properties properties) { Review Comment: Removed. ########## iceberg/iceberg-handler/src/main/java/org/apache/iceberg/mr/hive/HiveIcebergStorageHandler.java: ########## @@ -2021,4 +2023,34 @@ public List<Partition> getPartitionsByExpr(org.apache.hadoop.hive.ql.metadata.Ta throw new SemanticException(String.format("Error while fetching the partitions due to: %s", e)); } } + + @Override + public boolean supportsMergeFiles() { + return true; + } + + @Override + public List<FileStatus> getMergeInputFiles(Properties properties) throws IOException { + String tableName = properties.getProperty(Catalogs.NAME); + String snapshotRef = properties.getProperty(Catalogs.SNAPSHOT_REF); + Configuration configuration = SessionState.getSessionConf(); + List<JobContext> originalContextList = generateJobContext(configuration, tableName, snapshotRef); + List<JobContext> jobContextList = originalContextList.stream() + .map(TezUtil::enrichContextWithVertexId) + .collect(Collectors.toList()); + if (originalContextList.isEmpty()) { + return Collections.emptyList(); + } + List<FileStatus> dataFiles = Lists.newArrayList(); + for (DataFile dataFile : new HiveIcebergOutputCommitter().getWrittenFiles(jobContextList)) { + FileSystem fs = new Path(dataFile.path().toString()).getFileSystem(conf); + dataFiles.add(fs.getFileStatus(new Path(dataFile.path().toString()))); + } + return dataFiles; + } + + @Override + public StorageHandlerMergeProperties getStorageHandlerMergeProperties(Properties properties) { Review Comment: Renamed. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: gitbox-unsubscr...@hive.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: gitbox-unsubscr...@hive.apache.org For additional commands, e-mail: gitbox-h...@hive.apache.org