lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-26 08:45:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

maxapi: load http transport settings from env vars

Browse Source
This commit is contained in:
c9s 2021-06-08 00:08:03 +08:00
parent b31b830b2b
commit 684232041c
1 changed files with 28 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
pkg/exchange/max/maxapi/restapi.go
View File

@ -13,7 +13,6 @@ import (
"net/http" "net/http"
"net/http/httputil" "net/http/httputil"
"net/url" "net/url"
"os"
"reflect" "reflect"
"regexp" "regexp"
"strconv" "strconv"
@ -40,10 +39,23 @@ const (
var debugRequestDump = false var debugRequestDump = false
var debugMaxRequestPayload = false var debugMaxRequestPayload = false
var addUserAgentHeader = true
var httpTransportMaxIdleConnsPerHost = http.DefaultMaxIdleConnsPerHost
var httpTransportMaxIdleConns = 100
func init() { func init() {
debugMaxRequestPayload, _ = strconv.ParseBool(os.Getenv("DEBUG_MAX_REQUEST_PAYLOAD")) debugMaxRequestPayload, _ = util.GetEnvVarBool("DEBUG_MAX_REQUEST_PAYLOAD")
debugRequestDump, _ = strconv.ParseBool(os.Getenv("DEBUG_MAX_REQUEST")) debugRequestDump, _ = util.GetEnvVarBool("DEBUG_MAX_REQUEST")
addUserAgentHeader, _ = util.GetEnvVarBool("DISABLE_MAX_USER_AGENT_HEADER")
if val, ok := util.GetEnvVarInt("HTTP_TRANSPORT_MAX_IDLE_CONNS_PER_HOST"); ok {
httpTransportMaxIdleConnsPerHost = val
}
if val, ok := util.GetEnvVarInt("HTTP_TRANSPORT_MAX_IDLE_CONNS"); ok {
httpTransportMaxIdleConns = val
}
} }
var logger = log.WithField("exchange", "max") var logger = log.WithField("exchange", "max")
@ -106,9 +118,16 @@ func NewRestClientWithHttpClient(baseURL string, httpClient *http.Client) *RestC
} }
func NewRestClient(baseURL string) *RestClient { func NewRestClient(baseURL string) *RestClient {
return NewRestClientWithHttpClient(baseURL, &http.Client{ transport := http.DefaultTransport.(*http.Transport)
transport.MaxIdleConnsPerHost = httpTransportMaxIdleConnsPerHost
transport.MaxIdleConns = httpTransportMaxIdleConns
client := &http.Client{
Timeout: defaultHTTPTimeout, Timeout: defaultHTTPTimeout,
}) Transport: transport,
}
return NewRestClientWithHttpClient(baseURL, client)
} }
// Auth sets api key and secret for usage is requests that requires authentication. // Auth sets api key and secret for usage is requests that requires authentication.
@ -155,7 +174,9 @@ func (c *RestClient) newRequest(method string, refURL string, params url.Values,
return nil, err return nil, err
} }
if addUserAgentHeader {
req.Header.Add("User-Agent", UserAgent) req.Header.Add("User-Agent", UserAgent)
}
return req, nil return req, nil
} }