jeecgBoot(SpringBoot)中封装MInIO上传下载文件

在yml配置文件中配置minIO相关参数

可以选择编写一个Util工具类，来进行封装，这里我就直接写到了一个Controller里面来专门处理minIO上传和下载请求

@Slf4j
@RestController
@RequestMapping("/sys/common")
public class CommonController {

    @Autowired
    private ISysBaseAPI sysBaseAPI;

    @Value(value = "${jeecg.path.upload}")
    private String uploadpath;

    @Value(value = "${jeecg.minio.minio_url}")
    private String minioUrl;


    @Value(value = "${jeecg.minio.minio_name}")
    private String minioName;

    @Value(value = "${jeecg.minio.minio_pass}")
    private String minioPass;

    @Value(value = "${jeecg.minio.bucketName}")
    private String bucketName;

    /**
     * 本地：local minio：minio 阿里：alioss
     */
    @Value(value="${jeecg.uploadType}")
    private String uploadType;

    /**
     * @return
     */
    @GetMapping("/403")
    public Result<?> noauth()  {
        return Result.error("没有权限，请联系管理员授权");
    }

    /**
     * 文件上传统一方法
     * @param request
     * @param response
     * @return
     */
    @PostMapping(value = "/upload")
    public Result<?> upload(HttpServletRequest request, HttpServletResponse response) throws Exception {
        Result<?> result = new Result<>();
        String savePath = "";
        String bizPath = request.getParameter("biz");
        MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
        // 获取上传文件对象
        MultipartFile file = multipartRequest.getFile("file");
        if (file == null) {
            result.setMessage("未选择上传文件！");
            result.setSuccess(false);
        } else {
            // 实例化MinIO客户端
            MinioClient minioClient = MinioClient.builder().endpoint(minioUrl).credentials(minioName, minioPass).build();
            if (oConvertUtils.isEmpty(bizPath)) {
                if (CommonConstant.UPLOAD_TYPE_OSS.equals(uploadType)) {
                    //未指定目录，则用阿里云默认目录 upload
                    bizPath = "upload";
                    //result.setMessage("使用阿里云文件上传时，必须添加目录！");
                    //result.setSuccess(false);
                    //return result;
                } else {
                    bizPath = "";
                }
            }
            if (CommonConstant.UPLOAD_TYPE_LOCAL.equals(uploadType)) {
                //update-begin-author:lvdandan date:20200928 for:修改JEditor编辑器本地上传
                savePath = this.uploadLocal(file, bizPath);
            } else {
                // 使用MinIO客户端上传文件
                savePath = uploadMinio(minioClient, file, bizPath, bucketName);
            }
            if (oConvertUtils.isNotEmpty(savePath)) {
                result.setMessage(savePath);
                result.setSuccess(true);
            } else {
                result.setMessage("上传失败！");
                result.setSuccess(false);
            }
        }
        return result;
    }

    /**
     * 文件minIO下载
     *
     * @param request
     * @param response
     * @return
     */
    @GetMapping(value = "/download/*/{fileName}")
    public void download2(HttpServletRequest request, HttpServletResponse response, @PathVariable("fileName") String fileName) throws Exception {
        String uri = request.getRequestURI();
        String sub1 = "/download/";
        int startIndex = uri.indexOf(sub1) + sub1.length();
        int endIndex = uri.lastIndexOf("/");
        String filePath;
        if (endIndex - startIndex != 0) {
            filePath = uri.substring(startIndex, endIndex) + "/" + fileName;
        } else {
            filePath = fileName;
        }
        OutputStream outputStream = null;
        try {
            // 初始化MinIO客户端
            MinioClient minioClient = MinioClient.builder().endpoint(minioUrl).credentials(minioName, minioPass).build();
            // 下载文件
            try (InputStream inputStream = minioClient.getObject(GetObjectArgs.builder().bucket(bucketName).object(filePath).build())) {
                // 设置响应头
                response.setContentType("application/octet-stream");
                //中文乱码
                //fileName = URLEncoder.encode(fileName, "UTF-8");
                response.setHeader("Content-Disposition", "attachment; filename=\"" + new String(fileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1) + "\"");
//                response.setContentLength(inputStream.available());
//
//                // 设置Accept-Ranges
//                response.setHeader("Accept-Ranges", "bytes");
                // 将文件流写入响应流
                outputStream = response.getOutputStream();
                byte[] buffer = new byte[4096];
                int bytesRead;
                while ((bytesRead = inputStream.read(buffer)) != -1) {
                    outputStream.write(buffer, 0, bytesRead);
                }

            }
        } catch (MinioException e) {
            System.out.println("MinIO错误发生: " + e.getMessage());
            response.getWriter().write("MinIO错误发生：" + e.getMessage());
        } catch (IOException e) {
            System.out.println("IO错误发生: " + e.getMessage());
            response.getWriter().write("IO错误发生：" + e.getMessage());
        } finally {
            if (outputStream != null) {
                outputStream.flush();
                outputStream.close();
            }
        }
    }
    @GetMapping(value = "/download/{fileName}")
    public void download(HttpServletResponse response, @PathVariable("fileName") String fileName) throws Exception {
        OutputStream outputStream = null;
        try {
            // 初始化MinIO客户端
            MinioClient minioClient = MinioClient.builder().endpoint(minioUrl).credentials(minioName, minioPass).build();
            // 下载文件
            try (InputStream inputStream = minioClient.getObject(GetObjectArgs.builder().bucket(bucketName).object(fileName).build())) {
                // 设置响应头
                response.setContentType("application/octet-stream");
                //中文乱码
                //fileName = URLEncoder.encode(fileName, "UTF-8");
                response.setHeader("Content-Disposition", "attachment; filename=\"" + new String(fileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1) + "\"");
//                response.setContentLength(inputStream.available());
//
//                // 设置Accept-Ranges
//                response.setHeader("Accept-Ranges", "bytes");
                // 将文件流写入响应流
                outputStream = response.getOutputStream();
                byte[] buffer = new byte[4096];
                int bytesRead;
                while ((bytesRead = inputStream.read(buffer)) != -1) {
                    outputStream.write(buffer, 0, bytesRead);
                }

            }
        } catch (MinioException e) {
            System.out.println("MinIO错误发生: " + e.getMessage());
            response.getWriter().write("MinIO错误发生：" + e.getMessage());
        } catch (IOException e) {
            System.out.println("IO错误发生: " + e.getMessage());
            response.getWriter().write("IO错误发生：" + e.getMessage());
        } finally {
            if (outputStream != null) {
                outputStream.flush();
                outputStream.close();
            }
        }
    }

    /**
     * Minio文件上传
     *
     * @param minioClient
     * @param file
     * @param bizPath
     * @param bucketName
     * @return
     * @throws Exception
     */
    private String uploadMinio(MinioClient minioClient, MultipartFile file, String bizPath, String bucketName) throws Exception {
        // 生成文件保存路径等逻辑
        String fileName = generateFileName(file.getOriginalFilename());
        String savePath = bizPath + "/" + fileName;
        // 使用MinIO客户端上传文件,设置文件类型为上传文件的ContentType
        PutObjectArgs putObjectArgs = PutObjectArgs.builder()
                .bucket(bucketName)
                .object(savePath)
                .stream(file.getInputStream(), file.getSize(), -1)
                .contentType(file.getContentType())
                .build();
        minioClient.putObject(putObjectArgs);
        // 返回保存路径
        return savePath;
    }

    private String generateFileName(String originalFileName) {
        // 生成文件名的逻辑
        return originalFileName;
    }

}