LINK-SERVER/src/main/java/com/lqkj/link/module/base/service/BaseService.java

package com.lqkj.link.module.base.service;

import com.lqkj.link.message.MessageBean;
import com.lqkj.link.util.FileUtils;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.util.UUID;

@Service
public class BaseService {

    public MessageBean<String> uploadImg(MultipartFile file, String path) {
        String fileName = file.getOriginalFilename();
        String suffix = fileName == null ? "" : fileName.substring(fileName.lastIndexOf(".") + 1);
        if (!suffix.equals("png") && !suffix.equals("jpg"))
            return MessageBean.error("上传文件类型必须是png、jpg图片文件");
        String newFileName = UUID.randomUUID() + "." + suffix;
        String filePath = "./upload/" + path;
        FileUtils.saveFile(file, filePath, newFileName);
        return MessageBean.ok(filePath.substring(1) + newFileName, "上传图标文件");
    }

    public MessageBean<String> uploadVideo(MultipartFile file, String path) {
        String fileName = file.getOriginalFilename();
        String suffix = fileName == null ? "" : fileName.substring(fileName.lastIndexOf(".") + 1);
        if (!suffix.equals("mp4"))
            return MessageBean.error("上传文件类型必须是mp4视频文件");
        String newFileName = UUID.randomUUID() + "." + suffix;
        String filePath = "./upload/" + path;
        FileUtils.saveFile(file, filePath, newFileName);
        return MessageBean.ok(filePath.substring(1) + newFileName, "上传视频文件");
    }

    public MessageBean<String> uploadAudio(MultipartFile file, String path) {
        String fileName = file.getOriginalFilename();
        String suffix = fileName == null ? "" : fileName.substring(fileName.lastIndexOf(".") + 1);
        if (!suffix.equals("mp3"))
            return MessageBean.error("上传文件类型必须是mp3音频文件");
        String newFileName = UUID.randomUUID() + "." + suffix;
        String filePath = "./upload/" + path;
        FileUtils.saveFile(file, filePath, newFileName);
        return MessageBean.ok(filePath.substring(1) + newFileName, "上传音频文件");
    }
    public MessageBean<String> uploadModelCompress(MultipartFile file, String path) {
        String fileName = file.getOriginalFilename();
        String suffix = fileName == null ? "" : fileName.substring(fileName.lastIndexOf(".") + 1);
        if (!suffix.equals("zip") || !suffix.equals("rar")) {
            return MessageBean.error("上传文件类型必须是zip、rar格式的模型文件");
        }
        String newFileName = UUID.randomUUID() + "." + suffix;
        String filePath = "./upload/" + path;
        FileUtils.saveFile(file, filePath, newFileName);
        return MessageBean.ok(filePath.substring(1) + newFileName, "上传模型压缩文件");
    }

    public MessageBean<String> uploadModel(MultipartFile file, String path) {
        String fileName = file.getOriginalFilename();
        String suffix = fileName == null ? "" : fileName.substring(fileName.lastIndexOf(".") + 1);
        if (!suffix.equals("obj") && !suffix.equals("fbx"))
            return MessageBean.error("上传文件类型必须是obj、fbx格式的压缩文件");
        String newFileName = UUID.randomUUID() + "." + suffix;
        String filePath = "./upload/" + path;
        FileUtils.saveFile(file, filePath, newFileName);
        return MessageBean.ok(filePath.substring(1) + newFileName, "上传模型文件");
    }

    public MessageBean<String> uploadJsonFile(MultipartFile file, String path) {
        String fileName = file.getOriginalFilename();
        String suffix = fileName == null ? "" : fileName.substring(fileName.lastIndexOf(".") + 1);
        if (!suffix.equals("json"))
            return MessageBean.error("上传文件类型必须是json格式的json文件");
        String newFileName = UUID.randomUUID() + "." + suffix;
        String filePath = "./upload/" + path;
        FileUtils.saveFile(file, filePath, newFileName);
        return MessageBean.ok(filePath.substring(1) + newFileName, "上传模型文件");
    }
}