Revision 02e7f855cb4dcc54e2f1e1de41e8720cd1a9ed7f authored by pecornilleau on 04 April 2024, 16:11:12 UTC, committed by pecornilleau on 04 April 2024, 17:15:54 UTC
1 parent eaf7214
cluster.ml
(*****************************************************************************)
(* *)
(* Open Source License *)
(* Copyright (c) 2020 Nomadic Labs <contact@nomadic-labs.com> *)
(* *)
(* Permission is hereby granted, free of charge, to any person obtaining a *)
(* copy of this software and associated documentation files (the "Software"),*)
(* to deal in the Software without restriction, including without limitation *)
(* the rights to use, copy, modify, merge, publish, distribute, sublicense, *)
(* and/or sell copies of the Software, and to permit persons to whom the *)
(* Software is furnished to do so, subject to the following conditions: *)
(* *)
(* The above copyright notice and this permission notice shall be included *)
(* in all copies or substantial portions of the Software. *)
(* *)
(* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR*)
(* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, *)
(* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL *)
(* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER*)
(* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING *)
(* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER *)
(* DEALINGS IN THE SOFTWARE. *)
(* *)
(*****************************************************************************)
let next_name = ref 1
let fresh_name () =
let index = !next_name in
incr next_name ;
"cluster" ^ string_of_int index
let create ?path ?name count ?rpc_external arguments =
let name = match name with None -> fresh_name () | Some name -> name in
List.map
(fun i ->
Node.create
?path
~name:(name ^ "." ^ string_of_int i)
?rpc_external
arguments)
(range 1 count)
let symmetric_add_peer a b =
Node.add_peer a b ;
Node.add_peer b a
let meta_connect connect a b = List.iter (fun a -> List.iter (connect a) b) a
let rec meta_clique connect = function
| [] -> ()
| head :: tail ->
List.iter (connect head) tail ;
meta_clique connect tail
let rec meta_clique_lwt connect = function
| [] -> unit
| head :: tail ->
let* () = Lwt_list.iter_s (connect head) tail in
meta_clique_lwt connect tail
let meta_ring connect nodes =
match nodes with
| [] -> ()
| first :: _ ->
let rec loop = function
| [] ->
(* We checked that the list was not empty already. *)
assert false
| [last] -> connect last first
| a :: (b :: _ as tail) ->
connect a b ;
loop tail
in
loop nodes
let meta_star connect center other_nodes =
List.iter (connect center) other_nodes
let connect = meta_connect symmetric_add_peer
let clique = meta_clique symmetric_add_peer
let ring = meta_ring symmetric_add_peer
let star = meta_star symmetric_add_peer
let start ?(public = false) ?event_level ?event_sections_levels
?(wait_connections = false) nodes =
let start_node node =
let* () = Node.identity_generate node in
let n = Node.get_peers node |> List.length in
let* () = Node.config_init node [] in
let* () =
Node.run
?event_level
?event_sections_levels
node
(if public then [] else [Private_mode])
in
let waiter =
let* () =
if wait_connections then (
Node.on_event node (fun {name; _} ->
match name with
| "disconnection.v0" ->
Log.debug "The topology of the test has changed"
| _ -> ()) ;
Node.wait_for_connections node n)
else unit
in
Node.wait_for_ready node
in
waiter
in
Lwt_list.iter_p start_node nodes
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...