com.qiniu.api.resumableio.RandomAccessFileUploadBlock Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sdk Show documentation

Qiniu Resource (Cloud) Storage SDK demo for Java

There is a newer version: 6.1.9

package com.qiniu.api.resumableio;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.RandomAccessFile;
import java.util.concurrent.locks.Lock;

import org.apache.http.HttpEntity;
import org.apache.http.client.HttpClient;
import org.apache.http.entity.AbstractHttpEntity;


public class RandomAccessFileUploadBlock extends UploadBlock {
	protected RandomAccessFile file;
	private final Lock fileUploadLock;

	RandomAccessFileUploadBlock(SliceUpload sliceUpload,
			HttpClient httpClient, String host, int blockIdx, long offset,
			int len, RandomAccessFile file, Lock fileUploadLock) {
		super(sliceUpload, httpClient, host, blockIdx, offset, len);
		this.file = file;
		this.fileUploadLock = fileUploadLock;
	}

	@Override
	protected HttpEntity buildHttpEntity(final int start, final int len) {
		AbstractHttpEntity entity = new AbstractHttpEntity() {
			private boolean consumed = false;
			private long length = len;

			@Override
			public boolean isRepeatable() {
				return true;
			}

			@Override
			public long getContentLength() {
				return length;
			}

			@Override
			public InputStream getContent() throws IOException,
					IllegalStateException {
				return null;
			}

			@Override
			public void writeTo(OutputStream os) throws IOException {
				consumed = false;
				try {
					byte[] b = new byte[1024 * 4];
					int len = -1;
					long off = offset + start;

					fileUploadLock.lock();
					file.seek(off);
					while (true) {
						len = file.read(b);
						if (len == -1) {
							break;
						}

						os.write(b, 0, len);
						off += len;
					}
					os.flush();
				} finally {
					fileUploadLock.unlock();
					consumed = true;
				}
			}

			@Override
			public boolean isStreaming() {
				return !consumed;
			}
		};
		entity.setContentType("application/octet-stream");
		return entity;
	}

	@Override
	protected long buildCrc32(int start, int len) {
		return Util.crc32(copy2New(start, len));
	}

	private byte[] copy2New(int start, int len) {
		byte[] data = new byte[len];
		try {
			long off = offset + start;
			fileUploadLock.lock();
			file.seek(off);
			file.read(data);
			return data;
		} catch (IOException e) {
			throw new RuntimeException(e);
		} finally {
			fileUploadLock.unlock();
		}
	}

	@Override
	protected void clean() {

	}

}