从零学elk_从零学ELK系列（十一）:SpringBoot项目接入ELK超级版（超详细图文教程）...

/** Copyright (c) 2020. zhanghan_java@163.com All Rights Reserved.* 项目名称：SpringBoot项目接入ELK* 类名称：FileBeatLogUtil.java* 创建人：张晗* 联系方式：zhanghan_java@163.com* 开源地址: https://github.com/dangnianchuntian/springboot* 博客地址: https://zhanghan.blog.csdn.net*/

package com.zhanghan.zhelkboot.util;

import com.alibaba.fastjson.JSON;

import com.zhanghan.zhelkboot.util.wrapper.Wrapper;

import org.aspectj.lang.JoinPoint;

import org.aspectj.lang.reflect.MethodSignature;

import org.slf4j.Logger;

import org.slf4j.MDC;

import org.springframework.util.StringUtils;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import java.text.SimpleDateFormat;

import java.util.*;

public class FileBeatLogUtil {

public static void writeRequestInfo(HttpServletRequest request, String applicationName, String reqName, String requestParams) {

String requestURI = request.getRequestURI();

//获取requestHeader Enumeration requestHeaderNames = request.getHeaderNames();

Map reuestHeaderMap = new HashMap<>();

while (requestHeaderNames.hasMoreElements()) {

String name = requestHeaderNames.nextElement();

String value = request.getHeaders(name).nextElement();

reuestHeaderMap.put(name, value);

}

String requestHeader = "";

if (null != reuestHeaderMap && reuestHeaderMap.size() > 0) {

requestHeader = JSON.toJSONString(reuestHeaderMap);

}

//防止MDC值空指针，所有入参不为null applicationName = StringUtils.isEmpty(applicationName) ? "" : applicationName;

requestURI = StringUtils.isEmpty(requestURI) ? "" : requestURI;

reqName = StringUtils.isEmpty(reqName) ? "" : reqName;

requestParams = "null".equals(requestParams) ? "" : requestParams;

//MDC值为ES键值对JSON信息 MDC.put("applicationName", applicationName);

MDC.put("requestTime", getStringTodayTime());

MDC.put("requestURI", requestURI);

MDC.put("requestHeader", requestHeader);

MDC.put("sourceName", reqName);

MDC.put("requestParams", requestParams);

MDC.put("exceptionCount", "0");

}

public static void writeExceptionLog(String exceptionMethodName, String exceptionMethodArgs, String exceptionMessage) {

MDC.put("exceptionCount", "1");

exceptionMessage = String.format("MethodName:%s;Args:%s;Exception:%s", exceptionMethodName, exceptionMethodArgs, exceptionMessage);

//MDC值为ES键值对JSON信息 MDC.put("exceptionMessage", exceptionMessage);

}

public static void writeResponseLog(Object o, Logger log, HttpServletResponse response) {

//取responseHeader内容 Map responseHeaderMap = new HashMap<>();

Collection headerNames = response.getHeaderNames();

headerNames.forEach(name -> {

responseHeaderMap.put(name, response.getHeader(name));

});

String strResponseHeader = "";

if (null != responseHeaderMap && responseHeaderMap.size() > 0) {

strResponseHeader = JSON.toJSONString(responseHeaderMap);

}

//获取response内容 String responseCode = "";

String responseMsg = "";

String responseBody = "";

Wrapper wrapper;

if (null != o) {

wrapper = (Wrapper) o;

if (null != wrapper) {

responseCode = String.valueOf(wrapper.getCode());

responseMsg = wrapper.getMessage();

Object result = wrapper.getResult();

if (null != result) {

responseBody = result.toString();

}

}

}

//MDC值为ES键值对JSON信息 MDC.put("responseHeader", strResponseHeader);

MDC.put("responseCode", responseCode);

MDC.put("responseMsg", responseMsg);

MDC.put("responseBody", responseBody);

MDC.put("responseTime", getStringTodayTime());

Map copyOfContextMap = MDC.getCopyOfContextMap();

String reqInfoJsonStr = JSON.toJSONString(copyOfContextMap);

log.info(reqInfoJsonStr);

}

/*** 获取请求参数，处理为json字符串** @param joinPoint* @return*/

public static String getParams(JoinPoint joinPoint) {

Object[] argValues = joinPoint.getArgs();

String[] argNames = ((MethodSignature) joinPoint.getSignature()).getParameterNames();

LinkedHashMap linkedHashMap = new LinkedHashMap<>();

if (argNames != null && argNames.length > 0) {

for (int i = 0; i < argNames.length; i++) {

String thisArgName = argNames[i];

String thisArgValue = argValues[i].toString();

linkedHashMap.put(thisArgName, thisArgValue);

}

}

return JSON.toJSONString(linkedHashMap);

}

public static String getStringTodayTime() {

Date todat_date = new Date();

//将日期格式化 SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");

//转换成字符串格式 return simpleDateFormat.format(todat_date);

}

}