正文

如何用Java编写代码确保编号唯一性及避免重复

/0 浏览量
0710

在Java编程中，确保编号的唯一性是常见的需求，特别是在数据库操作、生成序列号或处理并发环境下的唯一标识符时。以下是一些实现编号唯一性和避免重复的方法：

1. 使用UUID类

java.util.UUID类可以生成全球唯一的标识符，通常称为UUID（通用唯一识别码）。UUID由128位组成，几乎可以保证在全球范围内不会重复。

import java.util.UUID;

public class UniqueIdGenerator {
    public static void main(String[] args) {
        UUID uniqueId = UUID.randomUUID();
        System.out.println("生成的UUID: " + uniqueId);
    }
}

2. 利用数据库自增字段

如果使用数据库，可以利用数据库的自增字段（如MySQL中的AUTO_INCREMENT）来确保每条记录的唯一性。

// 示例：MySQL中的自增字段
CREATE TABLE example (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(100)
);

// Java中通过JDBC插入数据
Connection conn = null;
PreparedStatement pstmt = null;
try {
    conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/dbname", "user", "password");
    pstmt = conn.prepareStatement("INSERT INTO example (name) VALUES (?)");
    pstmt.setString(1, "Alice");
    pstmt.executeUpdate();
} catch (SQLException e) {
    e.printStackTrace();
} finally {
    // 关闭资源
}

3. 使用原子类AtomicInteger

java.util.concurrent.atomic.AtomicInteger是一个线程安全的整数操作类，可以确保在多线程环境下编号的唯一性。

import java.util.concurrent.atomic.AtomicInteger;

public class AtomicIdGenerator {
    private static final AtomicInteger counter = new AtomicInteger(0);

    public static int generateId() {
        return counter.incrementAndGet();
    }

    public static void main(String[] args) {
        System.out.println("ID 1: " + generateId());
        System.out.println("ID 2: " + generateId());
    }
}

4. 利用雪花算法（Snowflake Algorithm）

雪花算法是一种分布式系统中常用的ID生成策略，可以生成64位的唯一ID。这个ID由一个数据中心ID、时间戳、序列号和数据机器ID组成。

import java.util.concurrent.atomic.AtomicLong;

public class SnowflakeIdGenerator {
    private final long twepoch = 1288834974657L;
    private final long workerIdBits = 5L;
    private final long datacenterIdBits = 5L;
    private final long maxWorkerId = -1L ^ (-1L << workerIdBits);
    private final long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
    private final long sequenceBits = 12L;

    private final long workerIdShift = sequenceBits;
    private final long datacenterIdShift = sequenceBits + workerIdBits;
    private final long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
    private final long sequenceMask = -1L ^ (-1L << sequenceBits);

    private long workerId;
    private long datacenterId;
    private long sequence = 0L;
    private long lastTimestamp = -1L;

    public SnowflakeIdGenerator(long workerId, long datacenterId) {
        if (workerId > maxWorkerId || workerId < 0) {
            throw new IllegalArgumentException(String.format("worker Id can't be greater than %d or less than 0", maxWorkerId));
        }
        if (datacenterId > maxDatacenterId || datacenterId < 0) {
            throw new IllegalArgumentException(String.format("datacenter Id can't be greater than %d or less than 0", maxDatacenterId));
        }
        this.workerId = workerId;
        this.datacenterId = datacenterId;
    }

    public synchronized long nextId() {
        long timestamp = timeGen();

        if (timestamp < lastTimestamp) {
            throw new RuntimeException(String.format("Clock moved backwards. Refusing to generate id for %d milliseconds", lastTimestamp - timestamp));
        }

        if (lastTimestamp == timestamp) {
            sequence = (sequence + 1) & sequenceMask;
            if (sequence == 0) {
                timestamp = tilNextMillis(lastTimestamp);
            }
        } else {
            sequence = 0L;
        }

        lastTimestamp = timestamp;

        return ((timestamp - twepoch) << timestampLeftShift) | (datacenterId << datacenterIdShift) | (workerId << workerIdShift) | sequence;
    }

    private long tilNextMillis(long lastTimestamp) {
        long timestamp = timeGen();
        while (timestamp <= lastTimestamp) {
            timestamp = timeGen();
        }
        return timestamp;
    }

    private long timeGen() {
        return System.currentTimeMillis();
    }

    public static void main(String[] args) {
        SnowflakeIdGenerator idGenerator = new SnowflakeIdGenerator(1, 1);
        System.out.println("生成的ID: " + idGenerator.nextId());
    }
}

以上代码提供了一个简单的雪花算法实现，可以根据实际情况调整参数。

这些方法各有优缺点，选择哪种方法取决于具体的应用场景和需求。在实际开发中，应根据实际业务需求选择合适的编号生成策略。

-- 展开阅读全文 --