SpringBoot 策略模式實現(xiàn)切換上傳文件模式

更新時間：2023年11月20日 12:03:48 作者：草木物語

策略模式是指有一定行動內(nèi)容的相對穩(wěn)定的策略名稱,這篇文章主要介紹了SpringBoot 策略模式切換上傳文件模式,需要的朋友可以參考下

SpringBoot 策略模式

策略模式是指有一定行動內(nèi)容的相對穩(wěn)定的策略名稱。

我們定義一個接口（就比如接下來要實現(xiàn)的文件上傳接口）
我們定義所需要實現(xiàn)的策略實現(xiàn)類 A、B、C、D（也就是項目中所使用的四種策略阿里云Oss上傳、騰訊云Cos上傳、七牛云Kodo上傳、本地上傳）
我們通過策略上下文來調(diào)用策略接口，并選擇所需要使用的策略

策略接口

public interface UploadStrategy {
    /**
     * 上傳文件
     *
     * @param file     文件
     * @param filePath 文件上傳露肩
     * @return {@link String} 文件上傳的全路徑
     */
    String uploadFile(MultipartFile file, final String filePath);

策略實現(xiàn)類內(nèi)部實現(xiàn)

@Getter
@Setter
public abstract class AbstractUploadStrategyImpl implements UploadStrategy {
    @Override
    public String uploadFile(MultipartFile file, String filePath) {
        try {
            //region 獲取文件md5值 -> 獲取文件后綴名 -> 生成相對路徑
            String fileMd5 = XcFileUtil.getMd5(file.getInputStream());
            String extName = XcFileUtil.getExtName(file.getOriginalFilename());
            String fileRelativePath = filePath + fileMd5 + extName;
            //endregion
            //region 初始化
            initClient();
            //endregion
            //region 檢測文件是否已經(jīng)存在，不存在則進行上傳操作
            if (!checkFileIsExisted(fileRelativePath)) {
                executeUpload(file, fileRelativePath);
            }
            //endregion
            return getPublicNetworkAccessUrl(fileRelativePath);
        } catch (IOException e) {
            throw new XcException("文件上傳失敗");
        }
    }
    /**
     * 初始化客戶端
     */
    public abstract void initClient();
    /**
     * 檢查文件是否已經(jīng)存在（文件MD5值唯一）
     *
     * @param fileRelativePath 文件相對路徑
     * @return true 已經(jīng)存在  false 不存在
     */
    public abstract boolean checkFileIsExisted(String fileRelativePath);
    /**
     * 執(zhí)行上傳操作
     *
     * @param file             文件
     * @param fileRelativePath 文件相對路徑
     * @throws IOException io異常信息
     */
    public abstract void executeUpload(MultipartFile file, String fileRelativePath) throws IOException;
    /**
     * 獲取公網(wǎng)訪問路徑
     *
     * @param fileRelativePath 文件相對路徑
     * @return 公網(wǎng)訪問絕對路徑
     */
    public abstract String getPublicNetworkAccessUrl(String fileRelativePath);
}

本地上傳策略具體實現(xiàn)

@Slf4j
@Getter
@Setter
@RequiredArgsConstructor
@Service("localUploadServiceImpl")
public class LocalUploadStrategyImpl extends AbstractUploadStrategyImpl {
    /**
     * 本地項目端口
     */
    @Value("${server.port}")
    private Integer port;
    /**
     * 前置路徑 ip/域名
     */
    private String prefixUrl;
    /**
     * 構(gòu)造器注入bean
     */
    private final ObjectStoreProperties properties;
    @Override
    public void initClient() {
        try {
            prefixUrl = ResourceUtils.getURL("classpath:").getPath() + "static/";
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            throw new XcException("文件不存在");
        }
        log.info("initClient Init Success...");
    }
    @Override
    public boolean checkFileIsExisted(String fileRelativePath) {
        return new File(prefixUrl + fileRelativePath).exists();
    }
    @Override
    public void executeUpload(MultipartFile file, String fileRelativePath) throws IOException {
        File dest = checkFolderIsExisted(fileRelativePath);
        try {
            file.transferTo(dest);
        } catch (IOException e) {
            e.printStackTrace();
            throw new XcException("文件上傳失敗");
        }
    }
    @Override
    public String getPublicNetworkAccessUrl(String fileRelativePath) {
        try {
            String host = InetAddress.getLocalHost().getHostAddress();
            if (StringUtils.isEmpty(properties.getLocal().getDomainUrl())) {
                return String.format("http://%s:%d%s", host, port, fileRelativePath);
            }
            return properties.getLocal().getDomainUrl() + fileRelativePath;
        } catch (UnknownHostException e) {
            throw new XcException("HttpCodeEnum.UNKNOWN_ERROR");
        }
    }
    /**
     * 檢查文件夾是否存在，若不存在則創(chuàng)建文件夾，最終返回上傳文件
     *
     * @param fileRelativePath 文件相對路徑
     * @return {@link  File} 文件
     */
    private File checkFolderIsExisted(String fileRelativePath) {
        File rootPath = new File(prefixUrl + fileRelativePath);
        if (!rootPath.exists()) {
            if (!rootPath.mkdirs()) {
                throw new XcException("文件夾創(chuàng)建失敗");
            }
        }
        return rootPath;
    }
}

策略上下文實現(xiàn)

@Component
@RequiredArgsConstructor
public class UploadStrategyContext {
    private final Map<String, UploadStrategy> uploadStrategyMap;
    /**
     * 執(zhí)行上傳策略
     *
     * @param file     文件
     * @param filePath 文件上傳路徑前綴
     * @return {@link String} 文件上傳全路徑
     */
    public String executeUploadStrategy(MultipartFile file, final String filePath, String uploadServiceName) {
        // 執(zhí)行特定的上傳策略
        return uploadStrategyMap.get(uploadServiceName).uploadFile(file, filePath);
    }
}

上傳測試