package com.vyloy.server;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.ReadableByteChannel;
import java.nio.charset.Charset;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import com.vyloy.server.db.File;
import com.vyloy.server.db.service.FileService;
import com.vyloy.server.fileupload.FileUpload;
/**
* Servlet implementation class UploadServlet
*/
public class UploadServlet extends HttpServlet {
private static final long serialVersionUID = 1L;
/**
* @see HttpServlet#HttpServlet()
*/
public UploadServlet() {
super();
}
/**
* @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)
*/
protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
ByteBuffer result = ByteBuffer.allocate(3*1024*1024);
try {
ReadableByteChannel in = Channels.newChannel(request
.getInputStream());
ByteBuffer tempbytes = ByteBuffer.allocate(1024*1024);
while ((in.read(tempbytes)) != -1) {
tempbytes.flip();
//重点--比较临时字节数是否比输出字节缓存剩余容量大
if(result.remaining()<tempbytes.limit()){
ByteBuffer _result=ByteBuffer.allocate(result.capacity()*2);
result.flip();
_result.put(result);
result=_result;
}
result.put(tempbytes);
tempbytes.clear();
}
result.flip();
} catch (IOException e) {
e.printStackTrace();
response.getWriter().write("0");
return;
}
try{
//重点--result.array()会返回底层数组,如果有效数据少于容量,
//则数据中会包含很多没用的值为0的字节
//所以这里创建一个新的数组再把有效数据存入
byte[] bytes=new byte[result.limit()];
result.get(bytes);
result.clear();
FileUpload fileUpload = new FileUpload(bytes);
String fileName=new String(fileUpload.get("Filename"),Charset.forName("utf-8"));
byte[] data=fileUpload.get("Filedata");
File file = new File(fileName);
FileService.getInstance().save(file, data);
response.getWriter().write("1");
}catch(Exception e){
e.printStackTrace();
response.getWriter().write("0");
}
}
}
不变式
标记、位置、限制和容量值遵守以下不变式:
0 <= 标记 <= 位置 <= 限制 <= 容量
新创建的缓冲区总有一个 0 位置和一个未定义的标记。初始限制可以为 0,也可以为其他值,这取决于缓冲区类型及其构建方式。一般情况下,缓冲区的初始内容是未定义的。
清除、反转和重绕
除了访问位置、限制、容量值的方法以及做标记和重置的方法外,此类还定义了以下可对缓冲区进行的操作:
clear() 使缓冲区为一系列新的通道读取或相对放置 操作做好准备:它将限制设置为容量大小,将位置设置为 0。
flip() 使缓冲区为一系列新的通道写入或相对获取 操作做好准备:它将限制设置为当前位置,然后将位置设置为 0。
rewind() 使缓冲区为重新读取已包含的数据做好准备:它使限制保持不变,将位置设置为 0。