mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2026-01-16 18:43:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

Reuse existing connections, as we don't need separate ones

Browse source
This commit is contained in:
Till Faelligen 2023-03-16 11:14:51 +01:00
parent 3a2e8606c7
commit d0a764a083
No known key found for this signature in database
GPG key ID: ACCDC9606D472758
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
setup/jetstream/nats.go
View file

@ -20,6 +20,8 @@ import (
type NATSInstance struct { type NATSInstance struct {
*natsserver.Server *natsserver.Server
nc *natsclient.Conn
js natsclient.JetStreamContext
} }
var natsLock sync.Mutex var natsLock sync.Mutex
@ -69,11 +71,18 @@ func (s *NATSInstance) Prepare(process *process.ProcessContext, cfg *config.JetS
if !s.ReadyForConnections(time.Second * 10) { if !s.ReadyForConnections(time.Second * 10) {
logrus.Fatalln("NATS did not start in time") logrus.Fatalln("NATS did not start in time")
} }
// reuse existing connections
if s.nc != nil {
return s.js, s.nc
}
nc, err := natsclient.Connect("", natsclient.InProcessServer(s)) nc, err := natsclient.Connect("", natsclient.InProcessServer(s))
if err != nil { if err != nil {
logrus.Fatalln("Failed to create NATS client") logrus.Fatalln("Failed to create NATS client")
} }
return setupNATS(process, cfg, nc) js, _ := setupNATS(process, cfg, nc)
s.js = js
s.nc = nc
return js, nc
} }
func setupNATS(process *process.ProcessContext, cfg *config.JetStream, nc *natsclient.Conn) (natsclient.JetStreamContext, *natsclient.Conn) { func setupNATS(process *process.ProcessContext, cfg *config.JetStream, nc *natsclient.Conn) (natsclient.JetStreamContext, *natsclient.Conn) {