101 lines
2.7 KiB
Go
101 lines
2.7 KiB
Go
|
// Package pool define concurrency call function in ants
|
|||
|
|
package pool
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"log"
|
|||
|
|
"net/http"
|
|||
|
|
"strings"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"modelRT/config"
|
|||
|
|
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var AnchorFunc = func(anchorConfig interface{}) {
|
|||
|
|
logger := zap.L()
|
|||
|
|
|
|||
|
|
paramConfig, ok := anchorConfig.(config.AnchorParamConfig)
|
|||
|
|
if !ok {
|
|||
|
|
logger.Error("conversion model anchor param config type failed")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
fmt.Println(paramConfig)
|
|||
|
|
// TODO 解析 paramConfig 轮询 dataRT http 接口
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type APIEndpoint struct {
|
|||
|
|
URL string `json:"url"`
|
|||
|
|
Method string `json:"method"` // HTTP 方法，如 "GET", "POST"
|
|||
|
|
Headers map[string]string `json:"headers"`
|
|||
|
|
QueryParams map[string]string `json:"query_params"`
|
|||
|
|
Body string `json:"body"` // 对于 POST 请求等，可能需要一个请求体
|
|||
|
|
Interval int `json:"interval"` // 轮询间隔时间（秒）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// fetchAPI 执行 HTTP 请求并返回响应体（作为字符串）或错误
|
|||
|
|
func fetchAPI(endpoint APIEndpoint) (string, error) {
|
|||
|
|
client := &http.Client{}
|
|||
|
|
|
|||
|
|
// 构建请求
|
|||
|
|
req, err := http.NewRequest(endpoint.Method, endpoint.URL, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置请求头
|
|||
|
|
for key, value := range endpoint.Headers {
|
|||
|
|
req.Header.Set(key, value)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置查询参数（如果需要）
|
|||
|
|
q := req.URL.Query()
|
|||
|
|
for key, value := range endpoint.QueryParams {
|
|||
|
|
q.Set(key, value)
|
|||
|
|
}
|
|||
|
|
req.URL.RawQuery = q.Encode()
|
|||
|
|
|
|||
|
|
// 设置请求体（如果需要，例如 POST 请求）
|
|||
|
|
if endpoint.Method == "POST" || endpoint.Method == "PUT" {
|
|||
|
|
req.Body = ioutil.NopCloser(strings.NewReader(endpoint.Body))
|
|||
|
|
req.Header.Set("Content-Type", "application/json") // 假设是 JSON 请求体
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 执行请求
|
|||
|
|
resp, err := client.Do(req)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
// 读取响应体
|
|||
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return string(body), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// pollAPIEndpoints 轮询 API 端点列表，并根据指定的间隔时间执行请求
|
|||
|
|
func pollAPIEndpoints(endpoints []APIEndpoint, interval int, wg *sync.WaitGroup, results chan<- string) {
|
|||
|
|
defer wg.Done()
|
|||
|
|
for _, endpoint := range endpoints {
|
|||
|
|
for {
|
|||
|
|
body, err := fetchAPI(endpoint)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("Error fetching from %s: %v", endpoint.URL, err)
|
|||
|
|
} else {
|
|||
|
|
results <- fmt.Sprintf("Response from %s: %s", endpoint.URL, body)
|
|||
|
|
}
|
|||
|
|
time.Sleep(time.Duration(interval) * time.Second)
|
|||
|
|
// 注意：这里使用了 endpoint.Interval 而不是传入的 interval，
|
|||
|
|
// 但为了示例简单，我们统一使用传入的 interval。
|
|||
|
|
// 如果要根据每个端点的不同间隔来轮询，应该使用 endpoint.Interval。
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|