node-gamedig/lib/QueryRunner.js

104 lines
3.2 KiB
JavaScript
Raw Normal View History

2019-01-12 11:43:36 +01:00
const GameResolver = require('./GameResolver'),
ProtocolResolver = require('./ProtocolResolver'),
GlobalUdpSocket = require('./GlobalUdpSocket');
2019-01-12 11:43:36 +01:00
const defaultOptions = {
socketTimeout: 2000,
attemptTimeout: 10000,
maxAttempts: 1
};
class QueryRunner {
constructor() {
this.udpSocket = new GlobalUdpSocket();
2019-01-12 11:43:36 +01:00
this.gameResolver = new GameResolver();
this.protocolResolver = new ProtocolResolver();
}
async run(userOptions) {
for (const key of Object.keys(userOptions)) {
const value = userOptions[key];
if (['port'].includes(key)) {
userOptions[key] = parseInt(value);
}
}
const {
port_query: gameQueryPort,
port_query_offset: gameQueryPortOffset,
...gameOptions
} = this.gameResolver.lookup(userOptions.type);
const attempts = [];
if (userOptions.port) {
if (gameQueryPortOffset) {
attempts.push({
...defaultOptions,
...gameOptions,
...userOptions,
port: userOptions.port + gameQueryPortOffset
});
}
if (userOptions.port === gameOptions.port && gameQueryPort) {
attempts.push({
...defaultOptions,
...gameOptions,
...userOptions,
port: gameQueryPort
});
}
attempts.push({
...defaultOptions,
...gameOptions,
...userOptions
});
} else if (gameQueryPort) {
attempts.push({
...defaultOptions,
...gameOptions,
...userOptions,
port: gameQueryPort
});
} else if (gameOptions.port) {
attempts.push({
...defaultOptions,
...gameOptions,
2019-01-12 12:45:09 +01:00
...userOptions,
port: gameOptions.port + (gameQueryPortOffset || 0)
2019-01-12 11:43:36 +01:00
});
} else {
throw new Error("Could not determine port to query. Did you provide a port or gameid?");
}
2019-02-14 05:46:13 +01:00
const numRetries = userOptions.maxAttempts || gameOptions.maxAttempts || defaultOptions.maxAttempts;
let attemptNum = 0;
const errors = [];
for (const attempt of attempts) {
for (let retry = 0; retry < numRetries; retry++) {
attemptNum++;
2019-01-12 11:43:36 +01:00
try {
return await this._attempt(attempt);
2019-02-14 05:46:13 +01:00
} catch (e) {
e.stack = 'Attempt #' + attemptNum + ' - Port=' + attempt.port + ' Retry=' + (retry) + ':\n' + e.stack;
errors.push(e);
2019-01-12 11:43:36 +01:00
}
}
2019-02-14 05:46:13 +01:00
}
2019-01-12 11:43:36 +01:00
2019-02-14 05:46:13 +01:00
const err = new Error('Failed all ' + errors.length + ' attempts');
for (const e of errors) {
err.stack += '\n' + e.stack;
2019-01-12 11:43:36 +01:00
}
2019-02-14 05:46:13 +01:00
throw err;
2019-01-12 11:43:36 +01:00
}
async _attempt(options) {
const core = this.protocolResolver.create(options.protocol);
core.options = options;
core.udpSocket = this.udpSocket;
2019-02-14 05:46:13 +01:00
return await core.runOnceSafe();
2019-01-12 11:43:36 +01:00
}
}
module.exports = QueryRunner;