mirror of
https://github.com/oneclickvirt/basics.git
synced 2025-10-08 01:50:28 +08:00
94 lines
2.7 KiB
Go
94 lines
2.7 KiB
Go
package utils
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"net"
|
||
"net/http"
|
||
"reflect"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/imroc/req/v3"
|
||
)
|
||
|
||
// FetchJsonFromURL 函数用于从指定的 URL 获取信息
|
||
// url 参数表示要获取信息的 URL
|
||
// netType 参数表示网络类型，只能为 "tcp4" 或 "tcp6"。
|
||
// enableHeader 参数表示是否启用请求头信息。
|
||
// additionalHeader 参数表示传入的额外的请求头信息(用于传输api的key)。
|
||
// 返回一个解析 json 得到的 map 和 一个可能发生的错误 。
|
||
func FetchJsonFromURL(url, netType string, enableHeader bool, additionalHeader string) (map[string]interface{}, error) {
|
||
// 检查网络类型是否有效
|
||
if netType != "tcp4" && netType != "tcp6" {
|
||
return nil, fmt.Errorf("Invalid netType: %s. Expected 'tcp4' or 'tcp6'.", netType)
|
||
}
|
||
|
||
// 创建 HTTP 客户端
|
||
client := req.C()
|
||
client.SetTimeout(6 * time.Second).
|
||
SetDial(func(ctx context.Context, network, addr string) (net.Conn, error) {
|
||
return (&net.Dialer{}).DialContext(ctx, netType, addr)
|
||
}).
|
||
SetTLSHandshakeTimeout(3 * time.Second).
|
||
SetResponseHeaderTimeout(3 * time.Second).
|
||
SetExpectContinueTimeout(3 * time.Second)
|
||
client.R().
|
||
SetRetryCount(2).
|
||
SetRetryBackoffInterval(1*time.Second, 2*time.Second).
|
||
SetRetryFixedInterval(1 * time.Second)
|
||
// 如果启用请求头，则设置请求头信息
|
||
if enableHeader {
|
||
client.Headers = make(http.Header)
|
||
client.ImpersonateChrome()
|
||
if additionalHeader != "" {
|
||
tempList := strings.Split(additionalHeader, ":")
|
||
if len(tempList) == 2 {
|
||
client.Headers.Set(tempList[0], tempList[1])
|
||
} else if len(tempList) > 2 {
|
||
client.Headers.Set(tempList[0], strings.Join(tempList[1:], ":"))
|
||
}
|
||
}
|
||
}
|
||
// 执行请求
|
||
resp, err := client.R().Get(url)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("Error fetching %s info: %v", url, err)
|
||
}
|
||
// 检查响应状态码
|
||
if !resp.IsSuccess() {
|
||
return nil, fmt.Errorf("Error fetching %s info: status code %d", url, resp.StatusCode)
|
||
}
|
||
// 解析 JSON 响应体
|
||
var data map[string]interface{}
|
||
err = json.Unmarshal(resp.Bytes(), &data)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("Error decoding %s info: %v", url, err)
|
||
}
|
||
// 返回解析后的数据和 nil 错误
|
||
return data, nil
|
||
}
|
||
|
||
// BoolToString 将布尔值转换为对应的字符串表示，true 则返回 "Yes"，false 则返回 "No"
|
||
func BoolToString(value bool) string {
|
||
if value {
|
||
return "Yes"
|
||
}
|
||
return "No"
|
||
}
|
||
|
||
// ExtractFieldNames 获取结构体的属性名字
|
||
func ExtractFieldNames(data interface{}) []string {
|
||
var fields []string
|
||
val := reflect.ValueOf(data).Elem()
|
||
for i := 0; i < val.NumField(); i++ {
|
||
field := val.Type().Field(i)
|
||
name := field.Name
|
||
if name != "Tag" {
|
||
fields = append(fields, name)
|
||
}
|
||
}
|
||
return fields
|
||
}
|