From eafe95f8cc03bc838970ad38a3ce1e44bedfcb5a Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 14 Jan 2026 09:41:56 +0800
Subject: [PATCH] feat(fileUpload): 添加公司隔离的文件上传功能
---
src/main/java/cc/mrbird/febs/ai/controller/fileUpload/FileUploadController.java | 36 +++++++++++++++++++++++++-----------
src/main/resources/templates/febs/views/modules/ai/fileUpload/index.html | 14 ++++++++++----
2 files changed, 35 insertions(+), 15 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/fileUpload/FileUploadController.java b/src/main/java/cc/mrbird/febs/ai/controller/fileUpload/FileUploadController.java
index 1f1c849..8170501 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/fileUpload/FileUploadController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/fileUpload/FileUploadController.java
@@ -28,11 +28,11 @@
@RequestMapping(value = "/fileUpload")
public class FileUploadController extends BaseController {
- // @Value("${file.upload.dir}")
- private String uploadDir = "d:/upload/files";
+ // 基础上传目录
+ private String baseUploadDir = "/home/javaweb/webresource/ai";
- // @Value("${file.chunk.dir}")
- private String chunkDir = "d:/upload/chunks";
+ // 分片存储目录
+ private String baseChunkDir = "/home/javaweb/webresource/ai/chunks";
/**
* 上传文件分片
@@ -42,8 +42,11 @@
@RequestParam("fileName") String fileName,
@RequestParam("chunk") int chunk,
@RequestParam("chunks") int chunks,
- @RequestParam("fileMd5") String fileMd5) {
+ @RequestParam("fileMd5") String fileMd5,
+ @RequestParam("companyId") String companyId) {
try {
+ // 构建公司专属分片目录
+ String chunkDir = baseChunkDir + "/" + companyId;
// 确保分片目录存在
Path chunkPath = Paths.get(chunkDir, fileMd5);
if (!Files.exists(chunkPath)) {
@@ -67,8 +70,11 @@
@PostMapping("/mergeChunks")
public FebsResponse mergeChunks(@RequestParam("fileName") String fileName,
@RequestParam("fileMd5") String fileMd5,
- @RequestParam("chunks") int chunks) {
+ @RequestParam("chunks") int chunks,
+ @RequestParam("companyId") String companyId) {
try {
+ // 构建公司专属上传目录
+ String uploadDir = baseUploadDir + "/" + companyId;
// 确保上传目录存在
Path uploadPath = Paths.get(uploadDir);
if (!Files.exists(uploadPath)) {
@@ -79,6 +85,8 @@
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)) {
@@ -115,9 +123,12 @@
/**
* 播放视频文件
*/
- @GetMapping("/play/{fileName}")
- public void playVideo(@PathVariable("fileName") String fileName, HttpServletResponse response) {
+ @GetMapping("/play/{companyId}/{fileName}")
+ public void playVideo(@PathVariable("companyId") String companyId,
+ @PathVariable("fileName") String fileName,
+ HttpServletResponse response) {
try {
+ String uploadDir = baseUploadDir + "/" + companyId;
Path filePath = Paths.get(uploadDir, fileName);
if (!Files.exists(filePath)) {
response.setStatus(HttpStatus.NOT_FOUND.value());
@@ -147,9 +158,10 @@
/**
* 获取文件列表
*/
- @GetMapping("/list")
- public FebsResponse getFileList() {
+ @GetMapping("/list/{companyId}")
+ public FebsResponse getFileList(@PathVariable("companyId") String companyId) {
try {
+ String uploadDir = baseUploadDir + "/" + companyId;
Path uploadPath = Paths.get(uploadDir);
if (!Files.exists(uploadPath)) {
return new FebsResponse().data(new ArrayList<>());
@@ -181,8 +193,10 @@
* 删除文件
*/
@PostMapping("/delete")
- public FebsResponse deleteFile(@RequestParam("fileName") String fileName) {
+ public FebsResponse deleteFile(@RequestParam("fileName") String fileName,
+ @RequestParam("companyId") String companyId) {
try {
+ String uploadDir = baseUploadDir + "/" + companyId;
Path filePath = Paths.get(uploadDir, fileName);
if (Files.exists(filePath)) {
Files.delete(filePath);
diff --git a/src/main/resources/templates/febs/views/modules/ai/fileUpload/index.html b/src/main/resources/templates/febs/views/modules/ai/fileUpload/index.html
index 613b626..07d6317 100644
--- a/src/main/resources/templates/febs/views/modules/ai/fileUpload/index.html
+++ b/src/main/resources/templates/febs/views/modules/ai/fileUpload/index.html
@@ -310,6 +310,7 @@
var isUploading = false;
var startTime = 0;
var uploadedSize = 0;
+ var companyId = 'default'; // 默认公司ID，实际应用中应从session或URL参数获取
// 初始化
init();
@@ -483,6 +484,7 @@
formData.append('chunk', currentChunk);
formData.append('chunks', chunks);
formData.append('fileMd5', fileMd5);
+ formData.append('companyId', companyId);
$.ajax({
url: '/fileUpload/uploadChunk',
@@ -535,7 +537,8 @@
data: {
fileName: file.name,
fileMd5: fileMd5,
- chunks: chunks
+ chunks: chunks,
+ companyId: companyId
},
success: function(response) {
if (response.code === 200 || response.success) {
@@ -589,7 +592,7 @@
function refreshFileList() {
layer.msg('正在获取文件列表...', {icon: 16, time: 0});
$.ajax({
- url: '/fileUpload/list',
+ url: '/fileUpload/list/' + companyId,
type: 'GET',
success: function(response) {
layer.closeAll();
@@ -632,7 +635,7 @@
var videoPlayer = document.getElementById('videoPlayer');
var videoSource = document.getElementById('videoSource');
- videoSource.src = '/fileUpload/play/' + encodeURIComponent(fileName);
+ videoSource.src = '/fileUpload/play/' + companyId + '/' + encodeURIComponent(fileName);
videoPlayer.load();
videoPlayer.play();
});
@@ -646,7 +649,10 @@
$.ajax({
url: '/fileUpload/delete',
type: 'POST',
- data: {fileName: fileName},
+ data: {
+ fileName: fileName,
+ companyId: companyId
+ },
success: function(response) {
if (response.code === 200 || response.success) {
layer.msg('删除成功', {icon: 1});
--
Gitblit v1.9.1