{"id":201255,"date":"2025-05-10T09:38:52","date_gmt":"2025-05-10T01:38:52","guid":{"rendered":"https:\/\/server.hk\/cnblog\/201255\/"},"modified":"2025-05-10T09:38:52","modified_gmt":"2025-05-10T01:38:52","slug":"%e5%9f%ba%e4%ba%8eredislua%e8%84%9a%e6%9c%ac%e5%ae%9e%e7%8e%b0%e5%88%86%e5%b8%83%e5%bc%8f%e9%99%90%e6%b5%81%e7%bb%84%e4%bb%b6%e5%b0%81%e8%a3%85%e7%9a%84%e6%96%b9%e6%b3%95","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/201255\/","title":{"rendered":"\u57fa\u4e8eRedis+Lua\u811a\u672c\u5b9e\u73b0\u5206\u5e03\u5f0f\u9650\u6d41\u7ec4\u4ef6\u5c01\u88c5\u7684\u65b9\u6cd5"},"content":{"rendered":"<p><b><\/b> <\/p>\n<h1>\u57fa\u4e8eRedis+Lua\u811a\u672c\u5b9e\u73b0\u5206\u5e03\u5f0f\u9650\u6d41\u7ec4\u4ef6\u5c01\u88c5\u7684\u65b9\u6cd5<\/h1>\n<p><span style=\"cursor: pointer\"><i><\/i>\u6536\u85cf<\/span> <\/p>\n<p>\u5728\u6570\u636e\u5e93\u5b9e\u6218\u5f00\u53d1\u7684\u8fc7\u7a0b\u4e2d\uff0c\u6211\u4eec\u7ecf\u5e38\u4f1a\u9047\u5230\u4e00\u4e9b\u8fd9\u6837\u90a3\u6837\u7684\u95ee\u9898\uff0c\u7136\u540e\u8981\u5361\u597d\u534a\u5929\uff0c\u7b49\u95ee\u9898\u89e3\u51b3\u4e86\u624d\u53d1\u73b0\u539f\u6765\u4e00\u4e9b\u7ec6\u8282\u77e5\u8bc6\u70b9\u8fd8\u662f\u6ca1\u6709\u638c\u63e1\u597d\u3002\u4eca\u5929golang\u5b66\u4e60\u7f51\u5c31\u6574\u7406\u5206\u4eab\u300a\u57fa\u4e8eRedis+Lua\u811a\u672c\u5b9e\u73b0\u5206\u5e03\u5f0f\u9650\u6d41\u7ec4\u4ef6\u5c01\u88c5\u7684\u65b9\u6cd5\u300b\uff0c\u804a\u804aRedislua\u811a\u672c\u3001\u5206\u5e03\u5f0f\u9650\u6d41\u7ec4\u4ef6\uff0c\u5e0c\u671b\u53ef\u4ee5\u5e2e\u52a9\u5230\u6b63\u5728\u52aa\u529b\u8d5a\u94b1\u7684\u4f60\u3002<\/p>\n<p><strong>\u521b\u5efa\u9650\u6d41\u7ec4\u4ef6\u9879\u76ee<\/strong><\/p>\n<p style=\"text-align: center\"><img decoding=\"async\" style=\"max-width:100%\" src=\"https:\/\/www.17golang.com\/uploads\/20230101\/167253661563b0e2273a6c9.png\" class=\"aligncenter\"><\/p>\n<p><strong>pom.xml\u6587\u4ef6\u4e2d\u5f15\u5165\u76f8\u5173\u4f9d\u8d56<\/strong><\/p>\n<pre>\n \n   \n    \n     \n      org.springframework.boot\n     \n     \n      spring-boot-starter-data-redis\n     \n    \n    \n     \n      org.springframework.boot\n     \n     \n      spring-boot-starter-aop\n     \n    \n    \n     \n      com.google.guava\n     \n     \n      guava\n     \n     \n      18.0\n     \n    \n   <\/pre>\n<p><strong>\u5728resources\u76ee\u5f55\u4e0b\u521b\u5efalua\u811a\u672c&nbsp;&nbsp;ratelimiter.lua<\/strong><\/p>\n<pre>\n--\n-- Created by IntelliJ IDEA.\n-- User: \u5bd2\u591c\n--\n \n-- \u83b7\u53d6\u65b9\u6cd5\u7b7e\u540d\u7279\u5f81\nlocal methodKey = KEYS[1]\nredis.log(redis.LOG_DEBUG, 'key is', methodKey)\n \n-- \u8c03\u7528\u811a\u672c\u4f20\u5165\u7684\u9650\u6d41\u5927\u5c0f\nlocal limit = tonumber(ARGV[1])\n \n-- \u83b7\u53d6\u5f53\u524d\u6d41\u91cf\u5927\u5c0f\nlocal count = tonumber(redis.call('get', methodKey) or \"0\")\n \n-- \u662f\u5426\u8d85\u51fa\u9650\u6d41\u9608\u503c\nif count + 1 &gt; limit then\n -- \u62d2\u7edd\u670d\u52a1\u8bbf\u95ee\n return false\nelse\n -- \u6ca1\u6709\u8d85\u8fc7\u9608\u503c\n -- \u8bbe\u7f6e\u5f53\u524d\u8bbf\u95ee\u7684\u6570\u91cf+1\n redis.call(\"INCRBY\", methodKey, 1)\n -- \u8bbe\u7f6e\u8fc7\u671f\u65f6\u95f4\n redis.call(\"EXPIRE\", methodKey, 1)\n -- \u653e\u884c\n return true\nend<\/pre>\n<p><strong>\u521b\u5efaRedisConfiguration \u7c7b<\/strong><\/p>\n<pre>\npackage com.imooc.springcloud;\n \nimport org.springframework.context.annotation.Bean;\nimport org.springframework.context.annotation.Configuration;\nimport org.springframework.core.io.ClassPathResource;\nimport org.springframework.data.redis.connection.RedisConnectionFactory;\nimport org.springframework.data.redis.core.RedisTemplate;\nimport org.springframework.data.redis.core.StringRedisTemplate;\nimport org.springframework.data.redis.core.script.DefaultRedisScript;\n \n\/**\n * @author \u5bd2\u591c\n *\/\n@Configuration\npublic class RedisConfiguration {\n \n @Bean\n public RedisTemplate\n   \n     redisTemplate(\n RedisConnectionFactory factory) {\n return new StringRedisTemplate(factory);\n }\n \n @Bean\n public DefaultRedisScript loadRedisScript() {\n DefaultRedisScript redisScript = new DefaultRedisScript();\n redisScript.setLocation(new ClassPathResource(\"ratelimiter.lua\"));\n redisScript.setResultType(java.lang.Boolean.class);\n return redisScript;\n }\n \n}\n   <\/pre>\n<p><strong>\u521b\u5efa\u4e00\u4e2a\u81ea\u5b9a\u4e49\u6ce8\u89e3&nbsp;<\/strong><\/p>\n<pre>\npackage com.hy.annotation;\n \nimport java.lang.annotation.*;\n \n\/**\n * @author \u5bd2\u591c\n *\/\n@Target({ElementType.METHOD})\n@Retention(RetentionPolicy.RUNTIME)\n@Documented\npublic @interface AccessLimiter {\n \n int limit();\n \n String methodKey() default \"\";\n \n}<\/pre>\n<p>\u521b\u5efa\u4e00\u4e2a\u5207\u5165\u70b9<\/p>\n<pre>\npackage com.hy.annotation;\n \nimport com.google.common.collect.Lists;\nimport lombok.extern.slf4j.Slf4j;\nimport org.aspectj.lang.JoinPoint;\nimport org.aspectj.lang.annotation.Aspect;\nimport org.aspectj.lang.annotation.Before;\nimport org.aspectj.lang.annotation.Pointcut;\nimport org.aspectj.lang.reflect.MethodSignature;\nimport org.springframework.data.redis.core.StringRedisTemplate;\nimport org.springframework.data.redis.core.script.RedisScript;\nimport org.springframework.stereotype.Component;\nimport org.springframework.util.StringUtils;\n \nimport java.lang.reflect.Method;\nimport java.util.Arrays;\nimport java.util.stream.Collectors;\n \n\/**\n * @author \u5bd2\u591c\n *\/\n@Slf4j\n@Aspect\n@Component\npublic class AccessLimiterAspect {\n \n private final StringRedisTemplate stringRedisTemplate;\n \n private final RedisScript\n   \n     rateLimitLua;\n \n public AccessLimiterAspect(StringRedisTemplate stringRedisTemplate, RedisScript\n    \n      rateLimitLua) { this.stringRedisTemplate = stringRedisTemplate; this.rateLimitLua = rateLimitLua; } @Pointcut(value = \"@annotation(com.hy.annotation.AccessLimiter)\") public void cut() { log.info(\"cut\"); } @Before(\"cut()\") public void before(JoinPoint joinPoint) { \/\/ 1. \u83b7\u5f97\u65b9\u6cd5\u7b7e\u540d\uff0c\u4f5c\u4e3amethod Key MethodSignature signature = (MethodSignature) joinPoint.getSignature(); Method method = signature.getMethod(); AccessLimiter annotation = method.getAnnotation(AccessLimiter.class); if (annotation == null) { return; } String key = annotation.methodKey(); int limit = annotation.limit(); \/\/ \u5982\u679c\u6ca1\u8bbe\u7f6emethodkey, \u4ece\u8c03\u7528\u65b9\u6cd5\u7b7e\u540d\u751f\u6210\u81ea\u52a8\u4e00\u4e2akey if (StringUtils.isEmpty(key)) { Class[] type = method.getParameterTypes(); key = method.getClass() + method.getName(); if (type != null) { String paramTypes = Arrays.stream(type) .map(Class::getName) .collect(Collectors.joining(\",\")); log.info(\"param types: \" + paramTypes); key += \"#\" + paramTypes; } } \/\/ 2. \u8c03\u7528Redis boolean acquired = stringRedisTemplate.execute( rateLimitLua, \/\/ Lua script\u7684\u771f\u8eab Lists.newArrayList(key), \/\/ Lua\u811a\u672c\u4e2d\u7684Key\u5217\u8868 Integer.toString(limit) \/\/ Lua\u811a\u672cValue\u5217\u8868 ); if (!acquired) { log.error(\"your access is blocked, key={}\", key); throw new RuntimeException(\"Your access is blocked\"); } } }\n    \n   <\/pre>\n<p><strong>\u521b\u5efa\u6d4b\u8bd5\u9879\u76ee<\/strong><\/p>\n<p style=\"text-align: center\"><img decoding=\"async\" style=\"max-width:100%\" src=\"https:\/\/www.17golang.com\/uploads\/20230101\/167253661563b0e22793c3c.png\" class=\"aligncenter\"><\/p>\n<p><strong>pom.xml\u4e2d\u5f15\u5165\u7ec4\u4ef6<\/strong><\/p>\n<p style=\"text-align: center\"><img decoding=\"async\" style=\"max-width:100%\" src=\"https:\/\/www.17golang.com\/uploads\/20230101\/167253661563b0e227ecff0.png\" class=\"aligncenter\"><\/p>\n<p><strong>application.yml\u914d\u7f6e<\/strong><\/p>\n<pre>\nspring:\n redis:\n host: 192.168.0.218\n port: 6379\n password: 123456\n database: 0\n application:\n name: ratelimiter-test\nserver:\n port: 10087<\/pre>\n<p><strong>\u521b\u5efacontroller<\/strong><\/p>\n<pre>\npackage com.hy;\n \nimport com.hy.annotation.AccessLimiter;\nimport lombok.extern.slf4j.Slf4j;\nimport org.springframework.web.bind.annotation.GetMapping;\nimport org.springframework.web.bind.annotation.RestController;\n \n\/**\n * @author \u5bd2\u591c\n *\/\n@RestController\n@Slf4j\npublic class Controller {\n \n private final com.hy.AccessLimiter accessLimiter;\n \n public Controller(com.hy.AccessLimiter accessLimiter) {\n this.accessLimiter = accessLimiter;\n }\n \n @GetMapping(\"test\")\n public String test() {\n accessLimiter.limitAccess(\"ratelimiter-test\", 3);\n return \"success\";\n }\n \n \/\/ \u63d0\u9192\uff01 \u6ce8\u610f\u914d\u7f6e\u626b\u5305\u8def\u5f84\uff08com.hy\u8def\u5f84\u4e0d\u540c\uff09\n @GetMapping(\"test-annotation\")\n @AccessLimiter(limit = 1)\n public String testAnnotation() {\n return \"success\";\n }\n \n}<\/pre>\n<p><strong>\u5f00\u59cb\u6d4b\u8bd5\uff0c\u5feb\u901f\u70b9\u51fb\u7ed3\u679c\u5982\u4e0b<\/strong><\/p>\n<p style=\"text-align: center\"><img decoding=\"async\" style=\"max-width:100%\" src=\"https:\/\/www.17golang.com\/uploads\/20230101\/167253661663b0e2284d9c8.png\" class=\"aligncenter\"><\/p>\n<p style=\"text-align: center\"><img decoding=\"async\" style=\"max-width:100%\" src=\"https:\/\/www.17golang.com\/uploads\/20230101\/167253661663b0e228a5e9b.png\" class=\"aligncenter\"><\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u57fa\u4e8eRedis+Lua\u811a\u672c\u5b9e\u73b0\u5206&#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-201255","post","type-post","status-publish","format-standard","hentry","category-database"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/201255","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=201255"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/201255\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=201255"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=201255"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=201255"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}