数据库的自增字段代码生成器——解决不同数据库自增字段的差异机制

数据库的ID自增字段代码生成器——解决不同数据库自增字段的差异机制

问题：

在设计数据库字段时，有时需要一个int型的id主键，让它能自动递增，每次插入一条数据，它都能够自动增1或者规定的自增数n。对于特定的数据库，要实现这个很简单，比如mysql是用auto_increment，Sql Server是用identity。但是如果在代码中使用和特定数据库有关的特性，那么代码就不能移植，比如把数据库从mysql换成sql server，那么代码就要更改，就不能实现移植扩展。

解决思路：

既然要实现代码的移植扩展，就不能使用和数据库相关的特性，那该怎么办呢？这就是本篇要介绍的ID生成器。

思路是这样的：在数据库中另外建立一张表ID_Main，其包含两个字段：一个字段是要实现ID自增的所属表名(table_name)，另一个字段是该表中自增字段对应的现在最大的记录值(id_max)。假设现在要在数据表(user_info)中插入一条数据时，先使表ID_Main中对应字段table_name为的user_info的id_max的值加1，然后再取出该值，就是自动增1后的值。以后每次要插入一条记录，都会先使ID_Main表中对应的id值加1，再返回，其也就是保持了id的自增。

表ID_Main的设计如下：

/*==============================================================*/

/* Table: ID_Main */

/*==============================================================*/

create table ID_Main (

TABLE_NAME VARCHAR2(20) not null,

ID_MAX NUMBER(10) not null,

constraint PK_ ID_Main primary key (TABLE_NAME)

);

comment on table ID_Main is '用于维护各个表中的记录条数';

以下就是代码层，以Java分析为例：

import java.sql.ResultSet;
import java.sql.SQLException;

import kane.DbUtility;

//ID生成器实体类
public class IDGenerator {
// 使用单例模式
private static IDGenerator instance = null;

private IDGenerator() {
}

public static synchronized IDGenerator getInstance() {
if (instance == null) {
instance = new IDGenerator();
}
return instance;
}

// 通过主键维护表ID_Main来获取表中的记录数
public int getTotal(Connection conn, String tableName) throws SQLException {
int i = 0;
PreparedStatement pstmt = null;
ResultSet rs = null;
try {
String sql = "select ID_MAX from ID_Main where table_name = ?";
pstmt = conn.prepareStatement(sql);
pstmt.setString(1, tableName);
rs = pstmt.executeQuery();
if (rs.next()) {
i = rs.getInt(1);
}
} catch (SQLException e) {
e.printStackTrace();
throw e;
} finally {
DbUtility.closeResultSet(rs);
DbUtility.closePreparedStatement(pstmt);
}
return i;
}

// 将主键维护表ID_Main中的记录数加1,然后再返回该值
public int getNextID(String tableName) {
int total = 0;
Connection conn = null;
PreparedStatement pstmt = null;
try {
conn = DbUtility.getConnection();
String sql = "update ID_Main set ID_MAX = ID_MAX +1 where table_name=?";
// 开启事务
conn.setAutoCommit(false);
pstmt = conn.prepareStatement(sql);
pstmt.setString(1, tableName);
pstmt.executeUpdate();
total = getTotal(conn, tableName);
// 执行事务
conn.commit();
} catch (SQLException e) {
e.printStackTrace();
// 事务回滚
try {
conn.rollback();
} catch (SQLException e1) {
e1.printStackTrace();
}
} finally {
DbUtility.closePreparedStatement(pstmt);
DbUtility.closeConnection(conn);
}
return total;
}
}

其中的DbUtility类是连接数据库操作的封装类，其代码详细参见《分页技术原理与实现(二)》。代码使用了connection的事务操作，保证事务的一致性，也就是当ID_MAX的值自增1时，应该在相应的表中插入一条记录，以保证自增值一致。如果出错，这应该回滚事务。

通过以上分析与代码示例，就简单的实现了一个IDGenerator即ID生成器，它与数据库的特性隔离，实现了代码层的移植性。