tunnel checker

Author: xixingde

etc/conf.yaml

@@ -94,3 +94,9 @@ Validation:
 TokenLimit:
   max_running_tasks: 10
   enabled: true
+
+# 探活接口配置
+HealthCheck:
+  Enabled: true
+  URL: "http://localhost:9001/codebase-indexer/api/v1/index/summary"
+  Timeout: 3s

internal/config/config.go

@@ -2,6 +2,7 @@ package config
 
 import (
 	"errors"
+	"time"
 
 	"github.com/zeromicro/go-zero/rest"
 )
@@ -18,6 +19,7 @@ type Config struct {
 	Cleaner     CleanerConf
 	Validation  ValidationConfig
 	TokenLimit  TokenLimitConf
+	HealthCheck HealthCheckConf
 }
 
 // TokenLimitConf token限流配置
@@ -26,6 +28,13 @@ type TokenLimitConf struct {
 	Enabled         bool `json:"enabled" yaml:"enabled"`
 }
 
+// HealthCheckConf 探活接口配置
+type HealthCheckConf struct {
+	Enabled bool          `json:"enabled" yaml:"enabled"`
+	URL     string        `json:"url" yaml:"url"`
+	Timeout time.Duration `json:"timeout" yaml:"timeout"`
+}
+
 // Validate 实现 Validator 接口
 func (c Config) Validate() error {
 	if len(c.Name) == 0 {

internal/handler/summary.go

@@ -1,9 +1,10 @@
 package handler
 
 import (
-	"github.com/zgsm-ai/codebase-indexer/internal/response"
 	"net/http"
 
+	"github.com/zgsm-ai/codebase-indexer/internal/response"
+
 	"github.com/zeromicro/go-zero/rest/httpx"
 	"github.com/zgsm-ai/codebase-indexer/internal/logic"
 	"github.com/zgsm-ai/codebase-indexer/internal/svc"
@@ -18,8 +19,17 @@ func summaryHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
 			return
 		}
 
+		// 从请求头获取 Authorization
+		authorization := r.Header.Get("Authorization")
+
+		// 验证 Authorization 头是否存在
+		if authorization == "" {
+			response.Error(w, response.NewAuthError("missing Authorization header"))
+			return
+		}
+
 		l := logic.NewSummaryLogic(r.Context(), svcCtx)
-		resp, err := l.Summary(&req)
+		resp, err := l.Summary(&req, authorization)
 		if err != nil {
 			response.Error(w, err)
 		} else {

internal/logic/health_check.go

@@ -0,0 +1,102 @@
+package logic
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"net/url"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zgsm-ai/codebase-indexer/internal/svc"
+	"github.com/zgsm-ai/codebase-indexer/internal/types"
+)
+
+type HealthCheckLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewHealthCheckLogic(ctx context.Context, svcCtx *svc.ServiceContext) *HealthCheckLogic {
+	return &HealthCheckLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+// CheckHealth 执行探活检查
+func (l *HealthCheckLogic) CheckHealth(authorization string, req *types.IndexSummaryRequest) error {
+	// 如果探活检查未启用，直接返回成功
+	if !l.svcCtx.Config.HealthCheck.Enabled {
+		l.Info("health check is disabled, skipping")
+		return nil
+	}
+
+	// 创建HTTP客户端
+	client := &http.Client{
+		Timeout: l.svcCtx.Config.HealthCheck.Timeout,
+	}
+
+	// 构建请求URL
+	healthCheckURL := l.svcCtx.Config.HealthCheck.URL
+
+	// 将req参数添加到查询参数中
+	if req != nil {
+		// 解析基础URL
+		parsedURL, err := url.Parse(healthCheckURL)
+		if err != nil {
+			l.Errorf("failed to parse health check URL: %v", err)
+			return fmt.Errorf("failed to parse health check URL: %w", err)
+		}
+
+		// 获取现有查询参数或创建新的
+		q := parsedURL.Query()
+
+		// 添加req中的参数
+		if req.ClientId != "" {
+			q.Set("clientId", req.ClientId)
+		}
+		if req.CodebasePath != "" {
+			q.Set("codebasePath", req.CodebasePath)
+		}
+
+		// 将查询参数设置回URL
+		parsedURL.RawQuery = q.Encode()
+		healthCheckURL = parsedURL.String()
+	}
+
+	// 创建请求
+	httpReq, err := http.NewRequestWithContext(l.ctx, "GET", healthCheckURL, nil)
+	if err != nil {
+		l.Errorf("failed to create health check request: %v", err)
+		return fmt.Errorf("failed to create health check request: %w", err)
+	}
+
+	// 设置请求头
+	httpReq.Header.Set("Authorization", authorization)
+	httpReq.Header.Set("X-Costrict-Version", "v1.0.6")
+
+	// 重试逻辑
+	var lastErr error
+
+	// 发送请求
+	resp, err := client.Do(httpReq)
+	if err != nil {
+		lastErr = fmt.Errorf("health check request failed: %w", err)
+		return lastErr
+	}
+
+	// 确保响应体被关闭
+	defer resp.Body.Close()
+
+	// 检查状态码
+	if resp.StatusCode == http.StatusOK {
+		l.Info("health check passed")
+		return nil
+	}
+
+	lastErr = fmt.Errorf("health check returned status code: %d", resp.StatusCode)
+
+	return fmt.Errorf("health check failed : %w", lastErr)
+}

internal/logic/summary.go

@@ -3,6 +3,7 @@ package logic
 import (
 	"context"
 	"errors"
+	"fmt"
 	"sync"
 	"time"
 
@@ -29,10 +30,17 @@ func NewSummaryLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SummaryLo
 	}
 }
 
-func (l *SummaryLogic) Summary(req *types.IndexSummaryRequest) (*types.IndexSummaryResonseData, error) {
+func (l *SummaryLogic) Summary(req *types.IndexSummaryRequest, authorization string) (*types.IndexSummaryResonseData, error) {
 
 	ctx := context.WithValue(l.ctx, tracer.Key, req.ClientId)
 
+	// 执行探活检查
+	healthCheckLogic := NewHealthCheckLogic(ctx, l.svcCtx)
+	if err := healthCheckLogic.CheckHealth(authorization, req); err != nil {
+		tracer.WithTrace(ctx).Errorf("health check failed: %v", err)
+		return nil, fmt.Errorf("health check failed: %w", err)
+	}
+
 	var (
 		wg                 sync.WaitGroup
 		embeddingSummary   *types.EmbeddingSummary