mirror of
https://github.com/bestnite/sub2clash.git
synced 2025-06-01 14:35:01 +08:00
103 lines
2.1 KiB
Go
103 lines
2.1 KiB
Go
package common
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"net/http"
|
||
"time"
|
||
|
||
"github.com/nitezs/sub2clash/config"
|
||
)
|
||
|
||
type GetConfig struct {
|
||
userAgent string
|
||
}
|
||
|
||
type GetOption func(*GetConfig)
|
||
|
||
func WithUserAgent(userAgent string) GetOption {
|
||
return func(config *GetConfig) {
|
||
config.userAgent = userAgent
|
||
}
|
||
}
|
||
|
||
func Get(url string, options ...GetOption) (resp *http.Response, err error) {
|
||
retryTimes := config.Default.RequestRetryTimes
|
||
haveTried := 0
|
||
retryDelay := time.Second
|
||
getConfig := GetConfig{}
|
||
for _, option := range options {
|
||
option(&getConfig)
|
||
}
|
||
var req *http.Request
|
||
var get *http.Response
|
||
for haveTried < retryTimes {
|
||
client := &http.Client{}
|
||
//client.Timeout = time.Second * 10
|
||
req, err = http.NewRequest("GET", url, nil)
|
||
if err != nil {
|
||
haveTried++
|
||
time.Sleep(retryDelay)
|
||
continue
|
||
}
|
||
if getConfig.userAgent != "" {
|
||
req.Header.Set("User-Agent", getConfig.userAgent)
|
||
}
|
||
get, err = client.Do(req)
|
||
if err != nil {
|
||
haveTried++
|
||
time.Sleep(retryDelay)
|
||
continue
|
||
} else {
|
||
if get != nil && get.ContentLength > config.Default.RequestMaxFileSize {
|
||
return nil, errors.New("文件过大")
|
||
}
|
||
return get, nil
|
||
}
|
||
|
||
}
|
||
return nil, fmt.Errorf("请求失败：%v", err)
|
||
}
|
||
|
||
func Head(url string, options ...GetOption) (resp *http.Response, err error) {
|
||
retryTimes := config.Default.RequestRetryTimes
|
||
haveTried := 0
|
||
retryDelay := time.Second
|
||
|
||
// 解析可选参数（如 User-Agent）
|
||
getConfig := GetConfig{}
|
||
for _, option := range options {
|
||
option(&getConfig)
|
||
}
|
||
|
||
var req *http.Request
|
||
var headResp *http.Response
|
||
|
||
for haveTried < retryTimes {
|
||
client := &http.Client{}
|
||
req, err = http.NewRequest("HEAD", url, nil)
|
||
if err != nil {
|
||
haveTried++
|
||
time.Sleep(retryDelay)
|
||
continue
|
||
}
|
||
|
||
// 设置 User-Agent（如果提供）
|
||
if getConfig.userAgent != "" {
|
||
req.Header.Set("User-Agent", getConfig.userAgent)
|
||
}
|
||
|
||
headResp, err = client.Do(req)
|
||
if err != nil {
|
||
haveTried++
|
||
time.Sleep(retryDelay)
|
||
continue
|
||
}
|
||
|
||
// HEAD 请求不检查 ContentLength，因为没有响应体
|
||
return headResp, nil
|
||
}
|
||
|
||
return nil, fmt.Errorf("HEAD 请求失败：%v", err)
|
||
}
|