/**
  * 保存站點(diǎn)數(shù)據(jù)模型
  * <p>
  * 站點(diǎn)統(tǒng)計(jì)hash：
  * - visit_info:
  * ---- pv: 站點(diǎn)的總pv
  * ---- uv: 站點(diǎn)的總uv
  * ---- pv_path: 站點(diǎn)某個(gè)資源的總訪問(wèn)pv
  * ---- uv_path: 站點(diǎn)某個(gè)資源的總訪問(wèn)uv
  * - visit_info_ip:
  * ---- pv: 用戶訪問(wèn)的站點(diǎn)總次數(shù)
  * ---- path_pv: 用戶訪問(wèn)的路徑總次數(shù)
  * - visit_info_20230822每日記錄, 一天一條記錄
  * ---- pv: 12  # field = 月日_pv, pv的計(jì)數(shù)
  * ---- uv: 5   # field = 月日_uv, uv的計(jì)數(shù)
  * ---- pv_path: 2 # 資源的當(dāng)前訪問(wèn)計(jì)數(shù)
  * ---- uv_path: # 資源的當(dāng)天訪問(wèn)uv
  * ---- pv_ip: # 用戶當(dāng)天的訪問(wèn)次數(shù)
  * ---- pv_path_ip: # 用戶對(duì)資源的當(dāng)天訪問(wèn)次數(shù)
  *
  * @param visitIp 訪問(wèn)者ip
  * @param path    訪問(wèn)的資源路徑
  */
 @Override
 public void saveVisitInfo(String visitIp, String path) {
     String globalKey = SitemapConstants.SITE_VISIT_KEY;
     String day = SitemapConstants.day(LocalDate.now());

     String todayKey = globalKey + "_" + day;

     // 用戶的全局訪問(wèn)計(jì)數(shù)+1
     Long globalUserVisitCnt = RedisClient.hIncr(globalKey + "_" + visitIp, "pv", 1);
     // 用戶的當(dāng)日訪問(wèn)計(jì)數(shù)+1
     Long todayUserVisitCnt = RedisClient.hIncr(todayKey, "pv_" + visitIp, 1);

     RedisClient.PipelineAction pipelineAction = RedisClient.pipelineAction();
     if (globalUserVisitCnt == 1) {
         // 站點(diǎn)新用戶
         // 今日的uv + 1
         pipelineAction.add(todayKey, "uv"
                 , (connection, key, field) -> {
                     connection.hIncrBy(key, field, 1);
                 });
         pipelineAction.add(todayKey, "uv_" + path
                 , (connection, key, field) -> connection.hIncrBy(key, field, 1));

         // 全局站點(diǎn)的uv
         pipelineAction.add(globalKey, "uv", (connection, key, field) -> connection.hIncrBy(key, field, 1));
         pipelineAction.add(globalKey, "uv_" + path, (connection, key, field) -> connection.hIncrBy(key, field, 1));
     } else if (todayUserVisitCnt == 1) {
         // 判斷是今天的首次訪問(wèn)，更新今天的uv+1
         pipelineAction.add(todayKey, "uv", (connection, key, field) -> connection.hIncrBy(key, field, 1));
         if (RedisClient.hIncr(todayKey, "pv_" + path + "_" + visitIp, 1) == 1) {
             // 判斷是否為今天首次訪問(wèn)這個(gè)資源，若是，則uv+1
             pipelineAction.add(todayKey, "uv_" + path, (connection, key, field) -> connection.hIncrBy(key, field, 1));
         }

         // 判斷是否是用戶的首次訪問(wèn)這個(gè)path，若是，則全局的path uv計(jì)數(shù)需要+1
         if (RedisClient.hIncr(globalKey + "_" + visitIp, "pv_" + path, 1) == 1) {
             pipelineAction.add(globalKey, "uv_" + path, (connection, key, field) -> connection.hIncrBy(key, field, 1));
         }
     }


     // 更新pv 以及 用戶的path訪問(wèn)信息
     // 今天的相關(guān)信息 pv
     pipelineAction.add(todayKey, "pv", (connection, key, field) -> connection.hIncrBy(key, field, 1));
     pipelineAction.add(todayKey, "pv_" + path, (connection, key, field) -> connection.hIncrBy(key, field, 1));
     if (todayUserVisitCnt > 1) {
         // 非當(dāng)天首次訪問(wèn)，則pv+1; 因?yàn)槭状卧L問(wèn)時(shí)，在前面更新uv時(shí)，已經(jīng)計(jì)數(shù)+1了
         pipelineAction.add(todayKey, "pv_" + path + "_" + visitIp, (connection, key, field) -> connection.hIncrBy(key, field, 1));
     }


     // 全局的 PV
     pipelineAction.add(globalKey, "pv", (connection, key, field) -> connection.hIncrBy(key, field, 1));
     pipelineAction.add(globalKey, "pv" + "_" + path, (connection, key, field) -> connection.hIncrBy(key, field, 1));

     // 保存訪問(wèn)信息
     pipelineAction.execute();
     if (log.isDebugEnabled()) {
         log.info("用戶訪問(wèn)信息更新完成! 當(dāng)前用戶總訪問(wèn): {}，今日訪問(wèn): {}", globalUserVisitCnt, todayUserVisitCnt);
     }
 }

