diff options
author | Neil Alexander <neilalexander@users.noreply.github.com> | 2021-05-10 16:58:48 +0100 |
---|---|---|
committer | Neil Alexander <neilalexander@users.noreply.github.com> | 2021-05-10 16:59:03 +0100 |
commit | a49d06138ec3bb35c3f5ee6e97c3c8ff0c198720 (patch) | |
tree | 7a6443583ad1970308d4e233adaf359c772ecca1 /cmd | |
parent | 81d60d54488a2e4b4bbcdbab0f7e01f780e5fbe8 (diff) |
Updates to dendrite-demo-pinecone
Diffstat (limited to 'cmd')
-rw-r--r-- | cmd/dendrite-demo-pinecone/conn/client.go | 38 |
1 files changed, 14 insertions, 24 deletions
diff --git a/cmd/dendrite-demo-pinecone/conn/client.go b/cmd/dendrite-demo-pinecone/conn/client.go index bf23085d..66f9ebfa 100644 --- a/cmd/dendrite-demo-pinecone/conn/client.go +++ b/cmd/dendrite-demo-pinecone/conn/client.go @@ -45,47 +45,37 @@ func (y *RoundTripper) RoundTrip(req *http.Request) (*http.Response, error) { return y.inner.RoundTrip(req) } -func CreateClient( - base *setup.BaseDendrite, s *pineconeSessions.Sessions, -) *gomatrixserverlib.Client { +func createTransport(s *pineconeSessions.Sessions) *http.Transport { tr := &http.Transport{} tr.RegisterProtocol( "matrix", &RoundTripper{ inner: &http.Transport{ - MaxIdleConns: 100, - MaxIdleConnsPerHost: 5, - Dial: s.Dial, - DialContext: s.DialContext, - DialTLS: s.DialTLS, - DialTLSContext: s.DialTLSContext, + DisableKeepAlives: false, + Dial: s.Dial, + DialContext: s.DialContext, + DialTLS: s.DialTLS, + DialTLSContext: s.DialTLSContext, }, }, ) + return tr +} + +func CreateClient( + base *setup.BaseDendrite, s *pineconeSessions.Sessions, +) *gomatrixserverlib.Client { return gomatrixserverlib.NewClient( - gomatrixserverlib.WithTransport(tr), + gomatrixserverlib.WithTransport(createTransport(s)), ) } func CreateFederationClient( base *setup.BaseDendrite, s *pineconeSessions.Sessions, ) *gomatrixserverlib.FederationClient { - tr := &http.Transport{} - tr.RegisterProtocol( - "matrix", &RoundTripper{ - inner: &http.Transport{ - MaxIdleConns: 100, - MaxIdleConnsPerHost: 5, - Dial: s.Dial, - DialContext: s.DialContext, - DialTLS: s.DialTLS, - DialTLSContext: s.DialTLSContext, - }, - }, - ) return gomatrixserverlib.NewFederationClient( base.Cfg.Global.ServerName, base.Cfg.Global.KeyID, base.Cfg.Global.PrivateKey, - gomatrixserverlib.WithTransport(tr), + gomatrixserverlib.WithTransport(createTransport(s)), ) } |