com.yuweix.kuafu.session.conf.PathPattern Maven / Gradle / Ivy
package com.yuweix.kuafu.session.conf;
import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.List;
/**
* @author yuwei
*/
public class PathPattern {
private static final String EXTENSION_MAPPING_PATTERN = "*.";
private static final String PATTERN_MATCH_ALL = "/*";
/**
* 精确匹配
*/
private final List exactMatches = new ArrayList<>();
/**
* 前置匹配
*/
private final List startsWithMatches = new ArrayList<>();
/**
* 后缀匹配
*/
private final List endsWithMatches = new ArrayList<>();
public PathPattern(String... urlPatterns) {
for (String urlPattern : urlPatterns) {
if (urlPattern == null) {
continue;
}
addUrlPattern(urlPattern.trim());
}
}
private void addUrlPattern(String urlPattern) {
if (urlPattern.startsWith(EXTENSION_MAPPING_PATTERN)) {
this.endsWithMatches.add(urlPattern.substring(1));
} else if (urlPattern.equals(PATTERN_MATCH_ALL)) {
this.startsWithMatches.add("");
} else if (urlPattern.endsWith(PATTERN_MATCH_ALL)) {
this.startsWithMatches.add(urlPattern.substring(0, urlPattern.length() - 1));
this.exactMatches.add(urlPattern.substring(0, urlPattern.length() - 1));
this.exactMatches.add(urlPattern.substring(0, urlPattern.length() - 2));
} else {
this.exactMatches.add(urlPattern);
}
}
public boolean matches(HttpServletRequest request) {
return matches(request.getRequestURI());
}
public boolean matches(String requestPath) {
for (String pattern : this.exactMatches) {
if (pattern.equals(requestPath)) {
return true;
}
}
if (!requestPath.startsWith("/")) {
return false;
}
for (String pattern : this.endsWithMatches) {
if (requestPath.endsWith(pattern)) {
return true;
}
}
for (String pattern : this.startsWithMatches) {
if (requestPath.startsWith(pattern)) {
return true;
}
}
return false;
}
}