package com.qxueyou.scc.web;
|
|
import java.io.IOException;
|
|
import javax.servlet.Filter;
|
import javax.servlet.FilterChain;
|
import javax.servlet.FilterConfig;
|
import javax.servlet.ServletException;
|
import javax.servlet.ServletRequest;
|
import javax.servlet.ServletResponse;
|
import javax.servlet.annotation.WebFilter;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import javax.servlet.http.HttpSession;
|
|
import org.apache.commons.lang3.ArrayUtils;
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
|
import com.alibaba.druid.util.StringUtils;
|
import com.alibaba.fastjson.JSONObject;
|
import com.qxueyou.scc.base.model.Result;
|
import com.qxueyou.scc.base.model.UserInfoWrapper;
|
import com.qxueyou.scc.base.service.ILicenseService;
|
import com.qxueyou.scc.config.SccConfig;
|
import com.qxueyou.scc.sys.service.IAuthorizeService;
|
|
@WebFilter(urlPatterns = "/*", filterName = "authorizeFilter")
|
@EnableConfigurationProperties(SccConfig.class)
|
public class AuthorizeFilter implements Filter {
|
private final Logger log = LogManager.getLogger("AuthorizeFilter");
|
|
@Autowired
|
private IAuthorizeService authorizeService;
|
|
@Autowired
|
private SccConfig sccConfig;
|
|
@Autowired
|
ILicenseService licenseService;
|
|
@Autowired
|
private StringRedisTemplate stringRedisTemplate;
|
|
private String[] excludeUrls;
|
|
/**
|
* ·Ç·¨ÇëÇ󷵻ؽá¹û
|
*/
|
private String forbidden = JSONObject.toJSONString(new Result(false,Result.CODE_FORBIDDEN));
|
|
private String forbiddenOfInvalidLicense = JSONObject.toJSONString(new Result(false,Result.CODE_FORBIDDEN,"license invalid"));
|
|
@Override
|
public void init(FilterConfig filterConfig) throws ServletException {
|
// ´ÓÅäÖÃÎļþÖлñÈ¡excludeUrls
|
String excludeUrl = sccConfig.getExcludeUrl();
|
|
if (!StringUtils.isEmpty(excludeUrl)) {
|
excludeUrls = excludeUrl.split(";");
|
String root = filterConfig.getServletContext().getContextPath();
|
for (int i = 0; i < excludeUrls.length; i++) {
|
excludeUrls[i] = excludeUrls[i].replace("\n", "").replace("\t", "").replace(" ", "");
|
excludeUrls[i] = root.concat(excludeUrls[i]);
|
}
|
}
|
|
log.debug("license info: valid:{0},companyName:{1},productName:{2},expiredDate{3}",
|
licenseService.isValid(),
|
licenseService.getCustomerName(),
|
licenseService.getProductName(),
|
licenseService.getExpiredDate()
|
);
|
}
|
|
@Override
|
public void doFilter(ServletRequest request, ServletResponse response, FilterChain filterChain)
|
throws IOException, ServletException {
|
|
HttpServletRequest httpRequest = (HttpServletRequest) request;
|
HttpServletResponse httpResponse = (HttpServletResponse) response;
|
|
String url = httpRequest.getRequestURI();
|
log.debug("request URI:" + url);
|
|
// ÊÇ·ñÂÔ¹ý
|
if (ArrayUtils.isEmpty(excludeUrls) || isExclude(httpRequest)) {
|
filterChain.doFilter(httpRequest, httpResponse);
|
return;
|
}
|
|
//ÊÇ·ñºÏ·¨
|
// if(!licenseService.isValid()) {
|
// log.debug("license invalid");
|
// forbiddenAsLicense(httpResponse);
|
// return;
|
// }
|
|
if (!addClientInfo(httpRequest)) {
|
log.debug("·Ç·¨ÇëÇó:" + httpRequest.getRequestURI());
|
forbidden(httpResponse);
|
return;
|
}
|
filterChain.doFilter(request, response);
|
}
|
|
private void forbidden(HttpServletResponse httpResponse) {
|
try {
|
httpResponse.getWriter().append(forbidden);
|
} catch (IOException e) {
|
log.error(e,e);
|
}
|
|
}
|
|
private void forbiddenAsLicense(HttpServletResponse httpResponse) {
|
try {
|
httpResponse.getWriter().append(forbiddenOfInvalidLicense);
|
} catch (IOException e) {
|
log.error(e,e);
|
}
|
|
}
|
|
|
|
/**
|
* °ó¶¨¿Í»§¶ËÐÅÏ¢
|
*
|
* @param httpRequest
|
* @param cookieKey
|
* @return
|
*/
|
private boolean addClientInfo(HttpServletRequest httpRequest) {
|
HttpSession session = httpRequest.getSession(true);
|
UserInfoWrapper wrapper = (UserInfoWrapper) session.getAttribute(UserInfoWrapper.SESSION_USER_INFO_KEY);
|
|
if (wrapper == null) {
|
return false;
|
}
|
|
//»ñÈ¡Óû§¶ÔÓ¦µÄsessionIdÊÇ·ñÓë±£´æÔÚredisÖеÄÒ»Ö£¬Èç¹û²»Ò»Ö£¬Èç¹û²»Ò»ÖÂÔòÌøתµ½µÇ¼ҳÃæ
|
//String currOnlineSessinId = (String) stringRedisTemplate.opsForHash().get(UserInfoWrapper.REDIS_USER_ONLINE_MAP_KEY, wrapper.getInfo(UserInfoWrapper.INF_USER_ID));
|
|
//Èç¹û²»´æÔÚ,Ôò·µ»Ø
|
// if(!session.getId().equalsIgnoreCase(currOnlineSessinId)){
|
// session.invalidate();
|
// return false;
|
// }
|
|
authorizeService.addClientInfo(wrapper);
|
return true;
|
}
|
|
/**
|
* ÊÇ·ñ¹ýÂ˵ô
|
*
|
* @param httpRequest
|
* @return
|
*/
|
private boolean isExclude(HttpServletRequest httpRequest) {
|
String url = httpRequest.getRequestURI();
|
for (String excludeUrl : excludeUrls) {
|
if (url.startsWith(excludeUrl)) {
|
return true;
|
}
|
}
|
return false;
|
}
|
|
|
@Override
|
public void destroy() {
|
System.out.println("destroy authorizeFilter");
|
}
|
|
}
|
