登陆记录切面

登陆记录切面

import javax.servlet.http.HttpServletRequest;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

/**
 * 登陆记录切面
 * @author wanlf
 */
public class LoginHistoryAop {

	private static Logger logger = LoggerFactory.getLogger(LoginHistoryAop.class);

	@Autowired
	private LoginHistoryService loginHistoryService;

	@Pointcut("execution(* com.xxx.controller.LoginController.login(..))")
	public void controller() {
	}

	@AfterReturning("controller()")
	private void afterLogin(JoinPoint joinPoint) {
		ServletRequestAttributes sra = (ServletRequestAttributes) 
				(RequestContextHolder.getRequestAttributes());
		HttpServletRequest request = sra.getRequest();

		// 获取用户
		User loginUser = (User) SecurityUtils.getSubject().getPrincipal();
		// 获取IP
		String ip = IPUtils.getIP(request);
		// 获取UA完整信息
		String ua = request.getHeader(HttpHeaders.USER_AGENT);
		UserAgent userAgent = UserAgent.parseUserAgentString(ua);
		// 从UA中提取操作系统
		String os = userAgent.getOperatingSystem().getName();
		// 从UA中提取浏览器
		String browserName = userAgent.getBrowser().getName();
		String browserVersion = userAgent.getBrowserVersion().getVersion();

		LoginHistory loginHistory = new LoginHistory();
		loginHistory.setUser(loginUser);
		loginHistory.setIp(ip);
		loginHistory.setUa(ua);
		loginHistory.setOs(os);
		loginHistory.setBrowserName(browserName);
		loginHistory.setBrowserVersion(browserVersion);

		loginHistoryService.insert(loginHistory);
	}
}