Mirroristas/pockethost
2
0
Fork 0
mirror of https://github.com/pockethost/pockethost.git synced 2025-11-24 14:35:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: move stats initialization into proxyservice

Browse Source
This commit is contained in:
Ben Allfree 2024-11-28 00:41:45 -08:00
parent 00f3a4dc0a
commit fec6d4fa59
1 changed files with 54 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
packages/pockethost/src/services/ProxyService.ts
View File

@ -27,57 +27,6 @@ export type ProxyMiddleware = (
logger: Logger, logger: Logger,
) => boolean | Promise<boolean> ) => boolean | Promise<boolean>
const stats = (() => {
const metrics = {
requests: 0,
errors: 0,
hosts: new Map<string, number>(),
ips: new Map<string, number>(),
countries: new Map<string, number>(),
}
setInterval(() => {
const top10Ips = Array.from(metrics.ips.entries())
.sort(([, a], [, b]) => b - a)
.slice(0, 10)
const topHosts = Array.from(metrics.hosts.entries())
.sort(([, a], [, b]) => b - a)
.slice(0, 10)
const top10Countries = Array.from(metrics.countries.entries())
.sort(([, a], [, b]) => b - a)
.slice(0, 10)
console.log({
...metrics,
ips: top10Ips,
hosts: topHosts,
countries: top10Countries,
})
metrics.requests = 0
metrics.errors = 0
metrics.ips.clear()
metrics.hosts.clear()
metrics.countries.clear()
}, 10000)
return {
addRequest: () => {
metrics.requests++
},
addError: () => {
metrics.errors++
},
addHost: (host: string) => {
metrics.hosts.set(host, (metrics.hosts.get(host) || 0) + 1)
},
addIp: (ip: string) => {
metrics.ips.set(ip, (metrics.ips.get(ip) || 0) + 1)
},
addCountry: (country: string) => {
metrics.countries.set(country, (metrics.countries.get(country) || 0) + 1)
},
}
})()
export type ProxyServiceConfig = SingletonBaseConfig & { export type ProxyServiceConfig = SingletonBaseConfig & {
coreInternalUrl: string coreInternalUrl: string
} }
@ -90,6 +39,60 @@ export const proxyService = mkSingleton(
const { coreInternalUrl } = config const { coreInternalUrl } = config
const stats = (() => {
const metrics = {
requests: 0,
errors: 0,
hosts: new Map<string, number>(),
ips: new Map<string, number>(),
countries: new Map<string, number>(),
}
setInterval(() => {
const top10Ips = Array.from(metrics.ips.entries())
.sort(([, a], [, b]) => b - a)
.slice(0, 10)
const topHosts = Array.from(metrics.hosts.entries())
.sort(([, a], [, b]) => b - a)
.slice(0, 10)
const top10Countries = Array.from(metrics.countries.entries())
.sort(([, a], [, b]) => b - a)
.slice(0, 10)
console.log({
...metrics,
ips: top10Ips,
hosts: topHosts,
countries: top10Countries,
})
metrics.requests = 0
metrics.errors = 0
metrics.ips.clear()
metrics.hosts.clear()
metrics.countries.clear()
}, 10000)
return {
addRequest: () => {
metrics.requests++
},
addError: () => {
metrics.errors++
},
addHost: (host: string) => {
metrics.hosts.set(host, (metrics.hosts.get(host) || 0) + 1)
},
addIp: (ip: string) => {
metrics.ips.set(ip, (metrics.ips.get(ip) || 0) + 1)
},
addCountry: (country: string) => {
metrics.countries.set(
country,
(metrics.countries.get(country) || 0) + 1,
)
},
}
})()
const proxy = httpProxy.createProxyServer({}) const proxy = httpProxy.createProxyServer({})
proxy.on('error', (err, req, res, target) => { proxy.on('error', (err, req, res, target) => {
warn(`Proxy error ${err} on ${req.url} (${req.headers.host})`) warn(`Proxy error ${err} on ${req.url} (${req.headers.host})`)