package com.ycl.aspect;
|
|
import com.ycl.annotation.LogSave;
|
import com.ycl.bo.AdminUserDetails;
|
import com.ycl.entity.user.UmsAdminLoginLog;
|
import com.ycl.mapper.user.UmsAdminLoginLogMapper;
|
import org.aspectj.lang.JoinPoint;
|
import org.aspectj.lang.annotation.Aspect;
|
import org.aspectj.lang.annotation.Before;
|
import org.aspectj.lang.annotation.Pointcut;
|
import org.aspectj.lang.reflect.MethodSignature;
|
import org.springframework.security.core.context.SecurityContextHolder;
|
import org.springframework.stereotype.Component;
|
import org.springframework.web.context.request.RequestContextHolder;
|
import org.springframework.web.context.request.ServletRequestAttributes;
|
|
import javax.annotation.Resource;
|
import javax.servlet.http.HttpServletRequest;
|
import java.lang.reflect.Method;
|
import java.util.Date;
|
|
@Component
|
@Aspect
|
public class LogAspect {
|
|
@Resource
|
UmsAdminLoginLogMapper umsAdminLoginLogMapper;
|
|
@Pointcut("@annotation(com.ycl.annotation.LogSave)")
|
public void pointCut() {
|
}
|
|
@Before("pointCut()")
|
public void handle(JoinPoint joinPoint) {
|
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
|
Method method = signature.getMethod();
|
LogSave annotation = method.getAnnotation(LogSave.class);
|
AdminUserDetails adminUserDetails = (AdminUserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
|
UmsAdminLoginLog loginLog = new UmsAdminLoginLog();
|
loginLog.setAdminId(adminUserDetails.getUserId());
|
loginLog.setCreateTime(new Date());
|
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
HttpServletRequest request = attributes.getRequest();
|
loginLog.setIp(request.getRemoteAddr());
|
String header = request.getHeader("user-agent");
|
if (header.contains("Windows NT")) {
|
loginLog.setUserAgent("Web端");
|
}
|
else if(header.contains("Android")) {
|
loginLog.setUserAgent("Android端");
|
}
|
else if(header.contains("iPhone")) {
|
loginLog.setUserAgent("ios端");
|
}
|
else if(header.contains("iPad")) {
|
loginLog.setUserAgent("iPad端");
|
}else {
|
loginLog.setUserAgent("Web端");
|
}
|
loginLog.setOperationType(annotation.operationType());
|
loginLog.setContain(annotation.contain());
|
umsAdminLoginLogMapper.insert(loginLog);
|
}
|
|
}
|
