Skip to content

Y
yzg-util
Commit 01bd4fe7 authored by yanzg's avatar yanzg
Browse files
Options

拦截器

parent e6a2add1
Hide whitespace changes
Inline Side-by-side
Showing with 214 additions and 114 deletions
yzg-util-cloud/src/main/java/com/yanzuoguang/cloud/aop/BaseRequestAspect.java 0 → 100644
View file @ 01bd4fe7
package com.yanzuoguang.cloud.aop;
import com.alibaba.fastjson.JSON;
import com.yanzuoguang.util.cache.MemoryCache;
import com.yanzuoguang.util.contants.ResultConstants;
import com.yanzuoguang.util.helper.StringHelper;
import com.yanzuoguang.util.thread.ThreadNext;
import com.yanzuoguang.util.vo.LogVo;
import com.yanzuoguang.util.vo.ResponseResult;
import org.aspectj.lang.ProceedingJoinPoint;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.ApplicationContext;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.LinkedBlockingQueue;
public class BaseRequestAspect implements ThreadNext.Next {
@Value("${spring.application.name}")
protected String applicationName;
@Value("${yzg.logAll:false}")
protected boolean logAll = false;
@Value("${yzg.cacheTime:120}")
protected int cacheTime = 120;
@Autowired
protected ApplicationContext context;
protected LogFeign logFeign;
/**
* 缓存队列
*/
protected volatile LinkedBlockingQueue<LogVo> cache = new LinkedBlockingQueue<LogVo>();
// 缓存的每次请求的结果
protected MemoryCache<RequestCacheResult> _cacheResult = new MemoryCache<>(cacheTime);
public BaseRequestAspect() {
ThreadNext.start(this, "save log error");
}
/**
* 添加日志到缓存中，并不是立即添加，而是通过线程执行，防止对环境造成影响
*
* @param logVo
*/
public void addLog(LogVo logVo) {
cache.add(logVo);
}
/**
* 初始化日志Vo
*
* @param url 请求路径
* @param joinPoint 请求参数
* @param responseResult 返回参数
* @return
*/
protected LogVo initLogInterVo(String url, ProceedingJoinPoint joinPoint, ResponseResult responseResult) {
LogVo logInterVo = new LogVo();
logInterVo.setLogId(StringHelper.getNewID());
logInterVo.setLogSources(applicationName);//平台名
logInterVo.setInterUrl(url);//请求URL
List<Object> para = new ArrayList<>();
for (Object item : joinPoint.getArgs()) {
if (item instanceof HttpServlet || item instanceof HttpServletResponse || item instanceof HttpServletRequest) {
continue;
}
para.add(item);
}
Object paraTo = para;
if (para.size() == 1) {
paraTo = para.get(0);
}
logInterVo.setContent(JSON.toJSONString(paraTo));//请求参数
logInterVo.setResult(JSON.toJSONString(responseResult));//返回参数
logInterVo.setStatus(responseResult.getCode() == ResultConstants.SUCCESS ? 1 : 0);
return logInterVo;
}
/**
* 执行下一个函数,出现异常会继续执行
*
* @return 是否继续执行
*/
@Override
public boolean next() {
while (cache.size() > 0) {
LogVo item = cache.poll();
if (item != null) {
if (logFeign == null) {
logFeign = context.getBean(LogFeign.class);
}
logFeign.save(item);
}
}
return true;
}
/**
* 沉睡时间
*
* @return
*/
@Override
public int getNextTime() {
return 1000;
}
}
yzg-util-cloud/src/main/java/com/yanzuoguang/cloud/aop/FeignAspect.java
View file @ 01bd4fe7
......@@ -17,7 +17,7 @@ import org.springframework.stereotype.Component;
*/
@Aspect
@Component
public class FeignAspect extends AbstractValidateAspect {
public class FeignAspect extends HttpAspectUtil {
private static final Logger logger = LoggerFactory.getLogger(FeignAspect.class);
......
yzg-util-cloud/src/main/java/com/yanzuoguang/cloud/aop/AbstractValidateAspect.java → yzg-util-cloud/src/main/java/com/yanzuoguang/cloud/aop/HttpAspectUtil.java
View file @ 01bd4fe7
......@@ -9,14 +9,14 @@ import javax.servlet.http.HttpServletRequest;
/**
* 虚拟拦截类
*/
public abstract class AbstractValidateAspect {
public class HttpAspectUtil {
/**
* 获取 http request
*
* @return
*/
protected static HttpServletRequest getRequest() {
public static HttpServletRequest getRequest() {
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
if (attributes == null) {
throw new CodeException("监视错误");
......@@ -29,7 +29,7 @@ public abstract class AbstractValidateAspect {
*
* @return
*/
protected static String getHttpRequestUrl() {
public static String getHttpRequestUrl() {
return getRequest().getServletPath();
}
}
yzg-util-cloud/src/main/java/com/yanzuoguang/cloud/aop/MqAspect.java 0 → 100644
View file @ 01bd4fe7
package com.yanzuoguang.cloud.aop;
import com.yanzuoguang.util.contants.ResultConstants;
import com.yanzuoguang.util.exception.ExceptionHelper;
import com.yanzuoguang.util.helper.StringHelper;
import com.yanzuoguang.util.log.Log;
import com.yanzuoguang.util.vo.LogVo;
import com.yanzuoguang.util.vo.ResponseResult;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
/**
* LogsAspect（接口请求日志切面）
*
* @author: Kang
* @time: 2018年04月25日 11:43
*/
@Aspect
@Component
public class MqAspect extends BaseRequestAspect {
private static final Logger logger = LoggerFactory.getLogger(MqAspect.class);
/**
* exec aop point aspect
*/
@Pointcut("execution(* *..mq..*Consumer.*(..))")
public void webAspect() {
}
/**
* 执行环形切面
*
* @param joinPoint
* @return
*/
@Around(value = "webAspect()")
public Object requestWebAround(ProceedingJoinPoint joinPoint) throws Throwable {
Log.threadBegin();
long start = System.currentTimeMillis();
String name = joinPoint.getSignature().getName();
try {
Object ret = executeMethod(joinPoint, name);
long end = System.currentTimeMillis();
logger.error("[ {} ] time ({})ms, result is {}", name, (end - start), ret);
return ret;
} catch (Exception e) {
long end = System.currentTimeMillis();
logger.error("[ {} ] time ({})ms, is error {}", name, (end - start), e);
// 消费消息出错
ResponseResult responseResult = ExceptionHelper.getError(e);
saveInterLogs(joinPoint, responseResult);
Log.error(joinPoint.getSignature().getDeclaringType(), e);
throw e;
} finally {
Log.threadCommit();
}
}
private Object executeMethod(ProceedingJoinPoint joinPoint, String name) throws Throwable {
return joinPoint.proceed();
}
/**
* deal with inter api logs
*
* @param joinPoint 请求参数
* @param responseResult 返回参数
*/
private void saveInterLogs(ProceedingJoinPoint joinPoint, ResponseResult responseResult) {
// 日志请求不记录,防止死循环递归
if (applicationName.indexOf("log") >= 0) {
return;
}
// 正常请求不记录
if (!logAll && responseResult.getCode() == ResultConstants.SUCCESS) {
return;
}
LogVo logVo = initLogInterVo(StringHelper.EMPTY, joinPoint, responseResult);
addLog(logVo);
}
}
yzg-util-cloud/src/main/java/com/yanzuoguang/cloud/aop/WebAspect.java
View file @ 01bd4fe7
package com.yanzuoguang.cloud.aop;
import com.alibaba.fastjson.JSON;
import com.yanzuoguang.util.cache.MemoryCache;
import com.yanzuoguang.util.contants.ResultConstants;
import com.yanzuoguang.util.exception.ExceptionHelper;
import com.yanzuoguang.util.helper.JSONHelper;
import com.yanzuoguang.util.helper.StringHelper;
import com.yanzuoguang.util.log.Log;
import com.yanzuoguang.util.base.ObjectHelper;
import com.yanzuoguang.util.thread.ThreadNext;
import com.yanzuoguang.util.vo.LogVo;
import com.yanzuoguang.util.vo.ResponseResult;
import org.aspectj.lang.ProceedingJoinPoint;
......@@ -19,21 +16,12 @@ import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.lang.reflect.Method;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.LinkedBlockingQueue;
/**
* LogsAspect（接口请求日志切面）
......@@ -43,35 +31,9 @@ import java.util.concurrent.LinkedBlockingQueue;
*/
@Aspect
@Component
public class WebAspect extends AbstractValidateAspect implements ThreadNext.Next {
public class WebAspect extends BaseRequestAspect {
private static final Logger logger = LoggerFactory.getLogger(WebAspect.class);
@Value("${spring.application.name}")
private String applicationName;
@Value("${yzg.logAll:false}")
private boolean logAll = false;
@Value("${yzg.cacheTime:120}")
private int cacheTime = 120;
@Autowired
private ApplicationContext context;
private LogFeign logFeign;
/**
* 缓存队列
*/
private volatile LinkedBlockingQueue<LogVo> cache = new LinkedBlockingQueue<LogVo>();
// 缓存的每次请求的结果
private MemoryCache<RequestCacheResult> _cacheResult = new MemoryCache<>(cacheTime);
public WebAspect() {
ThreadNext.start(this, "save log error");
}
private static final Logger logger = LoggerFactory.getLogger(BaseRequestAspect.class);
/**
* exec aop point aspect
......@@ -153,7 +115,7 @@ public class WebAspect extends AbstractValidateAspect implements ThreadNext.Next
// 请求编号和公司编号挂钩
reqId = StringHelper.getId(ObjectHelper.getString(firstArgs, "companyId"), reqId);
}
String reqFull = StringHelper.getId(reqId, getHttpRequestUrl());
String reqFull = StringHelper.getId(reqId, HttpAspectUtil.getHttpRequestUrl());
RequestCacheResult req = _cacheResult.get(reqFull, new RequestCacheResult(reqFull));
// 缓存的键值对
if (req.getResult() == null) {
......@@ -190,75 +152,7 @@ public class WebAspect extends AbstractValidateAspect implements ThreadNext.Next
if (!logAll && responseResult.getCode() == ResultConstants.SUCCESS) {
return;
}
LogVo logVo = initLogInterVo(getHttpRequestUrl(), joinPoint, responseResult);
LogVo logVo = initLogInterVo(HttpAspectUtil.getHttpRequestUrl(), joinPoint, responseResult);
addLog(logVo);
}
/**
* 添加日志到缓存中，并不是立即添加，而是通过线程执行，防止对环境造成影响
*
* @param logVo
*/
public void addLog(LogVo logVo) {
cache.add(logVo);
}
/**
* 初始化日志Vo
*
* @param url 请求路径
* @param joinPoint 请求参数
* @param responseResult 返回参数
* @return
*/
private LogVo initLogInterVo(String url, ProceedingJoinPoint joinPoint, ResponseResult responseResult) {
LogVo logInterVo = new LogVo();
logInterVo.setLogId(StringHelper.getNewID());
logInterVo.setLogSources(applicationName);//平台名
logInterVo.setInterUrl(url);//请求URL
List<Object> para = new ArrayList<>();
for (Object item : joinPoint.getArgs()) {
if (item instanceof HttpServlet || item instanceof HttpServletResponse || item instanceof HttpServletRequest) {
continue;
}
para.add(item);
}
Object paraTo = para;
if (para.size() == 1) {
paraTo = para.get(0);
}
logInterVo.setContent(JSON.toJSONString(paraTo));//请求参数
logInterVo.setResult(JSON.toJSONString(responseResult));//返回参数
logInterVo.setStatus(responseResult.getCode() == ResultConstants.SUCCESS ? 1 : 0);
return logInterVo;
}
/**
* 执行下一个函数,出现异常会继续执行
*
* @return 是否继续执行
*/
@Override
public boolean next() {
while (cache.size() > 0) {
LogVo item = cache.poll();
if (item != null) {
if (logFeign == null) {
logFeign = context.getBean(LogFeign.class);
}
logFeign.save(item);
}
}
return true;
}
/**
* 沉睡时间
*
* @return
*/
@Override
public int getNextTime() {
return 1000;
}
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment