13 changed files with 239 additions and 31 deletions
@ -0,0 +1,121 @@ |
|||||||
|
package com.biutag.supervision.util; |
||||||
|
|
||||||
|
import com.biutag.supervision.service.FileService; |
||||||
|
import jakarta.servlet.http.HttpServletRequest; |
||||||
|
import jakarta.servlet.http.HttpServletResponse; |
||||||
|
|
||||||
|
import java.io.*; |
||||||
|
import java.net.HttpURLConnection; |
||||||
|
import java.net.URL; |
||||||
|
import java.net.URLEncoder; |
||||||
|
import java.nio.file.*; |
||||||
|
import java.util.ArrayList; |
||||||
|
import java.util.List; |
||||||
|
import java.util.zip.ZipEntry; |
||||||
|
import java.util.zip.ZipOutputStream; |
||||||
|
|
||||||
|
public class DownloadAndZip { |
||||||
|
|
||||||
|
|
||||||
|
// 下载文件并保存到指定目录
|
||||||
|
public static void downloadFile(InputStream inputStream, String savePath) throws IOException { |
||||||
|
|
||||||
|
|
||||||
|
try ( |
||||||
|
FileOutputStream outputStream = new FileOutputStream(savePath)) { |
||||||
|
|
||||||
|
byte[] buffer = new byte[1024]; |
||||||
|
int bytesRead; |
||||||
|
while ((bytesRead = inputStream.read(buffer)) != -1) { |
||||||
|
outputStream.write(buffer, 0, bytesRead); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 将文件移动到指定文件夹
|
||||||
|
public static void moveFileToFolder(String sourcePath, String folderName) throws IOException { |
||||||
|
Path source = Paths.get(sourcePath); |
||||||
|
Path targetFolder = Paths.get("downloads/" + folderName); |
||||||
|
Path target = targetFolder.resolve(source.getFileName()); |
||||||
|
|
||||||
|
// 创建目标文件夹（如果不存在）
|
||||||
|
if (!Files.exists(targetFolder)) { |
||||||
|
Files.createDirectories(targetFolder); |
||||||
|
} |
||||||
|
|
||||||
|
// 移动文件
|
||||||
|
Files.move(source, target, StandardCopyOption.REPLACE_EXISTING); |
||||||
|
} |
||||||
|
|
||||||
|
// 将多个文件夹打包成zip
|
||||||
|
public static void zipFolders(List<String> folderNames, String zipFilePath) throws IOException { |
||||||
|
try (FileOutputStream fos = new FileOutputStream(zipFilePath); |
||||||
|
ZipOutputStream zos = new ZipOutputStream(fos)) { |
||||||
|
|
||||||
|
for (String folderName : folderNames) { |
||||||
|
Path folderPath = Paths.get("downloads/" + folderName); |
||||||
|
zipFolder(folderPath, folderName, zos); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public static void doGet(HttpServletResponse response,String filePath,String zipName) throws IOException { |
||||||
|
// 从请求参数中获取文件路径（这里假设文件路径是通过请求参数传递的，实际应用中可能需要更安全的方式）
|
||||||
|
|
||||||
|
if (filePath == null || filePath.isEmpty()) { |
||||||
|
response.sendError(HttpServletResponse.SC_BAD_REQUEST, "File path is required"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// 校验文件路径的合法性（这里只是简单示例，实际应用中需要更严格的校验）
|
||||||
|
File file = new File(filePath); |
||||||
|
if (!file.exists() || !file.isFile()) { |
||||||
|
response.sendError(HttpServletResponse.SC_NOT_FOUND, "File not found"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// 获取MIME类型
|
||||||
|
String mimeType = "application/octet-stream"; |
||||||
|
|
||||||
|
// 设置响应内容类型
|
||||||
|
response.setContentType(mimeType); |
||||||
|
response.setContentLength((int) file.length()); |
||||||
|
|
||||||
|
// 设置Content-Disposition头以提示浏览器下载文件
|
||||||
|
String headerKey = "Content-Disposition"; |
||||||
|
String headerValue = String.format("attachment; filename=\"%s\"", URLEncoder.encode(zipName, "UTF-8")); |
||||||
|
response.setHeader(headerKey, headerValue); |
||||||
|
|
||||||
|
// 读取文件内容并写入响应输出流
|
||||||
|
try (FileInputStream inputStream = new FileInputStream(file); |
||||||
|
OutputStream outputStream = response.getOutputStream()) { |
||||||
|
|
||||||
|
byte[] buffer = new byte[1024]; |
||||||
|
int bytesRead; |
||||||
|
while ((bytesRead = inputStream.read(buffer)) != -1) { |
||||||
|
outputStream.write(buffer, 0, bytesRead); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private static void zipFolder(Path folderPath, String parentFolderName, ZipOutputStream zos) throws IOException { |
||||||
|
for (File file : folderPath.toFile().listFiles()) { |
||||||
|
Path filePath = folderPath.resolve(file.getName()); |
||||||
|
if (Files.isDirectory(filePath)) { |
||||||
|
zipFolder(filePath, parentFolderName + "/" + file.getName(), zos); |
||||||
|
} else { |
||||||
|
try (FileInputStream fis = new FileInputStream(filePath.toFile())) { |
||||||
|
ZipEntry zipEntry = new ZipEntry(parentFolderName + "/" + file.getName()); |
||||||
|
zos.putNextEntry(zipEntry); |
||||||
|
byte[] buffer = new byte[1024]; |
||||||
|
int length; |
||||||
|
while ((length = fis.read(buffer)) > 0) { |
||||||
|
zos.write(buffer, 0, length); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue