1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#!/usr/bin/env node
if (process.argv.length != 7) {
console.log("Usage: ./run_tour.js <seed_file> <num_turns> <num_workers> <output_file> <logs_dir>");
console.log("<seed_file> should contain a 32-hex-digit seed on each line. One game will be run for each seed. The games will all have <num_turns> turns.");
console.log("The tournament will be run using <num_workers> parallel workers.")
console.log("Output results will be written to <output_file>.")
console.log("Game log files will be written to the directory <logs_dir>.");
process.exit(1);
}
const fs = require("fs");
const child_process = require("child_process");
const mkdirp = require("mkdirp");
const seed_file = process.argv[2];
const num_turns = +process.argv[3];
const num_workers = +process.argv[4];
const output_file = process.argv[5];
const logs_dir = process.argv[6];
mkdirp.sync(logs_dir);
const output_stream = fs.createWriteStream(output_file);
const seeds =
fs.readFileSync(seed_file).toString()
.split("\n").filter(s => s.length > 0);
console.log(seeds.length + " games to play using " + num_workers + " workers.");
function register_result(result) {
output_stream.write(result + "\n");
const seed = result.split(" ", 1)[0];
console.log("Finished game with seed " + seed);
}
const workers = [];
const current_seed = new Array(num_workers).fill(null);
const start_at = new Array(num_workers).fill(null);
function distribute_work() {
if (seeds.length == 0) return;
for (let i = 0; i < num_workers; i++) {
if (current_seed[i] == null) {
const seed = seeds.shift();
current_seed[i] = seed;
start_at[i] = new Date();
workers[i].stdin.write(seed + "\n");
console.log("Started game with seed " + seed);
}
}
}
function check_finished() {
for (let i = 0; i < num_workers; i++) {
if (current_seed[i] != null) return false;
}
console.log("\nFINISHED! Shutting down...");
output_stream.end();
for (const worker of workers) {
worker.stdin.end();
}
console.log("All done.");
setTimeout(() => process.exit(0), 500);
}
function spawn_worker(index) {
const worker = child_process.spawn(
"node",
["tour_worker.js", num_turns.toString(), logs_dir],
{ stdio: ["pipe", "pipe", "inherit"] }
);
let buffer = "";
worker.stdout.on("data", (data) => {
buffer += data;
let idx = buffer.indexOf("\n");
while (idx != -1) {
const line = buffer.slice(0, idx);
buffer = buffer.slice(idx + 1);
idx = buffer.indexOf("\n");
register_result(line);
current_seed[index] = null;
start_at[index] = null;
distribute_work();
if (check_finished()) return;
}
});
return worker;
}
for (let i = 0; i < num_workers; i++) {
workers.push(spawn_worker(workers.length));
}
distribute_work();
// Check for stuck games once in a while
setInterval(function() {
const max_game_time = 1350 * num_turns;
const now = new Date();
for (let i = 0; i < num_workers; i++) {
if (current_seed[i] != null && now - start_at[i] > max_game_time) {
console.log("Game for seed " + current_seed[i] + " is taking too long, KILLING!");
workers[i].kill("SIGKILL");
console.log("Spawning new worker and hoping everything's fine...");
workers[i] = spawn_worker(i);
}
}
}, 60 * 1000);
|
