基于Java的面向REST API高性能认证鉴权框架设计源码

package com.usthe.sureness.matcher.util; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.util.HashMap; import java.util.Map; import java.util.Queue; import java.util.Set; import java.util.LinkedList; import java.util.List; import java.util.regex.Pattern; /** * Improved dictionary matching tree * support regular: str*str, *, ** * the * in str*str can match zero or more character eg: *.html -> content.html * the * can match zero or one directory * the ** can match zero or more directories * Match priority: Raw string > str*str > * > ** * @author tomsun28 * @date 19:25 2019-01-18 */ public class TirePathTree { private static final Logger logger = LoggerFactory.getLogger(TirePathTree.class); private static final String NODE_TYPE_PATH_NODE = "pathNode"; private static final String NODE_TYPE_MAY_PATH_END = "mayPathEnd"; private static final String NODE_TYPE_METHOD = "methodNode"; private static final String NODE_TYPE_FILTER_ROLES = "rolesNode"; private static final String URL_PATH_SPLIT = "/"; private static final String MATCH_ALL_METHOD = "*"; private static final String MATCH_ONE_PATH = "*"; private static final String MATCH_MULTI_PATH = "**"; private static final int PATH_NODE_NUM_3 = 3; private static final int PATH_NODE_NUM_2 = 2; private static final Pattern PATH_SPLIT_PATTERN = Pattern.compile("/+"); /** * str - Pattern.compile(str) */ private static final Map<String, Pattern> PATTERN_MAP = new HashMap<>(8); /** * root node */ private volatile Node root; public TirePathTree() { this.root = new Node("root"); } /** * build dictionary matching tree * @param paths resource path set */ public synchronized void buildTree(Set<String> paths) { if (logger.isDebugEnabled()) { logger.debug("sureness - start buildTree..."); } clearTree(); for (String path : paths) { insertNode(path, this.root); } if (logger.isDebugEnabled()) { logger.debug("sureness - buildTree finish"); } } /** * rebuild and update dictionary matching tree * Concurrency type:RCU -- read copy update * Ensure that reading is not affected during reconstruction * @param paths resources paths set */ public synchronized void rebuildTree(Set<String> paths) { if (logger.isDebugEnabled()) { logger.debug("sureness - start rebuildTree..., try rcu current way"); } Node buildRoot = new Node("root"); for (String path : paths) { insertNode(path, buildRoot); } this.root = buildRoot; if (logger.isDebugEnabled()) { logger.debug("sureness - rebuildTree finish"); } } /** * clear dictionary matching tree */ public void clearTree() { if (logger.isDebugEnabled()) { logger.debug("sureness - clearTree"); } root.getChildren().clear(); } /** * Get the number of resources (URL+METHOD) in the current matching tree * @return int resource number */ public int getResourceNum() { int resourceNum = 0; // Breadth First Search - bfs Queue<Node> resourceList = new LinkedList<>(); resourceList.add(root); while (!resourceList.isEmpty()) { Node currentNode = resourceList.poll(); if (currentNode.getMethodChildren() != null && !currentNode.getMethodChildren().isEmpty()) { resourceNum += currentNode.getMethodChildren().size(); } if (currentNode.getChildren() != null && !currentNode.getChildren().isEmpty()) { resourceList.addAll(currentNode.getChildren().values()); } } if (logger.isDebugEnabled()) { logger.debug("sureness - current PathTree resource num is: {}", resourceNum); } return resourceNum; } /** * Use the resource path to match supported roles in tree * @param path /api/v2/host/detail===get * @return java.lang.String roles eg: [role1,role2] */ public String searchPathFilterRoles(String path) { if (path == null || "".equals(path)) { return null; } if (logger.isTraceEnabled()) { logger.trace("sureness - searchPathFilterRoles, path is {}", path); } if (!path.startsWith(URL_PATH_SPLIT)) { path = URL_PATH_SPLIT.concat(path); } path = PATH_SPLIT_PATTERN.matcher(path).replaceAll("/"); path = path.substring(1).toLowerCase(); String[] tmp = path.split("==="); if (tmp.length != PATH_NODE_NUM_2) { return null; } String[] urlPac = tmp[0].split("/"); String method = tmp[1]; // Pattern matching * ** Node current = root; return searchPathRoleInChildren(current, urlPac, -1, method); } /** * Find the leaf node of the corresponding branch from the current node * @param current current node * @param urlPac urlPath arr * @param currentFlow current Flow * @param method http method: post get delete put... * @return match return roles eg:[role,role2], else return null */ private String searchPathRole(Node current, String[] urlPac, int currentFlow, String method) { if (current == null || urlPac == null || currentFlow >= urlPac.length || currentFlow < 0 || method == null || "".equals(method)) { return null; } String currentNodeData = current.getData(); // fast fail if (isNoMatchString(currentNodeData, urlPac[currentFlow])) { return null; } if (currentFlow == urlPac.length - 1 && (NODE_TYPE_MAY_PATH_END.equals(current.getNodeType()))) { Node methodNode = current.getMethodChild(method); if (methodNode != null) { if (NODE_TYPE_METHOD.equals(methodNode.getNodeType())) { return methodNode.getChildren().keySet().iterator().next(); } } else { Node nextNode = current.getMethodChild(MATCH_ALL_METHOD); if (nextNode != null && NODE_TYPE_METHOD.equals(nextNode.getNodeType())) { return nextNode.getChildren().keySet().iterator().next(); } nextNode = current.getChildren().get(MATCH_ONE_PATH); if (nextNode != null && NODE_TYPE_MAY_PATH_END.equals(nextNode.getNodeType())) { methodNode = nextNode.getMethodChild(method); methodNode = methodNode == null ? nextNode.getMethodChild(MATCH_ALL_METHOD) : methodNode; if (methodNode != null && NODE_TYPE_METHOD.equals(methodNode.getNodeType())) { return methodNode.getChildren().keySet().iterator().next(); } } nextNode = current.getChildren().get(MATCH_MULTI_PATH); if (nextNode != null && NODE_TYPE_MAY_PATH_END.equals(nextNode.getNodeType())) { methodNode = nextNode.getMethodChild(method); methodNode = methodNode == null ? nextNode.getMethodChild(MATCH_ALL_METHOD) : methodNode; if (methodNode != null && NODE_TYPE_METHOD.equals(methodNode.getNodeType())) { return methodNode.getChildren().keySet().iterator().next(); } } } } String matchRole = null; if (currentNodeData.equals(urlPac[currentFlow]) || isPatternStr(currentNodeData)) { matchRole = searchPathRoleInChildren(current, urlPac, currentFlow, method); if (matchRole != null) { return matchRole; } } if (currentNodeData.equals(MATCH_