package com.ks.app.aop;
|
|
import com.ks.app.entity.AdminUser;
|
import com.ks.app.utils.ApiCodeConstant;
|
import com.ks.app.utils.SystemInfoUtil;
|
import com.ks.app.vo.AcceptAdminData;
|
import org.aspectj.lang.ProceedingJoinPoint;
|
import org.aspectj.lang.annotation.Around;
|
import org.aspectj.lang.annotation.Aspect;
|
import org.springframework.core.annotation.Order;
|
import org.springframework.security.core.Authentication;
|
import org.springframework.security.core.context.SecurityContext;
|
import org.springframework.stereotype.Component;
|
import org.springframework.web.context.request.RequestContextHolder;
|
import org.springframework.web.context.request.ServletRequestAttributes;
|
import org.yeshi.utils.JsonUtil;
|
|
import javax.servlet.http.HttpSession;
|
import java.io.IOException;
|
import java.io.PrintWriter;
|
|
//后台接口监控
|
@Component
|
@Aspect
|
@Order(2)
|
public class AdminApiFilter {
|
|
public static final String EDP = "execution(* com.ks.app.controller.admin..*.*(..))";
|
|
@Around(EDP)
|
public Object around(ProceedingJoinPoint joinPoint) throws IOException {
|
Object[] args = joinPoint.getArgs();
|
ServletRequestAttributes servletContainer = (ServletRequestAttributes) RequestContextHolder
|
.getRequestAttributes();
|
|
AcceptAdminData acceptAdminData = null;
|
for (Object obj : args) {
|
if (obj instanceof AcceptAdminData) {
|
acceptAdminData = (AcceptAdminData) obj;
|
break;
|
}
|
}
|
|
if (acceptAdminData != null) {
|
HttpSession session = servletContainer.getRequest().getSession();
|
|
acceptAdminData.setSystem(SystemInfoUtil.getAdminSelectedSystem(servletContainer.getRequest().getSession()));
|
if (acceptAdminData.getSystem() == null) {
|
PrintWriter out = servletContainer.getResponse().getWriter();
|
out.print(JsonUtil.loadFalseResult(ApiCodeConstant.CODE_ERROR_IN_SERVER, "未选择系统"));
|
out.close();
|
return null;
|
}
|
//如果用户登录了将用户信息加入进去
|
SecurityContext contextSession = (SecurityContext) session.getAttribute("SPRING_SECURITY_CONTEXT");
|
if (contextSession != null) {
|
Authentication authentication = contextSession.getAuthentication();
|
AdminUser adminUser = (AdminUser) authentication.getPrincipal();
|
acceptAdminData.setAdminUser(adminUser);
|
}
|
}
|
|
|
try {
|
return joinPoint.proceed(args);
|
} catch (Throwable e) {
|
PrintWriter out = servletContainer.getResponse().getWriter();
|
out.print(JsonUtil.loadFalseResult(ApiCodeConstant.CODE_ERROR_IN_SERVER, "服务器内部错误"));
|
out.close();
|
}
|
return null;
|
}
|
|
|
}
|
