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方法,然后再实现具体的业务逻辑;

案例代码

点击下载

相关推荐
冰帝海岸7 分钟前
01-spring security认证笔记
java·笔记·spring
世间万物皆对象32 分钟前
Spring Boot核心概念:日志管理
java·spring boot·单元测试
没书读了1 小时前
ssm框架-spring-spring声明式事务
java·数据库·spring
小二·1 小时前
java基础面试题笔记(基础篇)
java·笔记·python
开心工作室_kaic2 小时前
ssm161基于web的资源共享平台的共享与开发+jsp(论文+源码)_kaic
java·开发语言·前端
懒洋洋大魔王2 小时前
RocketMQ的使⽤
java·rocketmq·java-rocketmq
武子康2 小时前
Java-06 深入浅出 MyBatis - 一对一模型 SqlMapConfig 与 Mapper 详细讲解测试
java·开发语言·数据仓库·sql·mybatis·springboot·springcloud
转世成为计算机大神2 小时前
易考八股文之Java中的设计模式?
java·开发语言·设计模式
qq_327342733 小时前
Java实现离线身份证号码OCR识别
java·开发语言
阿龟在奔跑4 小时前
引用类型的局部变量线程安全问题分析——以多线程对方法局部变量List类型对象实例的add、remove操作为例
java·jvm·安全·list