lkddi/nexusphp
1
0
Fork 0
mirror of https://github.com/lkddi/nexusphp.git synced 2026-04-24 03:57:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

add cache for isIPSeedBoxFromASN

Browse Source
This commit is contained in:
xiaomlove
2025-01-21 13:24:10 +08:00
parent b3702cb065
commit 2ef6fc3c88
1 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/globalfunctions.php
+18 -2
View File
@@ -863,6 +863,9 @@ function do_action($name, ...$args)
function isIPSeedBoxFromASN($ip, $exceptionWhenYes = false): bool function isIPSeedBoxFromASN($ip, $exceptionWhenYes = false): bool
{ {
$redis = \Nexus\Database\NexusDB::redis();
$key = "nexus_asn";
$notFoundCacheValue = "__NOT_FOUND__";
try { try {
static $reader; static $reader;
$database = nexus_env('GEOIP2_ASN_DATABASE'); $database = nexus_env('GEOIP2_ASN_DATABASE');
@@ -878,10 +881,23 @@ function isIPSeedBoxFromASN($ip, $exceptionWhenYes = false): bool
if ($asn <= 0) { if ($asn <= 0) {
return false; return false;
} }
$cacheResult = $redis->hGet($key, $asn);
if ($cacheResult !== false) {
if ($cacheResult === $notFoundCacheValue) {
return false;
} else {
return true;
}
}
$row = \Nexus\Database\NexusDB::getOne("seed_box_records", "asn = $asn", "id"); $row = \Nexus\Database\NexusDB::getOne("seed_box_records", "asn = $asn", "id");
if (!empty($row)) {
$redis->hSet($key, $asn, $row['id']);
} else {
$redis->hSet($key, $asn, $notFoundCacheValue);
}
} catch (\Throwable $throwable) { } catch (\Throwable $throwable) {
do_log("ip: $ip, error: " . $throwable->getMessage(), "error"); do_log("ip: $ip, " . $throwable->getMessage());
return false; $redis->hSet($key, $asn, $notFoundCacheValue);
} }
$result = !empty($row); $result = !empty($row);
if ($result && $exceptionWhenYes) { if ($result && $exceptionWhenYes) {