lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-13 02:23:51 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

grpc: fix connect and add balance snapshot

Browse Source
This commit is contained in:
c9s 2022-04-15 14:28:35 +08:00
parent dd6588b1e9
commit 84d4f312fa
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/grpc/server.go
View File

@ -30,7 +30,7 @@ func (s *UserDataService) Subscribe(request *pb.UserDataRequest, server pb.UserD
if len(sessionName) == 0 { if len(sessionName) == 0 {
return fmt.Errorf("session name can not be empty") return fmt.Errorf("session name can not be empty")
} }
session, ok := s.Environ.Session(sessionName) session, ok := s.Environ.Session(sessionName)
if !ok { if !ok {
return fmt.Errorf("session %s not found", sessionName) return fmt.Errorf("session %s not found", sessionName)
@ -71,13 +71,31 @@ func (s *UserDataService) Subscribe(request *pb.UserDataRequest, server pb.UserD
userDataStream.OnBalanceUpdate(balanceHandler) userDataStream.OnBalanceUpdate(balanceHandler)
userDataStream.OnBalanceSnapshot(balanceHandler) userDataStream.OnBalanceSnapshot(balanceHandler)
ctx := server.Context()
balances, err := session.Exchange.QueryAccountBalances(ctx)
if err != nil {
return err
}
err = server.Send(&pb.UserData{
Channel: pb.Channel_BALANCE,
Event: pb.Event_SNAPSHOT,
Balances: transBalances(session, balances),
})
if err != nil {
log.WithError(err).Errorf("grpc: can not send user data")
}
go userDataStream.Connect(ctx)
defer func() { defer func() {
if err := userDataStream.Close(); err != nil { if err := userDataStream.Close(); err != nil {
log.WithError(err).Errorf("user data stream close error") log.WithError(err).Errorf("user data stream close error")
} }
}() }()
<-server.Context().Done() <-ctx.Done()
return nil return nil
} }