diff --git a/mhd-common/mhd-common-core/src/main/java/minio/MinioBootUtil.java b/mhd-common/mhd-common-core/src/main/java/minio/MinioBootUtil.java new file mode 100644 index 000000000..f0ab2e931 --- /dev/null +++ b/mhd-common/mhd-common-core/src/main/java/minio/MinioBootUtil.java @@ -0,0 +1,240 @@ +package minio; + +import com.aliyun.oss.ServiceException; +import io.minio.MinioClient; +import io.minio.errors.InvalidEndpointException; +import io.minio.errors.InvalidPortException; +import lombok.extern.slf4j.Slf4j; +import net.coobird.thumbnailator.Thumbnails; +import org.jeecg.modules.oss.dto.WaterMarkDTO; +import org.jeecg.modules.system.entity.SysDepartInterfaceInfo; +import org.springframework.web.multipart.MultipartFile; +import org.springframework.web.util.UriUtils; + +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.nio.charset.StandardCharsets; +import java.text.SimpleDateFormat; +import java.util.Calendar; +import java.util.Date; +import java.util.List; + +/** + * @Description: minio oss 上传工具类 + */ +@Slf4j +public class MinioBootUtil { + + private static String bucket; + private static String host; + private static String accessKey; + private static String secretKey; + /** + * minio 工具客户端 + */ + private static MinioClient minioClient = null; + + private static String FILE_URL; + + /** + * @Description 强制初始化minio + * @Author Alex + * @Date 2023/10/25 11:12 + */ + public static MinioClient forceInitMinio(List sysDepartInterfaceInfoList) throws InvalidPortException, InvalidEndpointException { + for (SysDepartInterfaceInfo sysDepartInterfaceInfo : sysDepartInterfaceInfoList) { + String interfaceTypeKey = sysDepartInterfaceInfo.getInterfaceTypeKey();//接口参数 + String interfaceTypeValue = sysDepartInterfaceInfo.getInterfaceTypeValue();//接口参数值 + switch (interfaceTypeKey) { + case "bucket": + bucket = interfaceTypeValue; + break; + case "host": + host = interfaceTypeValue; + break; + case "accessKey": + accessKey = interfaceTypeValue; + break; + case "secretKey": + secretKey = interfaceTypeValue; + break; + default: + break; + } + } + minioClient = new MinioClient(host, accessKey, secretKey); + return minioClient; + } + private static MinioClient initMinio() throws InvalidPortException, InvalidEndpointException { + if (minioClient == null) { + minioClient = new MinioClient(host, accessKey, secretKey); + } + return minioClient; + } + /** + * 上传文件 + * 文件上传成功,返回文件完整访问路径 + * 文件上传失败,返回 null + * + * @param multipartFile 待上传文件 + * @param fileDir 文件保存目录 + * @return oss 中的相对文件路径 + */ + public static String upload(MultipartFile multipartFile, String fileDir, List sysDepartInterfaceInfoList) { + for (SysDepartInterfaceInfo sysDepartInterfaceInfo : sysDepartInterfaceInfoList) { + String interfaceTypeKey = sysDepartInterfaceInfo.getInterfaceTypeKey();//接口参数 + String interfaceTypeValue = sysDepartInterfaceInfo.getInterfaceTypeValue();//接口参数值 + switch (interfaceTypeKey) { + case "bucket": + bucket = interfaceTypeValue; + break; + case "host": + host = interfaceTypeValue; + break; + case "accessKey": + accessKey = interfaceTypeValue; + break; + case "secretKey": + secretKey = interfaceTypeValue; + break; + default: + break; + } + } + + try (InputStream inputStream = multipartFile.getInputStream()) { + //初始化minio + initMinio(); + // bucket 不存在，创建 + if (!minioClient.bucketExists(bucket)) { + minioClient.makeBucket(bucket); + } + // 上传文件的名称 + String fileName = multipartFile.getOriginalFilename(); + long timestamp = Calendar.getInstance().getTimeInMillis(); + // PutObjectOptions，上传配置(文件大小，内存中文件分片大小) +// PutObjectOptions putObjectOptions = new PutObjectOptions(multipartFile.getSize(), PutObjectOptions.MIN_MULTIPART_SIZE); +// // 文件的ContentType +// putObjectOptions.setContentType(multipartFile.getContentType()); + + SimpleDateFormat sdYear = new SimpleDateFormat("yyyy"); + SimpleDateFormat sdMonth = new SimpleDateFormat("MM"); + SimpleDateFormat sdDay = new SimpleDateFormat("dd"); + String year = sdYear.format(new Date()); + String month = sdMonth.format(new Date()); + String day = sdDay.format(new Date()); + String filrDir = "wlhy/" + year + "/" + month + "/" + day + "/"; + +// minioClient.putObject(bucket, filrDir+timestamp+fileName, inputStream, putObjectOptions); + minioClient.putObject(bucket, filrDir+timestamp+fileName, inputStream, multipartFile.getContentType()); + // 返回访问路径 + String fileurl = host+ "/"+bucket + "/"+filrDir+timestamp+ UriUtils.encode(fileName, StandardCharsets.UTF_8); + System.out.println("fileurl===="+fileurl); + FILE_URL = fileurl; + return fileurl; + } catch (ServiceException e){ + log.error(e.getErrorMessage()); + } catch (Exception e){ + log.error(e.getMessage()); + } + return FILE_URL; + } + + /** + * 上传文件 带时间水印 + * 文件上传成功,返回文件完整访问路径 + * 文件上传失败,返回 null + * + * @param multipartFile 待上传文件 + * @param fileDir 文件保存目录 + * @return oss 中的相对文件路径 + */ + public static String uploadWaterMark(MultipartFile multipartFile, WaterMarkDTO waterMarkDTO, String fileDir, List sysDepartInterfaceInfoList) { + for (SysDepartInterfaceInfo sysDepartInterfaceInfo : sysDepartInterfaceInfoList) { + String interfaceTypeKey = sysDepartInterfaceInfo.getInterfaceTypeKey();//接口参数 + String interfaceTypeValue = sysDepartInterfaceInfo.getInterfaceTypeValue();//接口参数值 + switch (interfaceTypeKey) { + case "bucket": + bucket = interfaceTypeValue; + break; + case "host": + host = interfaceTypeValue; + break; + case "accessKey": + accessKey = interfaceTypeValue; + break; + case "secretKey": + secretKey = interfaceTypeValue; + break; + default: + break; + } + } + + // 压缩代码 + InputStream inputStream = null; + try { + //初始化minio + initMinio(); + // bucket 不存在，创建 + if (!minioClient.bucketExists(bucket)) { + minioClient.makeBucket(bucket); + } + String OriginalFilename = multipartFile.getOriginalFilename(); + inputStream = multipartFile.getInputStream(); + SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); + //水印内容 + String waterMarkContent = sdf.format(new Date()); + //以下两种方法加水印 + //方法1 + if (OriginalFilename.indexOf(".pdf") < 0) { + ByteArrayOutputStream byteArrayOutputStreamut = new ByteArrayOutputStream(); + Thumbnails.of(inputStream).width(1024).keepAspectRatio(true).outputQuality(0.9).toOutputStream(byteArrayOutputStreamut); + inputStream = new ByteArrayInputStream(byteArrayOutputStreamut.toByteArray()); + //添加水印 + inputStream = OssBootUtil.buildWatermarkOfText(inputStream, waterMarkDTO, OssBootUtil.WaterMarkLocation.SOUTH_WEST, null, 0.5f); + } + //方法2 + /*if (ImageUtil.isImage(multipartFile.getOriginalFilename())) { + inputStream = ImageUtil.getInputStream( + ImageUtil.setWatermark( + ImageUtil.compress( + ImageIO.read(inputStream)),waterMarkContent), + ImageUtil.getFileExtention(multipartFile.getOriginalFilename())); + }*/ + StringBuilder fileUrl = new StringBuilder(); + String suffix = multipartFile.getOriginalFilename().substring(multipartFile.getOriginalFilename().lastIndexOf('.')); + // 上传文件的名称 + String fileName = multipartFile.getOriginalFilename(); + long timestamp = Calendar.getInstance().getTimeInMillis(); + // PutObjectOptions，上传配置(文件大小，内存中文件分片大小) +// PutObjectOptions putObjectOptions = new PutObjectOptions(multipartFile.getSize(), PutObjectOptions.MIN_MULTIPART_SIZE); +// // 文件的ContentType +// putObjectOptions.setContentType(multipartFile.getContentType()); + + SimpleDateFormat sdYear = new SimpleDateFormat("yyyy"); + SimpleDateFormat sdMonth = new SimpleDateFormat("MM"); + SimpleDateFormat sdDay = new SimpleDateFormat("dd"); + String year = sdYear.format(new Date()); + String month = sdMonth.format(new Date()); + String day = sdDay.format(new Date()); + String filrDir = "wlhy/" + year + "/" + month + "/" + day + "/"; + +// minioClient.putObject(bucket, filrDir+timestamp+fileName, inputStream, putObjectOptions); + minioClient.putObject(bucket, filrDir+timestamp+fileName, inputStream, multipartFile.getContentType()); + // 返回访问路径 + String fileurl = host+ "/"+bucket + "/"+filrDir+timestamp+ UriUtils.encode(fileName, StandardCharsets.UTF_8); + System.out.println("fileurl===="+fileurl); + FILE_URL = fileurl; + return fileurl; + } catch (IOException e) { + log.error(e.getMessage()); + } catch (Exception e) { + log.error(e.getMessage()); + } + return FILE_URL; + } + +}