/* Copyright (C) 2002-2004 MySQL AB This program is free software; you can redistribute it and/or modify it under the terms of version 2 of the GNU General Public License as  published by the Free Software Foundation. There are special exceptions to the terms and conditions of the GPL  as it is applied to this software. View the full text of the  exception in file EXCEPTIONS-CONNECTOR-J in the directory of this  software distribution. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */package testsuite.perf;import java.sql.PreparedStatement;import java.sql.SQLException;import java.text.NumberFormat;import testsuite.BaseTestCase;/** * Simple performance testing unit test. *  * @author Mark Matthews */public class LoadStorePerfTest extends BasePerfTest {	/** The table type to use (only for MySQL), 'HEAP' by default */	private String tableType = "HEAP";	private boolean takeMeasurements = false;	private boolean useColumnNames = false;	private boolean largeResults = false;	/**	 * Constructor for LoadStorePerfTest.	 * 	 * @param name	 *            the name of the test to run	 */	public LoadStorePerfTest(String name) {		super(name);		String newTableType = System				.getProperty("com.mysql.jdbc.test.tabletype");		this.largeResults = "TRUE"				.equalsIgnoreCase(System						.getProperty("com.mysql.jdbc.testsuite.loadstoreperf.useBigResults"));		if ((newTableType != null) && (newTableType.length() > 0)) {			this.tableType = newTableType;			System.out.println("Using specified table type of '"					+ this.tableType + "'");		}	}	/**	 * Runs all tests in this test case	 * 	 * @param args	 *            ignored	 * 	 * @throws Exception	 *             if an error occurs	 */	public static void main(String[] args) throws Exception {		new LoadStorePerfTest("test1000Transactions").run();	}	/**	 * @see junit.framework.TestCase#setUp()	 */	public void setUp() throws Exception {		super.setUp();		try {			this.stmt.executeUpdate("DROP TABLE perfLoadStore");		} catch (SQLException sqlEx) {			// ignore		}		String dateTimeType = "DATETIME";		if (BaseTestCase.dbUrl.indexOf("oracle") != -1) {			dateTimeType = "TIMESTAMP";		}		//		// Approximate a run-of-the-mill entity in a business application		//		String query = "CREATE TABLE perfLoadStore (priKey INT NOT NULL, "				+ "fk1 INT NOT NULL, " + "fk2 INT NOT NULL, " + "dtField "				+ dateTimeType + ", " + "charField1 CHAR(32), "				+ "charField2 CHAR(32), " + "charField3 CHAR(32), "				+ "charField4 CHAR(32), " + "intField1 INT, "				+ "intField2 INT, " + "intField3 INT, " + "intField4 INT, "				+ "doubleField1 DECIMAL," + "doubleField2 DOUBLE,"				+ "doubleField3 DOUBLE," + "doubleField4 DOUBLE,"				+ "PRIMARY KEY (priKey))";		if (BaseTestCase.dbUrl.indexOf("mysql") != -1) {			query += (" TYPE=" + this.tableType);		}		this.stmt.executeUpdate(query);		String currentDateValue = "NOW()";		if (BaseTestCase.dbUrl.indexOf("sqlserver") != -1) {			currentDateValue = "GETDATE()";		}		if (BaseTestCase.dbUrl.indexOf("oracle") != -1) {			currentDateValue = "CURRENT_TIMESTAMP";		}		int numLoops = 1;		if (this.largeResults) {			numLoops = 32;		}		System.out.println("Inserting " + numLoops + " rows to retrieve...");		for (int i = 0; i < numLoops; i++) {			this.stmt.executeUpdate("INSERT INTO perfLoadStore (" + "priKey, "					+ "fk1, " + "fk2, " + "dtField, " + "charField1, "					+ "charField2, " + "charField3, " + "charField4, "					+ "intField1, " + "intField2, " + "intField3, "					+ "intField4, " + "doubleField1," + "doubleField2,"					+ "doubleField3," + "doubleField4" + ") VALUES (" + i + "," // priKey					+ "2," // fk1					+ "3," // fk2					+ currentDateValue + "," // dtField					+ "'0123456789ABCDEF0123456789ABCDEF'," // charField1					+ "'0123456789ABCDEF0123456789ABCDEF'," // charField2					+ "'0123456789ABCDEF0123456789ABCDEF'," // charField3					+ "'0123456789ABCDEF0123456789ABCDEF'," // charField4					+ "7," // intField1					+ "8," // intField2					+ "9," // intField3					+ "10," // intField4					+ "1.20," // doubleField1					+ "2.30," // doubleField2					+ "3.40," // doubleField3					+ "4.50" // doubleField4					+ ")");		}	}	/**	 * @see junit.framework.TestCase#tearDown()	 */	public void tearDown() throws Exception {		try {			this.stmt.executeUpdate("DROP TABLE perfLoadStore");		} catch (SQLException sqlEx) {			// ignore		}		super.tearDown();	}	/**	 * Tests and times 1000 load/store type transactions	 * 	 * @throws Exception	 *             if an error occurs	 */	public void test1000Transactions() throws Exception {		this.takeMeasurements = false;		warmUp();		this.takeMeasurements = true;		doIterations(29);		reportResults("\n\nResults for instance # 1: ");	}	/**	 * Runs one iteration of the test.	 * 	 * @see testsuite.perf.BasePerfTest#doOneIteration()	 */	protected void doOneIteration() throws Exception {		PreparedStatement pStmtStore = this.conn				.prepareStatement("UPDATE perfLoadStore SET " + "priKey = ?, "						+ "fk1 = ?, " + "fk2 = ?, " + "dtField = ?, "						+ "charField1 = ?, " + "charField2 = ?, "						+ "charField3 = ?, " + "charField4 = ?, "						+ "intField1 = ?, " + "intField2 = ?, "						+ "intField3 = ?, " + "intField4 = ?, "						+ "doubleField1 = ?," + "doubleField2 = ?,"						+ "doubleField3 = ?," + "doubleField4 = ?"						+ " WHERE priKey=?");		PreparedStatement pStmtCheck = this.conn				.prepareStatement("SELECT COUNT(*) FROM perfLoadStore WHERE priKey=?");		PreparedStatement pStmtLoad = null;		if (this.largeResults) {			pStmtLoad = this.conn.prepareStatement("SELECT " + "priKey, "					+ "fk1, " + "fk2, " + "dtField, " + "charField1, "					+ "charField2, " + "charField3, " + "charField4, "					+ "intField1, " + "intField2, " + "intField3, "					+ "intField4, " + "doubleField1," + "doubleField2, "					+ "doubleField3," + "doubleField4" + " FROM perfLoadStore");		} else {			pStmtLoad = this.conn.prepareStatement("SELECT " + "priKey, "					+ "fk1, " + "fk2, " + "dtField, " + "charField1, "					+ "charField2, " + "charField3, " + "charField4, "					+ "intField1, " + "intField2, " + "intField3, "					+ "intField4, " + "doubleField1," + "doubleField2, "					+ "doubleField3," + "doubleField4"					+ " FROM perfLoadStore WHERE priKey=?");		}		NumberFormat numFormatter = NumberFormat.getInstance();		numFormatter.setMaximumFractionDigits(4);		numFormatter.setMinimumFractionDigits(4);		int transactionCount = 5000;		if (this.largeResults) {			transactionCount = 50;		}		long begin = System.currentTimeMillis();		for (int i = 0; i < transactionCount; i++) {			this.conn.setAutoCommit(false);			pStmtCheck.setInt(1, 1);			this.rs = pStmtCheck.executeQuery();			while (this.rs.next()) {				this.rs.getInt(1);			}			this.rs.close();			if (!this.largeResults) {				pStmtLoad.setInt(1, 1);			}			this.rs = pStmtLoad.executeQuery();			if (this.rs.next()) {				int key = this.rs.getInt(1);				if (!this.useColumnNames) {					pStmtStore.setInt(1, key); // priKey					pStmtStore.setInt(2, this.rs.getInt(2)); // fk1					pStmtStore.setInt(3, this.rs.getInt(3)); // fk2					pStmtStore.setTimestamp(4, this.rs.getTimestamp(4)); // dtField					pStmtStore.setString(5, this.rs.getString(5)); // charField1					pStmtStore.setString(6, this.rs.getString(7)); // charField2					pStmtStore.setString(7, this.rs.getString(7)); // charField3					pStmtStore.setString(8, this.rs.getString(8)); // charField4					pStmtStore.setInt(9, this.rs.getInt(9)); // intField1					pStmtStore.setInt(10, this.rs.getInt(10)); // intField2					pStmtStore.setInt(11, this.rs.getInt(11)); // intField3					pStmtStore.setInt(12, this.rs.getInt(12)); // intField4					pStmtStore.setDouble(13, this.rs.getDouble(13)); // doubleField1					pStmtStore.setDouble(14, this.rs.getDouble(14)); // doubleField2					pStmtStore.setDouble(15, this.rs.getDouble(15)); // doubleField3					pStmtStore.setDouble(16, this.rs.getDouble(16)); // doubleField4					pStmtStore.setInt(17, key);				} else {					/*					 * "UPDATE perfLoadStore SET " + "priKey = ?, " + "fk1 = ?, " +					 * "fk2 = ?, " + "dtField = ?, " + "charField1 = ?, " +					 * "charField2 = ?, " + "charField3 = ?, " + "charField4 = ?, " +					 * "intField1 = ?, " + "intField2 = ?, " + "intField3 = ?, " +					 * "intField4 = ?, " + "doubleField1 = ?," + "doubleField2 =					 * ?," + "doubleField3 = ?," + "doubleField4 = ?" + " WHERE					 * priKey=?");					 */					pStmtStore.setInt(1, key); // priKey					pStmtStore.setInt(2, this.rs.getInt("fk1")); // fk1					pStmtStore.setInt(3, this.rs.getInt("fk2")); // fk2					pStmtStore.setTimestamp(4, this.rs.getTimestamp("dtField")); // dtField					pStmtStore.setString(5, this.rs.getString("charField1")); // charField1					pStmtStore.setString(6, this.rs.getString("charField2")); // charField2					pStmtStore.setString(7, this.rs.getString("charField3")); // charField3					pStmtStore.setString(8, this.rs.getString("charField4")); // charField4					pStmtStore.setInt(9, this.rs.getInt("intField1")); // intField1					pStmtStore.setInt(10, this.rs.getInt("intField2")); // intField2					pStmtStore.setInt(11, this.rs.getInt("intField3")); // intField3					pStmtStore.setInt(12, this.rs.getInt("intField4")); // intField4					pStmtStore.setDouble(13, this.rs.getDouble("doubleField1")); // doubleField1					pStmtStore.setDouble(14, this.rs.getDouble("doubleField2")); // doubleField2					pStmtStore.setDouble(15, this.rs.getDouble("doubleField3")); // doubleField3					pStmtStore.setDouble(16, this.rs.getDouble("doubleField4")); // doubleField4					pStmtStore.setInt(17, key);				}				pStmtStore.executeUpdate();			}			this.rs.close();			this.conn.commit();			this.conn.setAutoCommit(true);		}		pStmtStore.close();		pStmtCheck.close();		pStmtLoad.close();		long end = System.currentTimeMillis();		long timeElapsed = (end - begin);		double timeElapsedSeconds = (double) timeElapsed / 1000;		double tps = transactionCount / timeElapsedSeconds;		if (this.takeMeasurements) {			addResult(tps);			System.out.print("1 [ " + numFormatter.format(getMeanValue())					+ " ] ");		} else {			System.out.println("Warm-up: " + tps + " trans/sec");		}	}	/**	 * Runs the test 10 times to get JIT going, and GC going	 * 	 * @throws Exception	 *             if an error occurs.	 */	protected void warmUp() throws Exception {		try {			System.out.print("Warm-up period (10 iterations)");			for (int i = 0; i < 10; i++) {				doOneIteration();				System.out.print(".");			}			System.out.println();			System.out.println("Warm-up period ends");			System.out.println("\nUnits for this test are transactions/sec.");		} catch (Exception ex) {			ex.printStackTrace();			throw ex;		}	}}