命令统计使用流量 完成(测试)

application/api/controller/Index.php

@@ -36,7 +36,7 @@ class Index extends Api
         }
         $ret = $this->auth->login($account, $password);
         if ($ret) {
-            $user = \app\common\model\User::where(['mobile'=>$account])->find();
+            $user = \app\common\model\User::where(['username'=>$account])->find();
             if($user->total_traffic < $user->traffic){
                 // 流量已用完
                 return $this->err();

application/common/command/SettlementTraffic.php

@@ -6,6 +6,7 @@ namespace app\common\command;
 
 use app\common\model\FlowLogs;
 use app\common\model\User;
+use app\common\model\UserTraffic;
 use think\console\Command;
 use think\console\Input;
 use think\console\Output;
@@ -20,6 +21,15 @@ class SettlementTraffic extends Command
         $this->setName('settlement')->setDescription('后台统计用户流量使用, 并扣除相关流量');
     }
 
+    /**
+     * 功能: 执行任务, 计算统计使用的流量
+     * @param Input $input
+     * @param Output $output
+     * @return int|void|null
+     * @throws \think\db\exception\DataNotFoundException
+     * @throws \think\db\exception\ModelNotFoundException
+     * @throws \think\exception\DbException
+     */
     protected function execute(Input $input, Output $output)
     {
 //        parent::execute($input, $output); // TODO: Change the autogenerated stub
@@ -53,8 +63,41 @@ class SettlementTraffic extends Command
         }
     }
 
-    protected function usedTraffic($uid, $bytes)
+
+    /**
+     * 功能: 处理用户使用过的流量
+     * @param $uid
+     * @param $bytes
+     * @param $dateTime
+     * @return bool
+     * @throws \think\db\exception\DataNotFoundException
+     * @throws \think\db\exception\ModelNotFoundException
+     * @throws \think\exception\DbException
+     */
+    protected function usedTraffic($uid, $bytes, $dateTime)
     {
+        $usable = UserTraffic::where('uid', $uid)
+            ->where('is_usable', '1')
+            ->where('start_time', '<', $dateTime)
+            ->where('expired_time', '>', $dateTime)
+            ->order('expired_time', 'ASC')
+            ->select();
+
+        if (empty($usable)) {
+            return true;
+        }
 
+        foreach ($usable as $item) {
+            if ($bytes == 0) break;
+            if ($item->traffic > bcadd($item->used, $bytes)) {
+                $item->setInc('used', $bytes);
+            } else {
+                $bytes = bcsub($bytes, bcsub($item->traffic, $item->used));
+                $item->used = $item->traffic;
+                $item->is_usable = 0;
+                $item->save();
+            }
+        }
+        return true;
     }
 }

application/common/model/UserTraffic.php

@@ -0,0 +1,12 @@
+<?php
+
+
+namespace app\common\model;
+
+
+use think\Model;
+
+class UserTraffic extends Model
+{
+
+}