mirror of
https://github.com/v2board/v2board.git
synced 2024-11-15 00:09:13 +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
|
|
]);
|
|
}
|
|
}
|