在Spring Boot中整合Redis并使用Lua脚本:
首先,在Spring Boot项目的pom.xml
文件中添加Spring Boot和Spring Data Redis的依赖:
org.springframework.boot
spring-boot-starter-data-redis
在application.properties
或application.yml
中配置Redis的连接信息,以及 redis 配置:
server:
port: 8080
spring:
redis:
host: localhost
port: 6379
RedisConfig.java
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;
@Configuration
public class RedisConfig {
@Bean
public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
RedisTemplate<String, Object> template = new RedisTemplate<>();
template.setConnectionFactory(redisConnectionFactory);
// 设置键的序列化方式为String
template.setKeySerializer(new StringRedisSerializer());
template.setHashKeySerializer(new StringRedisSerializer());
// 设置值的序列化方式为JSON
template.setValueSerializer(new GenericJackson2JsonRedisSerializer());
template.setHashValueSerializer(new GenericJackson2JsonRedisSerializer());
template.afterPropertiesSet(); // 必须调用该方法,完成初始化
return template;
}
}
-- src/main/resources/scripts/incrementBy.lua
local key = KEYS[1] -- 获取第一个参数作为键名
local incrementBy = tonumber(ARGV[1]) -- 获取第二个参数作为增量值,并将其转换为数字类型
-- 存储key和incrementBy的值:仅作debug用
-- redis.call("SET", "debug_key", key)
-- redis.call("SET", "debug_incrementBy", incrementBy)
local currentValue = redis.call("GET", key) -- 通过GET命令获取键的当前值
if not currentValue then
currentValue = 0 -- 如果键不存在,将当前值初始化为0
else
currentValue = tonumber(currentValue) -- 如果键存在,将当前值转换为数字类型
end
local newValue = currentValue + incrementBy -- 计算新的值
redis.call("SET", key, newValue) -- 通过SET命令将新值存储到键中
return newValue -- 返回新值
创建一个服务类,用于执行Lua脚本:
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Service;
import java.io.IOException;
import java.nio.file.Files;
import java.util.Collections;
@Service
public class RedisLuaService {
@Autowired
private RedisTemplate redisTemplate;
public Long executeLuaScript(String key, long incrementBy) {
ClassPathResource luaScriptResource = new ClassPathResource("scripts/incrementBy.lua");
String luaScript;
try {
luaScript = new String(Files.readAllBytes(luaScriptResource.getFile().toPath()));
} catch (IOException e) {
throw new RuntimeException("Failed to read Lua script file", e);
}
DefaultRedisScript redisScript = new DefaultRedisScript<>(luaScript, Long.class);
return redisTemplate.execute(redisScript, Collections.singletonList(key), incrementBy);
}
}
创建一个Controller类,以便从Web应用程序中触发执行Lua脚本的操作:
import com.lfsun.demolfsunstudyredislua.service.RedisLuaService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
@RestController
public class LuaController {
@Autowired
private RedisLuaService redisLuaService;
@GetMapping("/executeLua")
public Long executeLuaScript(@RequestParam("key") String key, @RequestParam("incrementBy") long incrementBy) {
return redisLuaService.executeLuaScript(key, incrementBy);
}
}
现在,启动Spring Boot应用程序。使用浏览器或API调用来触发执行Lua脚本的操作,例如访问http://localhost:8080/executeLua?key=money&incrementBy=99999
我们的money就会蹭蹭蹭的往上涨。
第0次访问,还没有money:
第1次访问,money 多了99999:
第n次访问,千万富翁:
debug_key、debug_incrementBy为debug调试用的,实际使用中删除。