基于Java11全新架构的Redkale微服务开源框架设计源码

/* * To change this license header, choose License Headers in Project Properties. * To change this template file, choose Tools | Templates * and open the template in the editor. */ package org.redkale.net.http; import java.io.*; import java.lang.annotation.*; import static java.lang.annotation.ElementType.TYPE; import static java.lang.annotation.RetentionPolicy.RUNTIME; import java.lang.reflect.*; import java.net.InetSocketAddress; import java.nio.channels.CompletionHandler; import java.util.*; import java.util.concurrent.CompletionStage; import org.redkale.annotation.*; import org.redkale.annotation.Comment; import org.redkale.asm.*; import static org.redkale.asm.ClassWriter.COMPUTE_FRAMES; import static org.redkale.asm.Opcodes.*; import org.redkale.asm.Type; import org.redkale.convert.*; import org.redkale.convert.json.*; import org.redkale.inject.ResourceFactory; import org.redkale.mq.spi.MessageAgent; import org.redkale.net.*; import org.redkale.net.sncp.Sncp; import org.redkale.service.*; import org.redkale.source.Flipper; import org.redkale.util.*; import org.redkale.util.RedkaleClassLoader.DynBytesClassLoader; import static org.redkale.util.Utility.isEmpty; /** * 详情见: https://redkale.org * * @author zhangjx */ @SuppressWarnings("unchecked") public final class Rest { // 请求是否为rpc协议，值类型: 布尔，取值为true、false public static final String REST_HEADER_RPC = "Rest-Rpc"; // traceid，值类型: 字符串 public static final String REST_HEADER_TRACEID = "Rest-Traceid"; // 当前用户ID值，值类型: 字符串 public static final String REST_HEADER_CURRUSERID = "Rest-Curruserid"; // 请求所需的RestService的资源名，值类型: 字符串 public static final String REST_HEADER_RESNAME = "Rest-Resname"; // 请求参数的反序列化种类，值类型: 字符串，取值为ConvertType枚举值名 public static final String REST_HEADER_REQ_CONVERT = "Rest-Req-Convert"; // 响应结果的序列化种类，值类型: 字符串，取值为ConvertType枚举值名 public static final String REST_HEADER_RESP_CONVERT = "Rest-Resp-Convert"; // --------------------------------------------------------------------------------------------------- static final String REST_TOSTRINGOBJ_FIELD_NAME = "_redkale_toStringSupplier"; static final String REST_CONVERT_FIELD_PREFIX = "_redkale_restConvert_"; static final String REST_SERVICE_FIELD_NAME = "_redkale_service"; static final String REST_SERVICEMAP_FIELD_NAME = "_redkale_serviceMap"; // 如果只有name=""的Service资源，则实例中_servicemap必须为null private static final String REST_PARAMTYPES_FIELD_NAME = "_redkale_paramTypes"; // 存在泛型的参数数组 Type[][] 第1维度是方法的下标， 第二维度是参数的下标 private static final String REST_RETURNTYPES_FIELD_NAME = "_redkale_returnTypes"; // 存在泛型的结果数组 private static final java.lang.reflect.Type TYPE_RETRESULT_STRING = new TypeToken<RetResult<String>>() {}.getType(); private static final Set<String> EXCLUDERMETHODS = new HashSet<>(); static { for (Method m : Object.class.getMethods()) { EXCLUDERMETHODS.add(m.getName()); } } /** 用于标记由Rest.createRestServlet 方法创建的RestServlet */ @Inherited @Documented @Target({TYPE}) @Retention(RUNTIME) public static @interface RestDyn { // 是否不需要解析HttpHeader，对应HttpContext.lazyHeaders boolean simple() default false; // 动态生成的类的子类需要关联一下，否则在运行过程中可能出现NoClassDefFoundError Class[] types() default {}; } /** 用于标记由Rest.createRestServlet 方法创建的RestServlet */ @Inherited @Documented @Target({TYPE}) @Retention(RUNTIME) public static @interface RestDynSourceType { Class value(); } private Rest() {} public static JsonFactory createJsonFactory(RestConvert[] converts, RestConvertCoder[] coders) { return createJsonFactory(-1, converts, coders); } public static JsonFactory createJsonFactory(int features, RestConvert[] converts, RestConvertCoder[] coders) { if (Utility.isEmpty(converts) && Utility.isEmpty(coders)) { return JsonFactory.root(); } final JsonFactory childFactory = JsonFactory.create(); if (features > -1) { childFactory.withFeatures(features); } List<Class> types = new ArrayList<>(); Set<Class> reloadTypes = new HashSet<>(); if (coders != null) { for (RestConvertCoder rcc : coders) { reloadTypes.add(rcc.type()); childFactory.register(rcc.type(), rcc.field(), (SimpledCoder) Creator.create(rcc.coder()).create()); } } if (converts != null) { for (RestConvert rc : converts) { if (rc.type() == void.class || rc.type() == Void.class) { return JsonFactory.create().skipAllIgnore(true); } if (types.contains(rc.type())) { throw new RestException("@RestConvert type(" + rc.type() + ") repeat"); } if (rc.skipIgnore()) { childFactory.registerSkipIgnore(rc.type()); childFactory.reloadCoder(rc.type()); } else if (rc.onlyColumns().length > 0) { childFactory.registerIgnoreAll(rc.type(), rc.onlyColumns()); childFactory.reloadCoder(rc.type()); } else { childFactory.register(rc.type(), false, rc.convertColumns()); childFactory.register(rc.type(), true, rc.ignoreColumns()); childFactory.reloadCoder(rc.type()); } types.add(rc.type()); if (rc.features() > -1) { childFactory.withFeatures(rc.features()); } } } for (Class type : reloadTypes) { childFactory.reloadCoder(type); } return childFactory; } static String getWebModuleNameLowerCase(Class<? extends Service> serviceType) { final RestService controller = serviceType.getAnnotation(RestService.class); if (controller == null) { return serviceType.getSimpleName().replaceAll("Service.*$", "").toLowerCase(); } if (controller.ignore()) { return null; } return (!controller.name().isEmpty()) ? controller.name().trim() : serviceType.getSimpleName().replaceAll("Service.*$", "").toLowerCase(); } static String getWebModuleName(Class<? extends Service> serviceType) { final RestService controller = serviceType.getAnnotation(RestService.class); if (controller == null) { return serviceType.getSimpleName().replaceAll("Service.*$", ""); } if (controller.ignore()) { return null; } return (!controller.name().isEmpty()) ? controller.name().trim() : serviceType.getSimpleName().replaceAll("Service.*$", ""); } /** * 判断HttpServlet是否为Rest动态生成的 * * @param servlet 检测的HttpServlet * @return 是否是动态生成的RestHttpServlet */ public static boolean isRestDyn(HttpServlet servlet) { return servlet.getClass().getAnnotation(RestDyn.class) != null; } /** * 判断HttpServlet是否为Rest动态生成的,且simple, 不需要读取http-header的方法视为simple=true * * @param servlet 检测的HttpServlet * @return 是否是动态生成的RestHttpServlet */ static boolean isSimpleRestDyn(HttpServlet servlet) {