• 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏吧

记一次使用策略模式优化代码的经历

开发技术 开发技术 2周前 (11-21) 11次浏览

一、背景

之前接手了一个 springboot 项目。在我负责的模块中,有一块用户注册的功能,但是比较特别的是这个注册并不是重新注册,而是从以前的旧系统的数据库中同步旧数据到新系统的数据库中。由于这些用户角色来自于不同的系统,所以我需要在注册的时候先判断类型(这个类型由一个专门的枚举类提供),再去调用已经写好的同步方法同步数据。

伪代码大概是这样的:

public void register(String type, String userId, String projectId, String declareId){
    // 判断用户类型
    if (UserSynchronizeTyeEnum.A.type.equals(type)) {
        // 同步A类型的数据
    } else if (UserSynchronizeTyeEnum.A.type.equals(type)) {
        // 同步B类型的数据
    } else {
		throw new RuntimeException("不存在的用户类型");
    }
    ... ...
}

由于用户的类型比较多,所以当我接手的时候已经有8个 if-esle 了,由于这个项目会逐步的跟其他平台对接,要同步的用户类型会越来越多,而且也不能排除什么时候不新增,反而要取消一部分类型的同步情况。

就这个情况来说,一方面每一次新增类型都会让 if-else 串越来越长,取消一些类型的同步还要直接删除 if-else 里的对应代码;另一方面,这个业务的需求相对稳定,同步方法会不一样,但是一定会根据类型来判断。出于以上考虑,我决定趁现在牵扯范围不大的时候重构一下。

二、思路

1.抽取策略接口和策略类

首先,由于每种用户类型的同步方法是由各模块自己提供的,其实已经抽出了策略,只是没有实现一个统一的策略接口。

但是我在这一步遇上了问题:

  • 各模块的同步方法的名称不全部一样;
  • 由于年代久远,旧代码是不允许改的。

代码不让改,就没法通过为旧实现类新增接口实现多态,方法名不一样,那么反射这条路子也走不通。我想到了装饰器,为每个实现类新增一个装饰器类,注册的时候通过装饰器去调用同步方法,但是这样缺点很明显,会引入一个装饰器接口+n多个装饰器类,为了优化这一个方法,反而要引入十几个类,实在是脱裤子放屁。

但是好在天无绝人之路,他们并不是完全没有相同点:

  • 虽然参数名不一样,但是每个同步方法都需要的参数数量和类型都是一样的;
  • 他们都返回一个布尔值

这让我想起了 JDK8 的函数式接口,将策略接口改造为函数式接口,由于同步方法的参数和返回值类型都是一样的,就可以直接以 Lambda 表达式的形式将各个模块的同步方法放进去,这样就不需要改动模块的代码了。

新增的接口如下:

@FunctionalInterface
public interface IUserSynchronizeSerivice {

    /**
     * 同步方法
     */
    public boolean sync(String userId, String projectId, String declareId);

}

2.策略池的实现

接着,为了实现原本 if-else 的逻辑,我需要一个策略池,能够建立起一个用户类型跟对应的同步策略的映射关系,一开始,我打算直接写在 register()方法所在的类中加入以下代码:

@Autowired
private AUserService aUserService;
@Autowired
private BUserService bUserService;

private static final Map<String, UserSynchronizeTyeEnum.IUserSynchronizeService> synchronizeServiceStrategy = new HashMap<>();
@PostConstruct
private void strategyInit(){
    // spring容器启动后将策略装入策略池
    synchronizeServiceStrategy.put(UserSynchronizeTyeEnum.A.type, aUserService::synchronization);
    synchronizeServiceStrategy.put(UserSynchronizeTyeEnum.B.type, bUserService::sync);
}

但是这样在添加新的用户类型时,需要先去枚举类添加新枚举,然后再回到register()所在的类为策略池添加策略,这个两个逻辑上相连的过程被分散到了两个地方,而且仍然要修改register()所在类的代码。所以决定不用上述的代码,而是去对枚举类下手。

原本的枚举类是这样的:

/**
 * 老系统用户注册,用户类型与同步方法的枚举类
 */
public enum UserSynchronizeTyeEnum {
    /**
     * 类型A的用户
     */
    A("a"),
    
    /**
     * 类型B的用户
     */
    B("b");

    /**
     * 用户类型
     */
    private final String type;

    UserSynchronizeTyeEnum(String type) {
        this.type = type;
    }

    public String getType() {
        return type;
    }
}

为了保证逻辑能够集中,我决定将添加策略这一过程一起放到到枚举类里,在添加枚举的时候就把策略一起放进去:

