Skip to content

Instantly share code, notes, and snippets.

@greenyleaf
Last active June 19, 2020 11:47
Show Gist options
  • Save greenyleaf/086a88fd2783a9830d18eb772cbb55d0 to your computer and use it in GitHub Desktop.
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.
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