neiyou_user_web/app/Console/Commands/AutoJob.php

<?php

namespace App\Console\Commands;

use App\Components\Helpers;
use App\Models\Config;
use App\Models\Coupon;
use App\Models\Invite;
use App\Models\Node;
use App\Models\Order;
use App\Models\User;
use App\Models\VerifyCode;
use App\Models\UserHourlyDataFlow;
use App\Models\UserDataFlowLog;
use Illuminate\Console\Command;
use Log;

class AutoJob extends Command
{
    protected $signature = 'autoJob';
    protected $description = '自动化任务';

    public function handle(): void
    {
        $jobStartTime = microtime(true);

        // 关闭超时未支付本地订单
       // Order::query()->recentUnPay()->update(['status' => -1]);
        Log::info('---开启实执行自动化任务');
        // 封禁账号
        $this->blockUsers();
        Log::info('---开启实执行封禁账号');
        // 解封被封禁的账号
        $this->unblockUsers();
        Log::info('---开启实执行解封被封禁的账号');
        // 端口回收与分配
        if (sysConfig('auto_release_port')) {
            $this->dispatchPort();
            Log::info('---开启实执行端口回收与分配');
        }

        $jobEndTime = microtime(true);
        $jobUsedTime = round(($jobEndTime - $jobStartTime), 4);

        Log::info('---【'.$this->description.'】完成---，耗时'.$jobUsedTime.'秒');
    }

    private function blockUsers(): void
    {
        // 封顶禁用：已使用流量 >= 配额
        User::activeUser()
            ->whereRaw('u + d >= transfer_enable')
            ->chunk(100, function ($users) {
                foreach ($users as $user) {
                    try {
                        $user->update(['enable' => 0]);
                        Helpers::addUserBanLog($user->id, 0, '【封禁代理】-流量已用完');
                        \Log::info('用户流量超限被封禁', [
                            'user_id' => $user->id,
                            'email' => $user->email,
                            'used_traffic' => $user->u + $user->d,
                            'transfer_enable' => $user->transfer_enable,
                            'ban_time' => now()
                        ]);
                    } catch (\Throwable $e) {
                        \Log::error("封禁超限用户失败 [ID: {$user->id}]", [
                            'error' => $e->getMessage()
                        ]);
                    }
                }
            });

        // 加权封禁：1小时内加权流量超限
        if (sysConfig('is_traffic_ban')) {
            $trafficBanTime = sysConfig('traffic_ban_time');
            $baseTrafficBanValue = sysConfig('traffic_ban_value') * GB;
            $oneHourAgo = time() - 3600;

            // 提前加载所有节点到内存，避免 N+1 查询
            $nodes = Node::all()->keyBy('id');

            // 加权流量累加器
            $userWeightedTotals = [];

            UserDataFlowLog::where('log_time', '>=', $oneHourAgo)
                ->chunk(1000, function ($records) use (&$userWeightedTotals, $nodes) {
                    foreach ($records as $record) {
                        $user_id = $record->user_id;
                        $node = $nodes->get($record->node_id);
                        $rate = $node ? ($node->traffic_rate > 0 ? $node->traffic_rate : 1) : 1;

                        $weighted = ($record->u + $record->d) / $rate;

                        if (!isset($userWeightedTotals[$user_id])) {
                            $userWeightedTotals[$user_id] = 0;
                        }
                        $userWeightedTotals[$user_id] += $weighted;
                    }
                });

            $abnormalUserIds = [];
            foreach ($userWeightedTotals as $user_id => $weightedTotal) {
                if ($weightedTotal > $baseTrafficBanValue) {
                    $user = User::find($user_id);
                    if (!$user) {
                        \Log::warning("⚠️ 找不到用户 [ID: $user_id]，跳过封禁");
                        continue;
                    }
                    \Log::warning("⚠️ 用户 [ID: $user_id, 邮箱: {$user->email}] 1小时流量为 ".flowAutoShow($weightedTotal)."，超出阈值 ".flowAutoShow($baseTrafficBanValue));
                    $abnormalUserIds[] = $user_id;
                }
            }

            if (!empty($abnormalUserIds)) {
                \Log::info('发现流量异常用户', [
                    'count' => count($abnormalUserIds),
                    'user_ids' => $abnormalUserIds,
                ]);

                User::activeUser()
                    ->whereIn('id', $abnormalUserIds)
                    ->whereBanTime(null)
                    ->chunk(100, function ($users) use ($trafficBanTime) {
                        foreach ($users as $user) {
                            try {
                                $user->update([
                                    'enable' => 0,
                                    'ban_time' => strtotime("+$trafficBanTime minutes"),
                                ]);
                                $user->refresh();
                                Helpers::addUserBanLog($user->id, $trafficBanTime, '【临时封禁代理】-1小时内流量异常');
                            } catch (\Throwable $e) {
                                \Log::error("封禁加权流量异常用户失败 [ID: {$user->id}]", [
                                    'error' => $e->getMessage(),
                                ]);
                            }
                        }
                    });
            } else {
                \Log::info('未发现流量异常用户');
            }
        }
    }

    private function unblockUsers(): void
    {
        // 解封被临时封禁的账号
        User::whereEnable(0)
            ->where('status', '>=', 0)
            ->whereNotNull('ban_time')
            ->where('ban_time', '<', time())
            ->chunk(100, function ($users) {
                foreach ($users as $user) {
                    $user->update(['status'=>1,'enable' => 1, 'ban_time' => null]);
                    Helpers::addUserBanLog($user->id, 0, '【自动解封】-临时封禁到期');
                    \Log::info('用户临时封禁到期自动解封', [
                        'user_id' => $user->id,
                        'email' => $user->email,
                        'ban_time' => $user->ban_time,
                        'unblock_time' => now(),
                        'reason' => '临时封禁到期'
                    ]);
                }
            });

        // 解封流量恢复的账号
        User::whereEnable(0)
            ->where('status', '>=', 0)
            ->whereBanTime(null)
            ->where('expired_at', '>=', date('Y-m-d H:i:s'))
            ->whereRaw('u + d < transfer_enable')
            ->chunk(100, function ($users) {
                foreach ($users as $user) {
                    $user->update(['status'=>1,'enable' => 1]);
                    Helpers::addUserBanLog($user->id, 0, '【自动解封】-有流量解封');
                    \Log::info('用户流量恢复自动解封', [
                        'user_id' => $user->id,
                        'email' => $user->email,
                        'used_traffic' => $user->u + $user->d,
                        'transfer_enable' => $user->transfer_enable,
                        'expired_at' => $user->expired_at,
                        'unblock_time' => now(),
                        'reason' => '流量恢复'
                    ]);
                }
            });
    }

    private function dispatchPort(): void
    {
        // 被封禁 / 过期一个月 的账号自动释放端口
        User::where('port', '<>', 0)
            ->where(function ($query) {
                $query->where('status', -1)
                    ->orWhere('expired_at', '<=', date('Y-m-d H:i:s', strtotime('-1 months')));
            })
            ->update(['port' => 0]);
    }
}