transport: conn, abort OpenShardConn faster when context is done

Author: Kulezi

transport/conn.go

@@ -378,6 +378,10 @@ func OpenShardConn(ctx context.Context, addr string, si ShardInfo, cfg ConnConfi
 	for i := 0; i < maxTries; i++ {
 		conn, err := OpenLocalPortConn(ctx, addr, it(), cfg)
 		if err != nil {
+			if err := ctx.Err(); err != nil {
+				return nil, fmt.Errorf("aborted opening connection on shard %d: %w", si.Shard, err)
+			}
+
 			log.Printf("%s dial error: %s (try %d/%d)", addr, err, i, maxTries)
 			if conn != nil {
 				conn.Close()