python redis lua 脚本 实战一个限流或抢红包等业务

> yum install -y readline
> yum install -y readline-devel
> wget http://www.lua.org/ftp/lua-5.1.1.tar.gz
> tar -zxvf lua-5.1.1.tar.gz -C /usr/local/
> cd /usr/local/lua-5.1.1
> make linux
> make install

> lua
> print("hello world")
> hello world

EVAL script numkeys key [key ...] arg [arg ...]

> eval "return {KEYS[1],KEYS[2],ARGV[1],ARGV[2]}" 2 key1 key2 first second
1) "key1"
2) "key2"
3) "first"
4) "second"

local result=0
    local id= redis.call(‘get‘,KEYS[1])
if id then
    result= redis.call(‘sismember‘,id,KEYS[2])
end
 
if result==1 then
    redis.call(‘expire‘,KEYS[1],ARGV[1])
end
  
return result

redis-cli  --eval 1.lua 2 user1 user:list 1800

  redis-cli  eval "$(1.lua)" 2 user1 user:list 1800

  redis-cli  script load "$(cat 1.lua)"
  redis-cli  evalsha 2cfe726fe60678e8ed4d689a3e98727d1e1e2339 1 test 1

redis.call(‘set‘,‘age‘,‘18‘)
local age = redis.call(‘get‘,‘age‘)
return age;

> /usr/local/redis-3.2.9/src/redis-server /usr/local/redis-3.2.9/redis.conf

> /usr/local/redis-3.2.9/src/redis-cli -h 192.168.1.30 -p 6379 --eval /usr/local/luadir/redis.lua
>"18"     #set了age，并返回了value = 18

--获取key1 的值，并赋值给local age
local age = redis.call(‘get‘, KEYS[1]);
if age == 18 then
redis.call(‘set‘,KEYS[2],ARGV[1])
end

> /usr/local/redis-3.2.9/src/redis-cli -h 192.168.1.30 -p 6379 --eval /usr/local/luadir/redis-arg.lua age name , mawenxia

import redis
pool = redis.ConnectionPool(host="localhost", port=6379, db=0)
r = redis.Redis(connection_pool=pool)

lua_2 = """
   local ret = redis.call("get", KEYS[1])
   return ret
"""
script_2 = r.register_script(lua_2)

print(script_2(keys=["hugo"]))

public class TestSingleRedis {

    private static Jedis jedis;     //单实例[]
    private static ShardedJedis shard;      //分片[]
    private static ShardedJedisPool pool;   //池化[apache common - pool2]

    @BeforeClass
    public static void setUpBeforeClass() throws Exception {
        //单个节点
        jedis = new Jedis("192.168.1.30", 6379);
        //分片
        List<JedisShardInfo> shards = Arrays.asList(
                new JedisShardInfo("192.168.1.30",6379));

        shard = new ShardedJedis(shards);
        //池化
        GenericObjectPoolConfig goConfig = new GenericObjectPoolConfig();
        goConfig.setMaxTotal(100);
        goConfig.setMaxIdle(20);
        goConfig.setMaxWaitMillis(-1);
        goConfig.setTestOnBorrow(true);        
        pool = new ShardedJedisPool(goConfig, shards);
    }

    @AfterClass
    public static void tearDownAfterClass() throws Exception {
        jedis.disconnect();
        shard.disconnect();
        pool.destroy();
    }
  /**
     * <B>方法名称：</B>Redis使用Lua执行应用程序<BR>
     * <B>概要说明：</B>
     * 我们可以使用Redis+Lua的方式，实现一个完整的事务、保证事务的原子性。
     * 如何使用Redis+Lua?
     * 我们使用scriptLoad方法，把我们写好的lua脚本加载到Redis的内存中（注意是内存，每次重启则失效）。
     * scriptLoad方法会返回一个索引Key，我们只需要通过这个索引Key就可以找到我们之前放到Redis里的脚本。
     * 调用evalsha方法，传入索引key，以及操作键、参数值。进行返回 <BR>
     */
    /**
     * lua script：
     *      local t1 = redis.call(‘hgetall‘,KEYS[1]);
     *      if type(t1) == ‘table‘ then 
     *          return t1;
     *      end;
     *
     *  type(t1) == ‘table‘: lua的表类型,redis的hash类型属于这个类型
     */
    public static final String SCRIPT = 
            "local t1 = redis.call(‘hgetall‘,KEYS[1]);" + "\n" +
            "if type(t1) == ‘table‘ then" + "\n" +
            "return t1;" + "\n" +
            "end;" + "\n" ;
    @Test
    public void  testLua(){
        /**
         * 把string类型的脚本放入redis的scriptLoad中,变成一个可执行的脚本
         * 并且会返回一个shakey,使用这个shakey来调用脚本
         */
        String shakey = jedis.scriptLoad(SCRIPT);//加载脚本，获取sha索引 
        System.out.println("shakey: " + shakey);
        //要获取的key值
        List<String> keys = new ArrayList<>();
        keys.add("myhash");
        //传入的参数。这里没有加参数
        List<String> args = new ArrayList<>();
        // /usr/local/bin/redis-cli -h 192.168.1.115 -p 6379 --eval /usr/local/luadir/03.lua name age , baihezhuo
        List<String> ret = (List<String>)jedis.evalsha(shakey, keys, args);

        System.out.println(ret);
        jedis.close();
    }

}

 //定义空表
T1={} 
//赋值
T1["name"] = "mawenxia"
T1["age"] = 18
//其他赋值方式
T1.name = "mawenxia"
T1.age = 18
//其他赋值方式
T1.person = {name="mawenxia",age=18}

> wget http://www.kyne.com.au/~mark/software/download/lua-cjson-2.1.0.tar.gz
> tar -zvxf lua-cjson-2.1.0.tar.gz -C /usr/local/
> cd /usr/local/lua-cjson-2.1.0/
> vim Makefile
> 修改PREFIX 变量为/usr/local  保存
> make install
> cp cjson.so /usr/local/lib/lua/5.1/