v2board/app/Console/Commands/ResetTraffic.php

93 lines
2.1 KiB
PHP
Raw Normal View History

2019-12-15 15:17:32 +08:00
<?php
namespace App\Console\Commands;
use Illuminate\Console\Command;
2020-02-28 01:21:14 +08:00
use App\Models\User;
use Illuminate\Support\Facades\DB;
2019-12-15 15:17:32 +08:00
2019-12-15 15:20:05 +08:00
class ResetTraffic extends Command
2019-12-15 15:17:32 +08:00
{
2020-09-14 17:43:47 +08:00
protected $builder;
2019-12-15 15:17:32 +08:00
/**
* 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();
2020-09-14 17:43:47 +08:00
$this->builder = User::where('expired_at', '!=', NULL)
2020-08-19 12:52:49 +08:00
->where('expired_at', '>', time());
2019-12-15 15:17:32 +08:00
}
/**
* Execute the console command.
*
* @return mixed
*/
public function handle()
{
DB::beginTransaction();
2020-03-02 23:36:19 +08:00
$resetTrafficMethod = config('v2board.reset_traffic_method', 0);
switch ((int)$resetTrafficMethod) {
2020-03-02 00:04:07 +08:00
// 1 a month
case 0:
2020-08-19 12:52:49 +08:00
$this->resetByMonthFirstDay();
2020-03-02 00:04:07 +08:00
break;
// expire day
case 1:
2020-08-19 12:52:49 +08:00
$this->resetByExpireDay();
2020-03-02 23:21:27 +08:00
break;
}
DB::commit();
2020-03-02 23:21:27 +08:00
}
2020-09-14 17:41:26 +08:00
private function resetByMonthFirstDay():void
2020-03-02 23:21:27 +08:00
{
2020-09-14 17:43:47 +08:00
$builder = $this->builder;
2020-03-06 16:23:49 +08:00
if ((string)date('d') === '01') {
2020-09-14 17:43:47 +08:00
$builder->update([
2020-03-06 16:23:49 +08:00
'u' => 0,
'd' => 0
]);
}
2020-03-02 23:21:27 +08:00
}
2020-08-19 12:52:49 +08:00
private function resetByExpireDay():void
2020-03-02 23:21:27 +08:00
{
2020-09-14 17:43:47 +08:00
$builder = $this->builder;
2020-03-02 23:36:19 +08:00
$lastDay = date('d', strtotime('last day of +0 months'));
2020-03-15 20:21:31 +08:00
$users = [];
2020-09-14 17:43:47 +08:00
foreach ($builder->get() as $item) {
2020-03-15 20:12:52 +08:00
$expireDay = date('d', $item->expired_at);
2020-04-01 00:51:45 +08:00
$today = date('d');
if ($expireDay === $today) {
array_push($users, $item->id);
}
2020-04-01 02:15:07 +08:00
if (($today === $lastDay) && $expireDay >= $lastDay) {
2020-03-15 20:21:31 +08:00
array_push($users, $item->id);
2020-03-15 20:12:52 +08:00
}
2020-03-02 23:21:27 +08:00
}
2020-04-01 00:51:45 +08:00
User::whereIn('id', $users)->update([
2020-03-15 20:21:31 +08:00
'u' => 0,
'd' => 0
]);
2019-12-15 15:17:32 +08:00
}
}