利用minio 进行word转PDF

利用minio 进行word转PDF

/**
* minioPath  word文件在minio中的地址
* from   转换pdf文件来源工程名称，用于在mybucket里面创建目录
**/
@Override
    public  CommonResult<String> wordToPdf(String minioPath, String from) {
        try {
            if(lock(minioPath)){
                S3FileClient fileClient = new S3FileClient(1L, s3FileClientConfig);
                fileClient.init();
                // 读取内容
                byte[] content = fileClient.getContent(minioPath);
                File tmpFile = File.createTempFile("pdf", ".pdf");
                //libreoffice转换文件必须先加锁，不然并行情况下会有问题
                documentConverter.convert(new ByteArrayInputStream(content)).as(DefaultDocumentFormatRegistry.DOCX).to(tmpFile).as(DefaultDocumentFormatRegistry.PDF).execute();

                InputStream in = new FileInputStream(tmpFile);
                content = in.readAllBytes();
                String originalFilename = tmpFile.getName();
                if (Objects.equals(originalFilename, "")) {
                    throw exception(FILE_NOT_EXISTS);
                }
                String path = FileUtils.generatePath(content, originalFilename);
                //将文件根据日期进行分类管理
                String currentDate = LocalDate.now().format(DateTimeFormatter.BASIC_ISO_DATE);
                path = from + "/" + currentDate + "/" + path;
                // 文件类型 非后缀
                String type = FileTypeUtils.getMineType(content, originalFilename);
                in.close();
                //文件大小
                String url = fileClient.upload(content, path, type);
                tmpFile.delete();
                return CommonResult.success(url);
            }
        } catch (Exception e) {
            log.error(e.getMessage());
        }finally {
            unLock(minioPath);
        }
        return CommonResult.error();
    }

@Data
public class S3FileClientConfig implements FileClientConfig {

    public static final String ENDPOINT_QINIU = "qiniucs.com";
    public static final String ENDPOINT_ALIYUN = "aliyuncs.com";
    public static final String ENDPOINT_TENCENT = "myqcloud.com";
    
    @NotNull(message = "endpoint 不能为空")
    private String endpoint;

    @URL(message = "domain 必须是 URL 格式")
    private String domain;
    /**
     * 存储 Bucket
     */
    @NotNull(message = "bucket 不能为空")
    private String bucket;

    @NotNull(message = "accessKey 不能为空")
    private String accessKey;
    /**
     * 访问 Secret
     */
    @NotNull(message = "accessSecret 不能为空")
    private String accessSecret;

    @SuppressWarnings("RedundantIfStatement")
    @AssertTrue(message = "domain 不能为空")
    @JsonIgnore
    public boolean isDomainValid() {
        // 如果是七牛，必须带有 domain
        if (StrUtil.contains(endpoint, ENDPOINT_QINIU) && StrUtil.isEmpty(domain)) {
            return false;
        }
        return true;
    }

    /**
     * 外网访问域名
     */
    private String address;
}

@JsonTypeInfo(use = JsonTypeInfo.Id.CLASS)
// @JsonTypeInfo 注解的作用，Jackson 多态
// 1. 序列化到时数据库时，增加 @class 属性。
// 2. 反序列化到内存对象时，通过 @class 属性，可以创建出正确的类型
public interface FileClientConfig {
}

public class FileTypeUtils {

    private static final ThreadLocal<Tika> TIKA = TransmittableThreadLocal.withInitial(Tika::new);

    /**
     * 获得文件的 mineType，对于doc，jar等文件会有误差
     *
     * @param data 文件内容
     * @return mineType 无法识别时会返回“application/octet-stream”
     */
    @SneakyThrows
    public static String getMineType(byte[] data) {
        return TIKA.get().detect(data);
    }

    /**
     * 已知文件名，获取文件类型，在某些情况下比通过字节数组准确，例如使用jar文件时，通过名字更为准确
     *
     * @param name 文件名
     * @return mineType 无法识别时会返回“application/octet-stream”
     */
    public static String getMineType(String name) {
        return TIKA.get().detect(name);
    }

    /**
     * 在拥有文件和数据的情况下，最好使用此方法，最为准确
     *
     * @param data 文件内容
     * @param name 文件名
     * @return mineType 无法识别时会返回“application/octet-stream”
     */
    public static String getMineType(byte[] data, String name) {
        return TIKA.get().detect(data, name);
    }

}

/**
 * 文件工具类
 *
 */
public class FileUtils {

    /**
     * 创建临时文件
     * 该文件会在 JVM 退出时，进行删除
     *
     * @param data 文件内容
     * @return 文件
     */
    @SneakyThrows
    public static File createTempFile(String data) {
        File file = createTempFile();
        // 写入内容
        FileUtil.writeUtf8String(data, file);
        return file;
    }

    /**
     * 创建临时文件
     * 该文件会在 JVM 退出时，进行删除
     *
     * @param data 文件内容
     * @return 文件
     */
    @SneakyThrows
    public static File createTempFile(byte[] data) {
        File file = createTempFile();
        // 写入内容
        FileUtil.writeBytes(data, file);
        return file;
    }

    /**
     * 创建临时文件，无内容
     * 该文件会在 JVM 退出时，进行删除
     *
     * @return 文件
     */
    @SneakyThrows
    public static File createTempFile() {
        // 创建文件，通过 UUID 保证唯一
        File file = File.createTempFile(IdUtil.simpleUUID(), null);
        // 标记 JVM 退出时，自动删除
        file.deleteOnExit();
        return file;
    }

    /**
     * 生成文件路径
     *
     * @param content      文件内容
     * @param originalName 原始文件名
     * @return path，唯一不可重复
     */
    public static String generatePath(byte[] content, String originalName) {
        String sha256Hex = DigestUtil.sha256Hex(content);
        // 情况一：如果存在 name，则优先使用 name 的后缀
        if (StrUtil.isNotBlank(originalName)) {
            String extName = FileNameUtil.extName(originalName);
            return StrUtil.isBlank(extName) ? sha256Hex : sha256Hex + "." + extName;
        }
        // 情况二：基于 content 计算
        return sha256Hex + '.' + FileTypeUtil.getType(new ByteArrayInputStream(content));
    }

}