Spring的getBeansOfType函数结合策略及工厂模式简化if else逻辑

        我们在开发过程中经常会用到if else这样的判断,可有的场景比较复杂,若嵌套多层的if语句不太友好,降低了代码的可读性、扩展性;这时就可以考虑使用工厂+策略模式解决

大家可看一下优化之前的代码真是一言难尽 ......

 

 再看优化后的,差距一目了然

AbstractWebSocketHandle webSocketHandle = webSocketFactory.create(url);
log.info("create返回值:{}",webSocketHandle);
if (null == webSocketHandle){
	log.error("doMessage webSocketHandle is null url={}",url);
		return;
	}
webSocketHandle.doMessage(jsonObject,userId,url);

 1、策略模式简介

        定义:策略模式(Strategy Pattern)也叫政策模式(Policy Pattern),它是将定义的算法家族、分别封装起来,让它们之间可以互相替换,从而让算法的变化不会影响到使用算法的用户,属于行为型模式。

        应用场景:同一个问题,有多种处理模式,每一个都能独立使用;算法需要自由切换时等等

2、getBeansOfType        

<T> Map<String, T> getBeansOfType(@Nullable Class<T> var1) throws BeansException;

         该方法用于获取 Spring 容器中指定类型的所有 JavaBean 对象,前提是这些对象必须由Spring容器管理

3、定义策略类(WebSocketHandle)

public interface WebSocketHandle {

    /**
     *  处理消息
     * @param webSocketSession 会话
     * @param userId 用户ID
     * @param url 地址
     */
    void doMessage(JSONObject jsonObject, String userId, String url) throws Exception;

    /**
     * 获取地址
     * @return
     */
    String getUrl();
}

4、定义抽象类(AbstractWebSocketHandle)

public abstract class AbstractWebSocketHandle implements WebSocketHandle{

    @Autowired
    StringRedisTemplate stringRedisTemplate;

    @Autowired
    RocketApiService rocketApiService;

    /**
     * 发送广播消息
     * @param userId
     * @param message
     */
    public void sendMessage(String userId,String message){
        Map<String,String> map=new HashMap<>();
        map.put("userId",userId);
        map.put("message",message);
        map.put("flag","send");
        stringRedisTemplate.convertAndSend(Constants.REDIS_CHANNEL_SEND, JSONObject.toJSONString(map));
    }

    // 获取在线用户ID
    public List<String> getOnLineUserIds(){
        // 模糊查询key
        Set<String> keys = stringRedisTemplate.keys("*"+"online" + "*");
        List<String> result=new ArrayList<>();
        keys.forEach(s -> {
            String s1 = stringRedisTemplate.opsForValue().get(s);
            result.add(s1);
        });
        return result;
    }

    // 发送消息失败时调用
    public void sendMessageError(Message mes, JSONObject jsonObject, String url, String userId) throws IOException {
        ResultBack error = ResultUtils.rocketError(false, url);
        JSONObject parseObject = JSONObject.parseObject(JSONObject.toJSONString(error));
        parseObject.put("rid",jsonObject.getString("rid"));
        parseObject.put("msg",mes.getMsg());
        parseObject.put("timestamp",jsonObject.getString("timestamp"));
        sendMessage(userId,parseObject.toString());
    }


}

AbstractWebSocketHandle类实现WebSocketHandle接口,方法可不重写;可将公共的方法放入抽象类中方便使用

5、实现类(例GetRooms)

@Slf4j
@Service
public class GetRooms extends AbstractWebSocketHandle{

    @Override
    public void doMessage(JSONObject jsonObject, String userId, String url) {
        Object getRooms = rocketApiService.GetRooms(userId);
        log.info("作战室列表返回:"+getRooms);
        ResultBack success = ResultUtils.rocketSuccess(true, url, getRooms);
        Object json = JSONObject.toJSON(success);
        sendMessage(userId,json.toString());
    }

    @Override
    public String getUrl() {
        return "GetRooms";
    }
}

实现类继承抽象类并重写策略类的接口,不同的业务逻辑在doMessage中处理,我写了有将近二十个实现类,每一个处理不同的业务

注意:不要忘记加@Service注解;getUrl中的返回值要唯一

6、工厂类(WebSocketFactory)

@Slf4j
@Component
public class WebSocketFactory implements ApplicationContextAware {

    private static ApplicationContext context;

    public AbstractWebSocketHandle create(String url){
        // 返回WebSocketHandle接口的全部实现类(前提是所有实现类都必须由Spring IoC容器管理)
        Map<String, ?> beansMap = context.getBeansOfType(WebSocketHandle.class);
        log.info("beansMap:{}",beansMap);
        if (beansMap == null || beansMap.isEmpty()) {
            return null;
        }
        Iterator<?> iterator = beansMap.entrySet().iterator();
        Map.Entry<String, WebSocketHandle> entry;
        // 迭代,根据URL判断返回对应的bean对象
        while (iterator.hasNext()) {
            entry = (Map.Entry<String, WebSocketHandle>) iterator.next();
            if (url.equals(entry.getValue().getUrl())) {
                return (AbstractWebSocketHandle)entry.getValue();
            }
        }
        log.warn("AbstractWebSocketHandle url:{}not found!", url);
        return null;
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        WebSocketFactory.context = applicationContext;
    }
}

ApplicationContextAware接口封装了Spring的上下文,通过getBeansOfType函数可以得到所有的实现了策略类的实现类,再根据参数url进行比对返回我们想要的对象;保证url的唯一性是确定我们拿到的对象是正确的

         希望对大家能够有所帮助!

  • 4
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值