大文分片上传 多线程上传文件(接收端)

package com.controller;

import java.io.File;

import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.util.Date;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.io.FileUtils;
import org.apache.log4j.Logger;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import com.response.BaseResponse;

@Controller
public class ReceiveFileController {

	private static final Logger logger = Logger.getLogger(ReceiveFileController.class);
	@RequestMapping(value = "/receive/receivefile", method = RequestMethod.POST)
	/*
	 * 参数名称与前面客户端对应 multipartFile:传到后台的具体文件 blockNumber:将文件分成的总块数
	 * blockIndex:这次请求的是第几块 targetFilePath:要根据这个路径制定块文件的路径和生成文件的路径
	 */
	@ResponseBody
	public BaseResponse fileUpload(
			HttpServletResponse response,
			@RequestParam(value = "blockIndex", required = false) String blockIndex,
			@RequestParam(value = "blockNumber", required = false) String blockNumber,
			@RequestParam(value = "targetFilePath", required = false) String targetFilePath,
			@RequestParam(value = "randomUUID", required = false) String randomUUID,
			@RequestParam(value = "file", required = false) MultipartFile multipartFile)
			 {
		// 文件不走分块上传

		System.out.println(new Date());
		BaseResponse baseResponse = new BaseResponse();
		baseResponse.setCode(0);
		
		if (Integer.valueOf(blockNumber) == 1) {
			File destTempFile = new File(targetFilePath);
			File fileParent = destTempFile.getParentFile();
			if (!fileParent.exists()) {
				fileParent.mkdirs();
			}
			try {
				destTempFile.createNewFile();
				if (multipartFile == null) {
					baseResponse.setCode(0);
					return baseResponse;
				}
				DigestUtils.md5Hex(multipartFile.getInputStream());
				FileUtils.copyInputStreamToFile(multipartFile.getInputStream(),
						destTempFile);
				baseResponse.setCode(1);
				return baseResponse;
			} catch (IOException e) {
				logger.error("文件写入异常",e);
				
				System.out.println("文件写入异常");
				e.printStackTrace();
			}
		} else if (Integer.valueOf(blockNumber) > 1) {
			
			try {
				String fileRealName = targetFilePath;
				String fileIndexPath = targetFilePath + ".index";
				targetFilePath = targetFilePath + ".part";
				File destTempFile = new File(targetFilePath);
				File fileParent = destTempFile.getParentFile();

				if (!fileParent.exists()) {
					fileParent.mkdirs();
				}
				try {
					destTempFile.createNewFile();
				} catch (IOException e) {
					System.out.println("新建目录异常");
					e.printStackTrace();
				}
				
				fileConsistent(randomUUID,fileIndexPath,targetFilePath);
				byte[] buf = new byte[1024 * 1024 * 2];
				int len = 0;
				RandomAccessFile fileWrite = new RandomAccessFile(destTempFile,
						"rw");
				InputStream fileReader = multipartFile.getInputStream();
				fileWrite.seek((Integer.valueOf(blockIndex) - 1) * 104857600);
				while ((len = fileReader.read(buf)) != -1) {
					fileWrite.write(buf, 0, len);
				}
				fileReader.close();
				fileWrite.close();

				if (fileComplete(randomUUID, fileIndexPath,
						Integer.valueOf(blockIndex),
						Integer.valueOf(blockNumber))) {
					File file = new File(targetFilePath);
					file.renameTo(new File(fileRealName));
					baseResponse.setCode(1);
					return baseResponse;
				}

			} catch (IOException e) {
				System.out.println("part文件写入异常");
				logger.error("part文件写入异常",e);
				e.printStackTrace();
			}
		}
		return baseResponse;
	}
	public static boolean fileComplete(String randomUUID, String fileIndexPath,
			int index, int blockNumber) throws IOException {
		File fileIndex = new File(fileIndexPath);

		RandomAccessFile fileIndexWrite = new RandomAccessFile(fileIndex, "rw");

		fileIndexWrite.seek(randomUUID.length() + index * 4);
		String number = String.valueOf(index) + ",";
		fileIndexWrite.write(number.getBytes());
		fileIndexWrite.seek(randomUUID.length());
		byte[] buff = new byte[4];
		// 用于保存实际读取的字节数
		int hasRead = 0;
		// 循环读取
		String fileContent = "";
		while ((hasRead = fileIndexWrite.read(buff)) > 0) {
			fileContent = fileContent + new String(buff, 0, hasRead);
		}
		fileIndexWrite.close();
		String[] list = fileContent.split(",");
		System.out.println(list.length);
		if (list.length == blockNumber) {
			fileIndex.delete();
			return true;
		}
		return false;
	}

	public static void fileConsistent(String randomUUID, String fileIndexPath,
			String filepath) throws IOException {
		File fileIndex = new File(fileIndexPath);
		if (!fileIndex.exists()) {
			RandomAccessFile fileIndexWrite = new RandomAccessFile(fileIndex,
					"rw");
			fileIndexWrite.seek(0);
			fileIndexWrite.write(randomUUID.getBytes());
			fileIndexWrite.close();
		} else {
			RandomAccessFile fileIndexWrite = new RandomAccessFile(fileIndex,
					"rw");
			fileIndexWrite.seek(0);
			byte bys[] = new byte[36];
			fileIndexWrite.read(bys);
			String udil = new String(bys);
			fileIndexWrite.close();
			if (!udil.equals(randomUUID)) {
				fileIndex.delete();
				File file = new File(filepath);
				file.delete();
				RandomAccessFile fileIndexWriter = new RandomAccessFile(
						fileIndex, "rw");
				fileIndexWriter.seek(0);
				fileIndexWriter.write(randomUUID.getBytes());
				fileIndexWriter.close();
			}
			
		}
		
	}
 
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值