溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章主要介紹“java如何實現統一打印入參出參等日志”的相關知識，小編通過實際案例向大家展示操作過程，操作方法簡單快捷，實用性強，希望這篇“java如何實現統一打印入參出參等日志”文章能幫助大家解決問題。
SpringBoot項目中，之前都是在controller方法的第一行手動打印 log，return之前再打印返回值。有多個返回點時，就需要出現多少重復代碼，過多的非業務代碼顯得十分凌亂。
本文將采用AOP 配置自定義注解實現 入參、出參的日志打印（方法的入參和返回值都采用 fastjson 序列化）。
將特定包下所有的controller生成代理類對象，并交由Spring容器管理，并重寫invoke方法進行增強(入參、出參的打印).
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Import({InteractRecordBeanPostProcessor.class})
public @interface EnableInteractRecord {
/**
* app對應controller包名
*/
String[] basePackages() default {};
/**
* 排除某些包
*/
String[] exclusions() default {};
}作用：獲取EnableInteractRecord注解對象，用于獲取需要創建代理對象的包名，以及需要排除的包名
@Component
public class InteractRecordFactoryPostProcessor implements BeanFactoryPostProcessor {
private static Logger logger = LoggerFactory.getLogger(InteractRecordFactoryPostProcessor.class);
private EnableInteractRecord enableInteractRecord;
@Override
public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
try {
String[] names = beanFactory.getBeanNamesForAnnotation(EnableInteractRecord.class);
for (String name : names) {
enableInteractRecord = beanFactory.findAnnotationOnBean(name, EnableInteractRecord.class);
logger.info("開啟交互記錄 ", enableInteractRecord);
}
} catch (Exception e) {
logger.error("postProcessBeanFactory() Exception ", e);
}
}
public EnableInteractRecord getEnableInteractRecord() {
return enableInteractRecord;
}
}作用：進行入參、出參打印，包含是否打印邏輯
@Component
public class ControllerMethodInterceptor implements MethodInterceptor {
private static Logger logger = LoggerFactory.getLogger(ControllerMethodInterceptor.class);
// 請求開始時間
ThreadLocal<Long> startTime = new ThreadLocal<>();
private String localIp = "";
@PostConstruct
public void init() {
try {
localIp = InetAddress.getLocalHost().getHostAddress();
} catch (UnknownHostException e) {
logger.error("本地IP初始化失敗 : ", e);
}
}
@Override
public Object invoke(MethodInvocation invocation) {
pre(invocation);
Object result;
try {
result = invocation.proceed();
post(invocation, result);
return result;
} catch (Throwable ex) {
logger.error("controller 執行異常: ", ex);
error(invocation, ex);
}
return null;
}
public void error(MethodInvocation invocation, Throwable ex) {
String msgText = ex.getMessage();
logger.info(startTime.get() + " 異常，請求結束");
logger.info("RESPONSE : " + msgText);
logger.info("SPEND TIME : " + (System.currentTimeMillis() - startTime.get()));
}
private void pre(MethodInvocation invocation) {
long now = System.currentTimeMillis();
startTime.set(now);
logger.info(now + " 請求開始");
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = attributes.getRequest();
logger.info("URL : " + request.getRequestURL().toString());
logger.info("HTTP_METHOD : " + request.getMethod());
logger.info("REMOTE_IP : " + getRemoteIp(request));
logger.info("LOCAL_IP : " + localIp);
logger.info("METHOD : " + request.getMethod());
logger.info("CLASS_METHOD : " + getTargetClassName(invocation) + "." + invocation.getMethod().getName());
// 獲取請求頭header參數
Map<String, String> map = new HashMap<String, String>();
Enumeration<String> headerNames = request.getHeaderNames();
while (headerNames.hasMoreElements()) {
String key = (String) headerNames.nextElement();
String value = request.getHeader(key);
map.put(key, value);
}
logger.info("HEADERS : " + JSONObject.toJSONString(map));
Date createTime = new Date(now);
// 請求報文
Object[] args = invocation.getArguments();// 參數
String msgText = "";
Annotation[][] annotationss = invocation.getMethod().getParameterAnnotations();
for (int i = 0; i < args.length; i++) {
Object arg = args[i];
if (!(arg instanceof ServletRequest)
&& !(arg instanceof ServletResponse)
&& !(arg instanceof Model)) {
RequestParam rp = null;
Annotation[] annotations = annotationss[i];
for (Annotation annotation : annotations) {
if (annotation instanceof RequestParam) {
rp = (RequestParam) annotation;
}
}
if (msgText.equals("")) {
msgText += (rp != null ? rp.value() + " = " : " ") + JSONObject.toJSONString(arg);
} else {
msgText += "," + (rp != null ? rp.value() + " = " : " ") + JSONObject.toJSONString(arg);
}
}
}
logger.info("PARAMS : " + msgText);
}
private void post(MethodInvocation invocation, Object result) {
logger.info(startTime.get() + " 請求結束");
if (!(result instanceof ModelAndView)) {
String msgText = JSONObject.toJSONString(result);
logger.info("RESPONSE : " + msgText);
}
logger.info("SPEND TIME : " + (System.currentTimeMillis() - startTime.get()));
}
private String getRemoteIp(HttpServletRequest request) {
String remoteIp = null;
String remoteAddr = request.getRemoteAddr();
String forwarded = request.getHeader("X-Forwarded-For");
String realIp = request.getHeader("X-Real-IP");
if (realIp == null) {
if (forwarded == null) {
remoteIp = remoteAddr;
} else {
remoteIp = remoteAddr + "/" + forwarded.split(",")[0];
}
} else {
if (realIp.equals(forwarded)) {
remoteIp = realIp;
} else {
if (forwarded != null) {
forwarded = forwarded.split(",")[0];
}
remoteIp = realIp + "/" + forwarded;
}
}
return remoteIp;
}
private String getTargetClassName(MethodInvocation invocation) {
String targetClassName = "";
try {
targetClassName = AopTargetUtils.getTarget(invocation.getThis()).getClass().getName();
} catch (Exception e) {
targetClassName = invocation.getThis().getClass().getName();
}
return targetClassName;
}
}AopTargetUtils：
public class AopTargetUtils {
/**
* 獲取 目標對象
* @param proxy 代理對象
* @return
* @throws Exception
*/
public static Object getTarget(Object proxy) throws Exception {
if(!AopUtils.isAopProxy(proxy)) {
return proxy;//不是代理對象
}
if(AopUtils.isJdkDynamicProxy(proxy)) {
return getJdkDynamicProxyTargetObject(proxy);
} else { //cglib
return getCglibProxyTargetObject(proxy);
}
}
private static Object getCglibProxyTargetObject(Object proxy) throws Exception {
Field h = proxy.getClass().getDeclaredField("CGLIB$CALLBACK_0");
h.setAccessible(true);
Object dynamicAdvisedInterceptor = h.get(proxy);
Field advised = dynamicAdvisedInterceptor.getClass().getDeclaredField("advised");
advised.setAccessible(true);
Object target = ((AdvisedSupport)advised.get(dynamicAdvisedInterceptor)).getTargetSource().getTarget();
return getTarget(target);
}
private static Object getJdkDynamicProxyTargetObject(Object proxy) throws Exception {
Field h = proxy.getClass().getSuperclass().getDeclaredField("h");
h.setAccessible(true);
AopProxy aopProxy = (AopProxy) h.get(proxy);
Field advised = aopProxy.getClass().getDeclaredField("advised");
advised.setAccessible(true);
Object target = ((AdvisedSupport)advised.get(aopProxy)).getTargetSource().getTarget();
return getTarget(target);
}
}作用：篩選出需要生成代理的類，并生成代理類，返回給Spring容器管理。
public class InteractRecordBeanPostProcessor implements BeanPostProcessor {
private static Logger logger = LoggerFactory.getLogger(InteractRecordBeanPostProcessor.class);
@Autowired
private InteractRecordFactoryPostProcessor interactRecordFactoryPostProcessor;
@Autowired
private ControllerMethodInterceptor controllerMethodInterceptor;
private String BASE_PACKAGES[];//需要攔截的包
private String EXCLUDING[];// 過濾的包
//一層目錄匹配
private static final String ONE_REGEX = "[a-zA-Z0-9_]+";
//多層目錄匹配
private static final String ALL_REGEX = ".*";
private static final String END_ALL_REGEX = "*";
@PostConstruct
public void init() {
EnableInteractRecord ir = interactRecordFactoryPostProcessor.getEnableInteractRecord();
BASE_PACKAGES = ir.basePackages();
EXCLUDING = ir.exclusions();
}
@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
try {
if (interactRecordFactoryPostProcessor.getEnableInteractRecord() != null) {
// 根據注解配置的包名記錄對應的controller層
if (BASE_PACKAGES != null && BASE_PACKAGES.length > 0) {
Object proxyObj = doEnhanceForController(bean);
if (proxyObj != null) {
return proxyObj;
}
}
}
} catch (Exception e) {
logger.error("postProcessAfterInitialization() Exception ", e);
}
return bean;
}
@Override
public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
return bean;
}
private Object doEnhanceForController(Object bean) {
String beanPackageName = getBeanPackageName(bean);
if (StringUtils.isNotBlank(beanPackageName)) {
for (String basePackage : BASE_PACKAGES) {
if (matchingPackage(basePackage, beanPackageName)) {
if (EXCLUDING != null && EXCLUDING.length > 0) {
for (String excluding : EXCLUDING) {
if (matchingPackage(excluding, beanPackageName)) {
return bean;
}
}
}
Object target = null;
try {
target = AopTargetUtils.getTarget(bean);
} catch (Exception e) {
logger.error("AopTargetUtils.getTarget() exception", e);
}
if (target != null) {
boolean isController = target.getClass().isAnnotationPresent(Controller.class);
boolean isRestController = target.getClass().isAnnotationPresent(RestController.class);
if (isController || isRestController) {
ProxyFactory proxy = new ProxyFactory();
proxy.setTarget(bean);
proxy.addAdvice(controllerMethodInterceptor);
return proxy.getProxy();
}
}
}
}
}
return null;
}
private static boolean matchingPackage(String basePackage, String currentPackage) {
if (StringUtils.isEmpty(basePackage) || StringUtils.isEmpty(currentPackage)) {
return false;
}
if (basePackage.indexOf("*") != -1) {
String patterns[] = StringUtils.split(basePackage, ".");
for (int i = 0; i < patterns.length; i++) {
String patternNode = patterns[i];
if (patternNode.equals("*")) {
patterns[i] = ONE_REGEX;
}
if (patternNode.equals("**")) {
if (i == patterns.length - 1) {
patterns[i] = END_ALL_REGEX;
} else {
patterns[i] = ALL_REGEX;
}
}
}
String basePackageRegex = StringUtils.join(patterns, "\\.");
Pattern r = Pattern.compile(basePackageRegex);
Matcher m = r.matcher(currentPackage);
return m.find();
} else {
return basePackage.equals(currentPackage);
}
}
private String getBeanPackageName(Object bean) {
String beanPackageName = "";
if (bean != null) {
Class<?> beanClass = bean.getClass();
if (beanClass != null) {
Package beanPackage = beanClass.getPackage();
if (beanPackage != null) {
beanPackageName = beanPackage.getName();
}
}
}
return beanPackageName;
}
}@EnableInteractRecord(basePackages = “com.test.test.controller”,exclusions = “com.test.demo.controller”)
以上即可實現入參、出參日志統一打印，并且可以將特定的controller集中管理，并不進行日志的打印（及不進生成代理類）。
實際開發中，特定不需要打印日志的接口，無法統一到一個包下。大部分需要打印的接口，和不需要打印的接口，大概率會參雜在同一個controller中，根據以上設計思路，無法進行區分。
解決辦法：
自定義排除入參打印注解
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface ExcludeReqLog {
}自定義排除出參打印注解
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface ExcludeRespLog {
}增加邏輯
// 1.在解析requestParam之前進行判斷
Method method = invocation.getMethod();
Annotation[] declaredAnnotations = method.getDeclaredAnnotations();
boolean flag = true;
for (Annotation annotation : declaredAnnotations) {
if (annotation instanceof ExcludeReqLog) {
flag = false;
}
}
if (!flag) {
logger.info("該方法已排除，不打印入參");
return;
}
// 2.在解析requestResp之前進行判斷
Method method = invocation.getMethod();
Annotation[] declaredAnnotations = method.getDeclaredAnnotations();
boolean flag = true;
for (Annotation annotation : declaredAnnotations) {
if (annotation instanceof ExcludeRespLog) {
flag = false;
}
}
if (!flag) {
logger.info("該方法已排除，不打印出參");
return;
}使用方法
// 1.不打印入參
@PostMapping("/uploadImg")
@ExcludeReqLog
public Result<List<Demo>> uploadIdeaImg(@RequestParam(value = "imgFile", required = false) MultipartFile[] imgFile) {
return demoService.uploadIdeaImg(imgFile);
}
//2.不打印出參
@PostMapping("/uploadImg")
@ExcludeRespLog
public Result<List<Demo>> uploadIdeaImg(@RequestParam(value = "imgFile", required = false) MultipartFile[] imgFile) {
return demoService.uploadIdeaImg(imgFile);
}關于“java如何實現統一打印入參出參等日志”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識，可以關注億速云行業資訊頻道，小編每天都會為大家更新不同的知識點。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
