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

connectivity: refactor and drop AnyDisconnected

Browse Source
This commit is contained in:
c9s 2024-11-18 14:08:35 +08:00
parent e497b8b5e5
commit d22d16ff95
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 3 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pkg/types/connectivitygroup.go
View File

@ -173,28 +173,7 @@ func (g *ConnectivityGroup) Add(con *Connectivity) {
})
}
func (g *ConnectivityGroup) AnyDisconnected(ctx context.Context) bool {
g.mu.Lock()
conns := g.connections
g.mu.Unlock()
for _, conn := range conns {
select {
case <-ctx.Done():
return false
case <-conn.connectedC:
continue
case <-conn.disconnectedC:
return true
}
}
return false
}
func (g *ConnectivityGroup) waitAllAuthed(ctx context.Context, c chan struct{}) {
func (g *ConnectivityGroup) waitForState(ctx context.Context, c chan struct{}, expected ConnectivityState) {
for {
select {
case <-ctx.Done():
@ -202,7 +181,7 @@ func (g *ConnectivityGroup) waitAllAuthed(ctx context.Context, c chan struct{})
default:
state := g.GetState()
if state == ConnectivityStateAuthed {
if state == expected {
close(c)
return
}
@ -217,6 +196,6 @@ func (g *ConnectivityGroup) waitAllAuthed(ctx context.Context, c chan struct{})
// and the channel can only be used once (because we can't close a channel twice)
func (g *ConnectivityGroup) AllAuthedC(ctx context.Context) <-chan struct{} {
c := make(chan struct{})
go g.waitAllAuthed(ctx, c)
go g.waitForState(ctx, c, ConnectivityStateAuthed)
return c
}