fix: 提升批量测IP并发与IP检查竞速

Author: cheluen

.env.example

@@ -31,6 +31,9 @@ SINGBOX_BINARY=
 # CORS_ALLOWED_ORIGINS=https://panel.example.com,https://ops.example.com
 CORS_ALLOWED_ORIGINS=
 
+# 批量测 IP 并发（前端批量请求并发数；留空使用默认值）
+SBPM_BATCH_CHECK_IP_CONCURRENCY=
+
 # 登录限速（防暴力破解）
 LOGIN_RATE_LIMIT_WINDOW_SECONDS=60
 LOGIN_RATE_LIMIT_MAX_ATTEMPTS=10

backend/main.go

@@ -401,6 +401,29 @@ const (
 	indexCacheControlHeader  = "no-cache, max-age=0, must-revalidate"
 )
 
+func upsertIndexMetaTag(indexContent []byte, metaName string, metaTag string) []byte {
+	metaMarker := []byte(fmt.Sprintf(`name="%s"`, metaName))
+	updated := false
+
+	if idx := bytes.Index(indexContent, metaMarker); idx >= 0 {
+		start := bytes.LastIndex(indexContent[:idx], []byte("<meta"))
+		if start >= 0 {
+			endRel := bytes.IndexByte(indexContent[idx:], '>')
+			if endRel >= 0 {
+				end := idx + endRel + 1
+				indexContent = append(indexContent[:start], append([]byte(metaTag), indexContent[end:]...)...)
+				updated = true
+			}
+		}
+	}
+
+	if !updated && bytes.Contains(indexContent, []byte("</head>")) {
+		indexContent = bytes.Replace(indexContent, []byte("</head>"), []byte(metaTag+"\n  </head>"), 1)
+	}
+
+	return indexContent
+}
+
 func registerFrontendRoutes(r *gin.Engine, frontendDistDir string, appVersion string) error {
 	assetsFS, sourceName, err := frontendAssetFS(frontendDistDir)
 	if err != nil {
@@ -417,23 +440,22 @@ func registerFrontendRoutes(r *gin.Engine, frontendDistDir string, appVersion st
 		log.Printf("Invalid SBPM_NODES_VIRTUAL_THRESHOLD=%d, using default 50", nodesVirtualThreshold)
 		nodesVirtualThreshold = 50
 	}
-	metaName := []byte(`name="sbpm-nodes-virtual-threshold"`)
-	meta := fmt.Sprintf(`    <meta name="sbpm-nodes-virtual-threshold" content="%d" />`, nodesVirtualThreshold)
-	thresholdUpdated := false
-	if idx := bytes.Index(indexContent, metaName); idx >= 0 {
-		start := bytes.LastIndex(indexContent[:idx], []byte("<meta"))
-		if start >= 0 {
-			endRel := bytes.IndexByte(indexContent[idx:], '>')
-			if endRel >= 0 {
-				end := idx + endRel + 1
-				indexContent = append(indexContent[:start], append([]byte(meta), indexContent[end:]...)...)
-				thresholdUpdated = true
-			}
-		}
-	}
-	if !thresholdUpdated && bytes.Contains(indexContent, []byte("</head>")) {
-		indexContent = bytes.Replace(indexContent, []byte("</head>"), []byte(meta+"\n  </head>"), 1)
-	}
+	nodesVirtualThresholdMeta := fmt.Sprintf(
+		`    <meta name="sbpm-nodes-virtual-threshold" content="%d" />`,
+		nodesVirtualThreshold,
+	)
+	indexContent = upsertIndexMetaTag(indexContent, "sbpm-nodes-virtual-threshold", nodesVirtualThresholdMeta)
+
+	batchCheckIPConcurrency := readIntEnv("SBPM_BATCH_CHECK_IP_CONCURRENCY", 10)
+	if batchCheckIPConcurrency < 1 {
+		log.Printf("Invalid SBPM_BATCH_CHECK_IP_CONCURRENCY=%d, using default 10", batchCheckIPConcurrency)
+		batchCheckIPConcurrency = 10
+	}
+	batchCheckIPConcurrencyMeta := fmt.Sprintf(
+		`    <meta name="sbpm-batch-check-ip-concurrency" content="%d" />`,
+		batchCheckIPConcurrency,
+	)
+	indexContent = upsertIndexMetaTag(indexContent, "sbpm-batch-check-ip-concurrency", batchCheckIPConcurrencyMeta)
 
 	indexFingerprint := calcContentFingerprint(indexContent)
 	indexETag := fmt.Sprintf("\"sbpm-%s\"", indexFingerprint)

backend/main_test.go

@@ -104,6 +104,33 @@ func TestRegisterFrontendRoutesServesIndexWithRevalidateHeaders(t *testing.T) {
 	}
 }
 
+func TestRegisterFrontendRoutes_InjectsBatchCheckIPConcurrencyMeta(t *testing.T) {
+	gin.SetMode(gin.TestMode)
+	r := gin.New()
+	distDir := writeTestFrontendDist(t)
+
+	t.Setenv("SBPM_BATCH_CHECK_IP_CONCURRENCY", "12")
+	if err := registerFrontendRoutes(r, distDir, "1.2.4"); err != nil {
+		t.Fatalf("register frontend routes: %v", err)
+	}
+
+	req := httptest.NewRequest(http.MethodGet, "/", nil)
+	rec := httptest.NewRecorder()
+	r.ServeHTTP(rec, req)
+
+	if rec.Code != http.StatusOK {
+		t.Fatalf("unexpected status: %d", rec.Code)
+	}
+
+	body := rec.Body.String()
+	if !strings.Contains(body, `name="sbpm-batch-check-ip-concurrency"`) {
+		t.Fatalf("missing batch check ip concurrency meta: %s", body)
+	}
+	if !strings.Contains(body, `content="12"`) {
+		t.Fatalf("missing batch check ip concurrency value: %s", body)
+	}
+}
+
 func TestRegisterFrontendRoutesServesAssetsWithImmutableCache(t *testing.T) {
 	gin.SetMode(gin.TestMode)
 	r := gin.New()