Redis之Lua脚本

目录

Lua脚本

编写Lua脚本

?springboot整合redis使用lua

---

Lua脚本

? ? ? ?Redis在2.6推出了脚本功能，允许开发者使用Lua语言编写脚本传到Redis中执行。使用脚本的好处如下:

1. 减少网络开销：本来5次网络请求的操作，可以用一个请求完成，原先5次请求的逻辑放在redis服务器上完成。使用脚本，减少了网络往返时延。

2. 原子操作：Redis会将整个脚本作为一个整体执行，中间不会被其他命令插入。管道不是原子的，不过redis的批量操作命令(类似mset)是原子的。

3. 替代redis的事务功能：redis自带的事务功能很鸡肋，而redis的lua脚本几乎实现了常规的事务功能，官方推荐如果要使用redis的事务功能可以用redis lua替代。

编写Lua脚本

使用?EVAL?命令

EVAL "your_lua_script_here" numkeys key [key ...] arg [arg ...]

your_lua_script_here?是你的Lua脚本。
numkeys?是键的数量。
key [key ...]?是键的列表。
arg [arg ...]?是参数的列表。

举例

?springboot整合redis使用lua

引入redis依赖

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-redis</artifactId>
        </dependency>

封装redis相关操作

@Service
public class RedisService {

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    public RedisService(RedisTemplate<String, String> redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    // 使用 RedisScript 执行 Lua 脚本
    public void setUsername(String username, String value) {
        // Lua 脚本
        String script = "redis.call('SET', KEYS[1], ARGV[1])";

        // 创建 RedisScript 对象
        DefaultRedisScript<String> redisScript = new DefaultRedisScript<>(script, String.class);
        redisScript.setResultType(String.class);

        // 执行 Lua 脚本
        redisTemplate.execute(redisScript, new StringRedisSerializer(), new StringRedisSerializer(),
                Collections.singletonList(username), value);
    }

    // 使用 RedisScript 执行 Lua 脚本
    public void setUsernames(List<String> keys, List<String> values) {
        // Lua 脚本
        String script = "for i, key in ipairs(KEYS) do\n" +
                "    redis.call('SET', key, ARGV[i])\n" +
                "end";

        // 创建 RedisScript 对象
        DefaultRedisScript<String> redisScript = new DefaultRedisScript<>(script, String.class);
        redisScript.setResultType(String.class);

        // 执行 Lua 脚本
        redisTemplate.execute(redisScript, new StringRedisSerializer(), new StringRedisSerializer(),
                keys, values.toArray(new String[0]));
    }
}

调用接口

@RestController
public class redisluaController {

    @Autowired
    private RedisService redisService;

    @RequestMapping("/test")
    public void executeLuaScript() {
        redisService.setUsername("username", "zhangsang");
    }

    @RequestMapping("/test02")
    public void executeLuaScript02() {
        ArrayList<String> keys = new ArrayList<>();
        keys.add("lisi");
        keys.add("wangwu");
        ArrayList<String> values = new ArrayList<>();
        values.add("20");
        values.add("18");
        redisService.setUsernames(keys, values);
    }
}

? ? ? ?test接口中通过lua脚本实现了set username zhangsan操作，传入参数是单个的key和value。执行结果如下：

? ? ? ?test02中通过lua脚本实现批量操作，传入参数是多个key和多个value，通过集合传入。执行结果如下：