Spring5 webFIux Demo

package com.example.demo.client.proxy; import com.example.demo.client.annotation.APIServer; import com.example.demo.client.beans.MethodInfo; import com.example.demo.client.beans.ServerInfo; import com.example.demo.client.client.RestHandler; import com.example.demo.client.client.WebClientRestHandler; import lombok.extern.slf4j.Slf4j; import org.springframework.http.HttpMethod; import org.springframework.web.bind.annotation.DeleteMapping; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.PutMapping; import org.springframework.web.bind.annotation.RequestBody; import reactor.core.publisher.Flux; import reactor.core.publisher.Mono; import java.lang.annotation.Annotation; import java.lang.reflect.InvocationHandler; import java.lang.reflect.Method; import java.lang.reflect.Parameter; import java.lang.reflect.ParameterizedType; import java.lang.reflect.Proxy; import java.lang.reflect.Type; import java.util.Arrays; import java.util.LinkedHashMap; import java.util.Map; /** * jdk动态代理类对象 */ @Slf4j public class JdkDynamicAopProxy implements ProxyCreator { RestHandler restHandler = new WebClientRestHandler(); @Override public Object createProxy(Class<?> type) { // 抽取服务器信息 ServerInfo serverInfo = extractServerInfo(type); log.info("serverInfo :{}", serverInfo); // 初始化服务器信息（webClient） restHandler.init(serverInfo); return Proxy.newProxyInstance(this.getClass().getClassLoader(), new Class[]{type}, new InvocationHandler() { @Override public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { // 抽取所有请求信息封装到 MethodInfo 对象 MethodInfo methodInfo = extractMethodInfo(method, args); log.info("methodInfo :{}", methodInfo); // 调用Rest客户端 处理请求 return restHandler.invokeRest(methodInfo); } private MethodInfo extractMethodInfo(Method method, Object[] args) { MethodInfo methodInfo = new MethodInfo(); // 封装url和请求方式 extractRequsetUrlAndMethod(method, methodInfo); // 封装请求参数和body extractRequestParamsAndBody(method, args, methodInfo); // 封装返回信息 extractReturnInfo(method, methodInfo); return methodInfo; } /** * 抽取返回的是Mono还是Flux和返回对象的类型 * @param method * @param methodInfo */ private void extractReturnInfo(Method method, MethodInfo methodInfo) { // 返回Mono还是Flux // isAssignableFrom() 判断类型是否是某个类的子类 vs instanceof 判断实例是否是某个的子类 boolean isFlux = method.getReturnType().isAssignableFrom(Flux.class); methodInfo.setReturnFlux(isFlux); // 得到返回对象的实际类型 Class<?> elementType = extractElementType(method.getGenericReturnType()); methodInfo.setReturnElementType(elementType); } /** *得到泛型的实际类型 * @param genericReturnType * @return */ private Class<?> extractElementType(Type genericReturnType) { Type[] actualTypeArguments = ((ParameterizedType) genericReturnType).getActualTypeArguments(); return (Class<?>) actualTypeArguments[0]; } /** * 抽取请求参数和请求body * @param method * @param args * @param methodInfo */ private void extractRequestParamsAndBody(Method method, Object[] args, MethodInfo methodInfo) { Map<String, Object> params = new LinkedHashMap<>(); Parameter[] parameters = method.getParameters(); for (int i = 0; i < parameters.length; i++) { // 参数信息 PathVariable annotation = parameters[i].getAnnotation(PathVariable.class); if (annotation != null) { params.put(annotation.value(), args[i]); methodInfo.setParams(params); } // 请求body和body类型 RequestBody requestBody = parameters[i].getAnnotation(RequestBody.class); if (requestBody != null) { methodInfo.setBody((Mono<?>) args[i]); methodInfo.setBodyElementType(extractElementType(parameters[i].getParameterizedType())); } } } /** * 抽取请求url 和请求方式 * @param method * @param methodInfo */ private void extractRequsetUrlAndMethod(Method method, MethodInfo methodInfo) { Annotation[] annotations = method.getAnnotations(); Arrays.stream(annotations).forEach(annotation -> { if (annotation instanceof GetMapping) { GetMapping getMethod = (GetMapping) annotation; methodInfo.setUrl(getMethod.value()[0]); methodInfo.setHttpMethod(HttpMethod.GET); } else if (annotation instanceof PostMapping) { PostMapping postMethod = (PostMapping) annotation; methodInfo.setUrl(postMethod.value()[0]); methodInfo.setHttpMethod(HttpMethod.POST); } else if (annotation instanceof DeleteMapping) { DeleteMapping deleteMethod = (DeleteMapping) annotation; methodInfo.setUrl(deleteMethod.value()[0]); methodInfo.setHttpMethod(HttpMethod.DELETE); } else if (annotation instanceof PutMapping) { PutMapping putMethod = (PutMapping) annotation; methodInfo.setUrl(putMethod.value()[0]); methodInfo.setHttpMethod(HttpMethod.PUT); } }); } }); } /** * 封装服务器信息 * * @param type * @return */ private ServerInfo extractServerInfo(Class<?> type) { String url = type.getAnnotation(APIServer.class).value(); return ServerInfo.builder().url(url).build(); } }