diff --git a/jOOQ-sql-console/src/main/java/org/jooq/debugger/SqlQueryDebuggerExecuteListener.java b/jOOQ-sql-console/src/main/java/org/jooq/debugger/SqlQueryDebuggerExecuteListener.java index 7ade47472b..edc3db6c0b 100644 --- a/jOOQ-sql-console/src/main/java/org/jooq/debugger/SqlQueryDebuggerExecuteListener.java +++ b/jOOQ-sql-console/src/main/java/org/jooq/debugger/SqlQueryDebuggerExecuteListener.java @@ -36,6 +36,7 @@ */ package org.jooq.debugger; +import java.sql.CallableStatement; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.util.List; @@ -78,10 +79,9 @@ public class SqlQueryDebuggerExecuteListener extends DefaultExecuteListener { } aggregatedPreparationDuration += System.currentTimeMillis() - startPreparationTime; PreparedStatement statement = ctx.statement(); - // TODO: How to wrap CallableStatement ? - // Wrapping it with UsageTrackingPreparedStatement will cause - // ClassCastExceptions in jOOQ. Maybe there is a design issue? - if (ctx.type() != ExecuteType.ROUTINE) { + if (ctx.type() == ExecuteType.ROUTINE) { + ctx.statement(new UsageTrackingCallableStatement((CallableStatement)statement)); + } else { ctx.statement(new UsageTrackingPreparedStatement(statement)); } } diff --git a/jOOQ-sql-console/src/main/java/org/jooq/debugger/UsageTrackingCallableStatement.java b/jOOQ-sql-console/src/main/java/org/jooq/debugger/UsageTrackingCallableStatement.java new file mode 100644 index 0000000000..bb58b22afc --- /dev/null +++ b/jOOQ-sql-console/src/main/java/org/jooq/debugger/UsageTrackingCallableStatement.java @@ -0,0 +1,625 @@ +/** + * Copyright (c) 2009-2012, Lukas Eder, lukas.eder@gmail.com + * Christopher Deckers, chrriis@gmail.com + * All rights reserved. + * + * This software is licensed to you under the Apache License, Version 2.0 + * (the "License"); You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * . Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * . Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * . Neither the name "jOOQ" nor the names of its contributors may be + * used to endorse or promote products derived from this software without + * specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ +package org.jooq.debugger; + +import java.io.InputStream; +import java.io.Reader; +import java.math.BigDecimal; +import java.net.URL; +import java.sql.Array; +import java.sql.Blob; +import java.sql.CallableStatement; +import java.sql.Clob; +import java.sql.Date; +import java.sql.NClob; +import java.sql.Ref; +import java.sql.RowId; +import java.sql.SQLException; +import java.sql.SQLXML; +import java.sql.Time; +import java.sql.Timestamp; +import java.util.Calendar; +import java.util.Map; + +/** + * @author Christopher Deckers + */ +public class UsageTrackingCallableStatement extends UsageTrackingPreparedStatement implements CallableStatement { + + private CallableStatement stmt; + + public UsageTrackingCallableStatement(CallableStatement stmt) { + super(stmt); + this.stmt = stmt; + } + + @Override + public void registerOutParameter(int parameterIndex, int sqlType) throws SQLException{ + stmt.registerOutParameter(parameterIndex, sqlType); + } + + @Override + public void registerOutParameter(int parameterIndex, int sqlType, int scale) throws SQLException { + stmt.registerOutParameter(parameterIndex, sqlType, scale); + } + + @Override + public boolean wasNull() throws SQLException { + return stmt.wasNull(); + } + + @Override + public String getString(int parameterIndex) throws SQLException { + return stmt.getString(parameterIndex); + } + + @Override + public boolean getBoolean(int parameterIndex) throws SQLException { + return stmt.getBoolean(parameterIndex); + } + + @Override + public byte getByte(int parameterIndex) throws SQLException { + return stmt.getByte(parameterIndex); + } + + @Override + public short getShort(int parameterIndex) throws SQLException { + return stmt.getShort(parameterIndex); + } + + @Override + public int getInt(int parameterIndex) throws SQLException { + return stmt.getInt(parameterIndex); + } + + @Override + public long getLong(int parameterIndex) throws SQLException { + return stmt.getLong(parameterIndex); + } + + @Override + public float getFloat(int parameterIndex) throws SQLException { + return stmt.getFloat(parameterIndex); + } + + @Override + public double getDouble(int parameterIndex) throws SQLException { + return stmt.getDouble(parameterIndex); + } + + @Override + public BigDecimal getBigDecimal(int parameterIndex, int scale) throws SQLException { + return stmt.getBigDecimal(parameterIndex, scale); + } + + @Override + public byte[] getBytes(int parameterIndex) throws SQLException { + return stmt.getBytes(parameterIndex); + } + + @Override + public Date getDate(int parameterIndex) throws SQLException { + return stmt.getDate(parameterIndex); + } + + @Override + public Time getTime(int parameterIndex) throws SQLException { + return stmt.getTime(parameterIndex); + } + + @Override + public Timestamp getTimestamp(int parameterIndex) throws SQLException { + return stmt.getTimestamp(parameterIndex); + } + + @Override + public Object getObject(int parameterIndex) throws SQLException { + return stmt.getObject(parameterIndex); + } + + @Override + public BigDecimal getBigDecimal(int parameterIndex) throws SQLException { + return stmt.getBigDecimal(parameterIndex); + } + + @Override + public Object getObject(int parameterIndex, Map> map) throws SQLException { + return stmt.getObject(parameterIndex, map); + } + + @Override + public Ref getRef(int parameterIndex) throws SQLException { + return stmt.getRef(parameterIndex); + } + + @Override + public Blob getBlob(int parameterIndex) throws SQLException { + return stmt.getBlob(parameterIndex); + } + + @Override + public Clob getClob(int parameterIndex) throws SQLException { + return stmt.getClob(parameterIndex); + } + + @Override + public Array getArray(int parameterIndex) throws SQLException { + return stmt.getArray(parameterIndex); + } + + @Override + public Date getDate(int parameterIndex, Calendar cal) throws SQLException { + return stmt.getDate(parameterIndex, cal); + } + + @Override + public Time getTime(int parameterIndex, Calendar cal) throws SQLException { + return stmt.getTime(parameterIndex, cal); + } + + @Override + public Timestamp getTimestamp(int parameterIndex, Calendar cal) throws SQLException { + return stmt.getTimestamp(parameterIndex, cal); + } + + @Override + public void registerOutParameter(int parameterIndex, int sqlType, String typeName) throws SQLException { + stmt.registerOutParameter(parameterIndex, sqlType, typeName); + } + + @Override + public void registerOutParameter(String parameterName, int sqlType) throws SQLException { + stmt.registerOutParameter(parameterName, sqlType); + } + + @Override + public void registerOutParameter(String parameterName, int sqlType, int scale) throws SQLException { + stmt.registerOutParameter(parameterName, sqlType, scale); + } + + @Override + public void registerOutParameter(String parameterName, int sqlType, String typeName) throws SQLException { + stmt.registerOutParameter(parameterName, sqlType, typeName); + } + + @Override + public URL getURL(int parameterIndex) throws SQLException { + return stmt.getURL(parameterIndex); + } + + @Override + public void setURL(String parameterName, URL val) throws SQLException { + stmt.setURL(parameterName, val); + } + + @Override + public void setNull(String parameterName, int sqlType) throws SQLException { + stmt.setNull(parameterName, sqlType); + } + + @Override + public void setBoolean(String parameterName, boolean x) throws SQLException { + stmt.setBoolean(parameterName, x); + } + + @Override + public void setByte(String parameterName, byte x) throws SQLException { + stmt.setByte(parameterName, x); + } + + @Override + public void setShort(String parameterName, short x) throws SQLException { + stmt.setShort(parameterName, x); + } + + @Override + public void setInt(String parameterName, int x) throws SQLException { + stmt.setInt(parameterName, x); + } + + @Override + public void setLong(String parameterName, long x) throws SQLException { + stmt.setLong(parameterName, x); + } + + @Override + public void setFloat(String parameterName, float x) throws SQLException { + stmt.setFloat(parameterName, x); + } + + @Override + public void setDouble(String parameterName, double x) throws SQLException { + stmt.setDouble(parameterName, x); + } + + @Override + public void setBigDecimal(String parameterName, BigDecimal x) throws SQLException { + stmt.setBigDecimal(parameterName, x); + } + + @Override + public void setString(String parameterName, String x) throws SQLException { + stmt.setString(parameterName, x); + } + + @Override + public void setBytes(String parameterName, byte[] x) throws SQLException { + stmt.setBytes(parameterName, x); + } + + @Override + public void setDate(String parameterName, Date x) throws SQLException { + stmt.setDate(parameterName, x); + } + + @Override + public void setTime(String parameterName, Time x) throws SQLException { + stmt.setTime(parameterName, x); + } + + @Override + public void setTimestamp(String parameterName, Timestamp x) throws SQLException { + stmt.setTimestamp(parameterName, x); + } + + @Override + public void setAsciiStream(String parameterName, InputStream x, int length) throws SQLException { + stmt.setAsciiStream(parameterName, x, length); + } + + @Override + public void setBinaryStream(String parameterName, InputStream x, int length) throws SQLException { + stmt.setBinaryStream(parameterName, x, length); + } + + @Override + public void setObject(String parameterName, Object x, int targetSqlType, int scale) throws SQLException { + stmt.setObject(parameterName, x, targetSqlType, scale); + } + + @Override + public void setObject(String parameterName, Object x, int targetSqlType) throws SQLException { + stmt.setObject(parameterName, x, targetSqlType); + } + + @Override + public void setObject(String parameterName, Object x) throws SQLException { + stmt.setObject(parameterName, x); + } + + @Override + public void setCharacterStream(String parameterName, Reader reader, int length) throws SQLException { + stmt.setCharacterStream(parameterName, reader, length); + } + + @Override + public void setDate(String parameterName, Date x, Calendar cal) throws SQLException { + stmt.setDate(parameterName, x, cal); + } + + @Override + public void setTime(String parameterName, Time x, Calendar cal) throws SQLException { + stmt.setTime(parameterName, x, cal); + } + + @Override + public void setTimestamp(String parameterName, Timestamp x, Calendar cal) throws SQLException { + stmt.setTimestamp(parameterName, x, cal); + } + + @Override + public void setNull(String parameterName, int sqlType, String typeName) throws SQLException { + stmt.setNull(parameterName, sqlType, typeName); + } + + @Override + public String getString(String parameterName) throws SQLException { + return stmt.getString(parameterName); + } + + @Override + public boolean getBoolean(String parameterName) throws SQLException { + return stmt.getBoolean(parameterName); + } + + @Override + public byte getByte(String parameterName) throws SQLException { + return stmt.getByte(parameterName); + } + + @Override + public short getShort(String parameterName) throws SQLException { + return stmt.getShort(parameterName); + } + + @Override + public int getInt(String parameterName) throws SQLException { + return stmt.getInt(parameterName); + } + + @Override + public long getLong(String parameterName) throws SQLException { + return stmt.getLong(parameterName); + } + + @Override + public float getFloat(String parameterName) throws SQLException { + return stmt.getFloat(parameterName); + } + + @Override + public double getDouble(String parameterName) throws SQLException { + return stmt.getDouble(parameterName); + } + + @Override + public byte[] getBytes(String parameterName) throws SQLException { + return stmt.getBytes(parameterName); + } + + @Override + public Date getDate(String parameterName) throws SQLException { + return stmt.getDate(parameterName); + } + + @Override + public Time getTime(String parameterName) throws SQLException { + return stmt.getTime(parameterName); + } + + @Override + public Timestamp getTimestamp(String parameterName) throws SQLException { + return stmt.getTimestamp(parameterName); + } + + @Override + public Object getObject(String parameterName) throws SQLException { + return stmt.getObject(parameterName); + } + + @Override + public BigDecimal getBigDecimal(String parameterName) throws SQLException { + return stmt.getBigDecimal(parameterName); + } + + @Override + public Object getObject(String parameterName, Map> map) throws SQLException { + return stmt.getObject(parameterName, map); + } + + @Override + public Ref getRef(String parameterName) throws SQLException { + return stmt.getRef(parameterName); + } + + @Override + public Blob getBlob(String parameterName) throws SQLException { + return stmt.getBlob(parameterName); + } + + @Override + public Clob getClob(String parameterName) throws SQLException { + return stmt.getClob(parameterName); + } + + @Override + public Array getArray(String parameterName) throws SQLException { + return stmt.getArray(parameterName); + } + + @Override + public Date getDate(String parameterName, Calendar cal) throws SQLException { + return stmt.getDate(parameterName); + } + + @Override + public Time getTime(String parameterName, Calendar cal) throws SQLException { + return stmt.getTime(parameterName, cal); + } + + @Override + public Timestamp getTimestamp(String parameterName, Calendar cal) throws SQLException { + return stmt.getTimestamp(parameterName, cal); + } + + @Override + public URL getURL(String parameterName) throws SQLException { + return stmt.getURL(parameterName); + } + + @Override + public RowId getRowId(int parameterIndex) throws SQLException { + return stmt.getRowId(parameterIndex); + } + + @Override + public RowId getRowId(String parameterName) throws SQLException { + return stmt.getRowId(parameterName); + } + + @Override + public void setRowId(String parameterName, RowId x) throws SQLException { + stmt.setRowId(parameterName, x); + } + + @Override + public void setNString(String parameterName, String value) throws SQLException { + stmt.setNString(parameterName, value); + } + + @Override + public void setNCharacterStream(String parameterName, Reader value, long length) throws SQLException { + stmt.setNCharacterStream(parameterName, value, length); + } + + @Override + public void setNClob(String parameterName, NClob value) throws SQLException { + stmt.setNClob(parameterName, value); + } + + @Override + public void setClob(String parameterName, Reader reader, long length) throws SQLException { + stmt.setClob(parameterName, reader, length); + } + + @Override + public void setBlob(String parameterName, InputStream inputStream, long length) throws SQLException { + stmt.setBlob(parameterName, inputStream, length); + } + + @Override + public void setNClob(String parameterName, Reader reader, long length) throws SQLException { + stmt.setNClob(parameterName, reader, length); + } + + @Override + public NClob getNClob(int parameterIndex) throws SQLException { + return stmt.getNClob(parameterIndex); + } + + @Override + public NClob getNClob(String parameterName) throws SQLException { + return stmt.getNClob(parameterName); + } + + @Override + public void setSQLXML(String parameterName, SQLXML xmlObject) throws SQLException { + stmt.setSQLXML(parameterName, xmlObject); + } + + @Override + public SQLXML getSQLXML(int parameterIndex) throws SQLException { + return stmt.getSQLXML(parameterIndex); + } + + @Override + public SQLXML getSQLXML(String parameterName) throws SQLException { + return stmt.getSQLXML(parameterName); + } + + @Override + public String getNString(int parameterIndex) throws SQLException { + return stmt.getNString(parameterIndex); + } + + @Override + public String getNString(String parameterName) throws SQLException { + return stmt.getNString(parameterName); + } + + @Override + public Reader getNCharacterStream(int parameterIndex) throws SQLException { + return stmt.getNCharacterStream(parameterIndex); + } + + @Override + public Reader getNCharacterStream(String parameterName) throws SQLException { + return stmt.getNCharacterStream(parameterName); + } + + @Override + public Reader getCharacterStream(int parameterIndex) throws SQLException { + return stmt.getCharacterStream(parameterIndex); + } + + @Override + public Reader getCharacterStream(String parameterName) throws SQLException { + return stmt.getCharacterStream(parameterName); + } + + @Override + public void setBlob(String parameterName, Blob x) throws SQLException { + stmt.setBlob(parameterName, x); + } + + @Override + public void setClob(String parameterName, Clob x) throws SQLException { + stmt.setClob(parameterName, x); + } + + @Override + public void setAsciiStream(String parameterName, InputStream x, long length) throws SQLException { + stmt.setAsciiStream(parameterName, x, length); + } + + @Override + public void setBinaryStream(String parameterName, InputStream x, long length) throws SQLException { + stmt.setBinaryStream(parameterName, x, length); + } + + @Override + public void setCharacterStream(String parameterName, Reader reader, long length) throws SQLException { + stmt.setCharacterStream(parameterName, reader, length); + } + + @Override + public void setAsciiStream(String parameterName, InputStream x) throws SQLException { + stmt.setAsciiStream(parameterName, x); + } + + @Override + public void setBinaryStream(String parameterName, InputStream x) throws SQLException { + stmt.setBinaryStream(parameterName, x); + } + + @Override + public void setCharacterStream(String parameterName, Reader reader) throws SQLException { + stmt.setCharacterStream(parameterName, reader); + } + + @Override + public void setNCharacterStream(String parameterName, Reader value) throws SQLException { + stmt.setNCharacterStream(parameterName, value); + } + + @Override + public void setClob(String parameterName, Reader reader) throws SQLException { + stmt.setClob(parameterName, reader); + } + + @Override + public void setBlob(String parameterName, InputStream inputStream) throws SQLException { + stmt.setBlob(parameterName, inputStream); + } + + @Override + public void setNClob(String parameterName, Reader reader) throws SQLException { + stmt.setNClob(parameterName, reader); + } + +} diff --git a/jOOQ-sql-console/src/main/java/org/jooq/debugger/UsageTrackingPreparedStatement.java b/jOOQ-sql-console/src/main/java/org/jooq/debugger/UsageTrackingPreparedStatement.java index f9e7f654b8..b9c7056360 100644 --- a/jOOQ-sql-console/src/main/java/org/jooq/debugger/UsageTrackingPreparedStatement.java +++ b/jOOQ-sql-console/src/main/java/org/jooq/debugger/UsageTrackingPreparedStatement.java @@ -61,6 +61,9 @@ import java.util.ArrayList; import java.util.Calendar; import java.util.List; +/** + * @author Christopher Deckers + */ public class UsageTrackingPreparedStatement implements PreparedStatement { private PreparedStatement stmt; @@ -642,7 +645,7 @@ public class UsageTrackingPreparedStatement implements PreparedStatement { sb.append("]"); } - private void logValue(int parameterIndex, Object o) { + protected void logValue(int parameterIndex, Object o) { // parameters in SQL are 1-based int missingLength = parameterIndex - paramList.size(); for(int i=0; i