feat: 🚀 访问量添加 iteration 配置项

Author: Kele-Bingtang

vitepress-theme-teek/src/components/ArticleAnalyze/src/index.vue

@@ -85,16 +85,25 @@ const docAnalysisConfig = computed<DocAnalysis>(() => {
 });
 
 const statisticsConfig = computed<NonNullable<DocAnalysis["statistics"]>>(() => {
-  const { provider = "", pageView = true, pageIteration = 2000 } = unref(docAnalysisConfig).statistics || {};
+  const {
+    provider = "",
+    pageView = true,
+    iteration = false,
+    pageIteration = 2000,
+  } = unref(docAnalysisConfig).statistics || {};
 
-  return { provider, pageView, pageIteration };
+  return { provider, pageView, iteration, pageIteration };
 });
 // 是否使用访问量功能
 const usePageView = computed(() => unref(statisticsConfig).provider && unref(statisticsConfig).pageView);
 
 const statisticsInfo: UseBuSunZi = { pagePv: ref(0), isGet: ref(false) };
 // 通过不蒜子获取访问量
-const { pagePv, isGet, request } = useBuSunZi(true, unref(statisticsConfig).pageIteration);
+const { pagePv, isGet, request } = useBuSunZi(
+  true,
+  unref(statisticsConfig).iteration,
+  unref(statisticsConfig).pageIteration
+);
 statisticsInfo.pagePv = pagePv;
 statisticsInfo.isGet = isGet;
 

vitepress-theme-teek/src/components/HomeDocAnalysisCard/src/index.vue

@@ -74,12 +74,12 @@ const statisticsInfo: UseBuSunZi = {
   siteUv: ref(0),
   isGet: ref(false),
 };
-const { provider = "", siteView = true, siteIteration = 2000 }: DocAnalysis["statistics"] = statistics;
+const { provider = "", siteView = true, iteration, siteIteration = 2000 }: DocAnalysis["statistics"] = statistics;
 const useSiteView = provider === "busuanzi" && siteView;
 
 if (useSiteView) {
   // 通过不蒜子获取访问量和访客数
-  const { sitePv, siteUv, isGet, request } = useBuSunZi(true, siteIteration);
+  const { sitePv, siteUv, isGet, request } = useBuSunZi(true, iteration, siteIteration);
   statisticsInfo.sitePv = sitePv;
   statisticsInfo.siteUv = siteUv;
   statisticsInfo.isGet = isGet;

vitepress-theme-teek/src/config/types.ts

@@ -762,6 +762,12 @@ export interface DocAnalysis {
      * @default true
      */
     pageView?: boolean;
+    /**
+     * 是否开启重试机制，当获取访问量失败会尝试重新获取
+     *
+     * @default false
+     */
+    iteration?: boolean;
     /**
      * 如果首页获取访问量失败，则每隔多少时间后获取一次访问量，直到获取成功或获取 5 次后
      *

vitepress-theme-teek/src/hooks/useBuSunZi.ts

@@ -38,7 +38,7 @@ export interface UseBuSunZi {
   isGet?: Ref<boolean | null>;
 }
 
-export const useBuSunZi = (initRequest = false, iterationTime = 2000) => {
+export const useBuSunZi = (initRequest = false, iteration = false, iterationTime = 2000) => {
   const sitePv = ref(9999);
   const siteUv = ref(9999);
   const pagePv = ref(9999);
@@ -60,21 +60,24 @@ export const useBuSunZi = (initRequest = false, iterationTime = 2000) => {
 
   // 第一次调用
   if (initRequest) request();
-  let intervalId: NodeJS.Timeout;
-  let i = 0;
 
-  // 如果第一次调用获取失败，每 3s 后重新调用，直至尝试 5 次或调用成功
-  intervalId = setInterval(() => {
-    if (!unref(isGet)) {
-      i += iterationTime;
-      if (i > iterationTime * 5) clearInterval(intervalId);
-      request();
-    } else clearInterval(intervalId);
-  }, iterationTime);
+  if (iteration) {
+    let intervalId: NodeJS.Timeout;
+    let i = 0;
 
-  onBeforeUnmount(() => {
-    if (intervalId) clearInterval(intervalId);
-  });
+    // 如果第一次调用获取失败，每 3s 后重新调用，直至尝试 5 次或调用成功
+    intervalId = setInterval(() => {
+      if (!unref(isGet)) {
+        i += iterationTime;
+        if (i > iterationTime * 5) clearInterval(intervalId);
+        request();
+      } else clearInterval(intervalId);
+    }, iterationTime);
+
+    onBeforeUnmount(() => {
+      if (intervalId) clearInterval(intervalId);
+    });
+  }
 
   return { sitePv, siteUv, pagePv, isGet, request };
 };