Redis 自动补全

Posted 2020-10-27

Redis自动补全：

1. 自动补全最近联系人

使用List列表结构有序地存储元素，redis主要用于记录联系人列表，而非实际的执行自动补全操作，数据的过滤及补全在客户端执行；对于较短的列表来说可行；

构建最近联系人自动补全列表，如果指定的联系人已经存在，将从列表移除该联系人，将他重新推入列表的最左端；对列表进行修剪，保留位于列表前的100个联系人：

public void addUpdateContact(Jedis conn, String user, String contact) {

        String acList = "recent:" + user;

        Transaction trans = conn.multi();

        trans.lrem(acList, 0, contact);

        trans.lpush(acList, contact);

        trans.ltrim(acList, 0, 99);

        trans.exec();

    }

当用户不想再联系某个人，将他从列表删除：

    public void removeContact(Jedis conn, String user, String contact) {

        conn.lrem("recent:" + user, 0, contact);

    }

获取自动补全列表，查找匹配的用户，但实际的自动补全处理是在java里完成的，所以操作首先需要获取整个列表结构：

    public List<String> fetchAutocompleteList(Jedis conn, String user, String prefix) {

        List<String> candidates = conn.lrange("recent:" + user, 0, -1);

        List<String> matches = new ArrayList<String>();

        for (String candidate : candidates) {

            if (candidate.toLowerCase().startsWith(prefix)){

                matches.add(candidate);

            }

        }

        return matches;

    }

2. 通讯录自动补全

对于包含非常多元素的列表，必须直接在redis内部完成匹配元素的工作，尽量减少服务器传输给客户端的数据量；使用ZSet有序集合来存储；

（把有序集合里的所有分值都设为0，利用其性质：当所有成员的分值都相同时，zset按照成员的名字来排序；当所有成员的分值都为0时，成员将按照字符串的二进制顺序进行排序。）

（用户输入abc，我们需要向zset插入两个元素，一个在abbz后面，即”abb{“，一个排在abd前面,即”abc{“；用watch监视有序集合zset；根据这两个元素的排名调用zrange命令，最后移除插入的两元素）

根据给定前缀，生成查找范围：

private static final String VALID_CHARACTERS = "`abcdefghijklmnopqrstuvwxyz{";

    public String[] findPrefixRange(String prefix) {

        int posn = VALID_CHARACTERS.indexOf(prefix.charAt(prefix.length() - 1));

        char suffix = VALID_CHARACTERS.charAt(posn > 0 ? posn - 1 : 0);

        String start = prefix.substring(0, prefix.length() - 1) + suffix + '{';

        String end = prefix + '{';

        return new String[]{start, end};

    }

自动补全函数：

public Set<String> autocompleteOnPrefix(Jedis conn, String guild, String prefix) {

        String[] range = findPrefixRange(prefix);

        String start = range[0];

        String end = range[1];

        String identifier = UUID.randomUUID().toString();

        start += identifier;

        end += identifier;

        String zsetName = "members:" + guild;

 

        conn.zadd(zsetName, 0, start);

        conn.zadd(zsetName, 0, end);

 

        Set<String> items = null;

        while (true){

            conn.watch(zsetName);

            int sindex = conn.zrank(zsetName, start).intValue();

            int eindex = conn.zrank(zsetName, end).intValue();

            int erange = Math.min(sindex + 9, eindex - 2);

 

            Transaction trans = conn.multi();

            trans.zrem(zsetName, start);

            trans.zrem(zsetName, end);

            trans.zrange(zsetName, sindex, erange);

            List<Object> results = trans.exec();

            if (results != null){

                items = (Set<String>)results.get(results.size() - 1);

                break;

            }

        }

 

        for (Iterator<String> iterator = items.iterator(); iterator.hasNext(); ){

            if (iterator.next().indexOf('{') != -1){

                iterator.remove();

            }

        }

        return items;

    }

加入通讯录或移除通讯录：

 public void joinGuild(Jedis conn, String guild, String user) {

        conn.zadd("members:" + guild, 0, user);

    }

 

    public void leaveGuild(Jedis conn, String guild, String user) {

        conn.zrem("members:" + guild, user);

    }