今天工作需要要写一个小工具来测试一下数据库的写入性能，需要支持多并发，并且支持调整事务提交记录大小，所以就用Java写了一个，就一个类，比较简单但是基本功能都有了，下面看看代码实现

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Statement;
import java.sql.Timestamp;
import java.sql.Types;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collections;
import java.util.Formatter;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Random;
import java.util.UUID;
import java.util.concurrent.CountDownLatch;
import java.util.logging.Level;
import java.util.logging.Logger;public class InsertTest {private static Logger logger = Logger.getLogger(InsertTest.class.getName());private static final String DB_DRIVER = "com.ibm.db2.jcc.DB2Driver";private static final String DB_URL = "jdbc:db2://9.111.251.152:50000/padb";private static final String DB_USERNAME = "db2inst";private static final String DB_PASSWORD = "Letmein";private static Random random = new Random(10000);private static final String TABLE_NAME = "BENCHMARK";private int batchSize;private int concurrent;private int sampling;public static void main(String[] args) throws Exception {printHeader();int[] concurrentList = new int[]{1, 5, 10, 20};int[] batchSizeList = new int[] {1000, 2000, 5000, 10000};for (int concurrent : concurrentList) {for (int batchSize : batchSizeList) {new InsertTest(batchSize, concurrent).run(true);}Thread.sleep(10000);}}public InsertTest(final int batchSize, final int concurrent) throws Exception {this.batchSize = batchSize;this.concurrent = concurrent;this.sampling = 100;}public InsertTest(final int batchSize, final int concurrent, final int sampling) throws Exception {this.batchSize = batchSize;this.concurrent = concurrent;this.sampling = sampling;}public void run(boolean printResult) throws Exception {final List<Long> results = Collections.synchronizedList(new ArrayList<Long>());final CountDownLatch startGate = new CountDownLatch(concurrent);final CountDownLatch endGate = new CountDownLatch(concurrent);for (int idxConcurrent = 0; idxConcurrent < concurrent; idxConcurrent++) {new Thread(new Runnable() {public void run() {startGate.countDown();try {long time = execute();long avg = batchSize * sampling * 1000 / time;;results.add(Long.valueOf(avg));} catch(Exception ex) {ex.printStackTrace();} finally {endGate.countDown();}}}).start();}endGate.await();Collections.sort(results);if (printResult) {printResult(batchSize, concurrent, results);}}public long execute() throws Exception {Connection conn = getConnection();Map<String, Integer> columns = queryTableColumns(conn);String insertSQL = generateInsertSQL(columns);PreparedStatement ps = conn.prepareStatement(insertSQL);try {long start = System.currentTimeMillis();for (int i = 0; i < sampling; i++) {execute(conn, ps, columns);}long stop = System.currentTimeMillis();return stop - start;} catch(Exception ex) {logger.log(Level.SEVERE, null, ex);conn.rollback();conn.close();throw ex;} finally {conn.close();}}public void execute(Connection conn, PreparedStatement ps, Map<String, Integer> columns) throws Exception {try {for (int idx = 0; idx < batchSize; idx++) {int idxColumn = 1;for (String column : columns.keySet()) {if (column.equalsIgnoreCase("ID")) {ps.setObject(idxColumn, UUID.randomUUID().toString());} else {ps.setObject(idxColumn, generateColumnValue(columns.get(column)));}idxColumn ++;}ps.addBatch();}ps.executeBatch();conn.commit();ps.clearBatch();} catch (SQLException ex) {logger.log(Level.SEVERE, null, ex);if (null != ex.getNextException()) {logger.log(Level.SEVERE, null, ex.getNextException());}conn.rollback();throw ex;}}private String generateInsertSQL(Map<String, Integer> columns) throws SQLException {StringBuilder sb = new StringBuilder();StringBuffer sbColumns = new StringBuffer();StringBuffer sbValues = new StringBuffer();sb.append("INSERT INTO ").append(TABLE_NAME);for (String column : columns.keySet()) {if (sbColumns.length() > 0) {sbColumns.append(",");sbValues.append(",");}sbColumns.append(column);sbValues.append("?");}sb.append("(").append(sbColumns).append(")");sb.append("VALUES");sb.append("(").append(sbValues).append(")");return sb.toString();}private Map<String, Integer> queryTableColumns(Connection conn) throws Exception {Map<String, Integer> columns = new LinkedHashMap<String, Integer>();String sql = "SELECT * FROM " + TABLE_NAME + " WHERE 1=0";Statement stmt = conn.createStatement();ResultSet rs = stmt.executeQuery(sql);ResultSetMetaData rsmd = rs.getMetaData();for (int i = 1; i <= rsmd.getColumnCount(); i++) {columns.put(rsmd.getColumnName(i), rsmd.getColumnType(i));}return columns;}private Object generateColumnValue(int type) {Object obj = null;switch (type) {case Types.DECIMAL:case Types.NUMERIC:case Types.DOUBLE:case Types.FLOAT:case Types.REAL:case Types.BIGINT:case Types.TINYINT:case Types.SMALLINT:case Types.INTEGER:obj = random.nextInt(10000);break;case Types.DATE:obj = Calendar.getInstance().getTime();break;case Types.TIMESTAMP:obj = new Timestamp(System.currentTimeMillis());break;default:obj = String.valueOf(random.nextInt(10000));break;}return obj;}private Connection getConnection() throws Exception {Class.forName(DB_DRIVER);Connection conn = DriverManager.getConnection(DB_URL, DB_USERNAME, DB_PASSWORD);conn.setAutoCommit(false);return conn;}private static void printHeader() {StringBuilder sb = new StringBuilder();sb.append("\n");sb.append(new Formatter().format("%15s|%15s|%15s|%15s|%15s", "BATCH_SIZE", "CONCURRENT", "AVG (r/s)", "MIN (r/s)", "MAX (r/s)"));System.out.println(sb.toString());}private static void printResult(int batch, int concurrent, List<Long> results) {Long total = Long.valueOf(0);for (Long result : results) {total += result;}StringBuilder sb = new StringBuilder();sb.append(new Formatter().format("%15s|%15s|%15s|%15s|%15s", batch, concurrent, (total/results.size()), results.get(0), results.get(results.size() - 1)));System.out.println(sb.toString());}
}

要运行测试需要
1. 修改数据库的配置信息
2. 修改TABLE_NAME指定数据库里需要测试的表名，测试程序会查询这个表的定义来生成写入SQL语句
3. 修改concurrentList指定需要测试并发数列表，默认测试1,5,10,20个并发
4. 修改batchSizeList指定每次测试的事务提交记录数据，默认是1000,2000,5000,10000

最后运行测试，会生成类似下面的结果

BATCH_SIZE|     CONCURRENT|      AVG (r/s)|      MIN (r/s)|      MAX (r/s)1000|              1|            ...|            ...|            ...2000|              1|            ...|            ...|            ...5000|              1|            ...|            ...|            ...10000|              1|            ...|            ...|            ...1000|              5|            ...|            ...|            ...2000|              5|            ...|            ...|            ...5000|              5|            ...|            ...|            ...10000|              5|            ...|            ...|            ...