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.HttpServletRequest;
|
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 {
|
|
// 基础上传目录
|
public static String baseUploadDir = "/home/javaweb/webresource/ai/file";
|
|
// 分片存储目录
|
public static String baseChunkDir = "/home/javaweb/webresource/ai/file/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 = fileName+ "_" + UUID.randomUUID().toString() ;
|
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, HttpServletRequest request, 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;
|
}
|
|
// 获取文件大小
|
long fileSize = Files.size(filePath);
|
|
// 检查是否有Range请求头
|
String rangeHeader = request.getHeader("Range");
|
if (rangeHeader != null) {
|
// 解析Range头，格式为"bytes=start-end"
|
String[] ranges = rangeHeader.replace("bytes=", "").split("-");
|
long start = 0, end = fileSize - 1;
|
|
if (!ranges[0].isEmpty()) {
|
start = Long.parseLong(ranges[0]);
|
}
|
if (ranges.length > 1 && !ranges[1].isEmpty()) {
|
end = Long.parseLong(ranges[1]);
|
} else {
|
end = fileSize - 1;
|
}
|
|
// 确保范围有效
|
if (start > end || start >= fileSize) {
|
response.setStatus(HttpStatus.REQUESTED_RANGE_NOT_SATISFIABLE.value());
|
return;
|
}
|
|
// 限制结束位置不超过文件大小
|
if (end >= fileSize) {
|
end = fileSize - 1;
|
}
|
|
// 计算内容长度
|
long contentLength = end - start + 1;
|
|
// 设置响应状态和头部信息
|
response.setStatus(HttpStatus.PARTIAL_CONTENT.value());
|
response.setContentType(getContentTypeByExtension(fileName));
|
response.setContentLengthLong(contentLength);
|
response.setHeader("Content-Range", "bytes " + start + "-" + end + "/" + fileSize);
|
response.setHeader("Accept-Ranges", "bytes");
|
response.setHeader("Content-Disposition", "inline; filename=\"" + fileName + "\"");
|
|
// 从指定位置开始读取文件
|
try (RandomAccessFile randomAccessFile = new RandomAccessFile(filePath.toFile(), "r");
|
OutputStream outputStream = response.getOutputStream()) {
|
|
randomAccessFile.seek(start);
|
byte[] buffer = new byte[8192];
|
long remaining = contentLength;
|
int bytesRead;
|
|
while (remaining > 0 && (bytesRead = randomAccessFile.read(buffer, 0, (int) Math.min(buffer.length, remaining))) != -1) {
|
outputStream.write(buffer, 0, bytesRead);
|
remaining -= bytesRead;
|
}
|
}
|
} else {
|
// 没有Range头，返回完整文件
|
response.setStatus(HttpStatus.OK.value());
|
response.setContentType(getContentTypeByExtension(fileName));
|
response.setContentLengthLong(fileSize);
|
response.setHeader("Accept-Ranges", "bytes");
|
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());
|
}
|
}
|
|
/**
|
* 根据文件扩展名获取MIME类型
|
*/
|
private String getContentTypeByExtension(String fileName) {
|
String extension = "";
|
int lastDotIndex = fileName.lastIndexOf('.');
|
if (lastDotIndex > 0) {
|
extension = fileName.substring(lastDotIndex).toLowerCase();
|
}
|
|
switch (extension) {
|
case ".mp4":
|
return "video/mp4";
|
case ".avi":
|
return "video/x-msvideo";
|
case ".mov":
|
return "video/quicktime";
|
case ".wmv":
|
return "video/x-ms-wmv";
|
case ".flv":
|
return "video/x-flv";
|
case ".webm":
|
return "video/webm";
|
case ".mkv":
|
return "video/x-matroska";
|
case ".m3u8":
|
return "application/vnd.apple.mpegurl";
|
case ".ts":
|
return "video/MP2T";
|
default:
|
return "application/octet-stream";
|
}
|
}
|
|
/**
|
* 获取文件列表
|
*/
|
@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;
|
}
|
}
|
}
|