/**
 * redis 管道執(zhí)行的封裝鏈路
 */
public static class PipelineAction {
    private List<Runnable> run = new ArrayList<>();

    private RedisConnection connection;

    public PipelineAction add(String key, BiConsumer<RedisConnection, byte[]> conn) {
        run.add(() -> conn.accept(connection, RedisClient.keyBytes(key)));
        return this;
    }

    public PipelineAction add(String key, String field, ThreeConsumer<RedisConnection, byte[], byte[]> conn) {
        run.add(() -> conn.accept(connection, RedisClient.keyBytes(key), valBytes(field)));
        return this;
    }

    public void execute() {
        template.executePipelined((RedisCallback<Object>) connection -> {
            PipelineAction.this.connection = connection;
            run.forEach(Runnable::run);
            return null;
        });
    }
}

@FunctionalInterface
public interface ThreeConsumer<T, U, P> {
    void accept(T t, U u, P p);
}

private HttpServletRequest initReqInfo(HttpServletRequest request, HttpServletResponse response) {
    if (isStaticURI(request)) {
        // 靜態(tài)資源直接放行
        return request;
    }

    StopWatch stopWatch = new StopWatch("請(qǐng)求參數(shù)構(gòu)建");
    try {
        stopWatch.start("traceId");
        // 添加全鏈路的traceId
        MdcUtil.addTraceId();
        stopWatch.stop();

        stopWatch.start("請(qǐng)求基本信息");
        // 手動(dòng)寫(xiě)入一個(gè)session，借助 OnlineUserCountListener 實(shí)現(xiàn)在線人數(shù)實(shí)時(shí)統(tǒng)計(jì)
        request.getSession().setAttribute("latestVisit", System.currentTimeMillis());

        ReqInfoContext.ReqInfo reqInfo = new ReqInfoContext.ReqInfo();
        reqInfo.setHost(request.getHeader("host"));
        reqInfo.setPath(request.getPathInfo());
        if (reqInfo.getPath() == null) {
            String url = request.getRequestURI();
            int index = url.indexOf("?");
            if (index > 0) {
                url = url.substring(0, index);
            }
            reqInfo.setPath(url);
        }
        reqInfo.setReferer(request.getHeader("referer"));
        reqInfo.setClientIp(IpUtil.getClientIp(request));
        reqInfo.setUserAgent(request.getHeader("User-Agent"));
        reqInfo.setDeviceId(getOrInitDeviceId(request, response));

        request = this.wrapperRequest(request, reqInfo);
        stopWatch.stop();

        stopWatch.start("登錄用戶信息");
        // 初始化登錄信息
        globalInitService.initLoginUser(reqInfo);
        stopWatch.stop();

        ReqInfoContext.addReqInfo(reqInfo);
        stopWatch.start("pv/uv站點(diǎn)統(tǒng)計(jì)");
        // 更新uv/pv計(jì)數(shù)
        AsyncUtil.execute(() -> SpringUtil.getBean(SitemapServiceImpl.class).saveVisitInfo(reqInfo.getClientIp(), reqInfo.getPath()));
        stopWatch.stop();

        stopWatch.start("回寫(xiě)traceId");
        // 返回頭中記錄traceId
        response.setHeader(GLOBAL_TRACE_ID_HEADER, Optional.ofNullable(MdcUtil.getTraceId()).orElse(""));
        stopWatch.stop();
    } catch (Exception e) {
        log.error("init reqInfo error!", e);
    } finally {
        if (!EnvUtil.isPro()) {
            log.info("{} -> 請(qǐng)求構(gòu)建耗時(shí): \n{}", request.getRequestURI(), stopWatch.prettyPrint(TimeUnit.MILLISECONDS));
        }
    }

    return request;
}

/**
 * 查詢站點(diǎn)某一天or總的訪問(wèn)信息
 *
 * @param date 日期，為空時(shí)，表示查詢所有的站點(diǎn)信息
 * @param path 訪問(wèn)路徑，為空時(shí)表示查站點(diǎn)信息
 * @return
 */
@Override
public SiteCntVo querySiteVisitInfo(LocalDate date, String path) {
    String globalKey = SitemapConstants.SITE_VISIT_KEY;
    String day = null, todayKey = globalKey;
    if (date != null) {
        day = SitemapConstants.day(date);
        todayKey = globalKey + "_" + day;
    }

    String pvField = "pv", uvField = "uv";
    if (path != null) {
        // 表示查詢對(duì)應(yīng)路徑的訪問(wèn)信息
        pvField += "_" + path;
        uvField += "_" + path;
    }

    Map<String, Integer> map = RedisClient.hMGet(todayKey, Arrays.asList(pvField, uvField), Integer.class);
    SiteCntVo siteInfo = new SiteCntVo();
    siteInfo.setDay(day);
    siteInfo.setPv(map.getOrDefault(pvField, 0));
    siteInfo.setUv(map.getOrDefault(uvField, 0));
    return siteInfo;
}