You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
cloudreve/pkg/request/request.go

107 lines
2.0 KiB

package request
import (
"github.com/HFO4/cloudreve/pkg/auth"
"io"
"net/http"
"time"
)
var GeneralClient Client = HTTPClient{}
// Response 请求的响应或错误信息
type Response struct {
Err error
Response *http.Response
}
// Client 请求客户端
type Client interface {
Request(method, target string, body io.Reader, opts ...Option) Response
}
// HTTPClient 实现 Client 接口
type HTTPClient struct {
}
// Option 发送请求的额外设置
type Option interface {
apply(*options)
}
type options struct {
timeout time.Duration
header http.Header
sign auth.Auth
signTTL int64
}
type optionFunc func(*options)
func (f optionFunc) apply(o *options) {
f(o)
}
func newDefaultOption() *options {
return &options{
header: http.Header{},
timeout: time.Duration(30) * time.Second,
}
}
// WithTimeout 设置请求超时
func WithTimeout(t time.Duration) Option {
return optionFunc(func(o *options) {
o.timeout = t
})
}
// WithCredential 对请求进行签名
func WithCredential(instance auth.Auth, ttl int64) Option {
return optionFunc(func(o *options) {
o.sign = instance
o.signTTL = ttl
})
}
// WithHeader 设置请求Header
func WithHeader(header http.Header) Option {
return optionFunc(func(o *options) {
o.header = header
})
}
// Request 发送HTTP请求
func (c HTTPClient) Request(method, target string, body io.Reader, opts ...Option) Response {
// 应用额外设置
options := newDefaultOption()
for _, o := range opts {
o.apply(options)
}
// 创建请求客户端
client := &http.Client{Timeout: options.timeout}
// 创建请求
req, err := http.NewRequest(method, target, body)
if err != nil {
return Response{Err: err}
}
// 添加请求header
req.Header = options.header
// 签名请求
if options.sign != nil {
auth.SignRequest(options.sign, req, time.Now().Unix()+options.signTTL)
}
// 发送请求
resp, err := client.Do(req)
if err != nil {
return Response{Err: err}
}
return Response{Err: nil, Response: resp}
}