[patch] add the possibility of issuing commit and rollback toward the database

JDBCSampler.patch (text/plain), 5.65 KB, created by Rubén Laguna on 2007-04-25 01:27:31 UTC

(hide)

Description: add the possibility of issuing commit and rollback toward the database

Filename:

MIME Type:

Creator: Rubén Laguna

Created: 2007-04-25 01:27:31 UTC

Size: 5.65 KB

patch

obsolete

>Index: C:/eclipse/eclipse322workspaces/eclipsercp/jmeter/src/protocol/jdbc/org/apache/jmeter/protocol/jdbc/sampler/JDBCSampler.java >=================================================================== >--- C:/eclipse/eclipse322workspaces/eclipsercp/jmeter/src/protocol/jdbc/org/apache/jmeter/protocol/jdbc/sampler/JDBCSampler.java (revision 532089) >+++ C:/eclipse/eclipse322workspaces/eclipsercp/jmeter/src/protocol/jdbc/org/apache/jmeter/protocol/jdbc/sampler/JDBCSampler.java (working copy) >@@ -147,47 +147,67 @@ > > // Based on query return value, get results > String _queryType = getQueryType(); >- if (SELECT.equals(_queryType)) { >- stmt = conn.createStatement(); >- ResultSet rs = null; >- try { >- rs = stmt.executeQuery(getQuery()); >- Data data = getDataFromResultSet(rs); >- res.setResponseData(data.toString().getBytes()); >- } finally { >- close(rs); >- } >- } else if (CALLABLE.equals(_queryType)) { >- CallableStatement cstmt = getCallableStatement(conn); >- setArguments(cstmt); >- // A CallableStatement can return more than 1 ResultSets >- // plus a number of update counts. >- boolean hasResultSet = cstmt.execute(); >- String sb = resultSetsToString(cstmt,hasResultSet); >- res.setResponseData(sb.toString().getBytes()); >- } else if (UPDATE.equals(_queryType)) { >- stmt = conn.createStatement(); >- stmt.executeUpdate(getQuery()); >- int updateCount = stmt.getUpdateCount(); >- String results = updateCount + " updates"; >- res.setResponseData(results.getBytes()); >- } else if (PREPARED_SELECT.equals(_queryType)) { >- PreparedStatement pstmt = getPreparedStatement(conn); >- setArguments(pstmt); >- pstmt.executeQuery(); >- String sb = resultSetsToString(pstmt,true); >- res.setResponseData(sb.toString().getBytes()); >- } else if (PREPARED_UPDATE.equals(_queryType)) { >- PreparedStatement pstmt = getPreparedStatement(conn); >- setArguments(pstmt); >- pstmt.executeUpdate(); >- String sb = resultSetsToString(pstmt,false); >- res.setResponseData(sb.toString().getBytes()); >- } else { // User provided incorrect query type >- String results="Unexpected query type: "+_queryType; >- res.setResponseMessage(results); >- res.setSuccessful(false); >- } >+ final String queryString = getQuery().trim().toLowerCase(); >+ if (queryString.startsWith("rollback")) { //rollback or rollback; >+ if (!conn.getAutoCommit()) { >+ conn.rollback(); >+ res.setResponseData("explicit transaction rollback".getBytes()); >+ } else { >+ res.setResponseData("explicit transaction rollback requested but the connection is in auto-commit mode. skipping it".getBytes()); >+ res.setSuccessful(false); >+ } >+ } else if(queryString.startsWith("commit")) { // commit or commit; >+ if (!conn.getAutoCommit()) { >+ conn.commit(); >+ res.setResponseData("explicit transaction commit".getBytes()); >+ } else { >+ res.setResponseData("explicit transaction commit requested but the connection is in auto-commit mode. skipping it".getBytes()); >+ res.setSuccessful(false); >+ } >+ } else { >+ //only if we are not in the case of a commit or a rollback. >+ if (SELECT.equals(_queryType)) { >+ stmt = conn.createStatement(); >+ ResultSet rs = null; >+ try { >+ rs = stmt.executeQuery(getQuery()); >+ Data data = getDataFromResultSet(rs); >+ res.setResponseData(data.toString().getBytes()); >+ } finally { >+ close(rs); >+ } >+ } else if (CALLABLE.equals(_queryType)) { >+ CallableStatement cstmt = getCallableStatement(conn); >+ setArguments(cstmt); >+ // A CallableStatement can return more than 1 ResultSets >+ // plus a number of update counts. >+ boolean hasResultSet = cstmt.execute(); >+ String sb = resultSetsToString(cstmt,hasResultSet); >+ res.setResponseData(sb.toString().getBytes()); >+ } else if (UPDATE.equals(_queryType)) { >+ stmt = conn.createStatement(); >+ stmt.executeUpdate(getQuery()); >+ int updateCount = stmt.getUpdateCount(); >+ String results = updateCount + " updates"; >+ res.setResponseData(results.getBytes()); >+ } else if (PREPARED_SELECT.equals(_queryType)) { >+ PreparedStatement pstmt = getPreparedStatement(conn); >+ setArguments(pstmt); >+ pstmt.executeQuery(); >+ String sb = resultSetsToString(pstmt,true); >+ res.setResponseData(sb.toString().getBytes()); >+ } else if (PREPARED_UPDATE.equals(_queryType)) { >+ PreparedStatement pstmt = getPreparedStatement(conn); >+ setArguments(pstmt); >+ pstmt.executeUpdate(); >+ String sb = resultSetsToString(pstmt,false); >+ res.setResponseData(sb.toString().getBytes()); >+ } else { // User provided incorrect query type >+ String results="Unexpected query type: "+_queryType; >+ res.setResponseMessage(results); >+ res.setSuccessful(false); >+ } >+ } > > } catch (SQLException ex) { > log.error("Error in JDBC sampling", ex); View Attachment As Diff View Attachment As Raw

This is ASF Bugzilla: the Apache Software Foundation bug system. In case of problems with the functioning of ASF Bugzilla, please contact bugzilla-admin@apache.org. Please Note: this e-mail address is only for reporting problems with ASF Bugzilla. Mail about any other subject will be silently ignored.

• • Home
  • | New
  • | Browse
  • | Search
  • |
    [?]
  • | Reports
  • | Help
  • | New Account
  • | Log In
    Remember [x]
  • | Forgot Password
    Login: [x]