Java文件下载限速

0

之前一直使用的Spring提供的ResponseEntity来做下载,主要代码如下:

headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
headers.setContentDispositionFormData("attachment", order.getSn() + "." + FilenameUtils.getExtension(file.getName()));
return new ResponseEntity<byte[]>(FileUtils.readFileToByteArray(file), headers, HttpStatus.OK);

上面的代码有一些问题,就是ResponseEntity加载的时候是一次性把整个文件load到内存。
那么就可能出现一些问题,例如几个用户同时下载一个200M的文件那么内存消耗非常快,而且没办法限速,也许是我没找到。

那么怎么限速呢?直接贴上代码:

response.setContentType("application/octet-stream");;
response.setHeader("Content-Disposition", "attachment; filename=" + order.getSn() + "." + FilenameUtils.getExtension(file.getName()));
OutputStream os = response.getOutputStream();
InputStream is = new FileInputStream(file);
response.setContentLength(is.available()); // 显示文件大小
byte[] bytes = new byte[128 * 1024];
int index = 0;
while((index = is.read(bytes)) != -1) {
    try {
        Thread.sleep(500);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    os.write(bytes, 0, index);
}
is.close();

这里主要使用线程休眠来达到控制速度的,代码请自行优化。

优化思路:下载人数少或者下载的文件比较大则把休眠时间调小等等