每日一例 | 手写RequestParameter注解，实现有参方法调用
共 7714字，需浏览 16分钟
· 2021-06-02
前言
昨天实现了requestMapping和controller注解，解决了请求地址与接口方法之间的映射问题，可以根据前端请求地址调用对应的接口方法。但是还不能解决有参方法的调用，今天我们就来啃下这个硬碴，解决这个最后一公里。
开肝
定义RequestParameter注解
springboot的注解是RequestParam，我们今天实现的需求就是参考RequestParam，但是我没有时间去研究springboot的源码（至少今天没有时间），就先按照自己的想法来实现了。和前面注解不一样的一点是，这个注解的target指定的是ElementType.PARAMETER，因为这个注解是加在方法的参数上的。
@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface RequestParameter {
String value();
}
加到方法的参数上是这样的：
@RequestMapping("/sayHello")
public String test(@RequestParameter("name") String name) {
return "hello," + name;
}
是不是看着还挺像像样的。
业务端使用
目前业务处理还是在doDispatcher方法，所以今天依然是动这个方法。
先说下思路，思路也很简单，就是服务器收到前端请求后，根据requestMapping（请求地址）拿到对应的方法，从方法中拿到参数注解RequestParameter（这个注解的作用就是标记参数名，让我们可以根据参数名拿到参数），然后根据注解的value拿到参数名，然后根据参数名从requestAttributeMap拿到请求参数的值，组装方法入参列表。
/**
* 请求分发处理
* @throws Exception
*/
public void doDispatcher() throws Exception{
RequestHear requestHear = request.getRequestHear();
Map<String, Object> requestAttributeMap = request.getRequestAttributeMap();
logger.info("请求头信息：{}", requestHear);
logger.info("请求信息：{}", requestAttributeMap);
if (Objects.isNull(requestHear) || Objects.isNull(requestAttributeMap)) {
return;
}
String requestMapping = requestHear.getRequestMapping();
if (requestMappingMap.containsKey(requestMapping)) {
Method method = requestMappingMap.get(requestMapping);
logger.debug("method:{}", method);
Class<?> declaringClass = method.getDeclaringClass();
Annotation[][] parameterAnnotations = method.getParameterAnnotations();
Object[] parameters = new Object[parameterAnnotations.length];
for (int i = 0; i < parameterAnnotations.length; i++) {
RequestParameter annotation = (RequestParameter)parameterAnnotations[i][0];
parameters[i] = requestAttributeMap.get(annotation.value());
}
Object o = declaringClass.newInstance();
Object invoke = method.invoke(o, parameters);
logger.info("invoke:{}", invoke);
response.write(String.format("hello syskeCat, dateTime:%d\n result = %s", System.currentTimeMillis(), invoke));
} else {
response.write(404, String.format("resources not found :%d", System.currentTimeMillis()));
}
socket.close();
}
我单独把修改部分拿出来，简单解释下。
Annotation[][] parameterAnnotations = method.getParameterAnnotations();
Object[] parameters = new Object[parameterAnnotations.length];
for (int i = 0; i < parameterAnnotations.length; i++) {
RequestParameter annotation = (RequestParameter)parameterAnnotations[i][0];
parameters[i] = requestAttributeMap.get(annotation.value());
}
Object o = declaringClass.newInstance();
Object invoke = method.invoke(o, parameters);
method.getParameterAnnotations的作用是获取方法的所有参数注解，返回结果是一个二维数组，因为一个方法可以有多个参数，每个参数都可以有多个注解，所以肯定是一个二维数组。parameterAnnotations[0][0]就是第一个参数的第一个注解，parameterAnnotations[1][0]就是第二个参数的第一个注解，其他以此类推。这张图更清楚地说明了这一点：
因为获取到的注解是Annotation，并非是我们的定义的注解，所以需要进行强制转换成我们的自定义注解RequestParameter。
然后通过注解的value()方法从注解中拿到参数名，根据参数名从参数map中拿到请求的值，组装成参数集合，然后反射调用。
请求参数处理这边我们也做了一些调整，主要是为了获取请求参数参数：
Map<String, Object> attributeMap = Maps.newHashMap();
if (requestMapping.contains("?")) {
int endIndex = requestMapping.lastIndexOf('?');
String requestParameterStr = requestMapping.substring(endIndex + 1);
requestMapping = requestMapping.substring(0, endIndex);
String[] split = requestParameterStr.split("&");
for (String s : split) {
String[] split1 = s.split("=");
attributeMap.put(StringUtil.trim(split1[0]), StringUtil.trim(split1[1]));
}
}
主要就是在增加了请求参数的处理，把sayHello?name=yunzhongzhi&age=12处理成requestMapping和请求参数map。这块后期还需要优化，要把get请求和post请求分开，分别处理，所以目前我们的服务器只支持get请求。
测试
完成以上调整，我们的方法就已经支持有参调用了，我们来测试下吧！
效果还不错，目标完美达成。
总结
感觉有思路的话，写东西还是比较快的，这些需求都是今天早上实现的，中间被Annotation卡住了，一直无法获取到参数名，后来发现只要强转一下就行了。千里之行，始于足下，感兴趣的小伙伴肝起来。
下面是项目的开源仓库，有兴趣的小伙伴可以去看看，如果有想法的小伙伴，我真心推荐你自己动个手，自己写一下，真的感觉不错：
https://github.com/Syske/syske-boot
