springboot 多线程下载

以下是一个使用Spring Boot实现多线程下载文件的示例代码：

@RestController
public class DownloadController {

    private static final int BUFFER_SIZE = 4096;

    @GetMapping("/download")
    public ResponseEntity<Resource> downloadFile(@RequestParam String fileUrl) throws IOException {
        URL url = new URL(fileUrl);
        HttpURLConnection httpConn = (HttpURLConnection) url.openConnection();
        int responseCode = httpConn.getResponseCode();

        if (responseCode == HttpURLConnection.HTTP_OK) {
            String fileName = "";
            String disposition = httpConn.getHeaderField("Content-Disposition");
            String contentType = httpConn.getContentType();
            int contentLength = httpConn.getContentLength();

            if (disposition != null) {
                int index = disposition.indexOf("filename=");
                if (index > 0) {
                    fileName = disposition.substring(index + 10, disposition.length() - 1);
                }
            } else {
                fileName = fileUrl.substring(fileUrl.lastIndexOf("/") + 1, fileUrl.length());
            }

            InputStream inputStream = httpConn.getInputStream();
            ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
            byte[] buffer = new byte[BUFFER_SIZE];
            int bytesRead = -1;

            while ((bytesRead = inputStream.read(buffer)) != -1) {
                outputStream.write(buffer, 0, bytesRead);
            }

            byte[] bytes = outputStream.toByteArray();
            HttpHeaders headers = new HttpHeaders();
            headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + fileName);
            headers.add(HttpHeaders.CONTENT_TYPE, contentType);
            headers.add(HttpHeaders.CONTENT_LENGTH, String.valueOf(contentLength));
            Resource resource = new ByteArrayResource(bytes);

            return ResponseEntity.ok()
                    .headers(headers)
                    .contentLength(bytes.length)
                    .contentType(MediaType.parseMediaType(contentType))
                    .body(resource);
        } else {
            throw new RuntimeException("Failed to download file: " + fileUrl);
        }
    }
}

该示例代码使用了Spring Boot的@RestController注解，将下载文件的方法映射到了“/download”路径上。该方法接收一个名为“fileUrl”的请求参数，该参数指定了要下载的文件的URL地址。

在方法中，我们首先使用Java标准库中的HttpURLConnection类打开了一个HTTP连接，并获取了响应码。如果响应码为HTTP_OK（即200），则说明连接成功，我们可以开始下载文件。

接下来，我们从HTTP连接中获取了文件的一些元数据，例如文件名、文件类型和文件大小等。如果响应头中包含了Content-Disposition字段，则说明服务器希望我们将文件保存为指定的文件名；否则，我们将使用URL中的文件名作为文件名。

然后，我们从HTTP连接中获取了文件的输入流，并使用Java标准库中的ByteArrayOutputStream类将文件内容读入内存中。在读取文件内容时，我们使用了一个大小为4096字节的缓冲区，以提高读取效率。

最后，我们将文件内容封装为一个Spring框架中的Resource对象，并将该对象作为响应体返回给客户端。在响应头中，我们设置了Content-Disposition、Content-Type和Content-Length等字段，以便客户端正确地处理响应。