package cn.ath.knowwikibackend.mdc;
import cn.hutool.core.lang.UUID;
import org.slf4j.MDC;
import java.util.Map;
import java.util.concurrent.Callable;
/**
 * 線程mdc打點(diǎn)工具類
 */
public class ThreadMdcUtil {
    public static void setTRaceIdIfAbsent() {
        if (MDC.get("traceId") == null) {
            MDC.put("traceId", UUID.fastUUID().toString());
        }
    }
    public static <T> Callable<T> wrap(final Callable<T> callable, final Map<String, String> context) {
        return () -> {
            if (context == null) {
                MDC.clear();
            } else {
                MDC.setContextMap(context);
            }
            setTRaceIdIfAbsent();
            try {
                //最終通過(guò) callable.call執(zhí)行線程任務(wù)
                return callable.call();
            } finally {
                MDC.clear();
            }
        };
    }
    public static Runnable wrap(final Runnable runnable,final Map<String,String> context){
        return ()->{
          if (context==null){
              MDC.clear(); //mdc上下文為空 就清掉mdc
          }else {
              MDC.setContextMap(context);  // mdc上下文不為空,要設(shè)置上下文為 context
          }
          setTRaceIdIfAbsent(); //設(shè)置mdc 記錄traceId
          try {
              //最終通過(guò) runnable.run 執(zhí)行線程任務(wù)
              runnable.run();
          }finally {
              MDC.clear();
          }
        };
    }
}

logging:
  file:
    path: ${user.home}/.${spring.application.name}/log/
  logback:
    rollingpolicy:
      max-file-size: 15MB
      max-history: 10
  pattern: # 注意這里配置的 [%X{traceId}] 即輸出mdc打點(diǎn)的traceId值
    console: "%date  %level [%thread] [%X{traceId}] %logger{10} [%file : %line] %msg%n"
    file: "%date  %level [%thread] [%X{traceId}] %logger{10} [%file : %line] %msg%n"
  level:
    cn.ath.knowwikibackend.rest: info

package cn.ath.knowwikibackend.mdc;
import cn.hutool.core.lang.UUID;
import org.slf4j.MDC;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
 * mvc trace攔截器，作用是 實(shí)現(xiàn)trace打點(diǎn)
 */
public class MvcTraceInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //在請(qǐng)求頭里拿一個(gè) traceId ,拿不到就創(chuàng)建1個(gè) traceId
        String traceId = request.getHeader("traceId");
        if (traceId==null){
            traceId = UUID.fastUUID().toString();
        }
        // traceId記錄到mdc中
        MDC.put("traceId",traceId);
        return true;
    }
    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
    }
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        //1個(gè)http請(qǐng)求 后臺(tái)處理完畢后 ,從mdc中移除 traceId
        MDC.remove("traceId");
    }
}

package cn.ath.knowwikibackend.mdc;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
@Configuration
public class MvcMdcConfig implements WebMvcConfigurer {
    /**
     * 配置spring mvc啟用 mvc trace攔截器，實(shí)現(xiàn)trace打點(diǎn)
     * @param registry InterceptorRegistry
     */
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new MvcTraceInterceptor())
                .addPathPatterns("/**");
    }
}

package cn.ath.knowwikibackend.mdc;
import org.slf4j.MDC;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import java.io.Serializable;
import java.util.concurrent.*;
/**
 * 異步線程池 的mdc 包裝器，作用是 讓 async 異步任務(wù) 實(shí)現(xiàn)trace打點(diǎn)
 */
public class AsyncThreadPoolMdcWrapper extends ThreadPoolTaskExecutor implements Serializable {
    private static final long serialVersionUID = -1530245553055682935L;
    @Override
    public void execute(Runnable task) {
        // Runnable 類型的線程運(yùn)行時(shí)  進(jìn)行 mdc 打點(diǎn)記錄
        super.execute(ThreadMdcUtil.wrap(task, MDC.getCopyOfContextMap()));
    }
    @Override
    public <T> Future<T> submit(Callable<T> task) {
        // Callable 類型的線程運(yùn)行時(shí)  進(jìn)行 mdc 打點(diǎn)記錄
        return super.submit(ThreadMdcUtil.wrap(task,MDC.getCopyOfContextMap()));
    }
    @Override
    public Future<?> submit(Runnable task) {
        //  Runnable 類型且返回Future的 線程運(yùn)行時(shí)  進(jìn)行 mdc 打點(diǎn)記錄
        return super.submit(ThreadMdcUtil.wrap(task,MDC.getCopyOfContextMap()));
    }
}

package cn.ath.knowwikibackend.mdc;
import lombok.extern.slf4j.Slf4j;
import org.apache.http.Header;
import org.apache.http.HttpException;
import org.apache.http.HttpRequest;
import org.apache.http.HttpRequestInterceptor;
import org.apache.http.protocol.HttpContext;
import org.slf4j.MDC;
import org.springframework.stereotype.Component;
import java.io.IOException;
/**
 * 自定義 HttpRequestInterceptor，
 * 這里可以對(duì)外發(fā)的http請(qǐng)求 追加請(qǐng)求頭/請(qǐng)求參數(shù)之類的配置
 *
 * 這里 在 Header 里追加上本應(yīng)用的traceId ,便于全局排查請(qǐng)求日志
 */
@Component
@Slf4j
public class HttpClientTracedInterceptor implements HttpRequestInterceptor {
    @Override
    public void process(HttpRequest httpRequest, HttpContext httpContext) throws HttpException, IOException {
        //從mdc 中取出traceId 然后追加到 外發(fā)的http請(qǐng)求頭里
        String traceId = MDC.get("traceId");
        if (traceId != null){
            httpRequest.addHeader("traceId",traceId);
        }
        //打印出所有的http請(qǐng)求頭
        for (Header header : httpRequest.getAllHeaders()) {
            log.info("req header item---->{}",header);
        }
    }
}

