protected void doPost(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
int flowChunkNumber = getFlowChunkNumber(request);
FlowInfo info = getFlowInfo(request);
RandomAccessFile raf = new RandomAccessFile(info.flowFilePath, "rw");
// Seek to position
long offset = ((long) (flowChunkNumber - 1)) * info.flowChunkSize;
raf.seek(offset);
// Save to file
InputStream is = request.getInputStream();
long readed = 0;
long content_length = request.getContentLength();
byte[] bytes = new byte[1024 * 100];
while (readed < content_length) {
int r = is.read(bytes);
if (r < 0) {
break;
}
raf.write(bytes, 0, r);
readed += r;
}
raf.close();
// Mark as uploaded.
info.uploadedChunks.add(new FlowInfo.FlowChunkNumber(flowChunkNumber));
// Print the number
System.out.println(
"Chunk #" + flowChunkNumber + " of " + info.flowFilename + " has been uploaded.");
if (info.checkIfUploadFinished()) { // Check if all chunks uploaded, and change filename
FlowInfoStorage.getInstance().remove(info);
response.getWriter().print("All finished.");
getOptionIsoName = GetIso();
list(info.flowFilename, getOptionIsoName);
// System.out.println(info.flowFilename + " is completed.");
} else {
response.getWriter().print("Upload");
}
}
private FlowInfo getFlowInfo(HttpServletRequest request) throws ServletException {
String base_dir = UPLOAD_DIR;
int flowChunkSize = HttpUtils.toInt(request.getParameter("flowChunkSize"), -1);
long flowTotalSize = HttpUtils.toLong(request.getParameter("flowTotalSize"), -1);
String flowIdentifier = request.getParameter("flowIdentifier");
String flowFilename = request.getParameter("flowFilename");
String flowRelativePath = request.getParameter("flowRelativePath");
// DEBUG
// System.out.println(
// "base_dir = " + base_dir + "\n"
// + "flowChunkSize = " + flowChunkSize + "\n"
// + "flowTotalSize = " + flowTotalSize + "\n"
// + "flowIdentifier = " + flowIdentifier + "\n"
// + "flowFilename = " + flowFilename + "\n"
// + "flowRelativePath = " + flowRelativePath
// );
// !DEBUG
File flowFile = new File(base_dir, flowFilename);
// Create the folder if it doesn't exist yet
flowFile.getParentFile().mkdirs();
// Here we add a ".temp" to every upload file to indicate NON-FINISHED
String flowFilePath = flowFile.getAbsolutePath() + ".temp";
FlowInfoStorage storage = FlowInfoStorage.getInstance();
FlowInfo info =
storage.get(
flowChunkSize,
flowTotalSize,
flowIdentifier,
flowFilename,
flowRelativePath,
flowFilePath);
if (!info.vaild()) {
storage.remove(info);
throw new ServletException("Invalid request params.");
}
return info;
}