java生成流水号根据redis递增

原创

mob649e8160f07c 2023-10-31 10:42:04 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8160f07c的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java生成流水号根据Redis递增

引言

在实际开发中，我们经常需要生成唯一的流水号来标识某个业务操作，如订单号、交易流水号等。为了保证生成的流水号不重复，通常会使用分布式锁或者数据库的自增字段来实现。但是，这种方式在高并发场景下可能会存在性能瓶颈，因为分布式锁会引入较高的延迟，并且对于数据库的自增字段也有一定的性能限制。

为了解决这个问题，我们可以借助Redis的原子性操作和自增功能，实现高效生成流水号的机制。本文将介绍如何使用Java结合Redis递增生成流水号，并给出相应的代码示例。

流程图

flowchart TD
    subgraph 生成流水号
        A[获取Redis连接]
        B[获取当前流水号]
        C[递增流水号]
        D[释放Redis连接]
    end

类图

classDiagram
    class RedisUtil{
        +get(String key):String
        +incr(String key):Long
        +release(Jedis jedis)
    }
    
    class SerialNumberGenerator{
        +generateSerialNumber():String
    }

代码示例

首先，我们需要一个Redis的工具类RedisUtil来封装与Redis的连接和操作。以下是RedisUtil的代码实现：

public class RedisUtil {
    private static final String REDIS_HOST = "localhost";
    private static final int REDIS_PORT = 6379;
    private static final String REDIS_PASSWORD = "password";

    private Jedis jedis;

    public RedisUtil() {
        jedis = new Jedis(REDIS_HOST, REDIS_PORT);
        jedis.auth(REDIS_PASSWORD);
    }

    public String get(String key) {
        return jedis.get(key);
    }

    public Long incr(String key) {
        return jedis.incr(key);
    }

    public void release() {
        jedis.close();
    }
}

接下来，我们创建一个SerialNumberGenerator类来生成流水号。在生成流水号的过程中，我们会使用RedisUtil来获取和递增流水号。以下是SerialNumberGenerator的代码实现：

public class SerialNumberGenerator {
    private static final String SERIAL_NUMBER_KEY = "serial_number";

    private RedisUtil redisUtil;

    public SerialNumberGenerator() {
        redisUtil = new RedisUtil();
    }

    public String generateSerialNumber() {
        String serialNumber = redisUtil.get(SERIAL_NUMBER_KEY);
        if (serialNumber == null) {
            serialNumber = "1";
        } else {
            Long nextNumber = redisUtil.incr(SERIAL_NUMBER_KEY);
            serialNumber = nextNumber.toString();
        }
        return serialNumber;
    }
}

在上述代码中，我们使用RedisUtil来获取当前流水号，并通过递增操作生成下一个流水号。如果当前流水号为空，我们默认设置为1。

示例运行

下面是一个示例的运行代码，用来测试上述的流水号生成机制：

public class Main {
    public static void main(String[] args) {
        SerialNumberGenerator generator = new SerialNumberGenerator();
        String serialNumber = generator.generateSerialNumber();
        System.out.println("生成的流水号：" + serialNumber);
        generator.release();
    }
}

输出结果为：