@Configuration
public class AsyncConfig {
     @Bean("asyncCountTestTaskExecutor")
    public AsyncThreadPoolMdcWrapper asyncCountTaskTest(){
        AsyncThreadPoolMdcWrapper executor = new AsyncThreadPoolMdcWrapper();
        //核心線程數(shù)5：線程池創(chuàng)建時(shí)候初始化的線程數(shù)
        executor.setCorePoolSize(5);
        //最大線程數(shù)10：線程池最大的線程數(shù)，只有在緩沖隊(duì)列滿了之后才會(huì)申請(qǐng)超過(guò)核心線程數(shù)的線程
        executor.setMaxPoolSize(10);
        //緩沖隊(duì)列100：用來(lái)緩沖執(zhí)行任務(wù)的隊(duì)列
        executor.setQueueCapacity(100);
        //允許線程的空閑時(shí)間60秒：當(dāng)超過(guò)了核心線程出之外的線程在空閑時(shí)間到達(dá)之后會(huì)被銷毀
        executor.setKeepAliveSeconds(60);
        //線程池名的前綴：設(shè)置好了之后可以方便我們定位處理任務(wù)所在的線程池
        executor.setThreadNamePrefix("countTestTaskAsync-");
        //線程池滿了后新任務(wù)由 任務(wù)發(fā)起者的線程執(zhí)行
        RejectedExecutionHandler callerRunsPolicy = new ThreadPoolExecutor.CallerRunsPolicy();
        executor.setRejectedExecutionHandler(callerRunsPolicy);
        executor.initialize();
        return executor;
    }
    @Bean("asyncVoidTestTaskExecutor")
    public AsyncThreadPoolMdcWrapper asyncVoidTaskTest(){
        AsyncThreadPoolMdcWrapper executor = new AsyncThreadPoolMdcWrapper();
        //核心線程數(shù)5：線程池創(chuàng)建時(shí)候初始化的線程數(shù)
        executor.setCorePoolSize(5);
        //最大線程數(shù)10：線程池最大的線程數(shù)，只有在緩沖隊(duì)列滿了之后才會(huì)申請(qǐng)超過(guò)核心線程數(shù)的線程
        executor.setMaxPoolSize(10);
        //緩沖隊(duì)列100：用來(lái)緩沖執(zhí)行任務(wù)的隊(duì)列
        executor.setQueueCapacity(100);
        //允許線程的空閑時(shí)間60秒：當(dāng)超過(guò)了核心線程出之外的線程在空閑時(shí)間到達(dá)之后會(huì)被銷毀
        executor.setKeepAliveSeconds(60);
        //線程池名的前綴：設(shè)置好了之后可以方便我們定位處理任務(wù)所在的線程池
        executor.setThreadNamePrefix("voidTestTaskAsync-");
        //線程池滿了后新任務(wù)由 任務(wù)發(fā)起者的線程執(zhí)行
        RejectedExecutionHandler callerRunsPolicy = new ThreadPoolExecutor.CallerRunsPolicy();
        executor.setRejectedExecutionHandler(callerRunsPolicy);
        executor.initialize();
        return executor;
    }
}

@Slf4j
@RestController
@RequestMapping("/third")
public class TestApi {
    @Autowired
    HttpClientTracedInterceptor httpClientTracedInterceptor;
    @Autowired
    private AsyncAllService asyncAllService
    @GetMapping("/t1")
    public String te() throws ExecutionException, InterruptedException, IOException {
        log.trace("trace---test!!!!!");
        log.info("info---test!!!!!");
        log.warn("warn---test!!!!!");
        log.debug("debug---test!!!!!");
        log.error("error---test!!!!!");
        //有返回值的async任務(wù)
        Future<Long> longFuture = asyncAllService.testCount1();
        //無(wú)返回值的async任務(wù)
        asyncAllService.testAsync2();
        //有返回值的async任務(wù),需要future執(zhí)行后,最后統(tǒng)一get返回值
        Long l605 = longFuture.get();
        log.info("async task res:{}",l605);
        //測(cè)試使用apache httpclient 外發(fā)1個(gè)http請(qǐng)求
        String content = "test.dhrth.xxx.zzzdfg.derferf.sregvreg.regetg.esrtg34gf3";
        String url = "http://10.1.5.212:8008/api/getresult";
        Map<String, String> mapData = new HashMap<String, String>();
        mapData.put("type", "NER_RE");
        mapData.put("text", content);
        Map<String, Object> map = new HashMap<String, Object>();
        map.put("sid", "re");
        map.put("data", mapData);
        String reqStr = JSON.toJSONString(map);
        // 獲取httpclient
        CloseableHttpClient httpclient = HttpClients.custom()
                .addInterceptorFirst(httpClientTracedInterceptor)
                .build();
        //創(chuàng)建post請(qǐng)求
        HttpPost httpPost = new HttpPost(url);
        RequestConfig requestConfig = RequestConfig.custom()
                .setSocketTimeout(20*1000)
                .setConnectionRequestTimeout(1000)
                .setConnectTimeout(1000).build();
        httpPost.setConfig(requestConfig);
        StringEntity entity = new StringEntity(reqStr, ContentType.APPLICATION_JSON);
        httpPost.setEntity(entity);
        CloseableHttpResponse response = httpclient.execute(httpPost);
        // 得到響應(yīng)信息
        int statusCode = response.getStatusLine().getStatusCode();
        log.info("http-res-statusCode:{}",statusCode);
        String respStr = EntityUtils.toString(response.getEntity(), "utf-8");
        log.info("http-resp:{}",respStr);
        return "thirdTestApi測(cè)試!";
    }
}