ZZZ-Plugin/utils/network.js
2024-07-10 14:47:23 +08:00

48 lines
1.2 KiB
JavaScript

import http from 'http';
import https from 'https';
export async function checkLatency(url) {
let request = http;
if (url.startsWith('https')) {
request = https;
}
return new Promise(resolve => {
const start = Date.now();
request
.get(url, res => {
res.on('data', () => {});
res.on('end', () => {
const latency = Date.now() - start;
resolve({ url, latency });
});
})
.on('error', err => {
logger.mark(`Error checking ${url}:`, err.message);
resolve({ url, latency: Infinity });
});
});
}
export async function findLowestLatencyUrl(urls) {
const results = await Promise.all(urls.map(checkLatency));
const lowestLatencyResult = results.reduce((prev, curr) =>
prev.latency < curr.latency ? prev : curr
);
return lowestLatencyResult.url;
}
export function getQueryVariable(url, variable) {
try {
const _url = new URL(url);
var query = _url.search.substring(1);
var vars = query.split('&');
for (var i = 0; i < vars.length; i++) {
var pair = vars[i].split('=');
if (pair[0] == variable) {
return pair[1];
}
}
return null;
} catch (error) {
return null;
}
}