f

internal/infrastructure/external/tianyancha/tianyancha_service.go

@@ -0,0 +1,183 @@
+package tianyancha
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+)
+
+var (
+	ErrDatasource   = errors.New("数据源异常")
+	ErrNotFound     = errors.New("查询为空")
+	ErrSystem       = errors.New("系统异常")
+	ErrInvalidParam = errors.New("参数错误")
+)
+
+// APIEndpoints 天眼查 API 端点映射
+var APIEndpoints = map[string]string{
+	"VerifyThreeElements": "/open/ic/verify/2.0",            // 企业三要素验证
+	"InvestHistory":       "/open/hi/invest/2.0",            // 对外投资历史
+	"FinancingHistory":    "/open/cd/findHistoryRongzi/2.0", // 融资历史
+	"PunishmentInfo":      "/open/mr/punishmentInfo/3.0",    // 行政处罚
+	"AbnormalInfo":        "/open/mr/abnormal/2.0",          // 经营异常
+	"OwnTax":              "/open/mr/ownTax/2.0",            // 欠税公告
+	"TaxContravention":    "/open/mr/taxContravention/2.0",  // 税收违法
+	"holderChange":        "/open/ic/holderChange/2.0",      // 股权变更
+	"baseinfo":            "/open/ic/baseinfo/normal",       // 企业基本信息
+	"investtree":          "/v3/open/investtree",            // 股权穿透
+}
+
+// TianYanChaConfig 天眼查配置
+type TianYanChaConfig struct {
+	BaseURL string
+	Token   string
+	Timeout time.Duration
+}
+
+// TianYanChaService 天眼查服务
+type TianYanChaService struct {
+	config TianYanChaConfig
+}
+
+// APIResponse 标准API响应结构
+type APIResponse struct {
+	Success bool        `json:"success"`
+	Code    int         `json:"code"`
+	Message string      `json:"message"`
+	Data    interface{} `json:"data"`
+}
+
+// TianYanChaResponse 天眼查原始响应结构
+type TianYanChaResponse struct {
+	ErrorCode int         `json:"error_code"`
+	Reason    string      `json:"reason"`
+	Result    interface{} `json:"result"`
+}
+
+// NewTianYanChaService 创建天眼查服务实例
+func NewTianYanChaService(baseURL, token string, timeout time.Duration) *TianYanChaService {
+	if timeout == 0 {
+		timeout = 60 * time.Second
+	}
+
+	return &TianYanChaService{
+		config: TianYanChaConfig{
+			BaseURL: baseURL,
+			Token:   token,
+			Timeout: timeout,
+		},
+	}
+}
+
+// CallAPI 调用天眼查API - 通用方法，由外部处理器传入具体参数
+func (t *TianYanChaService) CallAPI(ctx context.Context, apiCode string, params map[string]string) (*APIResponse, error) {
+	// 从映射中获取 API 端点
+	endpoint, exists := APIEndpoints[apiCode]
+	if !exists {
+		return nil, errors.Join(ErrInvalidParam, fmt.Errorf("未找到 API 代码对应的端点: %s", apiCode))
+	}
+
+	// 构建完整 URL
+	fullURL := strings.TrimRight(t.config.BaseURL, "/") + "/" + strings.TrimLeft(endpoint, "/")
+
+	// 检查 Token 是否配置
+	if t.config.Token == "" {
+		return nil, errors.Join(ErrSystem, fmt.Errorf("天眼查 API Token 未配置"))
+	}
+
+	// 构建查询参数
+	queryParams := url.Values{}
+	for key, value := range params {
+		queryParams.Set(key, value)
+	}
+
+	// 构建完整URL
+	requestURL := fullURL
+	if len(queryParams) > 0 {
+		requestURL += "?" + queryParams.Encode()
+	}
+
+	// 创建请求
+	req, err := http.NewRequestWithContext(ctx, "GET", requestURL, nil)
+	if err != nil {
+		return nil, errors.Join(ErrSystem, fmt.Errorf("创建请求失败: %v", err))
+	}
+
+	// 设置请求头
+	req.Header.Set("Authorization", t.config.Token)
+
+	// 发送请求
+	client := &http.Client{Timeout: t.config.Timeout}
+	resp, err := client.Do(req)
+	if err != nil {
+		// 检查是否是超时错误
+		isTimeout := false
+		if ctx.Err() == context.DeadlineExceeded {
+			isTimeout = true
+		} else if netErr, ok := err.(interface{ Timeout() bool }); ok && netErr.Timeout() {
+			isTimeout = true
+		} else if errStr := err.Error(); errStr == "context deadline exceeded" ||
+			errStr == "timeout" ||
+			errStr == "Client.Timeout exceeded" ||
+			errStr == "net/http: request canceled" {
+			isTimeout = true
+		}
+
+		if isTimeout {
+			return nil, errors.Join(ErrDatasource, fmt.Errorf("API请求超时: %v", err))
+		}
+		return nil, errors.Join(ErrDatasource, fmt.Errorf("API 请求异常: %v", err))
+	}
+	defer resp.Body.Close()
+
+	// 检查 HTTP 状态码
+	if resp.StatusCode != http.StatusOK {
+		return nil, errors.Join(ErrDatasource, fmt.Errorf("API 请求失败，状态码: %d", resp.StatusCode))
+	}
+
+	// 读取响应体
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, errors.Join(ErrSystem, fmt.Errorf("读取响应体失败: %v", err))
+	}
+
+	// 解析 JSON 响应
+	var tianYanChaResp TianYanChaResponse
+	if err := json.Unmarshal(body, &tianYanChaResp); err != nil {
+		return nil, errors.Join(ErrSystem, fmt.Errorf("解析响应 JSON 失败: %v", err))
+	}
+
+	// 检查天眼查业务状态码
+	if tianYanChaResp.ErrorCode != 0 {
+		// 特殊处理：ErrorCode 300000 表示查询为空，返回成功但数据为空数组
+		if tianYanChaResp.ErrorCode == 300000 {
+			return &APIResponse{
+				Success: true,
+				Code:    0,
+				Message: "",
+				Data:    []interface{}{}, // 返回空数组而不是nil
+			}, nil
+		}
+
+		return &APIResponse{
+			Success: false,
+			Code:    tianYanChaResp.ErrorCode,
+			Message: tianYanChaResp.Reason,
+			Data:    tianYanChaResp.Result,
+		}, nil
+	}
+
+	// 成功情况
+	return &APIResponse{
+		Success: true,
+		Code:    0,
+		Message: tianYanChaResp.Reason,
+		Data:    tianYanChaResp.Result,
+	}, nil
+}