《Redis Lua脚本实现ip限流示例.docx》由会员分享，可在线阅读，更多相关《Redis Lua脚本实现ip限流示例.docx（3页珍藏版）》请在taowenge.com淘文阁网|工程机械CAD图纸|机械工程制图|CAD装配图下载|SolidWorks_CaTia_CAD_UG_PROE_设计图分享下载上搜索。
1、引言分布式限流最关键的是要将限流服务做成原子化，而解决方案可以使使用redis+lua或者 nginx+lua技术进行实现，通过这两种技术可以实现的高并发和高性能。首先我们来使用redis+lua实现时间窗内某个接口的请求数限流，实现了该功能后可以改造 为限流总并发/请求数和限制总资源数。Lua本身就是一种编程语言，也可以使用它实现复杂 的令牌桶或漏桶算法。如下操作因是在一个lua脚本中(相当于原子操作)，又因Redis是单线程模型，因此是线 程安全的。相比Redis事务来说，Lua脚本有以下优点减少网络开销：不使用Lua的代码需要向Redis发送多次请求，而脚本只需一次即可，减少网络传输；原
2、子操作：Redis将整个脚本作为一个原子执行，无需担心并发，也就无需事务；复用：脚本会永久保存Redis中，其他客户端可继续使用.Lua脚本.local key = KEYS1一限流 KEY (一秒一个).local limit = tonumber(ARGV 1 ) 一限流大小. local current = tonumber (redis. call ( get , key) or 0).if current + 1 limit then 一如果超出限流大小.return 0.else 一请求数+1,并设置2秒过期.redis. call CINCRBY,key, 1).redis. c
3、all (expire”,key, 2).end.return 1java代码 import org.apache, commons, io. FileUtils;import redis.clients, jedis. Jedis;import java. io. File;import java. io. lOException;import java. net. URISyntaxException;import java. util. ArrayList;import java. util. List;import java. util, concurrent. CountDownLa
4、tch;public class RedisLimitRateWithLUA public static void main(String args) final CountDownLatch latch = new CountDownLatch(1);for (int i = 0; i 7; i+) new Thread (new Runnable() public void run() try latch, await ();System. out. printin (请求是否被执行：+accquire (); catch (Exception e) e. printStackTrace
5、();j). start ();)latch. countDown ();public static boolean accquire() throws lOException, URISyntaxExceptio n Jedis jedis = new Jedis(，z127. 0. 0. l，z);.File luaFile = new File (RedisLimitRateWithLUA. class. getResource (z，/，z). toURI (). getPath() + “limit. lua)；.String luaScript = FileUtils. readF
6、ileToString(luaFile);.String key = ip: + System. currentTimeMillis ()/1000; / 当前秒.String limit = 5； / 最大限制.List keys=newArrayList();.keys, add (key);List args=newArrayList();.args, add(limit);.Long result = (Long) (jedis. eval (luaScript, keys, args) ; / 执彳亍 lua脚本，传入参数 return result = 1; 运行结果请求是否被执行
7、：true请求是否被执行：true请求是否被执行：false请求是否被执行：true请求是否被执行：true请求是否被执行：true请求是否被执行：fals从结果可看出只有5个请求成功执行IP限流Lua脚本. local key = rate, limit: . KEYSTl. local limit = tonumber(ARGV1). local expire_time = ARGV2local is_exists = redis. call(EXISTS”, key) if is exists = 1 thenif redis. call (/ZTNCRZ，, key) limit then return 0elsereturn 1endelseredis. call (SET, key, 1)redis. call(EXPIRE”, key, expire time)return 1 end