package com.walker.tcp.util;
|
|
import com.walker.infrastructure.utils.StringUtils;
|
import com.walker.tcp.Message;
|
import com.walker.tcp.ProtocolResolver;
|
import com.walker.tcp.TcpRequest;
|
import com.walker.tcp.protocol.MessageProtocolResolver;
|
import org.reflections.Reflections;
|
|
import java.util.Arrays;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Set;
|
|
public class ConvertorUtils {
|
|
/**
|
* 返回制定包下面的所有<code>TcpRequest</code>注解类集合
|
* @param packageName
|
* @return
|
*/
|
public static Map<String, String> scanTcpRequestAnnotation(String packageName){
|
Reflections reflections = new Reflections(packageName);
|
Set<Class<?>> classesList = reflections.getTypesAnnotatedWith(TcpRequest.class);
|
|
if(classesList != null){
|
Map<String, String> map = new HashMap<>();
|
TcpRequest annotation = null;
|
for(Class<?> c : classesList){
|
annotation = c.getAnnotation(TcpRequest.class);
|
if(annotation == null){
|
continue;
|
}
|
if(StringUtils.isEmpty(annotation.value())){
|
throw new IllegalArgumentException("TcpRequest注解没有设置value，必须配置。class = " + c.getName());
|
}
|
map.put(annotation.value(), c.getName());
|
}
|
return map;
|
}
|
return null;
|
}
|
|
// /**
|
// * 根据报文内容，找到使用的协议解析器。</p>
|
// * 目前系统根据后缀来查找
|
// * @param message
|
// * @return
|
// */
|
// @Deprecated
|
//// public static final ProtocolResolver<?> getProtocolResolver(String message, List<ProtocolResolver<?>> resolverList){
|
// public static final ProtocolResolver<?> getProtocolResolver(String message, List<ProtocolResolver<?>> resolverList){
|
//// if(cacheList == null){
|
//// throw new IllegalArgumentException(ProtocolResolver.ERR_NOFOUND);
|
//// }
|
//// for(ProtocolResolver<?> pr : resolverList){
|
// for(ProtocolResolver<?> pr : resolverList){
|
// if(pr instanceof MessageProtocolResolver){
|
// continue;
|
// }
|
// if(message.endsWith(pr.getDelimiter())){
|
// if(pr.isRequireFeatureResolve()){
|
// if(pr.getProtocolFeature() == null){
|
// throw new IllegalArgumentException(ProtocolResolver.ERR_NO_FEATURE + pr.getName());
|
// }
|
// if(message.startsWith(pr.getProtocolFeature().toString())){
|
// return pr;
|
// } else {
|
// continue;
|
// }
|
// } else {
|
// return pr;
|
// }
|
// }
|
// }
|
// return null;
|
// }
|
|
public static final ProtocolResolver<?> getProtocolResolver(Object msg, List<ProtocolResolver<?>> resolverList){
|
if(msg == null){
|
throw new IllegalArgumentException("msg(Object) is required!");
|
}
|
|
Object feature = null;
|
String msgStr = null;
|
|
for(ProtocolResolver<?> pr : resolverList){
|
if(msg instanceof Message){
|
// 如果不行可以用这个判断类型
|
// Message.class.isAssignableFrom(msg.getClass())
|
Message message = (Message)msg;
|
if(pr.isOnlyMatchFeature()){
|
if(pr instanceof MessageProtocolResolver){
|
feature = pr.getProtocolFeature();
|
if(feature != null){
|
if(Arrays.equals(message.getFeature(), (byte[])feature)){
|
return pr;
|
}
|
}
|
continue;
|
|
} else {
|
// 如果是其他 MessageProtocolResolver 自定义实现，再扩展
|
throw new UnsupportedOperationException("需要扩展报文解析类型：" + message.toString());
|
}
|
} else {
|
// 可能需要报文分隔符来进一步确定类型，可能是长报文类型，暂未实现
|
throw new UnsupportedOperationException("未实现的报文解析类型：" + message.toString());
|
}
|
} else {
|
// 这里其他情况目前只有字符串类型
|
if(pr instanceof MessageProtocolResolver){
|
continue;
|
}
|
|
msgStr = msg.toString();
|
if(msgStr.endsWith(pr.getDelimiter())){
|
if(pr.isRequireFeatureResolve()){
|
if(pr.getProtocolFeature() == null){
|
throw new IllegalArgumentException(ProtocolResolver.ERR_NO_FEATURE + pr.getName());
|
}
|
if(msgStr.startsWith(pr.getProtocolFeature().toString())){
|
return pr;
|
} else {
|
continue;
|
}
|
} else {
|
return pr;
|
}
|
}
|
}
|
// if(message.endsWith(pr.getDelimiter())){
|
// if(pr.isRequireFeatureResolve()){
|
// if(pr.getProtocolFeature() == null){
|
// throw new IllegalArgumentException(ProtocolResolver.ERR_NO_FEATURE + pr.getName());
|
// }
|
// if(message.startsWith(pr.getProtocolFeature())){
|
// return pr;
|
// } else {
|
// continue;
|
// }
|
// } else {
|
// return pr;
|
// }
|
// }
|
}
|
return null;
|
}
|
|
public static void main(String[] args){
|
Map<String, String> map = scanTcpRequestAnnotation("com.walker.tcp");
|
if(map == null){
|
return;
|
}
|
for(Map.Entry<String, String> entry : map.entrySet()){
|
System.out.println(entry.getKey() + ", " + entry.getValue());
|
}
|
}
|
}
|
