package cc.mrbird.febs.ai.controller.fileUpload;
|
|
import cc.mrbird.febs.common.controller.BaseController;
|
import cc.mrbird.febs.common.entity.FebsResponse;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.http.HttpStatus;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.nio.file.Files;
|
import java.nio.file.Path;
|
import java.nio.file.Paths;
|
import java.text.SimpleDateFormat;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.UUID;
|
|
/**
|
* @author Administrator
|
*/
|
@Slf4j
|
@RestController
|
@RequiredArgsConstructor
|
@RequestMapping(value = "/fileUpload")
|
public class FileUploadController extends BaseController {
|
|
// 基础上传目录
|
private String baseUploadDir = "/home/javaweb/webresource/ai";
|
|
// 分片存储目录
|
private String baseChunkDir = "/home/javaweb/webresource/ai/chunks";
|
|
/**
|
* 上传文件分片
|
*/
|
@PostMapping("/uploadChunk")
|
public FebsResponse uploadChunk(@RequestParam("file") MultipartFile file,
|
@RequestParam("fileName") String fileName,
|
@RequestParam("chunk") int chunk,
|
@RequestParam("chunks") int chunks,
|
@RequestParam("fileMd5") String fileMd5) {
|
try {
|
String companyId = getCurrentUserCompanyId();
|
// 构建公司专属分片目录
|
String chunkDir = baseChunkDir + "/" + companyId;
|
// 确保分片目录存在
|
Path chunkPath = Paths.get(chunkDir, fileMd5);
|
if (!Files.exists(chunkPath)) {
|
Files.createDirectories(chunkPath);
|
}
|
|
// 保存分片文件
|
Path chunkFilePath = chunkPath.resolve(chunk + ".part");
|
Files.write(chunkFilePath, file.getBytes());
|
|
return new FebsResponse().success().message("分片上传成功");
|
} catch (Exception e) {
|
e.printStackTrace();
|
return new FebsResponse().fail().message("分片上传失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 合并文件分片
|
*/
|
@PostMapping("/mergeChunks")
|
public FebsResponse mergeChunks(@RequestParam("fileName") String fileName,
|
@RequestParam("fileMd5") String fileMd5,
|
@RequestParam("chunks") int chunks) {
|
try {
|
|
String companyId = getCurrentUserCompanyId();
|
// 构建公司专属上传目录
|
String uploadDir = baseUploadDir + "/" + companyId;
|
// 确保上传目录存在
|
Path uploadPath = Paths.get(uploadDir);
|
if (!Files.exists(uploadPath)) {
|
Files.createDirectories(uploadPath);
|
}
|
|
// 生成唯一文件名
|
String uniqueFileName = UUID.randomUUID().toString() + "_" + fileName;
|
Path targetFilePath = uploadPath.resolve(uniqueFileName);
|
|
// 构建公司专属分片目录
|
String chunkDir = baseChunkDir + "/" + companyId;
|
// 确保分片目录存在
|
Path chunkPath = Paths.get(chunkDir, fileMd5);
|
if (!Files.exists(chunkPath)) {
|
return new FebsResponse().fail().message("分片目录不存在");
|
}
|
|
// 合并分片
|
try (FileOutputStream outputStream = new FileOutputStream(targetFilePath.toFile())) {
|
for (int i = 0; i < chunks; i++) {
|
Path chunkFilePath = chunkPath.resolve(i + ".part");
|
if (!Files.exists(chunkFilePath)) {
|
return new FebsResponse().fail().message("分片文件不存在: " + chunkFilePath);
|
}
|
byte[] chunkBytes = Files.readAllBytes(chunkFilePath);
|
outputStream.write(chunkBytes);
|
// 删除已合并的分片
|
Files.deleteIfExists(chunkFilePath);
|
}
|
}
|
|
// 删除分片目录
|
Files.deleteIfExists(chunkPath);
|
|
// 保存文件信息到数据库或文件系统（这里简化处理）
|
saveFileInfo(uniqueFileName, targetFilePath.toFile().length());
|
|
return new FebsResponse().success().message("文件上传成功").data(uniqueFileName);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return new FebsResponse().fail().message("文件合并失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 播放视频文件
|
*/
|
@GetMapping("/play/{fileName}")
|
public void playVideo(@PathVariable("fileName") String fileName, HttpServletResponse response) {
|
try {
|
String companyId = getCurrentUserCompanyId();
|
String uploadDir = baseUploadDir + "/" + companyId;
|
Path filePath = Paths.get(uploadDir, fileName);
|
if (!Files.exists(filePath)) {
|
response.setStatus(HttpStatus.NOT_FOUND.value());
|
return;
|
}
|
|
// 设置响应头
|
response.setContentType("video/mp4");
|
response.setContentLengthLong(Files.size(filePath));
|
response.setHeader("Content-Disposition", "inline; filename=\"" + fileName + "\"");
|
|
// 流式传输文件
|
try (InputStream inputStream = Files.newInputStream(filePath);
|
OutputStream outputStream = response.getOutputStream()) {
|
byte[] buffer = new byte[8192];
|
int bytesRead;
|
while ((bytesRead = inputStream.read(buffer)) != -1) {
|
outputStream.write(buffer, 0, bytesRead);
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
response.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value());
|
}
|
}
|
|
/**
|
* 获取文件列表
|
*/
|
@GetMapping("/list")
|
public FebsResponse getFileList() {
|
try {
|
|
String companyId = getCurrentUserCompanyId();
|
String uploadDir = baseUploadDir + "/" + companyId;
|
Path uploadPath = Paths.get(uploadDir);
|
if (!Files.exists(uploadPath)) {
|
return new FebsResponse().data(new ArrayList<>());
|
}
|
|
List<FileInfo> fileList = new ArrayList<>();
|
Files.list(uploadPath).forEach(path -> {
|
if (Files.isRegularFile(path)) {
|
try {
|
FileInfo fileInfo = new FileInfo();
|
fileInfo.setFileName(path.getFileName().toString());
|
fileInfo.setFileSize(Files.size(path));
|
fileInfo.setUploadTime(new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date(Files.getLastModifiedTime(path).toMillis())));
|
fileList.add(fileInfo);
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
});
|
|
return new FebsResponse().success().data(fileList);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return new FebsResponse().fail().message("获取文件列表失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 删除文件
|
*/
|
@PostMapping("/delete")
|
public FebsResponse deleteFile(@RequestParam("fileName") String fileName) {
|
try {
|
|
String companyId = getCurrentUserCompanyId();
|
String uploadDir = baseUploadDir + "/" + companyId;
|
Path filePath = Paths.get(uploadDir, fileName);
|
if (Files.exists(filePath)) {
|
Files.delete(filePath);
|
}
|
return new FebsResponse().success().message("文件删除成功");
|
} catch (Exception e) {
|
e.printStackTrace();
|
return new FebsResponse().fail().message("文件删除失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 保存文件信息
|
*/
|
private void saveFileInfo(String fileName, long fileSize) {
|
// 这里可以实现保存文件信息到数据库的逻辑
|
// 简化处理，暂时不做数据库操作
|
}
|
|
/**
|
* 文件信息实体类
|
*/
|
public static class FileInfo {
|
private String fileName;
|
private long fileSize;
|
private String uploadTime;
|
|
public String getFileName() {
|
return fileName;
|
}
|
|
public void setFileName(String fileName) {
|
this.fileName = fileName;
|
}
|
|
public long getFileSize() {
|
return fileSize;
|
}
|
|
public void setFileSize(long fileSize) {
|
this.fileSize = fileSize;
|
}
|
|
public String getUploadTime() {
|
return uploadTime;
|
}
|
|
public void setUploadTime(String uploadTime) {
|
this.uploadTime = uploadTime;
|
}
|
}
|
}
|
