148 lines
3.6 KiB
Go
148 lines
3.6 KiB
Go
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
|
|
}
|