transport: disallow taking connections to nodes that are down

Kulezi · Kulezi · commit c06f70655da5 · 2022-09-23T10:20:09.000+02:00
Fixes #222
diff --git a/transport/node.go b/transport/node.go
@@ -54,9 +54,16 @@ func (n *Node) Close() {
 }
 
 func (n *Node) LeastBusyConn() (*Conn, error) {
+	if !n.IsUp() {
+		return nil, fmt.Errorf("node %v is down", n)
+	}
+
 	return n.pool.LeastBusyConn()
 }
 func (n *Node) Conn(qi QueryInfo) (*Conn, error) {
+	if !n.IsUp() {
+		return nil, fmt.Errorf("node %v is down", n)
+	}
 	if qi.tokenAware {
 		return n.pool.Conn(qi.token)
 	}

Original file line number	Diff line number	Diff line change
`@@ -54,9 +54,16 @@ func (n *Node) Close() {`
`54`	`54`	`}`
`55`	`55`
`56`	`56`	`func (n Node) LeastBusyConn() (Conn, error) {`
	`57`	`+ if !n.IsUp() {`
	`58`	`+ return nil, fmt.Errorf("node %v is down", n)`
	`59`	`+ }`
	`60`	`+`
`57`	`61`	`return n.pool.LeastBusyConn()`
`58`	`62`	`}`
`59`	`63`	`func (n Node) Conn(qi QueryInfo) (Conn, error) {`
	`64`	`+ if !n.IsUp() {`
	`65`	`+ return nil, fmt.Errorf("node %v is down", n)`
	`66`	`+ }`
`60`	`67`	`if qi.tokenAware {`
`61`	`68`	`return n.pool.Conn(qi.token)`
`62`	`69`	`}`