redis生成分布式id方案

分布式Id - redis方式

 

本篇分享内容是关于生成分布式Id的其中之一方案，除了redis方案之外还有如：数据库，雪花算法，mogodb(object_id也是数据库)等方案，对于redis来说是我们常用并接触比较多的，因此主要谈谈结合redis生成分布式id方案。

• 分布式Id设计流程图
• 基于redis的hash自动increment累加生成有序Id
• 定期删除无用hash列

分布式Id设计流程图(有点粗略)

基于redis的hash自动increment累加生成有序Id

使用redis方案生成id，其中之一的方式主要使用increment(递增)，不管是string、hash等都具有该方法，为了更方便管理我们id生成key这里建议使用hash的列的方式，以下内容都基于springboot分享；

当然，第一步我们需要创建一个hash和hkey才行，至于在业务第一次被访问来创建这个hash还是通过服务自动创建这个看业务和流量，这里的hkey是有一定规则的(当然不用局限性),这里我按照日期格式来做key，可以有如下代码：

 1     /**
 2      * 生成每天的初始Id
 3      * @param hashName
 4      * @return
 5      */
 6     public String initPrimaryId(String hashName) {
 7         Assert.hasLength(hashName, "hashName不能为空");
 8
 9         String hashCol = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
10         //自定义编号规则
11         String hashColVal = hashCol + "00001";
12         redisTemplate.opsForHash().putIfAbsent(hashName, hashCol, hashColVal);
13         return hashCol;
14     }

上面很容易理解，hash中key是有每天日期格式组成，意思每天都需要生成一个新的日期key，通过putIfAbsent达到不重复添加的原则，至于hval可以根据自定义编号规则来生成一串数字字符(注：一定要数字)；有了上面的基础，我们仅仅需要increment来累加，redis即帮我们完整hval+1的操作，当然可以自定义累加数，如下代码：

 1     /**
 2      * 获取分布式Id
 3      *
 4      * @param hashName
 5      * @return
 6      */
 7     public long getPrimaryId(String hashName) {
 8         try {
 9             String hashCol = initPrimaryId(hashName);
10             return redisTemplate.opsForHash().increment(hashName, hashCol, 1);
11         } catch (Exception ex) {
12             ex.printStackTrace();
13         }
14         return 0;
15     }

定期删除无用hash列

就上面我们通过hash来设置每天id只增初始值，hash的hkey布局用自动过期功能，因此我们需要代码中维护一套清除来hkey的机制，既然id是根据日期生成，我们可以就用往前推n天的方式达到清除老hkey目的：

 1     /**
 2      * 删除多少天之前的cols
 3      * @param hashName
 4      * @param lessDay
 5      * @return
 6      */
 7     public Long removePrimaryByLessDay(String hashName, int lessDay) {
 8         try {
 9             //当前日期
10             String hashCol = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
11             long idl = Long.valueOf(hashCol) - lessDay;
12
13             String[] removeCols = redisTemplate.opsForHash().entries(hashName).keySet().stream().
14                     map(key -> key.toString()).
15                     filter(key -> idl > Long.valueOf(key)).  //从+1开始，避免删除当天数据
16                     toArray(String[]::new);
17
18             if (ArrayUtils.isNotEmpty(removeCols)) {
19                 return redisTemplate.opsForHash().delete(hashName, removeCols);
20             }
21         } catch (Exception ex) {
22             ex.printStackTrace();
23         }
24         return 0L;
25     }

按照日期来生成分布式id，达到id不重复的目的，这也就是分布式id(不重复)，看起来简单其实如果在高流量冲击下，需要考虑的东西要很多，比如：什么时候生成初始Id、在多个服务器保证服务器时间尽可能一样情况下，该保留多少日期hkey等;

就上面代码对初始Id就做的不是很好，在业务获取Id时候，会去检测并创建id，这样与redis交互就多了一次，通常可以用服务来一次性生成当前日期往后推n天的hkey，这样就避免了在业务获取id时候，还要去putIfAbsent一次验证，减少了请求次数。实在不行可以使用lua脚本放在一次请求去做put和increment，你可能会用到：

1             RedisScript script = new DefaultRedisScript("");
2             redisTemplate.execute(script, Arrays.asList(""));

git地址： https://github.com/shenniubuxing3    云栖社区博客：https://yq.aliyun.com/users/xjexlr3no5xj4