import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.encoder.PatternLayoutEncoder;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.core.ConsoleAppender;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.MDC;

public class SimpleMDC {
    static public void main(String[] args) {
        Logger logger = LoggerFactory.getLogger(SimpleMDC.class);
        programmaticConfiguration();
        logger.info("今天的天气真不错");
        logger.info("是的, 雨好大啊");
        MDC.put("one", "one");
        MDC.put("two", "two");
        logger.info("今天的天气真不错");
        logger.debug("是的, 雨好大啊");
        MDC.put("one", "three");
        MDC.put("two", "four");
        logger.info("今天的天气真不错");
        logger.info("是的, 雨好大啊");
        // 实际使用,通常在finally中使用MDC.remove清除掉旧值
        MDC.remove("one");
        MDC.remove("two");
    }

    /**
     * 定义日志输出格式
     */
    static void programmaticConfiguration() {
        LoggerContext loggerContext = (LoggerContext) LoggerFactory.getILoggerFactory();
        loggerContext.reset();
        PatternLayoutEncoder layout = new PatternLayoutEncoder();
        layout.setContext(loggerContext);
        layout.setPattern("%X{one} %X{two} - %m%n");
        layout.start();
        ConsoleAppender<ILoggingEvent> appender = new ConsoleAppender<ILoggingEvent>();
        appender.setContext(loggerContext);
        appender.setEncoder(layout);
        appender.start();
        ch.qos.logback.classic.Logger root = (ch.qos.logback.classic.Logger) LoggerFactory.getLogger("root");
        root.addAppender(appender);
    }
}

/**
 * 日志跟踪
 */
public class TraceLogInterceptor extends HandlerInterceptorAdapter {

	/* 追踪ID */
	private static final String TRACE_KEY = "traceId";

	@Override
	public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)throws Exception {
		String a = UUID.randomUUID().toString().replace("-", "");
		MDC.put(TRACE_KEY, a);
		HttpLogLocalCache.setMsg( a);
		return true;
	}

	@Override
	public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
		MDC.clear();
	}
}

public class TraceConsumerFilter implements Filter {
    private static Logger logger = LoggerFactory.getLogger(TraceConsumerFilter.class);
    /* 追踪ID */
    private static final String TRACE_KEY = "traceId";
    @Override
    public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
        LoggerFactory.getLogger(invoker.getInterface().getName()).info(" [dubbo consumer] 服务调用-请求|method:{}|参数:{}| client[ip:{}-protocol:{}]", invocation.getMethodName(), invocation.getArguments(), RpcContext.getContext().getRemoteAddressString(), invoker.getUrl().getProtocol());
        try {
            logger.debug("MDC-traceId:" + MDC.get(TRACE_KEY));
            RpcContext.getContext().setAttachments(MDC.getCopyOfContextMap());
        } catch (Exception ex) {
            logger.error("dubbo consumer MDC-traceId:reset fail");
        }
        Result result = invoker.invoke(invocation);
        LoggerFactory.getLogger(invoker.getInterface().getName()).info(" [dubbo consumer] 服务调用-返回|method:{}|结果:{}| client[ip:{}-protocol:{}]", invocation.getMethodName(), result, RpcContext.getContext().getRemoteAddressString(), invoker.getUrl().getProtocol());
        return result;
    }

}

public class TraceProviderFilter implements Filter {

    @Override
    public Result invoke(Invoker<?> invoker, Invocation inv) throws RpcException {
        RpcContext context = RpcContext.getContext();
        MDC.setContextMap(context.getAttachments());
        LoggerFactory.getLogger(invoker.getInterface().getName()).info(" [dubbo provider] 服务调用-接收|method:{}|参数:{}| client[ip:{}-protocol:{}]", inv.getMethodName(), inv.getArguments(), RpcContext.getContext().getRemoteAddressString(), invoker.getUrl().getProtocol());
        long start = System.currentTimeMillis();
        Result result = invoker.invoke(inv);
        long time = System.currentTimeMillis() - start;
        Object rval = result.getValue();
        String logtxt = null;
        if (rval != null) {
            Class<?> clazz = rval.getClass();
            if (List.class.isAssignableFrom(clazz)) {
                int size = ((List) rval).size();
                if (size > 5) {
                    logtxt = size + "(SIZE)";
                } else {
                    logtxt = JSON.toJSONString(rval, new LogArgumentFilter());
                }
            } else if (Set.class.isAssignableFrom(clazz)) {
                int size = ((Set) rval).size();
                if (size > 5) {
                    logtxt = size + "(SIZE)";
                } else {
                    logtxt = JSON.toJSONString(rval, new LogArgumentFilter());
                }
            } else {
                logtxt = JSON.toJSONString(rval, new LogArgumentFilter());
            }
        }
        LoggerFactory.getLogger(invoker.getInterface().getName()).info(" [dubbo provider] 服务调用-返回|method:{}|结果:{}| client[ip:{}-protocol:{}]", inv.getMethodName(), logtxt, RpcContext.getContext().getRemoteAddressString(), invoker.getUrl().getProtocol());
        LoggerFactory.getLogger(invoker.getInterface().getName()).info(" [dubbo provider 服务性能] 统计|method:{}|耗时:{} ms|host:{}", inv.getMethodName(), time, RpcContext.getContext().getRemoteAddressString());
        try {
            MDC.clear();
        } catch (Exception ex) {
            LoggerFactory.getLogger(TraceProviderFilter.class).error("dubbo provider MDC-traceId:reset fail");
        }
        return result;
    }
}