aboutsummaryrefslogtreecommitdiff
path: root/cmd
diff options
context:
space:
mode:
authorNeil Alexander <neilalexander@users.noreply.github.com>2021-05-10 16:58:48 +0100
committerNeil Alexander <neilalexander@users.noreply.github.com>2021-05-10 16:59:03 +0100
commita49d06138ec3bb35c3f5ee6e97c3c8ff0c198720 (patch)
tree7a6443583ad1970308d4e233adaf359c772ecca1 /cmd
parent81d60d54488a2e4b4bbcdbab0f7e01f780e5fbe8 (diff)
Updates to dendrite-demo-pinecone
Diffstat (limited to 'cmd')
-rw-r--r--cmd/dendrite-demo-pinecone/conn/client.go38
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)),
)
}