From b3687e1eb2221b7c4166e223560a835732168378 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E9=9D=99?= <lym125@gmail.com>
Date: Wed, 5 Jan 2022 15:21:49 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BD=99=E9=A2=9D=E6=94=AF?=
 =?UTF-8?q?=E4=BB=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Controllers/Account/WalletController.php  | 24 +++++++++----------
 1 file changed, 11 insertions(+), 13 deletions(-)

diff --git a/app/Endpoint/Api/Http/Controllers/Account/WalletController.php b/app/Endpoint/Api/Http/Controllers/Account/WalletController.php
index 6b1060d8..ec18965c 100644
--- a/app/Endpoint/Api/Http/Controllers/Account/WalletController.php
+++ b/app/Endpoint/Api/Http/Controllers/Account/WalletController.php
@@ -126,22 +126,20 @@ class WalletController extends Controller
                                 ->lockForUpdate()
                                 ->first();
 
-                if ($payLog === null || !in_array($payLog->pay_way, [PayLog::PAY_WAY_WALLET, PayLog::PAY_WAY_BALANCE])) {
+                if ($payLog === null) {
                     throw new InvalidPaySerialNumberException();
                 }
 
-                switch ($payLog->pay_way) {
-                    case PayLog::PAY_WAY_WALLET:
-                        if ($user->wallet->is_frozen) {
-                            throw new WalletFrozenException();
-                        }
-                        break;
-
-                    case PayLog::PAY_WAY_BALANCE:
-                        if ($user->balance->is_frozen) {
-                            throw new BalanceFrozenException();
-                        }
-                        break;
+                if ($payLog->isBalance()) {
+                    if ($user->balance->is_frozen) {
+                        throw new BalanceFrozenException();
+                    }
+                } elseif ($payLog->isWallet()) {
+                    if ($user->wallet->is_frozen) {
+                        throw new WalletFrozenException();
+                    }
+                } else {
+                    throw new InvalidPaySerialNumberException();
                 }
 
                 $payable = $payLog->payable;