mirror of
https://github.com/lkddi/Xboard.git
synced 2026-04-28 06:47:24 +08:00
refactor: restructure device limit system
This commit is contained in:
xboard
@@ -0,0 +1,148 @@
|
||||
<?php
|
||||
|
||||
namespace App\Services;
|
||||
|
||||
use App\Models\User;
|
||||
use Illuminate\Support\Collection;
|
||||
use Illuminate\Support\Facades\Redis;
|
||||
|
||||
class DeviceStateService
|
||||
{
|
||||
private const PREFIX = 'user_devices:';
|
||||
private const TTL = 300; // device state ttl
|
||||
private const DB_THROTTLE = 10; // update db throttle
|
||||
|
||||
/**
|
||||
* 批量设置设备
|
||||
* 用于 HTTP /alive 和 WebSocket report.devices
|
||||
*/
|
||||
public function setDevices(int $userId, int $nodeId, array $ips): void
|
||||
{
|
||||
$key = self::PREFIX . $userId;
|
||||
$timestamp = time();
|
||||
|
||||
$this->clearNodeDevices($nodeId, $userId);
|
||||
|
||||
if (!empty($ips)) {
|
||||
$fields = [];
|
||||
foreach ($ips as $ip) {
|
||||
$fields["{$nodeId}:{$ip}"] = $timestamp;
|
||||
}
|
||||
Redis::hMset($key, $fields);
|
||||
Redis::expire($key, self::TTL);
|
||||
}
|
||||
|
||||
$this->notifyUpdate($userId);
|
||||
}
|
||||
|
||||
/**
|
||||
* clear node devices
|
||||
* - only nodeId: clear all devices of the node
|
||||
* - userId and nodeId: clear specific user's specific node device
|
||||
*/
|
||||
public function clearNodeDevices(int $nodeId, ?int $userId = null): void
|
||||
{
|
||||
if ($userId !== null) {
|
||||
$key = self::PREFIX . $userId;
|
||||
$prefix = "{$nodeId}:";
|
||||
foreach (Redis::hkeys($key) as $field) {
|
||||
if (str_starts_with($field, $prefix)) {
|
||||
Redis::hdel($key, $field);
|
||||
}
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
$keys = Redis::keys(self::PREFIX . '*');
|
||||
$prefix = "{$nodeId}:";
|
||||
|
||||
foreach ($keys as $key) {
|
||||
foreach (Redis::hkeys($key) as $field) {
|
||||
if (str_starts_with($field, $prefix)) {
|
||||
Redis::hdel($key, $field);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* get user device count (filter expired data)
|
||||
*/
|
||||
public function getDeviceCount(int $userId): int
|
||||
{
|
||||
$data = Redis::hgetall(self::PREFIX . $userId);
|
||||
$now = time();
|
||||
$count = 0;
|
||||
foreach ($data as $field => $timestamp) {
|
||||
// if ($now - $timestamp <= self::TTL) {
|
||||
$count++;
|
||||
// }
|
||||
}
|
||||
return $count;
|
||||
}
|
||||
|
||||
/**
|
||||
* get user device count (for alivelist interface)
|
||||
*/
|
||||
public function getAliveList(Collection $users): array
|
||||
{
|
||||
if ($users->isEmpty()) {
|
||||
return [];
|
||||
}
|
||||
|
||||
$result = [];
|
||||
foreach ($users as $user) {
|
||||
$count = $this->getDeviceCount($user->id);
|
||||
if ($count > 0) {
|
||||
$result[$user->id] = $count;
|
||||
}
|
||||
}
|
||||
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* get devices of multiple users (for sync.devices, filter expired data)
|
||||
*/
|
||||
public function getUsersDevices(array $userIds): array
|
||||
{
|
||||
$result = [];
|
||||
$now = time();
|
||||
|
||||
foreach ($userIds as $userId) {
|
||||
$data = Redis::hgetall(self::PREFIX . $userId);
|
||||
if (!empty($data)) {
|
||||
$ips = [];
|
||||
foreach ($data as $field => $timestamp) {
|
||||
// if ($now - $timestamp <= self::TTL) {
|
||||
$ips[] = substr($field, strpos($field, ':') + 1);
|
||||
// }
|
||||
}
|
||||
if (!empty($ips)) {
|
||||
$result[$userId] = array_unique($ips);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* notify update (throttle control)
|
||||
*/
|
||||
private function notifyUpdate(int $userId): void
|
||||
{
|
||||
$dbThrottleKey = "device:db_throttle:{$userId}";
|
||||
|
||||
if (Redis::setnx($dbThrottleKey, 1)) {
|
||||
Redis::expire($dbThrottleKey, self::DB_THROTTLE);
|
||||
|
||||
User::query()
|
||||
->whereKey($userId)
|
||||
->update([
|
||||
'online_count' => $this->getDeviceCount($userId),
|
||||
'last_online_at' => now(),
|
||||
]);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1,123 +0,0 @@
|
||||
<?php
|
||||
|
||||
|
||||
namespace App\Services;
|
||||
|
||||
use Illuminate\Support\Collection;
|
||||
use Illuminate\Support\Str;
|
||||
|
||||
class UserOnlineService
|
||||
{
|
||||
/**
|
||||
* 缓存相关常量
|
||||
*/
|
||||
private const CACHE_PREFIX = 'ALIVE_IP_USER_';
|
||||
|
||||
/**
|
||||
* 获取所有限制设备用户的在线数量
|
||||
*/
|
||||
public function getAliveList(Collection $deviceLimitUsers): array
|
||||
{
|
||||
if ($deviceLimitUsers->isEmpty()) {
|
||||
return [];
|
||||
}
|
||||
|
||||
$cacheKeys = $deviceLimitUsers->pluck('id')
|
||||
->map(fn(int $id): string => self::CACHE_PREFIX . $id)
|
||||
->all();
|
||||
|
||||
return collect(cache()->many($cacheKeys))
|
||||
->filter()
|
||||
->map(fn(array $data): ?int => $data['alive_ip'] ?? null)
|
||||
->filter()
|
||||
->mapWithKeys(fn(int $count, string $key): array => [
|
||||
(int) Str::after($key, self::CACHE_PREFIX) => $count
|
||||
])
|
||||
->all();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取指定用户的在线设备信息
|
||||
*/
|
||||
public static function getUserDevices(int $userId): array
|
||||
{
|
||||
$data = cache()->get(self::CACHE_PREFIX . $userId, []);
|
||||
if (empty($data)) {
|
||||
return ['total_count' => 0, 'devices' => []];
|
||||
}
|
||||
|
||||
$devices = collect($data)
|
||||
->filter(fn(mixed $item): bool => is_array($item) && isset($item['aliveips']))
|
||||
->flatMap(function (array $nodeData, string $nodeKey): array {
|
||||
return collect($nodeData['aliveips'])
|
||||
->mapWithKeys(function (string $ipNodeId) use ($nodeData, $nodeKey): array {
|
||||
$ip = Str::before($ipNodeId, '_');
|
||||
return [
|
||||
$ip => [
|
||||
'ip' => $ip,
|
||||
'last_seen' => $nodeData['lastupdateAt'],
|
||||
'node_type' => Str::before($nodeKey, (string) $nodeData['lastupdateAt'])
|
||||
]
|
||||
];
|
||||
})
|
||||
->all();
|
||||
})
|
||||
->values()
|
||||
->all();
|
||||
|
||||
return [
|
||||
'total_count' => $data['alive_ip'] ?? 0,
|
||||
'devices' => $devices
|
||||
];
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 批量获取用户在线设备数
|
||||
*/
|
||||
public function getOnlineCounts(array $userIds): array
|
||||
{
|
||||
$cacheKeys = collect($userIds)
|
||||
->map(fn(int $id): string => self::CACHE_PREFIX . $id)
|
||||
->all();
|
||||
|
||||
return collect(cache()->many($cacheKeys))
|
||||
->filter()
|
||||
->map(fn(array $data): int => $data['alive_ip'] ?? 0)
|
||||
->all();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取用户在线设备数
|
||||
*/
|
||||
public function getOnlineCount(int $userId): int
|
||||
{
|
||||
$data = cache()->get(self::CACHE_PREFIX . $userId, []);
|
||||
return $data['alive_ip'] ?? 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* 计算在线设备数量
|
||||
*/
|
||||
public static function calculateDeviceCount(array $ipsArray): int
|
||||
{
|
||||
$mode = (int) admin_setting('device_limit_mode', 0);
|
||||
|
||||
return match ($mode) {
|
||||
1 => collect($ipsArray)
|
||||
->filter(fn(mixed $data): bool => is_array($data) && isset($data['aliveips']))
|
||||
->flatMap(
|
||||
fn(array $data): array => collect($data['aliveips'])
|
||||
->map(fn(string $ipNodeId): string => Str::before($ipNodeId, '_'))
|
||||
->unique()
|
||||
->all()
|
||||
)
|
||||
->unique()
|
||||
->count(),
|
||||
0 => collect($ipsArray)
|
||||
->filter(fn(mixed $data): bool => is_array($data) && isset($data['aliveips']))
|
||||
->sum(fn(array $data): int => count($data['aliveips'])),
|
||||
default => throw new \InvalidArgumentException("Invalid device limit mode: $mode"),
|
||||
};
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user