注:下文的 SpringUtils 实现了 BeanFactoryPostProcessor 接口,是一个用于从 ConfigurableListableBeanFactory 获取对象的工具类。

/**
 * 老系统用户注册,用户类型与同步方法的枚举类
 * 添加新类型时,需要将模块对应的同步方法一并放入
 */
public enum UserSynchronizeTyeEnum {

    /**
     * 类型A的用户
     */
    A("a", (userId, projectId, declareId) -> {
        return SpringUtils.getBean(AUserService.class).synchronization(userId, projectId, declareId);
    }),

    /**
     * 类型B的用户
     */
    B("b", (userId, projectId, declareId) -> {
        return SpringUtils.getBean(BUserService.class).sync(userId, projectId, declareId);
    });

    /**
     * 用户类型
     */
    private final String type;

    /**
     * 同步方法
     */
    private final IUserSynchronizeService synchronizeService;

    UserSynchronizeTyeEnum(String type, IUserSynchronizeService synchronizeService) {
        this.type = type;
        this.synchronizeService = synchronizeService;
    }
}

由于由于枚举类已经相当于之前策略池的 Map 集合了,所以我们直接在里面添加一个 getSynchronizeService()方法,用于直接获取同步方法:

/**
 * 根据枚举值获取对应同步方法
 */
public Optional<IUserSynchronizeService> getSynchronizeService(String type) {
    for (UserSynchronizeTyeEnum tyeEnum : UserSynchronizeTyeEnum.values()) {
        if (tyeEnum.type.equals(type)) {
            return Optional.of(tyeEnum.synchronizeService);
        }
    }
    return Optional.empty();
}

到目前为止,策略池已经基本完成了,但是我们不难发现,现在为策略接口添加实现的地方也变成了枚举类中,策略接口 IUserSynchronizeService 一般也不会被用在其他地方,因此不妨把策略接口也一并引入枚举类中,让他成为一个枚举类的内部接口

现在,枚举类是这样的:

记一次使用策略模式优化代码的经历

枚举类堆外只暴露根据类型获取方法的IUserSynchronizeService() 方法,以及 A 和 B 两个枚举。

完整的 UserSynchronizeTyeEnum枚举类代码如下:

/**
 * 老系统用户注册,用户类型与同步方法的枚举类
 * 添加新类型时,需要将模块对应的同步方法一并放入。待用户注册时,会遍历枚举对象并根据类型获取对应的同步方法执行。
 */
public enum UserSynchronizeTyeEnum {

    /**
     * 类型A的用户
     */
    A("a", (userId, projectId, declareId) -> {
        return SpringUtils.getBean(AUserService.class).synchronization(userId, projectId, declareId);
    }),

    /**
     * 类型B的用户
     */
    B("b", (userId, projectId, declareId) -> {
        return SpringUtils.getBean(BUserService.class).sync(userId, projectId, declareId);
    });

    /**
     * 用户类型
     */
    public String type;

    /**
     * 同步方法
     */
    public IUserSynchronizeService synchronizeService;

    UserSynchronizeTyeEnum(String type, IUserSynchronizeService synchronizeService) {
        this.type = type;
        this.synchronizeService = synchronizeService;
    }

    /**
     * 根据枚举值获取对应同步方法
     */
    public static Optional<IUserSynchronizeService> getSynchronizeService(String type) {
        for (UserSynchronizeTyeEnum tyeEnum : UserSynchronizeTyeEnum.values()) {
            if (tyeEnum.type.equals(type)) {
                return Optional.of(tyeEnum.synchronizeService);
            }
        }
        return Optional.empty();
    }

    /**
     * 同步方法需要符合函数式接口
     */
    @FunctionalInterface
    public interface IUserSynchronizeService {
        boolean sync(List<Map<String, Object>> dateList, String userId, String projectId, String declareId);
    }
}

三、使用

现在,改造完毕,可以开始使用了,对于原先的 register()方法,现在改为:

public void register(String type, String userId, String projectId, String declareId){
    // 获取同步方法,没有就抛异常
    UserSynchronizeTyeEnum.IUserSynchronizeService synchronizeService = UserSynchronizeTyeEnum.getSynchronizeService(type)
        .orElseThrow(() -> new RuntimeException("类型不存在"));
    // 同步用户数据
    synchronizeService.sync(userId, projectId, declareId);
}

当我们需要再添加一个 C 类用户的同步注册的时候,只需要前往枚举类添加:

/**
 * 类型C的用户
 */
C("c", (userId, projectId, declareId) -> {
    return SpringUtils.getBean(CUserService.class).sync(userId, projectId, declareId);
});

即可,register()方法就不需要再做修改了。


喜欢 (0)