117 lines
3.0 KiB
Java
117 lines
3.0 KiB
Java
/*
|
||
* To change this license header, choose License Headers in Project Properties.
|
||
* To change this template file, choose Tools | Templates
|
||
* and open the template in the editor.
|
||
*/
|
||
package org.redkale.net.http;
|
||
|
||
import java.io.*;
|
||
import java.util.concurrent.atomic.AtomicLong;
|
||
|
||
/**
|
||
*
|
||
* <p> 详情见: http://www.redkale.org
|
||
* @author zhangjx
|
||
*/
|
||
public final class MultiPart {
|
||
|
||
private final String filename;
|
||
|
||
private final String name;
|
||
|
||
private final String contentType;
|
||
|
||
private final InputStream in;
|
||
|
||
private final AtomicLong received;
|
||
|
||
MultiPart(String filename, String name, String contentType, AtomicLong received, InputStream in) {
|
||
this.filename = filename;
|
||
this.name = name;
|
||
this.in = in;
|
||
this.contentType = contentType;
|
||
this.received = received;
|
||
}
|
||
|
||
@Override
|
||
public String toString() {
|
||
return this.getClass().getSimpleName() + "{" + "name=" + name + ", filename=" + filename + ", contentType=" + contentType + ", received=" + received + '}';
|
||
}
|
||
|
||
public boolean save(File file) throws IOException {
|
||
return save(Long.MAX_VALUE, file);
|
||
}
|
||
|
||
public boolean save(long max, File file) throws IOException {
|
||
OutputStream out = new FileOutputStream(file);
|
||
boolean rs = save(max, out);
|
||
out.close();
|
||
return rs;
|
||
}
|
||
|
||
public byte[] getContentBytes() throws IOException {
|
||
return getContentBytes(Long.MAX_VALUE);
|
||
}
|
||
|
||
/**
|
||
* 将文件流读进bytes， 如果超出max指定的值则返回null
|
||
*
|
||
* @param max 最大长度限制
|
||
* @return 内容
|
||
* @throws IOException 异常
|
||
*/
|
||
public byte[] getContentBytes(long max) throws IOException {
|
||
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
||
return save(max, out) ? out.toByteArray() : null;
|
||
}
|
||
|
||
public boolean save(OutputStream out) throws IOException {
|
||
return save(Long.MAX_VALUE, out);
|
||
}
|
||
|
||
/**
|
||
* 将文件流写进out， 如果超出max指定的值则中断并返回false
|
||
*
|
||
* @param max 最大长度限制
|
||
* @param out 输出流
|
||
* @return 是否成功
|
||
* @throws IOException 异常
|
||
*/
|
||
public boolean save(long max, OutputStream out) throws IOException {
|
||
byte[] bytes = new byte[4096];
|
||
int pos;
|
||
InputStream in0 = this.getInputStream();
|
||
while ((pos = in0.read(bytes)) != -1) {
|
||
if (max < 0) return false;
|
||
out.write(bytes, 0, pos);
|
||
max -= pos;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
public String getContentType() {
|
||
return contentType;
|
||
}
|
||
|
||
public String getFilename() {
|
||
return filename;
|
||
}
|
||
|
||
public String getName() {
|
||
return name;
|
||
}
|
||
|
||
public InputStream getInputStream() {
|
||
return in;
|
||
}
|
||
|
||
public long getReceived() {
|
||
return received.get();
|
||
}
|
||
|
||
public void skip() throws IOException {
|
||
in.skip(Long.MAX_VALUE);
|
||
}
|
||
|
||
}
|