Replaces use of time.Sleep with time.Ticker in echovault and echovault tests

This commit is contained in:
Kelvin Clement Mwinuka
2024-06-02 17:34:20 +08:00
parent b7e691bbf1
commit bc6537ad8f
6 changed files with 601 additions and 569 deletions

View File

@@ -447,8 +447,13 @@ func GetConnection(addr string, port int) (net.Conn, error) {
done <- struct{}{}
}()
ticker := time.NewTicker(10 * time.Second)
defer func() {
ticker.Stop()
}()
select {
case <-time.After(10 * time.Second):
case <-ticker.C:
return nil, errors.New("connection timeout")
case <-done:
return conn, err
@@ -472,8 +477,13 @@ func GetTLSConnection(addr string, port int, config *tls.Config) (net.Conn, erro
done <- struct{}{}
}()
ticker := time.NewTicker(10 * time.Second)
defer func() {
ticker.Stop()
}()
select {
case <-time.After(10 * time.Second):
case <-ticker.C:
return nil, errors.New("connection timeout")
case <-done:
return conn, err