tests: support to stop the multiple members concurrently

Signed-off-by: Wei Fu <fuweid89@gmail.com>
dependabot/go_modules/go.uber.org/atomic-1.10.0
Wei Fu 2023-02-03 22:23:06 +08:00
parent 712bd8a802
commit fa527c5e54
1 changed files with 10 additions and 3 deletions

View File

@ -823,11 +823,17 @@ func (epc *EtcdProcessCluster) rollingStart(f func(ep EtcdProcess) error) error
}
func (epc *EtcdProcessCluster) Stop() (err error) {
for _, p := range epc.Procs {
if p == nil {
errCh := make(chan error, len(epc.Procs))
for i := range epc.Procs {
if epc.Procs[i] == nil {
errCh <- nil
continue
}
if curErr := p.Stop(); curErr != nil {
go func(n int) { errCh <- epc.Procs[n].Stop() }(i)
}
for range epc.Procs {
if curErr := <-errCh; curErr != nil {
if err != nil {
err = fmt.Errorf("%v; %v", err, curErr)
} else {
@ -835,6 +841,7 @@ func (epc *EtcdProcessCluster) Stop() (err error) {
}
}
}
close(errCh)
return err
}