策略模式实现

策略模式

策略模式就是定义一组策略，每一个策略之间可以相互的替换，可以根据客户端的不同需求调用不同的策略。

实现

定义接口与实现类

public interface UpateStatusChangeStrategy {public Boolean updateUserStatus(Long id);}@Service
public class UserStatusImpl implements UpateStatusChangeStrategy {@Resourceprivate UserMapper userMapper;@Overridepublic Boolean updateUserStatus(Long id) {//校验参数if (id < 0 || id == 0){throw new BusinessException(ErrorCode.PARAMS_ERROR);}User user = userMapper.selectById(id);if (user == null){throw new BusinessException(ErrorCode.PARAMS_ERROR);}//修改状态值user.setUserStatus(0);int  result = userMapper.updateById(user);if (result != 0){return true;}return false;}
}@Service
public class AdminStatusImpl implements UpateStatusChangeStrategy {@Resourceprivate UserMapper userMapper;@Overridepublic Boolean updateUserStatus(Long id) {//校验参数if (id < 0 || id == 0){throw new BusinessException(ErrorCode.PARAMS_ERROR);}User user = userMapper.selectById(id);if (user == null){throw new BusinessException(ErrorCode.PARAMS_ERROR);}//修改状态值user.setUserStatus(1);int  result = userMapper.updateById(user);if (result != 0){return true;}return false;}
}

map + 函数式编程

使用 postConstruct 初始化map

@Service
@Slf4j
public class UserServiceImpl extends ServiceImpl<UserMapper, User>implements UserService {private Map<Integer, Function<Long, Boolean>> strategyMap = new HashMap<>();@Resourceprivate UserStatusImpl userStatusImpl;@Resourceprivate AdminStatusImpl adminStatusImpl;@PostConstructpublic void strategyMapInit() {strategyMap.put(1, id -> userStatusImpl.updateUserStatus(id));strategyMap.put(2, id -> adminStatusImpl.updateUserStatus(id));}
}

使用

    /*** 修改用户状态位 -- 使用策略模式** @return*/@PutMapping("/adminStatus")public BaseResponse<Boolean> adminStatus(Long id, Integer type) {if (id <= 0) {throw new BusinessException(ErrorCode.PARAMS_ERROR);}Boolean result = userService.updateUserStatusStrategy(id, type);return ResultUtils.success(result);}}@Overridepublic Boolean updateUserStatusStrategy(Long id, Integer type) {Function<Long, Boolean> function = strategyMap.get(type);if (function == null)throw new BusinessException(ErrorCode.SYSTEM_ERROR);Boolean result = function.apply(id);if (result == null)throw new BusinessException(ErrorCode.SYSTEM_ERROR);return result;}

总结

定义一组策略（通过一个接口+多个实现类），为了减少 if - else 的时候，使用 map + 函数式编程存储策略方法，并且进行初始化。这样就不要使用 if - else 了，如果后期需要使用单例模式的形式，只创建使用的对象，这就下一步的优化的地方