1.定义枚举BusinessTypeEnum
@Getter
public enum BusinesTypeEnum{
SUCCEED(0, "成功"),
P_ADDRESS_IS_INCORRECT(1, "地址不正确"),
THE_ACCESS_CODE_INCORRECT(2, "接入码不正确"),
USER_NO_NOTE(3, "用户没有注册"),
PASSWORD_ERROR(4, "密码错误"),
RESOURCES_ARE_INSUFFICIENT_TO_CONNECT_LATER_ALREADY_OCCUPIED(5, "资源紧张稍后再连接(已经占用)"),
OTHER(6, "其他");
private final Integer id;
private final String value;
BusinesTypeEnum(Integer id, String value) {
this.id = id;
this.value = value;
}
public static BusinesTypeEnum getBusinesTypeEnumById(Integer id){
if(id == null){
return null;
}
for (BusinesTypeEnum businesTypeEnum :BusinesTypeEnum.values()){
if(businesTypeEnum .id.equals(id)){
return businesTypeEnum ;
}
}
return null;
}
}
2.新建BaseBusinessExecute 接口
public interface BaseBusinesExecute <T> {
/**
* 获取消息体
* @param sourceId 来源ID
* @param data解析后的消息
* @return
*/
void execute(String sourceId, T data);
/**
* 获取业务类型
* @return
*/
BusinesTypeEnum getBusinessType();
}
3.实现BaseBusinessExecute接口
@Component
public class BaseBusinesExecuteImpl implements BaseBusinesExecute<Object>{
@Override
public BusinesTypeEnum getBusinessType() {
return BusinessTypeEnum.SUCCEED;
}
@Override
public void execute(String sourceId, Object data) {
//处理业务
}
}
4.利用ApplicationListener获取对象
@Component
public class BusinesSendDataService implements ApplicationListener<ContextRefreshedEvent> {
private Map<Integer, BaseBusinesExecute> map=new ConcurrentHashMap<>();
public void executeSend(String sourceId, WebsocketDTO dto) {
if(map.containsKey(dto.getBusinessType())){
map.get(dto.getBusinessType()).execute(sourceId,dto);
}
}
public Map<Integer, BaseBusinesExecute> getMap(){
return map;
}
@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
Map<String, BaseBusinesExecute> beansOfType = event.getApplicationContext().getBeansOfType(BaseBusinesExecute.class);
beansOfType.forEach((k,v)->{
map.put(v.getMsgType().getId(),v);
});
}
}
``