add relaysBatchesOf function to implemenent rolling deploy

input-output-hk · Aug 11, 2020 · f1c7a07 · f1c7a07
1 parent 3e20964
commit f1c7a07
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nix/topology-lib.nix b/nix/topology-lib.nix
@@ -133,4 +133,7 @@ pkgs: with pkgs; with lib; rec {
         in withinOneHop relaysForRegion
       ) indexedRegions));
 
+  relaysBatchesOf = n:
+    let byRegions = attrValues (mapAttrs (_: rs: let irs = imap0 (i: mergeAttrs {inherit i;}) rs; in genList (i: (map (r: r.name) (filter (r: mod r.i n == i) irs))) n) (groupBy (r: r.region) globals.topology.relayNodes));
+    in genList (i: concatMap (rs: elemAt rs i) byRegions) n;
 }