MySQL增加Sequence管理功能

出自：http://xm-koma.iteye.com/blog/1961793

项目应用中，曾有以下一个场景：

接口中要求发送一个int类型的流水号，由于多线程模式，如果用时间戳，可能会有重复的情况(当然概率很小)。

所以想到了利用一个独立的自增的sequence来解决该问题。

当前数据库为：mysql

由于mysql和oracle不太一样，不支持直接的sequence，所以需要创建一张table来模拟sequence的功能，理由sql语句如下：

第一步：创建--Sequence 管理表

Java代码  

1. DROP TABLE IF EXISTS sequence;  
2. CREATE TABLE sequence (  
3.          name VARCHAR(50) NOT NULL,  
4.          current_value INT NOT NULL,  
5.          increment INT NOT NULL DEFAULT 1,  
6.          PRIMARY KEY (name)  
7. ) ENGINE=InnoDB;  

 

第二步：创建--取当前值的函数

Java代码  

1. DROP FUNCTION IF EXISTS currval;  
2. DELIMITER $  
3. CREATE FUNCTION currval (seq_name VARCHAR(50))  
4.          RETURNS INTEGER  
5.          LANGUAGE SQL  
6.          DETERMINISTIC  
7.          CONTAINS SQL  
8.          SQL SECURITY DEFINER  
9.          COMMENT ''  
10. BEGIN  
11.          DECLARE value INTEGER;  
12.          SET value = 0;  
13.          SELECT current_value INTO value  
14.                    FROM sequence  
15.                    WHERE name = seq_name;  
16.          RETURN value;  
17. END  
18. $  
19. DELIMITER ;  

 

第三步：创建--取下一个值的函数

Java代码  

1. DROP FUNCTION IF EXISTS nextval;  
2. DELIMITER $  
3. CREATE FUNCTION nextval (seq_name VARCHAR(50))  
4.          RETURNS INTEGER  
5.          LANGUAGE SQL  
6.          DETERMINISTIC  
7.          CONTAINS SQL  
8.          SQL SECURITY DEFINER  
9.          COMMENT ''  
10. BEGIN  
11.          UPDATE sequence  
12.                    SET current_value = current_value + increment  
13.                    WHERE name = seq_name;  
14.          RETURN currval(seq_name);  
15. END  
16. $  
17. DELIMITER ;  

 

第四步：创建--更新当前值的函数

Java代码  

1. DROP FUNCTION IF EXISTS setval;  
2. DELIMITER $  
3. CREATE FUNCTION setval (seq_name VARCHAR(50), value INTEGER)  
4.          RETURNS INTEGER  
5.          LANGUAGE SQL  
6.          DETERMINISTIC  
7.          CONTAINS SQL  
8.          SQL SECURITY DEFINER  
9.          COMMENT ''  
10. BEGIN  
11.          UPDATE sequence  
12.                    SET current_value = value  
13.                    WHERE name = seq_name;  
14.          RETURN currval(seq_name);  
15. END  
16. $  
17. DELIMITER ;  

 

第五步：测试函数功能

当上述四步完成后，可以用以下数据设置需要创建的sequence名称以及设置初始值和获取当前值和下一个值。

 

INSERT INTO sequence VALUES ('TestSeq', 0, 1);----添加一个sequence名称和初始值，以及自增幅度
SELECT SETVAL('TestSeq', 10);---设置指定sequence的初始值
SELECT CURRVAL('TestSeq');--查询指定sequence的当前值
SELECT NEXTVAL('TestSeq');--查询指定sequence的下一个值

 

 

在java代码中，可直接创建sql语句查询下一个值，这样就解决了流水号唯一的问题。

贴出部分代码（已测试通过）

Java代码  

1. public void testGetSequence() {  
2.     Connection conn = JDBCUtils.getConnection(url, userName, password);  
3.     String sql = "SELECT CURRVAL('TestSeq');";  
4.     PreparedStatement ptmt = null;  
5.     ResultSet rs = null;  
6.     try {  
7.         ptmt = conn.prepareStatement(sql);  
8.         rs = ptmt.executeQuery();  
9.         int count = 0;  
10.         while (rs.next()) {  
11.             count = rs.getInt(1);  
12.         }  
13.         System.out.println(count);  
14.     } catch (SQLException e) {  
15.         e.printStackTrace();  
16.     } finally {  
17.         JDBCUtils.close(rs, ptmt, conn);  
18.     }  
19. }  

 

 

ps：在应用中，还有一种用java代码去实现模拟自增sequence的方式，具体思路是创建一张存放sequence的table，然后通过java调用sql语句去查询和修改这个table中指定sequence名称的值，这种方式请加上synchronized。具体代码这里就不上传了，因为实现了，未去测试过。