Fix non existing clients getting stacked up in overload scenarios

anexia · Oct 9, 2023 · 4d44184 · 4d44184
1 parent 51c592b
commit 4d44184
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/connection/connection_pool.go b/connection/connection_pool.go
@@ -26,6 +26,7 @@
 package connection
 
 import (
+ "errors"
  "rmux/graphite"
  "rmux/log"
  "strings"
@@ -117,7 +118,8 @@ func (cp *ConnectionPool) GetConnection() (connection *Connection, err error) {
  }
 
  return connection, nil
- // TODO: Maybe a while/timeout/graphiteping loop?
+ case <-time.After(1 * time.Second):
+ return nil, errors.New("timeout while waiting for a new connection")
  }
 }