75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"io"
|
|||
|
|
"net/http"
|
|||
|
|
"qnc-server/app/user/cmd/api/internal/config"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/tidwall/gjson"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type TianjuService struct {
|
|||
|
|
config config.TianjuConfig
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewTianjuService(c config.Config) *TianjuService {
|
|||
|
|
return &TianjuService{
|
|||
|
|
config: c.TianjuConfig,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *TianjuService) Request(apiPath string, params map[string]interface{}) ([]byte, error) {
|
|||
|
|
// 确保params中包含key参数
|
|||
|
|
reqParams := make(map[string]interface{})
|
|||
|
|
|
|||
|
|
// 复制用户参数
|
|||
|
|
for k, v := range params {
|
|||
|
|
reqParams[k] = v
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果未提供key，则使用配置中的ApiKey
|
|||
|
|
if _, ok := reqParams["key"]; !ok {
|
|||
|
|
reqParams["key"] = t.config.ApiKey
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建完整的URL，假设BaseURL已包含https://前缀
|
|||
|
|
fullURL := fmt.Sprintf("%s/%s/index", strings.TrimRight(t.config.BaseURL, "/"), apiPath)
|
|||
|
|
|
|||
|
|
// 将请求数据转换为JSON
|
|||
|
|
messageBytes, err := json.Marshal(reqParams)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("参数序列化失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发起HTTP请求
|
|||
|
|
resp, err := http.Post(fullURL, "application/json", strings.NewReader(string(messageBytes)))
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("发送请求失败: %w", err)
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
// 读取响应体
|
|||
|
|
body, err := io.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("读取响应失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查响应状态码
|
|||
|
|
code := gjson.GetBytes(body, "code").Int()
|
|||
|
|
if code != 200 {
|
|||
|
|
msg := gjson.GetBytes(body, "msg").String()
|
|||
|
|
return nil, fmt.Errorf("天聚请求失败: 状态码 %d, 信息: %s", code, msg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取结果数据
|
|||
|
|
result := gjson.GetBytes(body, "result")
|
|||
|
|
if !result.Exists() {
|
|||
|
|
return nil, fmt.Errorf("天聚请求result为空: %s", string(body))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return []byte(result.Raw), nil
|
|||
|
|
}
|