pavibhai commented on code in PR #6293:
URL: https://github.com/apache/iceberg/pull/6293#discussion_r1042005551
##########
orc/src/main/java/org/apache/iceberg/orc/ORC.java:
##########
@@ -789,7 +808,210 @@ static Reader newFileReader(InputFile file, Configuration
config) {
ReaderOptions readerOptions =
OrcFile.readerOptions(config).useUTCTimestamp(true);
if (file instanceof HadoopInputFile) {
readerOptions.filesystem(((HadoopInputFile) file).getFileSystem());
+ } else {
+ readerOptions.filesystem(new
InputFileSystem(file)).maxLength(file.getLength());
}
return newFileReader(file.location(), readerOptions);
}
+
+ static Writer newFileWriter(
+ OutputFile file, OrcFile.WriterOptions options, Map<String, byte[]>
metadata) {
+ if (file instanceof HadoopOutputFile) {
+ options.fileSystem(((HadoopOutputFile) file).getFileSystem());
+ } else {
+ options.fileSystem(new OutputFileSystem(file));
+ }
+ final Path locPath = new Path(file.location());
+ final Writer writer;
+
+ try {
+ writer = OrcFile.createWriter(locPath, options);
+ } catch (IOException ioe) {
+ throw new RuntimeIOException(ioe, "Can't create file %s", locPath);
+ }
+
+ metadata.forEach((key, value) -> writer.addUserMetadata(key,
ByteBuffer.wrap(value)));
+
+ return writer;
+ }
+
+ private static class WrappedSeekableInputStream extends FSInputStream {
+ private final SeekableInputStream inputStream;
+ private boolean closed;
+ private final StackTraceElement[] createStack;
+
+ private WrappedSeekableInputStream(SeekableInputStream inputStream) {
+ this.inputStream = inputStream;
+ this.createStack = Thread.currentThread().getStackTrace();
+ this.closed = false;
+ }
+
+ @Override
+ public void seek(long pos) throws IOException {
+ inputStream.seek(pos);
+ }
+
+ @Override
+ public long getPos() throws IOException {
+ return inputStream.getPos();
+ }
+
+ @Override
+ public boolean seekToNewSource(long targetPos) throws IOException {
+ throw new UnsupportedOperationException("seekToNewSource not supported");
+ }
+
+ @Override
+ public int read() throws IOException {
+ return inputStream.read();
+ }
+
+ @Override
+ public int read(@NotNull byte[] b, int off, int len) throws IOException {
+ return inputStream.read(b, off, len);
+ }
+
+ @Override
+ public void close() throws IOException {
+ inputStream.close();
+ closed = true;
+ }
+
+ @SuppressWarnings("checkstyle:NoFinalizer")
+ @Override
+ protected void finalize() throws Throwable {
+ super.finalize();
+ if (!closed) {
+ close(); // releasing resources is more important than printing the
warning
+ String trace =
+ Joiner.on("\n\t").join(Arrays.copyOfRange(createStack, 1,
createStack.length));
+ LOG.warn("Unclosed input stream created by:\n\t{}", trace);
+ }
+ }
+ }
+
+ private static class NullFileSystem extends FileSystem {
Review Comment:
Sure. Moved to FileIOFSUtil class
--
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: issues-unsubscr...@iceberg.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@iceberg.apache.org
For additional commands, e-mail: issues-h...@iceberg.apache.org
