77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package httpclient
|
||
|
||
import (
|
||
"time"
|
||
|
||
"github.com/xinliangnote/go-gin-api/internal/pkg/trace"
|
||
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
// Trace 记录内部流转信息
|
||
type Trace = trace.T
|
||
|
||
// Option 自定义设置http请求
|
||
type Option func(*option)
|
||
|
||
type option struct {
|
||
TTL time.Duration
|
||
Header map[string]string
|
||
Trace *trace.Trace
|
||
Dialog *trace.Dialog
|
||
Logger *zap.Logger
|
||
RetryTimes int
|
||
RetryDelay time.Duration
|
||
}
|
||
|
||
func newOption() *option {
|
||
return &option{
|
||
Header: make(map[string]string),
|
||
}
|
||
}
|
||
|
||
// WithTTL 本次http请求最长执行时间
|
||
func WithTTL(ttl time.Duration) Option {
|
||
return func(opt *option) {
|
||
opt.TTL = ttl
|
||
}
|
||
}
|
||
|
||
// WithHeader 设置http header，可以调用多次设置多对key-value
|
||
func WithHeader(key, value string) Option {
|
||
return func(opt *option) {
|
||
opt.Header[key] = value
|
||
}
|
||
}
|
||
|
||
// WithTrace 设置trace信息
|
||
func WithTrace(t Trace) Option {
|
||
return func(opt *option) {
|
||
if t != nil {
|
||
opt.Trace = t.(*trace.Trace)
|
||
opt.Dialog = new(trace.Dialog)
|
||
}
|
||
}
|
||
}
|
||
|
||
// WithLogger 设置logger以便打印关键日志
|
||
func WithLogger(logger *zap.Logger) Option {
|
||
return func(opt *option) {
|
||
opt.Logger = logger
|
||
}
|
||
}
|
||
|
||
// WithRetryTimes 如果请求失败，最多重试N次
|
||
func WithRetryTimes(retryTimes int) Option {
|
||
return func(opt *option) {
|
||
opt.RetryTimes = retryTimes
|
||
}
|
||
}
|
||
|
||
// WithRetryDelay 在重试前，延迟等待一会
|
||
func WithRetryDelay(retryDelay time.Duration) Option {
|
||
return func(opt *option) {
|
||
opt.RetryDelay = retryDelay
|
||
}
|
||
}
|