com.qiniu.api.io.IoApi Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdk Show documentation
Show all versions of sdk Show documentation
Qiniu Resource (Cloud) Storage SDK demo for Java
package com.qiniu.api.io;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.nio.charset.Charset;
import java.util.Map;
import java.util.zip.CRC32;
import java.util.zip.CheckedInputStream;
import org.apache.http.entity.mime.MultipartEntity;
import org.apache.http.entity.mime.content.AbstractContentBody;
import org.apache.http.entity.mime.content.FileBody;
import org.apache.http.entity.mime.content.InputStreamBody;
import org.apache.http.entity.mime.content.StringBody;
import com.qiniu.api.config.Config;
import com.qiniu.api.net.CallRet;
import com.qiniu.api.net.Client;
public class IoApi {
public static final String UNDEFINED_KEY = null;
public static final int NO_CRC32 = 0;
public static final int AUTO_CRC32 = 1;
public static final int WITH_CRC32 = 2;
private static PutRet put(String uptoken, String key, File file,
PutExtra extra) {
if (!file.exists() || !file.canRead()) {
return new PutRet(new CallRet(Config.ERROR_CODE, new Exception(
"File does not exist or not readable.")));
}
extra = extra == null ? new PutExtra() : extra;
MultipartEntity requestEntity = new MultipartEntity();
try {
requestEntity.addPart("token", new StringBody(uptoken));
AbstractContentBody fileBody = buildFileBody(file, extra);
requestEntity.addPart("file", fileBody);
setKey(requestEntity, key);
setParam(requestEntity, extra.params);
if (extra.checkCrc != NO_CRC32) {
if (extra.crc32 == 0) {
return new PutRet(new CallRet(Config.ERROR_CODE, new Exception("no crc32 specified!")));
}
requestEntity.addPart("crc32", new StringBody(extra.crc32 + ""));
}
if (extra.params != null) {
for (Map.Entry xvar : extra.params.entrySet()) {
requestEntity.addPart(xvar.getKey(), new StringBody(xvar.getValue(), Charset.forName(Config.CHARSET)));
}
}
} catch (Exception e) {
e.printStackTrace();
return new PutRet(new CallRet(Config.ERROR_CODE, e));
}
String url = Config.UP_HOST;
CallRet ret = new Client().callWithMultiPart(url, requestEntity);
return new PutRet(ret);
}
private static FileBody buildFileBody(File file,PutExtra extra){
if(extra.mimeType != null){
return new FileBody(file, extra.mimeType);
}else{
return new FileBody(file);
}
}
private static void setKey(MultipartEntity requestEntity, String key) throws UnsupportedEncodingException{
if(key != null){
requestEntity.addPart("key", new StringBody(key,Charset.forName(Config.CHARSET)));
}
}
private static void setParam(MultipartEntity requestEntity, Map params) throws UnsupportedEncodingException{
if(params == null){
return;
}
for(String name : params.keySet()){
requestEntity.addPart(name, new StringBody(params.get(name),Charset.forName(Config.CHARSET)));
}
}
private static PutRet putStream(String uptoken, String key, InputStream reader,
PutExtra extra, long length) {
extra = extra == null ? new PutExtra() : extra;
MultipartEntity requestEntity = new MultipartEntity();
try {
requestEntity.addPart("token", new StringBody(uptoken));
String fileName = key != null ? key : "null";
AbstractContentBody inputBody = buildInputStreamBody(reader, extra, fileName, length);
requestEntity.addPart("file", inputBody);
setKey(requestEntity, key);
setParam(requestEntity, extra.params);
if (extra.checkCrc != NO_CRC32) {
if (extra.crc32 == 0) {
return new PutRet(new CallRet(Config.ERROR_CODE, new Exception("no crc32 specified!")));
}
requestEntity.addPart("crc32", new StringBody(extra.crc32 + ""));
}
} catch (Exception e) {
e.printStackTrace();
return new PutRet(new CallRet(Config.ERROR_CODE, e));
}
String url = Config.UP_HOST;
CallRet ret = new Client().callWithMultiPart(url, requestEntity);
return new PutRet(ret);
}
private static AbstractContentBody buildInputStreamBody(InputStream reader,
PutExtra extra, String fileName, final long length){
if(extra.mimeType != null){
return new InputStreamBody(reader, extra.mimeType, fileName){
public long getContentLength() {
return length;
}
};
}else{
return new InputStreamBody(reader, fileName){
public long getContentLength() {
return length;
}
};
}
}
private static PutRet putStream0(String uptoken, String key, InputStream reader,
PutExtra extra, long length){
length = length <= 0 ? getLength(reader) : length;
if(length != -1) {
return putStream(uptoken,key,reader,extra, length);
}else{
return toPutFile(uptoken, key, reader, extra);
}
}
private static long getLength(InputStream is){
try {
return is.available();
} catch (Exception e) {
return -1;
}
}
private static PutRet toPutFile(String uptoken, String key,
InputStream reader, PutExtra extra) {
File file = null;
try{
file = copyToTmpFile(reader);
return put(uptoken, key, file, extra);
}finally{
if(file != null){
try{file.delete();}catch(Exception e){}
}
}
}
private static File copyToTmpFile(InputStream from){
FileOutputStream os = null;
try{
File to = File.createTempFile("qiniu_", ".tmp");
os = new FileOutputStream(to);
byte[] b = new byte[64 * 1024];
int l;
while ((l = from.read(b)) != -1) {
os.write(b, 0, l);
}
os.flush();
return to;
}catch(Exception e){
throw new RuntimeException("create tmp file failed.", e);
}finally{
if (os != null){
try{os.close();}catch(Exception e){}
}
if (from != null){
try{from.close();}catch(Exception e){}
}
}
}
/**
* @param uptoken
* @param key
* @param reader
* @param extra
* @param length 部分流 is.available() == 0，此时可指定内容长度
* @return
*/
public static PutRet Put(String uptoken,String key,InputStream reader,PutExtra extra, long length){
return putStream0(uptoken,key,reader,extra, length);
}
public static PutRet Put(String uptoken,String key,InputStream reader,PutExtra extra)
{
return Put(uptoken,key,reader,extra, -1);
}
public static PutRet putFile(String uptoken, String key, String fileName, PutExtra extra) {
File file=new File(fileName);
return putFile(uptoken, key, file, extra);
}
public static PutRet putFile(String uptoken, String key, File file, PutExtra extra) {
if (extra.checkCrc == AUTO_CRC32) {
try {
extra.crc32 = getCRC32(file);
} catch (Exception e) {
return new PutRet(new CallRet(Config.ERROR_CODE, e));
}
}
return put(uptoken, key, file, extra);
}
private static long getCRC32(File file) throws Exception {
CRC32 crc32 = new CRC32();
FileInputStream in = null;
CheckedInputStream checkedInputStream = null;
long crc = 0;
try {
in = new FileInputStream(file);
checkedInputStream = new CheckedInputStream(in, crc32);
while (checkedInputStream.read() != -1) {
}
crc = crc32.getValue();
} finally {
if (in != null) {
in.close();
in = null;
}
if (checkedInputStream != null) {
checkedInputStream.close();
checkedInputStream = null;
}
}
return crc;
}
}