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

grpc: allocate a stream pool

Browse Source
This commit is contained in:
c9s 2022-04-12 17:48:30 +08:00
parent fb5703bf13
commit 8705f38220
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/grpc/server.go
View File

@ -58,6 +58,7 @@ func (s *Server) Subscribe(request *pb.SubscribeRequest, server pb.MarketDataSer
}
}
streamPool := map[string]types.Stream{}
for sessionName, subs := range exchangeSubscriptions {
if session, ok := s.Environ.Session(sessionName); ok {
stream := session.Exchange.NewStream()
@ -83,10 +84,15 @@ func (s *Server) Subscribe(request *pb.SubscribeRequest, server pb.MarketDataSer
log.WithError(err).Error("grpc stream send error")
}
})
go stream.Connect(server.Context())
streamPool[sessionName] = stream
}
}
for sessionName, stream := range streamPool {
log.Infof("connecting stream %s", sessionName)
go stream.Connect(server.Context())
}
return nil
}