Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
b2c5d8f41c | |||
4d2c1622d3 | |||
82d1c5a8c8 |
5
main.go
5
main.go
@@ -3,7 +3,6 @@ package cyutils
|
||||
import (
|
||||
"net/http"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"golang.org/x/time/rate"
|
||||
)
|
||||
@@ -58,11 +57,10 @@ func (t *RateLimitedTransport) RoundTrip(req *http.Request) (*http.Response, err
|
||||
}
|
||||
return t.base.RoundTrip(req)
|
||||
}
|
||||
func LimitedHttp(rps float64, burst int, timeout time.Duration) *http.Client {
|
||||
func LimitedHttp(rps float64, burst int) *http.Client {
|
||||
baseTransport := &http.Transport{
|
||||
MaxIdleConns: 100,
|
||||
MaxIdleConnsPerHost: 10,
|
||||
IdleConnTimeout: 10 * time.Second,
|
||||
}
|
||||
|
||||
// Initialize rate limiter
|
||||
@@ -77,6 +75,5 @@ func LimitedHttp(rps float64, burst int, timeout time.Duration) *http.Client {
|
||||
// Return configured client
|
||||
return &http.Client{
|
||||
Transport: transport,
|
||||
Timeout: timeout,
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user