diff --git a/app/Middleware/Rule/LockRequestMiddleware.php b/app/Middleware/Rule/LockRequestMiddleware.php new file mode 100644 index 0000000..5cfcb05 --- /dev/null +++ b/app/Middleware/Rule/LockRequestMiddleware.php @@ -0,0 +1,48 @@ +getAttribute("userInfo"); + if (empty($user_info)){ + return $handler->handle($request); + } + + // 获取请求路径 + $path_info = $request->getUri()->getPath(); + if (empty($path_info)){ + return $handler->handle($request); + } + + $redis = $this->container->get(Redis::class); + $redis_key = "lock_request_" . $path_info . $user_info['user_id']; + $redis_lock = $redis->setnx($redis_key,1); + // 设置过期时间 + $redis->expire($redis_key,1); + if (!$redis_lock){ + // 设置失败,表示已经设置该值 + sleep(1); + } + + return $handler->handle($request); + } +} diff --git a/config/routes.php b/config/routes.php index 1b13fa3..81f36f5 100644 --- a/config/routes.php +++ b/config/routes.php @@ -32,6 +32,7 @@ use App\Controller\UserController; use App\Controller\UserDoctorController; use App\Controller\UserPatientController; use App\Controller\UserPharmacistController; +use App\Middleware\Rule\LockRequestMiddleware; use App\Services\SafeService; use Hyperf\HttpServer\Router\Router; @@ -137,7 +138,11 @@ Router::addGroup('/doctor', function () { Router::get('/record', [DoctorAccountController::class, 'getDoctorWithdrawalRecordList']); // 发起提现 - Router::post('', [DoctorAccountController::class, 'addDoctorWithdrawal']); + Router::post('', [DoctorAccountController::class, 'addDoctorWithdrawal'], + [ + 'middleware' => [LockRequestMiddleware::class] // 锁定重复请求 + ] + ); }); // 订单