| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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 org.apache.commons.vfs2.provider.http5; |
| |
| import java.io.DataInputStream; |
| import java.io.FilterInputStream; |
| import java.io.IOException; |
| import java.net.HttpURLConnection; |
| |
| import org.apache.commons.vfs2.FileSystemException; |
| import org.apache.commons.vfs2.provider.AbstractRandomAccessStreamContent; |
| import org.apache.commons.vfs2.util.MonitorInputStream; |
| import org.apache.commons.vfs2.util.RandomAccessMode; |
| import org.apache.hc.client5.http.classic.methods.HttpGet; |
| import org.apache.hc.core5.http.ClassicHttpResponse; |
| |
| /** |
| * RandomAccess content using {@code Http5FileObject}. |
| */ |
| final class Http5RandomAccessContent<FS extends Http5FileSystem> extends AbstractRandomAccessStreamContent { |
| |
| protected long filePointer; |
| |
| private final Http5FileObject<FS> fileObject; |
| |
| private DataInputStream dataInputStream; |
| private MonitorInputStream monitorInputStream; |
| |
| Http5RandomAccessContent(final Http5FileObject<FS> fileObject, final RandomAccessMode mode) { |
| super(mode); |
| this.fileObject = fileObject; |
| } |
| |
| @Override |
| public void close() throws IOException { |
| if (dataInputStream != null) { |
| dataInputStream.close(); |
| dataInputStream = null; |
| monitorInputStream = null; |
| } |
| } |
| |
| @Override |
| protected DataInputStream getDataInputStream() throws IOException { |
| if (dataInputStream != null) { |
| return dataInputStream; |
| } |
| |
| final HttpGet httpGet = new HttpGet(fileObject.getInternalURI()); |
| httpGet.setHeader("Range", "bytes=" + filePointer + "-"); |
| final ClassicHttpResponse httpResponse = fileObject.executeHttpUriRequest(httpGet); |
| final int status = httpResponse.getCode(); |
| |
| if (status != HttpURLConnection.HTTP_PARTIAL && status != HttpURLConnection.HTTP_OK) { |
| throw new FileSystemException("vfs.provider.http/get-range.error", fileObject.getName(), |
| Long.valueOf(filePointer), Integer.valueOf(status)); |
| } |
| |
| monitorInputStream = new MonitoredHttpResponseContentInputStream(httpResponse); |
| |
| // If the range request was ignored |
| if (status == HttpURLConnection.HTTP_OK) { |
| final long skipped = monitorInputStream.skip(filePointer); |
| if (skipped != filePointer) { |
| throw new FileSystemException("vfs.provider.http/get-range.error", fileObject.getName(), |
| Long.valueOf(filePointer), Integer.valueOf(status)); |
| } |
| } |
| |
| dataInputStream = new DataInputStream(new FilterInputStream(monitorInputStream) { |
| @Override |
| public int read() throws IOException { |
| final int ret = super.read(); |
| if (ret > -1) { |
| filePointer++; |
| } |
| return ret; |
| } |
| |
| @Override |
| public int read(final byte[] b) throws IOException { |
| final int ret = super.read(b); |
| if (ret > -1) { |
| filePointer += ret; |
| } |
| return ret; |
| } |
| |
| @Override |
| public int read(final byte[] b, final int off, final int len) throws IOException { |
| final int ret = super.read(b, off, len); |
| if (ret > -1) { |
| filePointer += ret; |
| } |
| return ret; |
| } |
| }); |
| |
| return dataInputStream; |
| } |
| |
| @Override |
| public long getFilePointer() throws IOException { |
| return filePointer; |
| } |
| |
| @Override |
| public long length() throws IOException { |
| return fileObject.getContent().getSize(); |
| } |
| |
| @Override |
| public void seek(final long pos) throws IOException { |
| if (pos == filePointer) { |
| // no change |
| return; |
| } |
| |
| if (pos < 0) { |
| throw new FileSystemException("vfs.provider/random-access-invalid-position.error", Long.valueOf(pos)); |
| } |
| |
| if (dataInputStream != null) { |
| close(); |
| } |
| |
| filePointer = pos; |
| } |
| } |