SpringBoot+策略模式实现多种文件存储模式

一、策略模式

背景

针对某种业务可能存在多种实现方式;传统方式是通过传统if...else...或者switch代码判断;

弊端:

  • 代码可读性差
  • 扩展性差
  • 难以维护

策略模式简介

策略模式是一种行为型模式,它将对象和行为分开,将行为定义为一个行为接口和具体行为的实现。

组成:

  • 抽象策略类(Strategy):策略的抽象;
  • 具体策略类(ConcreteStrategy):具体的策略实现,一种具体实现;
  • 环境类(Context):用来操作策略的上下文环境;

二、代码实现

以文件存储方式为例

抽象策略类

java 复制代码
package com.qiangesoft.file.core;

import org.springframework.web.multipart.MultipartFile;

/**
 * 文件存储策略接口
 *
 * @author qiangesoft
 * @date 2024-01-06
 */
public interface FileStoreStrategy {

    /**
     * 存储方式
     *
     * @return
     */
    String getType();

    /**
     * 上传文件
     *
     * @param file
     * @return
     * @throws Exception
     */
    void uploadFile(MultipartFile file) throws Exception;
}

具体策略实现

本地存储

java 复制代码
package com.qiangesoft.file.core.local;

import com.qiangesoft.file.core.FileStoreStrategy;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

/**
 * 文件本地存储
 *
 * @author qiangesoft
 * @date 2024-01-06
 */
@Component
@Slf4j
public class LocalStoreStrategyImpl implements FileStoreStrategy {

    @Override
    public String getType() {
        return "LOCAL";
    }

    @Override
    public void uploadFile(MultipartFile file) throws Exception {
        log.info("upload file by local");
    }
}

minio存储

java 复制代码
package com.qiangesoft.file.core.minio;

import com.qiangesoft.file.core.FileStoreStrategy;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

/**
 * 文件minio对象存储
 *
 * @author qiangesoft
 * @date 2024-01-06
 */
@Component
@Slf4j
public class MinioStoreStrategyImpl implements FileStoreStrategy {

    @Override
    public String getType() {
        return "MINIO";
    }

    @Override
    public void uploadFile(MultipartFile file) throws Exception {
        log.info("upload file by minio");
    }
}

策略上下文环境

java 复制代码
package com.qiangesoft.file.core;

import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 文件存储策略上下文
 *
 * @author qiangesoft
 * @date 2024-01-06
 */
@Component
public class FileStoreStrategyContext {

    /**
     * 策略实现类上下文
     */
    private final Map<String, FileStoreStrategy> strategyMap = new ConcurrentHashMap<>();

    /**
     * 利用Spring的发现机制,将实现了FileStoreStrategy的类都放到strategyMap里面
     */
    public FileStoreStrategyContext(List<FileStoreStrategy> fileStoreStrategyList) {
        for (FileStoreStrategy strategy : fileStoreStrategyList) {
            String type = strategy.getType();
            if (this.strategyMap.containsKey(type)) {
                throw new RuntimeException(String.format("Strategy [%s] is repeat!", type));
            }
            this.strategyMap.put(type, strategy);
        }
    }

    /**
     * 获取具体的策略实现类
     *
     * @param storeType
     * @return
     */
    public FileStoreStrategy getStrategy(String storeType) {
        FileStoreStrategy strategy = strategyMap.get(storeType);
        if (strategy == null) {
            throw new RuntimeException(String.format("Strategy [%s] is not found!", storeType));
        }
        return strategy;
    }
}

调用

java 复制代码
package com.qiangesoft.file.demo;

import com.qiangesoft.file.core.FileStoreStrategy;
import com.qiangesoft.file.core.FileStoreStrategyContext;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * Demo
 *
 * @author qiangesoft
 * @date 2024-01-08
 */
@RestController
@RequestMapping("/demo")
public class DemoController {

    @Autowired
    private FileStoreStrategyContext fileStoreStrategyContext;

    @GetMapping("/upload")
    public String upload(String type) throws Exception {
        FileStoreStrategy fileStoreStrategy = fileStoreStrategyContext.getStrategy(type);
        fileStoreStrategy.uploadFile(null);
        return fileStoreStrategy.getType();
    }

}



扩展

如果新增一种存储方式,只需增加一个策略实现类,实现FileStoreStrategy接口,重写getType方法,然后再实现具体的业务逻辑;

案例代码

点击下载

相关推荐
为将者,自当识天晓地。18 分钟前
c++多线程
java·开发语言
daqinzl26 分钟前
java获取机器ip、mac
java·mac·ip
激流丶42 分钟前
【Kafka 实战】如何解决Kafka Topic数量过多带来的性能问题?
java·大数据·kafka·topic
Themberfue1 小时前
Java多线程详解⑤(全程干货!!!)线程安全问题 || 锁 || synchronized
java·开发语言·线程·多线程·synchronized·
让学习成为一种生活方式1 小时前
R包下载太慢安装中止的解决策略-R语言003
java·数据库·r语言
晨曦_子画1 小时前
编程语言之战:AI 之后的 Kotlin 与 Java
android·java·开发语言·人工智能·kotlin
南宫生2 小时前
贪心算法习题其三【力扣】【算法学习day.20】
java·数据结构·学习·算法·leetcode·贪心算法
Heavydrink2 小时前
HTTP动词与状态码
java
ktkiko112 小时前
Java中的远程方法调用——RPC详解
java·开发语言·rpc
计算机-秋大田2 小时前
基于Spring Boot的船舶监造系统的设计与实现,LW+源码+讲解
java·论文阅读·spring boot·后端·vue