add qygl23t7

internal/infrastructure/external/tianyancha/tianyancha_service.go

@@ -0,0 +1,147 @@
+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", // 企业三要素验证
+}
+
+// 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 = 30 * 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, fmt.Errorf("%w: 未找到 API 代码对应的端点: %s", ErrInvalidParam, apiCode)
+	}
+
+	// 构建完整 URL
+	fullURL := strings.TrimRight(t.config.BaseURL, "/") + "/" + strings.TrimLeft(endpoint, "/")
+
+	// 检查 Token 是否配置
+	if t.config.Token == "" {
+		return nil, fmt.Errorf("%w: 天眼查 API Token 未配置", ErrSystem)
+	}
+
+	// 构建查询参数
+	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, fmt.Errorf("%w: 创建请求失败: %v", ErrSystem, err)
+	}
+
+	// 设置请求头
+	req.Header.Set("Authorization", t.config.Token)
+
+	// 发送请求
+	client := &http.Client{Timeout: t.config.Timeout}
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, fmt.Errorf("%w: API 请求异常: %v", ErrDatasource, err)
+	}
+	defer resp.Body.Close()
+
+	// 检查 HTTP 状态码
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("%w: API 请求失败，状态码: %d", ErrDatasource, resp.StatusCode)
+	}
+
+	// 读取响应体
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, fmt.Errorf("%w: 读取响应体失败: %v", ErrSystem, err)
+	}
+
+	// 解析 JSON 响应
+	var tianYanChaResp TianYanChaResponse
+	if err := json.Unmarshal(body, &tianYanChaResp); err != nil {
+		return nil, fmt.Errorf("%w: 解析响应 JSON 失败: %v", ErrSystem, err)
+	}
+
+	// 检查天眼查业务状态码
+	if tianYanChaResp.ErrorCode != 0 {
+		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
+}