我们在开发过程中经常会用到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的唯一性是确定我们拿到的对象是正确的
希望对大家能够有所帮助!