HXAPIGate基于Netty+Shiro开发的一款高性能API网关，对基于REST服务的细粒度API资源的权限管理平台

package hx.apigate.util; import java.util.Iterator; import java.util.List; import java.util.Map; import hx.apigate.circuitBreaker.CBManager; import hx.apigate.databridge.CircleBreakException; import org.apache.ignite.IgniteSemaphore; import org.apache.ignite.binary.BinaryObjectException; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import hx.apigate.databridge.NodeInfo; import hx.apigate.databridge.SemphareException; import hx.apigate.databridge.xmlBean.Route; import hx.apigate.databridge.xmlBean.RouteAll; import hx.apigate.databridge.xmlBean.RouteNode; import hx.apigate.socket.handlers.GatewayServerHandler; import io.netty.handler.codec.http.HttpMethod; /** * <p>Description: 路由选择工具类</p> 　 * <p>Copyright: Copyright (c) 2019</p> 　 * <p>Company: www.uiotcp.com</p> 　 * @author yangcheng 　 * @date 2019年10月29日 　 * @version 1.0 */ public class RouteSelectUtil { private static Logger logger = LoggerFactory.getLogger(RouteSelectUtil.class); public static String HTTP = "http"; public static String DUBBO = "dubbo"; public static String CIRCLE = "circle"; public static String WEIGHT = "weight"; public static String pathSeparator = "/"; /** * 根据url以及轮寻策略 选择路由的Node * @param httpMethod * @param sourceUrl 路径名称 * @return * @throws Exception */ public static Object[] selectOneNode(String sourceUrl, HttpMethod httpMethod) throws SemphareException, CircleBreakException { //获取与当前url匹配的route url String sourceUrlTemp = sourceUrl.contains("?") ? sourceUrl.split("\\?")[0] : sourceUrl; String pattern = getMatchedPattern(sourceUrlTemp,httpMethod); if(pattern == null) { return null; } Object[] ret =new Object[2]; ret[0] = pattern; NodeInfo info = getRouteByPattern(sourceUrl,sourceUrlTemp,pattern); if(info != null) { ret[1] = info; } return ret; } /** * 通过uri获取路由（不是节点node） * @param patternUri * @return */ public static IgniteSemaphore selectRouteByUri(String patternUri,String version){ //获取与当前url匹配的route url Map<String, RouteAll> map = IgniteUtil.getAPIRouteCache().get("ALL_ROUTE"); RouteAll routeAll = map.get(patternUri); int len = routeAll.getRoutes().size(); for(int i = 0 ; i < len ; i ++) { if(version.equals(routeAll.getRoutes().get(i).getVersion())) { return routeAll.getRoutes().get(i).getTps(); } } return null; } /** * 根据获取到的url的pattern 获取规约类型并返回路由信息 * @param sourceUrl 请求的url(全路径，包含get请求的参数) * @param sourceUrlTemp 将get请求的url中的参数截取掉，剩下简单的路径<br> * @param pattern 网关配置的url模版 包含请求方式 * @return */ public static NodeInfo getRouteByPattern(String sourceUrl,String sourceUrlTemp,String pattern) throws SemphareException,CircleBreakException{ //根据负载策略选择微服务routeNode RouteAll routeAll = getRouteAll4lvs(pattern) ;//IgniteUtil.getAPIRouteCache().get("ALL_ROUTE").get(pattern); Route route = routeAll.nextRoute(); //获取路由信号量 if(!route.getTps().removed() && route.getTps().tryAcquire()) { if(HTTP.equals(route.getProtocal())) { int routeNum = route.getRouteNodes().size(); for(int i = 0 ; i < routeNum ; i ++) { if(RouteSelectUtil.WEIGHT.equals(route.getStratege()) || route.getStratege() == null){ RouteNode node = route.nextNodeByWeight(); if(!route.getTps().removed() && node.getTps().tryAcquire()) { String circleBreakKey = new StringBuilder(route.getMatchUrl()).append(route.getVersion()) .append(node.getIp()).append(node.getPort()).toString(); CBManager manager = null; if(!IgniteUtil.getCircleBreakCache().containsKey(circleBreakKey)) { manager = new CBManager(circleBreakKey,node.getIntTps() < 2 ? 1 : (node.getIntTps() > 100 ? 50 : node.getIntTps() >> 1), node.getIntTps() < 5? 1 : (node.getIntTps() > 100 ? 20 : node.getIntTps() >> 2), 1000); IgniteUtil.getCircleBreakCache().putIfAbsentAsync(circleBreakKey, manager); }else { manager = IgniteUtil.getCircleBreakCache().get(circleBreakKey); } try { manager.getState().preMethodExecute(); } catch (CircleBreakException e) { node.getTps().release(); route.getTps().release(); throw e; } return new NodeInfo(route.getVersion(),route.getProtocal(), node,sourceUrl,route.isNeedAuth(),circleBreakKey); } }else{ if(routeNum > 0){ int nextIndex = route.getIndex().addAndGet(1) % routeNum; route.getIndex().set(nextIndex); RouteNode node = route.getRouteNodes().get(nextIndex); System.out.println(sourceUrl+"的node信号量为"+node.getTps().availablePermits() + "; host = " + node.getIp() +":"+ node.getPort()); if(!route.getTps().removed() && node.getTps().tryAcquire()) { String circleBreakKey = new StringBuilder(route.getMatchUrl()).append(route.getVersion()) .append(node.getIp()).append(node.getPort()).toString(); CBManager manager = null; if(!IgniteUtil.getCircleBreakCache().containsKey(circleBreakKey)) { manager = new CBManager(circleBreakKey,node.getIntTps() < 2 ? 1 : (node.getIntTps() > 100 ? 50 : node.getIntTps() >> 1), node.getIntTps() < 5? 1 : (node.getIntTps() > 100 ? 20 : node.getIntTps() >> 2), 1000); IgniteUtil.getCircleBreakCache().putIfAbsentAsync(circleBreakKey, manager); }else { manager = IgniteUtil.getCircleBreakCache().get(circleBreakKey); } try { manager.getState().preMethodExecute(); } catch (CircleBreakException e) { node.getTps().release(); route.getTps().release(); throw e; } return new NodeInfo(route.getVersion(),route.getProtocal(), node,sourceUrl,route.isNeedAuth(),circleBreakKey); } } } } }else if(DUBBO.equals(route.getProtocal())) { String[] urls = sourceUrlTemp.substring(1).split("\\/"); int size = route.getRouteNodes().size(); if(size>1) { for(int i = 0 ; i < size ; i ++) { int nextIndex = route.getIndex().addAndGet(1) % size; route.getIndex().set(nextIndex); RouteNode node = route.getRouteNodes().get(nextIndex); if(