使用Redis的脚本功能实现Redis中数据简单查询

转：http://blog.csdn.net/kungstriving4/article/details/20219069

Redis不仅仅是一个简单的key-value内存数据库，Redis官网对自身的定义是“数据结构服务器”。通过用心设计各种数据结构类型的数据存储，可以实现部分的数据查询功能。因为在Redis的设计中，key是一切，对于Redis是可见的，而value对于Redis来说就是一个字节数组，Redis并不知道你的value中存储的是什么，所以要想实现比如

‘select * from users where user.location="shanghai"’

这样的查询，在Redis是没办法通过value进行比较得出结果的。但是可以通过不同的数据结构类型来做到这一点。比如如下的数据定义

[javascript]  view plain  copy

1. users:1 {name:Jack,age:28,location:shanghai}  
2. users:2 {name:Frank,age:30,location:beijing}  
3. users:location:shanghai [1]  

其中users:1 users:2 分别定义了两个用户信息，通过Redis中的hash数据结构，而users:location:shanghai 记录了所有上海的用户id，通过集合数据结构实现。这样通过两次简单的Redis命令调用就可以实现我们上面的查询。

[java]  view plain  copy

1. Jedis jedis = jedisPool.getResource();  
2. Set<String> shanghaiIDs = jedis.smembers("users:location:shanghai");  
3. //遍历该set  
4. //...  
5. //通过hgetall获取对应的user信息  
6. jedis.hgetAll("users:" + shanghaiIDs[0]);  

通过诸如以上的设计，可以实现简单的条件查询。但是这样的问题也很多，首先需要多维护一个ID索引的集合，其次对于一些复杂查询无能为力(当然也不能期望Redis实现像关系数据库那样的查询，Redis不是干这的)。

但是Redis2.6集成了Lua脚本，可以通过eval命令，直接在RedisServer环境中执行Lua脚本，并且可以在Lua脚本中调用Redis命令。其实，就是说可以让你用Lua这种脚本语言，对Redis中存储的key value进行操作，这个意义就大了，甚至可以将你们系统所需的各种业务写成一个个lua脚本，提前加载进入Redis，然后对于请求的响应，只需要调用一个个lua脚本就行。当然这样说有点夸张，但是意思就是这样的。

比如，现在我们要实现一个‘所有age大于28岁的user’这样一个查询，那么通过以下的Lua脚本就可以实现

[java]  view plain  copy

1. public static final String SCRIPT =  
2.         "local resultKeys={};"  
3.         + "for k,v in ipairs(KEYS) do "  
4.         + " local tmp = redis.call('hget', v, 'age');"  
5.         + " if tmp > ARGV[1] then "  
6.         + "     table.insert(resultKeys,v);"  
7.         + " end;"  
8.         + "end;"  
9.         + "return resultKeys;";  

执行脚本代码

[java]  view plain  copy

1.               Jedis jedis = jedisPool.getResource();  
2. jedis.auth(auth);  
3. List<String> keys = Arrays.asList(allUserKeys);  
4.   
5. List<String> args = new ArrayList<>();  
6. args.add("28");  
7.   
8. List<String> resultKeys = (List<String>)jedis.evalsha(funcKey, keys, args);  
9. return resultKeys;  

注意，以上的代码中使用的是evalsha命令，该命令参数的不是直接Lua脚本字符串，而是提前已经加载到Redis中的函数的一个SHA索引，通过以下的代码将系统中所有需要执行的函数提前加载到Redis中，我们的系统维护一个函数哈希表，后续需要实现什么功能，就从函数表中获取对应功能的SHA索引，通过evalsha调用就行。

[java]  view plain  copy

1. String shaFuncKey = jedis.scriptLoad(SCRIPT);//加载脚本，获取sha索引  
2. funcName_age, shaFuncKey);//添加到函数表中  

通过以上的方法，便可以使较为复杂的查询放到Redis中去执行，提高效率。