redis技巧--自动完成功能实现

自动完成功能通常都伴随搜索框出现，就是用户在输入时帮助其自动补全。

好比对成语进行补全，现有以下成语：一心一意，一心二用，一路顺风。

两种实现方式：

实现方式一：

为每一个成语的每一个前缀都使用一个集合类型键来存储该前缀对应的成语名，而且为了实现排序，咱们使用有序集合，并score都为0，这样就按元素值的字典序排序。若是想要实现按照词的热度排序，须要再建立一个有序集合，存放词和score，最后把查询结果和这个集合作交集便可，这样能够避免更新一个词的热度须要更新多个集合的状况，由于一个词会出如今多个集合中。

作法：

对于上边的成语就会建立以下形式的有序集合（key : value,...）,score都为0：

一 ：一心一意，一心二用，一路顺风

一心：一心一意，一心二用

一心一：一心一意

一心二：一心二用

一帆：一路顺风

一帆风：一路顺风

下边实现生成这些前缀对应的有序集合的代码

$valueArr = ['一心一意','一心二用','一路顺风']; //中文字符注意指定编码,不能用$str[$i]方式获取中文
foreach ($valueArr as $item) { for ($i = 1,$len = mb_strlen($item, "utf-8"); $i < $len; $i++) { $k = $prefix.mb_substr($item,0, $i, "utf-8"); $redis->zadd($k, 0, $item); } }

在经过对应的关键字获取对应的有序集合便可

$search1 = "一"; $search2 = "一帆"; $ret1 = $redis->zRange($$prefix.$search1, 0, -1); $ret2 = $redis->zRange($$prefix.$search2, 0, -1);

若是须要实现热度排序，只须要取关键字集合和所有成语的热度集合的交集便可，经过参数指定两个集合的分数组合方式

$redis->zInter($retZset, [$prefix.$search, $hotZset]); //取交集
$ret = $redis->zRevRange($retZset, 0, -1); //倒序

 

实现方式二：

经过有序集合实现，该方法由redis做者介绍。由于有序集合当元素的score相同时，按照元素的字典序排序，利用这个特性只用一个有序集合就能实现标签补全。

作法：

把所有成语前缀存入有序集合，再把所有成语后边加上*号后存入有序集合，分数均为0。

获取时先取关键字在zset中的排名，而后在获取排名后的N个元素，最后遍历结果，找出*结尾而且以对应关键字开头的元素便是结果。

先实现存入代码

//遍历所有的前缀和成语加入有序集合
foreach ($valueArr as $item) { for ($i = 1,$len = mb_strlen($item, "utf-8"); $i <= $len; $i++) { $member = mb_substr($item,0, $i, "utf-8"); if ($i == $len) { $member .= "*"; //非前缀须要在后边补*
 } $redis->zadd("auto:zset", 0, $member); } }

再实现获取代码

//查询
$search = "一帆"; $rank = $redis->zRank("auto:zset", $search); $ret = $redis->zRange("auto:zset", $rank + 1, $rank + 10); //获取10个
$ret = array_filter($ret, function($d){ return ("*" === mb_substr($d, -1)); //此处没有去掉*,没有匹配前缀,多出来不匹配条数能够做为推荐
}); echo "<pre>";print_r($ret);exit;

 

总结：

方式一须要N个集合存储，但每一个集合数据量较小，方式二只用一个集合实现，但数据量大，同时实现热度排序比较困难，这个看具体使用场景和数据量选择吧。