feat(fileUpload): 添加公司隔离的文件上传功能

Administrator

3 days ago eafe95f8cc03bc838970ad38a3ce1e44bedfcb5a

feat(fileUpload): 添加公司隔离的文件上传功能

- 修改控制器添加公司ID参数支持多租户文件管理
- 更新上传目录和分片存储路径为公司专属目录
- 在前端界面中集成公司ID参数传递机制
- 调整文件播放、删除和列表接口以支持公司隔离
- 实现基于公司ID的文件访问权限控制
- 更新前端AJAX请求以包含公司ID参数

 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);

 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});