001: /*
002: * $Id: MultiColumnPrimaryKeyInserts.java,v 1.3 2005/05/02 22:32:03 ahimanikya Exp $
003: * =======================================================================
004: * Copyright (c) 2005 Axion Development Team. All rights reserved.
005: *
006: * Redistribution and use in source and binary forms, with or without
007: * modification, are permitted provided that the following conditions
008: * are met:
009: *
010: * 1. Redistributions of source code must retain the above
011: * copyright notice, this list of conditions and the following
012: * disclaimer.
013: *
014: * 2. Redistributions in binary form must reproduce the above copyright
015: * notice, this list of conditions and the following disclaimer in
016: * the documentation and/or other materials provided with the
017: * distribution.
018: *
019: * 3. The names "Tigris", "Axion", nor the names of its contributors may
020: * not be used to endorse or promote products derived from this
021: * software without specific prior written permission.
022: *
023: * 4. Products derived from this software may not be called "Axion", nor
024: * may "Tigris" or "Axion" appear in their names without specific prior
025: * written permission.
026: *
027: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
028: * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
029: * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
030: * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
031: * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
032: * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
033: * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
034: * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
035: * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
036: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
037: * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
038: * =======================================================================
039: */
040: package org.axiondb.profile;
041:
042: import java.sql.Connection;
043: import java.sql.DriverManager;
044: import java.sql.PreparedStatement;
045: import java.sql.SQLException;
046: import java.sql.Statement;
047: import java.util.Random;
048:
049: /**
050: * Derived from Chas Emerick's 25-Jan-2005 post to the user's list.
051: * @author Chas Emerick
052: * @author Rodney Waldhoff
053: */
054: public class MultiColumnPrimaryKeyInserts {
055: private static Random _random = new Random();
056:
057: static {
058: try {
059: Class.forName("org.axiondb.jdbc.AxionDriver");
060: } catch (Exception e) {
061: e.printStackTrace();
062: }
063: }
064:
065: public static void main(String[] args) throws Exception {
066: String connectString = "jdbc:axiondb:test:/tmp/testdb"
067: + _random.nextLong();
068: int numLoops = 10000;
069: int reportEvery = 1000;
070: boolean commitEveryBatch = true;
071: boolean useConstraint = false;
072:
073: if (args.length > 0 && "--help".equals(args[0])) {
074: System.out
075: .println("args: <total # of inserts> <batch size> <commit after every batch> <use constraint> <connect string>");
076: System.out.println("defaults: " + numLoops + " "
077: + reportEvery + " " + commitEveryBatch + " "
078: + useConstraint + " " + connectString);
079: return;
080: }
081:
082: switch (args.length) {
083: case 5:
084: connectString = args[4];
085: case 4:
086: useConstraint = Boolean.valueOf(args[3]).booleanValue();
087: case 3:
088: commitEveryBatch = Boolean.valueOf(args[2]).booleanValue();
089: case 2:
090: reportEvery = Integer.parseInt(args[1]);
091: case 1:
092: numLoops = Integer.parseInt(args[0]);
093: break;
094: }
095:
096: System.out.println("Total number of inserts: " + numLoops);
097: System.out.println("Reporting every: " + reportEvery);
098: System.out.println("Commit after each batch? "
099: + commitEveryBatch);
100: System.out.println("Apply constraint? " + useConstraint);
101: System.out.println("Connect String: " + connectString);
102:
103: System.out.println("Running...");
104: System.out.println("");
105:
106: Connection conn = DriverManager.getConnection(connectString);
107: conn.setAutoCommit(false);
108: run(conn, numLoops, reportEvery, commitEveryBatch,
109: useConstraint);
110: conn.close();
111: }
112:
113: private static void run(Connection conn, int loops,
114: int reportEvery, boolean commitEveryBatch,
115: boolean useConstraint) throws SQLException {
116: String data = String.valueOf(_random.nextLong())
117: + String.valueOf(_random.nextLong())
118: + String.valueOf(_random.nextLong());
119: {
120: Statement stmt = conn.createStatement();
121: stmt
122: .executeUpdate("create table data_test (a int, b int, c varchar(1000) "
123: + (useConstraint ? ", constraint pk primary key (a, b)"
124: : "") + ")");
125: stmt.close();
126: conn.commit();
127: }
128: PreparedStatement pstmt = conn
129: .prepareStatement("insert into data_test (a, b, c) values (?, ?, ?)");
130: long start = System.currentTimeMillis();
131: long time = start;
132: for (int i = 1; i <= loops; i++) {
133: pstmt.setInt(1, i / 10);
134: pstmt.setInt(2, i % 10);
135: pstmt.setString(3, data);
136: pstmt.executeUpdate();
137: if (i % reportEvery == 0) {
138: if (commitEveryBatch) {
139: conn.commit();
140: }
141: long stop = System.currentTimeMillis();
142: System.out.println("Batch of " + reportEvery
143: + " inserts took " + (stop - time) + " ms ["
144: + (((double) stop - time) / reportEvery)
145: + " ms/insert]");
146: time = System.currentTimeMillis();
147: }
148: }
149: conn.commit();
150: long stop = System.currentTimeMillis();
151: System.out.println("Total of " + loops + " inserts took "
152: + (stop - start) + " ms ["
153: + (((double) stop - start) / loops) + " ms/insert]");
154: }
155:
156: }
|