Merge branch 'master' into experimental

This commit is contained in:
DebaucheryLibrarian 2021-03-15 04:16:36 +01:00
commit 1fb7d384fb
4 changed files with 28 additions and 5 deletions

4
package-lock.json generated
View File

@ -1,11 +1,11 @@
{ {
"name": "traxxx", "name": "traxxx",
"version": "1.184.1", "version": "1.184.2",
"lockfileVersion": 2, "lockfileVersion": 2,
"requires": true, "requires": true,
"packages": { "packages": {
"": { "": {
"version": "1.184.1", "version": "1.184.2",
"license": "ISC", "license": "ISC",
"dependencies": { "dependencies": {
"@casl/ability": "^5.2.2", "@casl/ability": "^5.2.2",

View File

@ -1,6 +1,6 @@
{ {
"name": "traxxx", "name": "traxxx",
"version": "1.184.1", "version": "1.184.2",
"description": "All the latest porn releases in one place", "description": "All the latest porn releases in one place",
"main": "src/app.js", "main": "src/app.js",
"scripts": { "scripts": {

View File

@ -1,6 +1,7 @@
'use strict'; 'use strict';
const config = require('config'); const config = require('config');
const Promise = require('bluebird');
const bhttp = require('bhttp'); const bhttp = require('bhttp');
const util = require('util'); const util = require('util');
const stream = require('stream'); const stream = require('stream');
@ -15,7 +16,12 @@ const argv = require('../argv');
const pipeline = util.promisify(stream.pipeline); const pipeline = util.promisify(stream.pipeline);
const limiters = {}; const limiters = {};
Promise.config({
cancellation: true,
});
const defaultOptions = { const defaultOptions = {
timeout: 60000,
encodeJSON: true, encodeJSON: true,
headers: { headers: {
'user-agent': 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:15.0) Gecko/20100101 Firefox/15.0.1', 'user-agent': 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:15.0) Gecko/20100101 Firefox/15.0.1',
@ -67,6 +73,7 @@ function getLimiter(options = {}, url) {
limiters[interval][concurrency] = new Bottleneck({ limiters[interval][concurrency] = new Bottleneck({
minTime: interval, minTime: interval,
maxConcurrent: concurrency, maxConcurrent: concurrency,
timeout: (options.timeout || defaultOptions.timeout) + 10000, // timeout 10 seconds after bhttp should
}); });
} }
@ -83,7 +90,7 @@ async function request(method = 'get', url, body, requestOptions = {}, limiter)
...defaultOptions.headers, ...defaultOptions.headers,
...requestOptions.headers, ...requestOptions.headers,
}, },
responseTimeout: requestOptions.responseTimeout || requestOptions.timeout || 60000, responseTimeout: requestOptions.responseTimeout || requestOptions.timeout || defaultOptions.timeout,
stream: !!requestOptions.destination, stream: !!requestOptions.destination,
session: null, session: null,
}; };
@ -131,10 +138,25 @@ async function request(method = 'get', url, body, requestOptions = {}, limiter)
}; };
} }
function getTimeout(options, url) {
return new Promise((resolve, reject, onCancel) => {
const timeoutId = setTimeout(() => {
reject(new Error(`URL ${url} timed out`));
}, (options?.timeout || defaultOptions.timeout) + 10000);
onCancel(() => clearTimeout(timeoutId));
});
}
async function scheduleRequest(method = 'get', url, body, options) { async function scheduleRequest(method = 'get', url, body, options) {
const limiter = getLimiter(options, url); const limiter = getLimiter(options, url);
const timeout = getTimeout(options, url);
return limiter.schedule(() => request(method, url, body, options, limiter)); const result = await limiter.schedule(() => Promise.race([request(method, url, body, options, limiter), timeout]));
timeout.cancel();
return result;
} }
async function get(url, options) { async function get(url, options) {

View File

@ -45,4 +45,5 @@ module.exports = postgraphile(
], ],
pgSettings, pgSettings,
}, },
pgSettings,
); );