mirror of
				https://github.com/v2board/v2board.git
				synced 2025-11-04 03:11:46 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			114 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			114 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
namespace App\Console\Commands;
 | 
						|
 | 
						|
use App\Models\Plan;
 | 
						|
use Illuminate\Console\Command;
 | 
						|
use App\Models\User;
 | 
						|
use Illuminate\Support\Facades\DB;
 | 
						|
 | 
						|
class ResetTraffic extends Command
 | 
						|
{
 | 
						|
    protected $builder;
 | 
						|
    /**
 | 
						|
     * The name and signature of the console command.
 | 
						|
     *
 | 
						|
     * @var string
 | 
						|
     */
 | 
						|
    protected $signature = 'reset:traffic';
 | 
						|
 | 
						|
    /**
 | 
						|
     * The console command description.
 | 
						|
     *
 | 
						|
     * @var string
 | 
						|
     */
 | 
						|
    protected $description = '流量清空';
 | 
						|
 | 
						|
    /**
 | 
						|
     * Create a new command instance.
 | 
						|
     *
 | 
						|
     * @return void
 | 
						|
     */
 | 
						|
    public function __construct()
 | 
						|
    {
 | 
						|
        parent::__construct();
 | 
						|
        $this->builder = User::where('expired_at', '!=', NULL)
 | 
						|
            ->where('expired_at', '>', time());
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Execute the console command.
 | 
						|
     *
 | 
						|
     * @return mixed
 | 
						|
     */
 | 
						|
    public function handle()
 | 
						|
    {
 | 
						|
        ini_set('memory_limit', -1);
 | 
						|
        foreach (Plan::get() as $plan) {
 | 
						|
            switch ($plan->reset_traffic_method) {
 | 
						|
                case null: {
 | 
						|
                    $resetTrafficMethod = config('v2board.reset_traffic_method', 0);
 | 
						|
                    switch ((int)$resetTrafficMethod) {
 | 
						|
                        // month first day
 | 
						|
                        case 0:
 | 
						|
                            $this->resetByMonthFirstDay($this->builder);
 | 
						|
                            break;
 | 
						|
                        // expire day
 | 
						|
                        case 1:
 | 
						|
                            $this->resetByExpireDay($this->builder);
 | 
						|
                            break;
 | 
						|
                        // no action
 | 
						|
                        case 2:
 | 
						|
                            break;
 | 
						|
                    }
 | 
						|
                    break;
 | 
						|
                }
 | 
						|
                case 0: {
 | 
						|
                    $builder = with(clone($this->builder))->where('plan_id', $plan->id);
 | 
						|
                    $this->resetByMonthFirstDay($builder);
 | 
						|
                    break;
 | 
						|
                }
 | 
						|
                case 1: {
 | 
						|
                    $builder = with(clone($this->builder))->where('plan_id', $plan->id);
 | 
						|
                    $this->resetByExpireDay($builder);
 | 
						|
                    break;
 | 
						|
                }
 | 
						|
                case 2: {
 | 
						|
                    break;
 | 
						|
                }
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    private function resetByMonthFirstDay($builder):void
 | 
						|
    {
 | 
						|
        if ((string)date('d') === '01') {
 | 
						|
            $builder->update([
 | 
						|
                'u' => 0,
 | 
						|
                'd' => 0
 | 
						|
            ]);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    private function resetByExpireDay($builder):void
 | 
						|
    {
 | 
						|
        $lastDay = date('d', strtotime('last day of +0 months'));
 | 
						|
        $users = [];
 | 
						|
        foreach ($builder->get() as $item) {
 | 
						|
            $expireDay = date('d', $item->expired_at);
 | 
						|
            $today = date('d');
 | 
						|
            if ($expireDay === $today) {
 | 
						|
                array_push($users, $item->id);
 | 
						|
            }
 | 
						|
 | 
						|
            if (($today === $lastDay) && $expireDay >= $lastDay) {
 | 
						|
                array_push($users, $item->id);
 | 
						|
            }
 | 
						|
        }
 | 
						|
        User::whereIn('id', $users)->update([
 | 
						|
            'u' => 0,
 | 
						|
            'd' => 0
 | 
						|
        ]);
 | 
						|
    }
 | 
						|
}
 |