Last active
June 19, 2020 11:47
-
-
Save greenyleaf/086a88fd2783a9830d18eb772cbb55d0 to your computer and use it in GitHub Desktop.
an interceptor in Spring MVC for logging user access, using DAO persistence and Logback logging. Process the forwarded request.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package top.sdrkyj.custom.config; | |
import top.sdrkyj.custom.entity.AccessLog; | |
import top.sdrkyj.custom.entity.Account; | |
import top.sdrkyj.custom.service.AccessLogService; | |
import org.apache.shiro.SecurityUtils; | |
import org.apache.shiro.subject.PrincipalCollection; | |
import org.slf4j.Logger; | |
import org.slf4j.LoggerFactory; | |
import org.springframework.stereotype.Component; | |
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter; | |
import javax.servlet.RequestDispatcher; | |
import javax.servlet.http.HttpServletRequest; | |
import javax.servlet.http.HttpServletResponse; | |
import javax.servlet.http.HttpSession; | |
import java.text.MessageFormat; | |
import java.util.Date; | |
@Component | |
public class AccessInterceptor extends HandlerInterceptorAdapter { | |
private static final Logger logger = LoggerFactory.getLogger(AccessInterceptor.class); | |
private static final String FORWARD_MARK = "top.sdrkyj.custom.config.AccessInterceptor_afterCompletion"; | |
private final AccessLogService accessLogService; | |
public AccessInterceptor(AccessLogService accessLogService) { | |
this.accessLogService = accessLogService; | |
} | |
@Override | |
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) { | |
if (request.getAttribute(FORWARD_MARK) != null) { | |
return; | |
} | |
request.setAttribute(FORWARD_MARK, true); | |
HttpSession session = request.getSession(false); | |
PrincipalCollection principals = SecurityUtils.getSubject().getPrincipals(); | |
String username = null; | |
if (principals != null) { | |
Account account = principals.oneByType(Account.class); | |
if (account != null) { | |
username = account.getUsername(); | |
} | |
} | |
String xForwardedFor = request.getHeader("X-Forwarded-For"); | |
String userId = request.getRemoteUser(); | |
String referer = request.getHeader("Referer"); | |
String userAgent = request.getHeader("User-Agent"); | |
String forwardRequestUri = (String) request.getAttribute(RequestDispatcher.FORWARD_REQUEST_URI); | |
String requestUri = forwardRequestUri != null ? forwardRequestUri : request.getRequestURI(); | |
String forwardQueryString = (String) request.getAttribute(RequestDispatcher.FORWARD_QUERY_STRING); | |
String queryString = forwardQueryString != null ? forwardQueryString : request.getQueryString(); | |
logger.info("{} {} " + | |
"{} " + | |
"[{}] " + | |
"\"{} {} {} {}\" " + | |
"{} {} " + | |
"{} " + | |
"\"{}\" \"{}\"", | |
xForwardedFor != null ? xForwardedFor : request.getRemoteAddr(), userId != null ? userId : "", | |
username == null ? "" : username, | |
MessageFormat.format("{0,date,dd/MM/yyyy:HH:mm:ss Z}", System.currentTimeMillis()), | |
request.getMethod(), requestUri, queryString != null ? queryString : "", request.getProtocol(), | |
response.getStatus(), request.getContentLength(), | |
session != null ? session.getId() : '-', | |
referer != null ? referer : "", userAgent != null ? userAgent : ""); | |
accessLogService.create(new AccessLog(request.getRemoteAddr(), userId, | |
username, new Date(), request.getMethod(), requestUri, queryString, request.getProtocol(), | |
response.getStatus(), request.getContentLength(), | |
session != null ? session.getId() : null, referer, userAgent, xForwardedFor)); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment