{"id":201107,"date":"2025-05-10T08:41:54","date_gmt":"2025-05-10T00:41:54","guid":{"rendered":"https:\/\/server.hk\/cnblog\/201107\/"},"modified":"2025-05-10T08:41:54","modified_gmt":"2025-05-10T00:41:54","slug":"%e5%9f%ba%e4%ba%8eredislua%e8%bf%9b%e8%a1%8c%e9%99%90%e6%b5%81%e7%9a%84%e6%96%b9%e6%b3%95","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/201107\/","title":{"rendered":"\u57fa\u4e8eredis+lua\u8fdb\u884c\u9650\u6d41\u7684\u65b9\u6cd5"},"content":{"rendered":"<p><b><\/b> <\/p>\n<h1>\u57fa\u4e8eredis+lua\u8fdb\u884c\u9650\u6d41\u7684\u65b9\u6cd5<\/h1>\n<p><span style=\"cursor: pointer\"><i><\/i>\u6536\u85cf<\/span> <\/p>\n<p>IT\u884c\u4e1a\u76f8\u5bf9\u4e8e\u4e00\u822c\u4f20\u7edf\u884c\u4e1a\uff0c\u53d1\u5c55\u66f4\u65b0\u901f\u5ea6\u66f4\u5feb\uff0c\u4e00\u65e6\u505c\u6b62\u4e86\u5b66\u4e60\uff0c\u5f88\u5feb\u5c31\u4f1a\u88ab\u884c\u4e1a\u6240\u6dd8\u6c70\u3002\u6240\u4ee5\u6211\u4eec\u9700\u8981\u8e0f\u8e0f\u5b9e\u5b9e\u7684\u4e0d\u65ad\u5b66\u4e60\uff0c\u7cbe\u8fdb\u81ea\u5df1\u7684\u6280\u672f\uff0c\u5c24\u5176\u662f\u521d\u5b66\u8005\u3002\u4eca\u5929golang\u5b66\u4e60\u7f51\u7ed9\u5927\u5bb6\u6574\u7406\u4e86\u300a\u57fa\u4e8eredis+lua\u8fdb\u884c\u9650\u6d41\u7684\u65b9\u6cd5\u300b\uff0c\u804a\u804a\u9650\u6d41\u3001redislua\uff0c\u6211\u4eec\u4e00\u8d77\u6765\u770b\u770b\u5427\uff01<\/p>\n<p>1\uff0c\u9996\u5148\u6211\u4eecredis\u6709\u5f88\u591a\u9650\u6d41\u7684\u7b97\u6cd5(\u6bd4\u5982\uff1a\u4ee4\u724c\u6876\uff0c\u8ba1\u6570\u5668\uff0c\u65f6\u95f4\u7a97\u53e3)\u7b49\uff0c\u4f46\u662f\u90fd\u6709\u4e00\u5b9a\u7684\u7f3a\u70b9\uff0c\u4ee4\u724c\u6876\u5728\u5355\u9879\u76ee\u4e2d\u76f8\u5bf9\u6765\u8bf4\u6bd4\u8f83\u7a33\u5b9a\uff0c\u4f46\u662f\u5728\u5206\u5e03\u5f0f\u96c6\u7fa4\u91cc\u9762\u7f3a\u663e\u7684\u4e0d\u90a3\u4e48\u53cb\u597d\uff0c\u8fd9\u65f6\u5019\uff0c\u5728\u5206\u5e03\u5f0f\u91cc\u9762\u8fdb\u884c\u9650\u6d41\u7684\u8bdd\uff0c\u6211\u4eec\u5219\u53ef\u4ee5\u4f7f\u7528redis+lua\u811a\u672c\u8fdb\u884c\u9650\u6d41\uff0c\u80fd\u6297\u4f4f\u4ebf\u7ea7\u5e76\u53d1<\/p>\n<p>2\uff0c\u4e0b\u9762\u8bf4\u8bf4lua+redis\u8fdb\u884c\u9650\u6d41\u7684\u505a\u6cd5<br \/>\u5f00\u53d1\u73af\u5883\uff1aidea+redis+lua<br \/>\u7b2c\u4e00\uff1a<br \/>\u6253\u5f00idea\u7684\u63d2\u4ef6\u5e02\u573a\uff0c\u7136\u540e\u641c\u7d22lua\uff0c\u70b9\u51fb\u53f3\u8fb9\u7684\u5b89\u88c5\uff0c\u7136\u540e\u5b89\u88c5\u597d\u4e86\uff0c\u91cd\u542f\u5373\u53ef<\/p>\n<p style=\"text-align:center\"><img decoding=\"async\" src=\"https:\/\/www.17golang.com\/uploads\/20221229\/167231610663ad84cae32e4.jpg\" class=\"aligncenter\"><\/p>\n<p>\u7b2c\u4e8c\uff1a\u5199\u4e00\u4e2a\u81ea\u5b9a\u4e49\u9650\u6d41\u6ce8\u89e3<\/p>\n<pre>package com.sport.sportcloudmarathonh5.config;\n\nimport java.lang.annotation.*;\n\n\/**\n * @author zdj\n * @version 1.0.0\n * @description \u81ea\u5b9a\u4e49\u6ce8\u89e3\u5b9e\u73b0\u5206\u5e03\u5f0f\u9650\u6d41\n *\/\n@Target(value = ElementType.METHOD)\n@Retention(RetentionPolicy.RUNTIME)\n@Documented\npublic @interface RedisLimitStream {\n    \/**\n     * \u8bf7\u6c42\u9650\u5236\uff0c\u4e00\u79d2\u5185\u53ef\u4ee5\u5141\u8bb8\u597d\u591a\u4e2a\u8fdb\u5165(\u9ed8\u8ba4\u4e00\u79d2\u53ef\u4ee5\u652f\u6301100\u4e2a)\n     * @return\n     *\/\n    int reqLimit() default 1000;\n\n    \/**\n     * \u6a21\u5757\u540d\u79f0\n     * @return\n     *\/\n    String reqName() default \"\";\n}<\/pre>\n<p>\u7b2c\u4e09\uff1a\u5728\u6307\u5b9a\u7684\u65b9\u6cd5\u4e0a\u9762\u6dfb\u52a0\u8be5\u6ce8\u89e3<\/p>\n<pre>\/**\n     * \u538b\u6d4b\u63a5\u53e3\n     * @return\n     *\/\n    @Login(isLogin = false)\n    @RedisLimitStream(reqName = \"\u540d\u989d\u79d2\u6740\", reqLimit = 1000)\n    @ApiOperation(value = \"\u538b\u6d4b\u63a5\u53e3\", notes = \"\u538b\u6d4b\u63a5\u53e3\", httpMethod = \"GET\")\n    @RequestMapping(value = \"\/pressure\", method = RequestMethod.GET)\n    public ResultVO pressure(){\n        return ResultVO.success(\"\u62a2\u8d2d\u6210\u529f\uff01\");\n    }\n<\/pre>\n<p>\u7b2c\u56db\uff1a\u6dfb\u52a0\u4e00\u4e2a\u62e6\u622a\u5668\u5bf9\u8bbf\u95ee\u7684\u65b9\u6cd5\u5728\u8bbf\u95ee\u4e4b\u524d\u8fdb\u884c\u62e6\u622a\uff1a<\/p>\n<pre>package com.sport.sportcloudmarathonh5.config;\n\nimport com.alibaba.fastjson.JSONObject;\nimport com.sport.sportcloudmarathonh5.service.impl.RedisService;\nimport org.aspectj.lang.ProceedingJoinPoint;\nimport org.aspectj.lang.annotation.Around;\nimport org.aspectj.lang.annotation.Aspect;\nimport org.aspectj.lang.annotation.Pointcut;\nimport org.aspectj.lang.reflect.MethodSignature;\nimport org.slf4j.Logger;\nimport org.slf4j.LoggerFactory;\nimport org.springframework.beans.factory.annotation.Autowired;\nimport org.springframework.data.redis.core.script.RedisScript;\nimport org.springframework.stereotype.Component;\nimport org.springframework.util.ObjectUtils;\n\nimport javax.servlet.http.HttpServletResponse;\nimport java.io.PrintWriter;\nimport java.util.ArrayList;\nimport java.util.List;\n\n\/**\n * @author zdj\n * @version 1.0.0\n * @description MyRedisLimiter\u6ce8\u89e3\u7684\u5207\u9762\u7c7b\n *\/\n@Aspect\n@Component\npublic class RedisLimiterAspect {\n    private final Logger logger = LoggerFactory.getLogger(RedisLimitStream.class);\n    \/**\n     * \u5f53\u524d\u54cd\u5e94\u8bf7\u6c42\n     *\/\n    @Autowired\n    private HttpServletResponse response;\n\n    \/**\n     * redis\u670d\u52a1\n     *\/\n    @Autowired\n    private RedisService redisService;\n\n    \/**\n     * \u6267\u884credis\u7684\u811a\u672c\u6587\u4ef6\n     *\/\n    @Autowired\n    private RedisScript\n   \n     rateLimitLua;\n\n    \/**\n     * \u5bf9\u6240\u6709\u63a5\u53e3\u8fdb\u884c\u62e6\u622a\n     *\/\n    @Pointcut(\"execution(public * com.sport.sportcloudmarathonh5.controller.*.*(..))\")\n    public void pointcut(){}\n\n    \/**\n     * \u5bf9\u5207\u70b9\u8fdb\u884c\u7ee7\u7eed\u5904\u7406\n     *\/\n    @Around(\"pointcut()\")\n    public Object process(ProceedingJoinPoint proceedingJoinPoint) throws Throwable{\n        \/\/\u4f7f\u7528\u53cd\u5c04\u83b7\u53d6RedisLimitStream\u6ce8\u89e3\n        MethodSignature signature = (MethodSignature) proceedingJoinPoint.getSignature();\n        \/\/\u6ca1\u6709\u6dfb\u52a0\u9650\u6d41\u6ce8\u89e3\u7684\u65b9\u6cd5\u76f4\u63a5\u653e\u884c\n        RedisLimitStream redisLimitStream = signature.getMethod().getDeclaredAnnotation(RedisLimitStream.class);\n        if(ObjectUtils.isEmpty(redisLimitStream)){\n            return proceedingJoinPoint.proceed();\n        }\n\n        \/\/List\u8bbe\u7f6eLua\u7684KEYS[1]\n        List\n    \n      keyList = new ArrayList(); keyList.add(\"ip:\" + (System.currentTimeMillis() \/ 1000)); \/\/\u83b7\u53d6\u6ce8\u89e3\u4e0a\u7684\u53c2\u6570\uff0c\u83b7\u53d6\u914d\u7f6e\u7684\u901f\u7387 \/\/List\u8bbe\u7f6eLua\u7684ARGV[1] int value = redisLimitStream.reqLimit(); \/\/ \u8c03\u7528Redis\u6267\u884clua\u811a\u672c\uff0c\u672a\u62ff\u5230\u4ee4\u724c\u7684\uff0c\u76f4\u63a5\u8fd4\u56de\u63d0\u793a boolean acquired = redisService.execute(rateLimitLua, keyList, value); logger.info(\"\u6267\u884clua\u7ed3\u679c\uff1a\" + acquired); if(!acquired){ this.limitStreamBackMsg(); return null; } \/\/\u83b7\u53d6\u5230\u4ee4\u724c\uff0c\u7ee7\u7eed\u5411\u4e0b\u6267\u884c return proceedingJoinPoint.proceed(); } \/** * \u88ab\u62e6\u622a\u7684\u4eba\uff0c\u63d0\u793a\u6d88\u606f *\/ private void limitStreamBackMsg() { response.setHeader(\"Content-Type\", \"text\/html;charset=UTF8\"); PrintWriter writer = null; try { writer = response.getWriter(); writer.println(\"{\\\"code\\\":503,\\\"message\\\":\\\"\u5f53\u524d\u6392\u961f\u4eba\u8f83\u591a\uff0c\u8bf7\u7a0d\u540e\u518d\u8bd5\uff01\\\",\\\"data\\\":\\\"null\\\"}\"); writer.flush(); } catch (Exception e) { e.printStackTrace(); } finally { if (writer != null) { writer.close(); } } } } \n    \n   <\/pre>\n<p>\u7b2c\u4e94\uff1a\u5199\u4e2a\u914d\u7f6e\u7c7b\uff0c\u5728\u542f\u52a8\u7684\u65f6\u5019\u5c06\u6211\u4eec\u7684lua\u811a\u672c\u4ee3\u7801\u52a0\u8f7d\u5230redisscript\u4e2d<\/p>\n<pre>package com.sport.sportcloudmarathonh5.config;\nimport org.springframework.context.annotation.Bean;\nimport org.springframework.context.annotation.Configuration;\nimport org.springframework.core.io.ClassPathResource;\nimport org.springframework.data.redis.core.script.DefaultRedisScript;\n\n\/**\n * @author zdj\n * @version 1.0.0\n * @description \u5b9e\u73b0redis\u7684\u7f16\u7801\u65b9\u5f0f\n *\/\n@Configuration\npublic class RedisConfiguration {\n\n    \/**\n     * \u521d\u59cb\u5316\u5c06lua\u811a\u672c\u52a0\u8f7d\u5230redis\u811a\u672c\u4e2d\n     * @return\n     *\/\n    @Bean\n    public DefaultRedisScript loadRedisScript() {\n        DefaultRedisScript redisScript = new DefaultRedisScript();\n        redisScript.setLocation(new ClassPathResource(\"limit.lua\"));\n        redisScript.setResultType(Boolean.class);\n        return redisScript;\n    }\n}\n\n<\/pre>\n<p>\u7b2c\u516d\uff1aredis\u6267\u884clua\u7684\u65b9\u6cd5<\/p>\n<pre>  \/**\n     * \u6267\u884clua\u811a\u672c\n     * @param redisScript lua\u6e90\u4ee3\u7801\u811a\u672c\n     * @param keyList\n     * @param value\n     * @return\n     *\/\n    public boolean execute(RedisScript\n   \n     redisScript, List\n    \n      keyList, int value) { return redisTemplate.execute(redisScript, keyList, String.valueOf(value)); } \n    \n   <\/pre>\n<p>\u7b2c\u4e03\uff1a\u5728resources\u76ee\u5f55\u4e0b\u9762\u65b0\u52a0\u4e00\u4e2alua\u811a\u672c\u6587\u4ef6\uff0c\u5c06\u4e0b\u9762\u4ee3\u7801\u62f7\u8d1d\u8fdb\u53bb\u5373\u53ef\uff1a<\/p>\n<pre>local key = KEYS[1] --\u9650\u6d41KEY\uff08\u4e00\u79d2\u4e00\u4e2a\uff09\nlocal limit = tonumber(ARGV[1]) --\u9650\u6d41\u5927\u5c0f\nlocal current = tonumber(redis.call('get', key) or \"0\")\nif current + 1 &gt; limit then --\u5982\u679c\u8d85\u51fa\u9650\u6d41\u5927\u5c0f\n    return false\nelse --\u8bf7\u6c42\u6570+1\uff0c\u5e76\u8bbe\u7f6e2\u79d2\u8fc7\u671f\n    redis.call(\"INCRBY\", key, \"1\")\n    redis.call(\"expire\", key, \"2\")\nend\nreturn true\n<\/pre>\n<p style=\"text-align:center\"><img decoding=\"async\" src=\"https:\/\/www.17golang.com\/uploads\/20221229\/167231610763ad84cba7447.jpg\" class=\"aligncenter\"><\/p>\n<p>\u6700\u540e\u6267\u884c\u5373\u53ef\uff1a<br \/>\u53ef\u4ee5\u4f7f\u7528jemster\u8fdb\u884c\u6d4b\u8bd5\uff1a<\/p>\n<p style=\"text-align:center\"><img decoding=\"async\" src=\"https:\/\/www.17golang.com\/uploads\/20221229\/167231610863ad84cc29751.png\" class=\"aligncenter\"><\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u57fa\u4e8eredis+lua\u8fdb\u884c\u9650\u6d41\u7684&#46;&#46;&#46;<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[101],"tags":[],"class_list":["post-201107","post","type-post","status-publish","format-standard","hentry","category-database"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/201107","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/comments?post=201107"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/201107\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=201107"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=201107"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=201107"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}