BlockerMiddleware.php 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. <?php
  2. // +----------------------------------------------------------------------
  3. // | CRMEB [ CRMEB赋能开发者,助力企业发展 ]
  4. // +----------------------------------------------------------------------
  5. // | Copyright (c) 2016~2026 https://www.crmeb.com All rights reserved.
  6. // +----------------------------------------------------------------------
  7. // | Licensed CRMEB并不是自由软件,未经许可不能去掉CRMEB相关版权
  8. // +----------------------------------------------------------------------
  9. // | Author: CRMEB Team <admin@crmeb.com>
  10. // +----------------------------------------------------------------------
  11. namespace app\api\middleware;
  12. use app\Request;
  13. use crmeb\exceptions\ApiException;
  14. use crmeb\interfaces\MiddlewareInterface;
  15. use crmeb\services\CacheService;
  16. use think\facade\Config;
  17. /**
  18. * Redis锁
  19. * Class BlockerMiddleware
  20. * @author 等风来
  21. * @email 136327134@qq.com
  22. * @date 2023/2/8
  23. * @package app\api\middleware
  24. */
  25. class BlockerMiddleware implements MiddlewareInterface
  26. {
  27. /**
  28. * @param Request $request
  29. * @param \Closure $next
  30. * @author 等风来
  31. * @email 136327134@qq.com
  32. * @date 2022/11/21
  33. */
  34. public function handle(Request $request, \Closure $next)
  35. {
  36. if (Config::get('cache.default') == 'file') {
  37. return $next($request);
  38. }
  39. $uid = $request->uid();
  40. $key = md5($request->rule()->getRule() . $uid);
  41. if (!CacheService::setMutex($key)) {
  42. throw new ApiException('请求太过频繁,请稍后再试');
  43. }
  44. $response = $next($request);
  45. $this->after($response, $key);
  46. return $response;
  47. }
  48. /**
  49. * @param $response
  50. * @param $key
  51. * @author 等风来
  52. * @email 136327134@qq.com
  53. * @date 2022/11/22
  54. */
  55. public function after($response, $key)
  56. {
  57. CacheService::delMutex($key);
  58. }
  59. }