package com.yammer.telemetry.agent.jdbc; import com.yammer.telemetry.tracing.Span; import com.yammer.telemetry.tracing.SpanHelper; import java.io.InputStream; import java.io.Reader; import java.math.BigDecimal; import java.net.URL; import java.sql.*; import java.util.Calendar; public class InstrumentedPreparedStatement extends InstrumentedStatement implements PreparedStatement { private final PreparedStatement underlying; public InstrumentedPreparedStatement(InstrumentedConnection connection, PreparedStatement underlying, String sql) { super(connection, underlying, sql); this.underlying = underlying; } @Override public ResultSet executeQuery() throws SQLException { try (Span ignored = SpanHelper.startSpan("Execute Query: " + getSql())) { return underlying.executeQuery(); } } @Override public int executeUpdate() throws SQLException { try (Span ignored = SpanHelper.startSpan("Execute Update: " + getSql())) { return underlying.executeUpdate(); } } @Override public void setNull(int parameterIndex, int sqlType) throws SQLException { underlying.setNull(parameterIndex, sqlType); } @Override public void setBoolean(int parameterIndex, boolean x) throws SQLException { underlying.setBoolean(parameterIndex, x); } @Override public void setByte(int parameterIndex, byte x) throws SQLException { underlying.setByte(parameterIndex, x); } @Override public void setShort(int parameterIndex, short x) throws SQLException { underlying.setShort(parameterIndex, x); } @Override public void setInt(int parameterIndex, int x) throws SQLException { underlying.setInt(parameterIndex, x); } @Override public void setLong(int parameterIndex, long x) throws SQLException { underlying.setLong(parameterIndex, x); } @Override public void setFloat(int parameterIndex, float x) throws SQLException { underlying.setFloat(parameterIndex, x); } @Override public void setDouble(int parameterIndex, double x) throws SQLException { underlying.setDouble(parameterIndex, x); } @Override public void setBigDecimal(int parameterIndex, BigDecimal x) throws SQLException { underlying.setBigDecimal(parameterIndex, x); } @Override public void setString(int parameterIndex, String x) throws SQLException { underlying.setString(parameterIndex, x); } @Override public void setBytes(int parameterIndex, byte[] x) throws SQLException { underlying.setBytes(parameterIndex, x); } @Override public void setDate(int parameterIndex, Date x) throws SQLException { underlying.setDate(parameterIndex, x); } @Override public void setTime(int parameterIndex, Time x) throws SQLException { underlying.setTime(parameterIndex, x); } @Override public void setTimestamp(int parameterIndex, Timestamp x) throws SQLException { underlying.setTimestamp(parameterIndex, x); } @Override public void setAsciiStream(int parameterIndex, InputStream x, int length) throws SQLException { underlying.setAsciiStream(parameterIndex, x, length); } @Override public void setUnicodeStream(int parameterIndex, InputStream x, int length) throws SQLException { underlying.setUnicodeStream(parameterIndex, x, length); } @Override public void setBinaryStream(int parameterIndex, InputStream x, int length) throws SQLException { underlying.setBinaryStream(parameterIndex, x, length); } @Override public void clearParameters() throws SQLException { underlying.clearParameters(); } @Override public void setObject(int parameterIndex, Object x, int targetSqlType) throws SQLException { underlying.setObject(parameterIndex, x, targetSqlType); } @Override public void setObject(int parameterIndex, Object x) throws SQLException { underlying.setObject(parameterIndex, x); } @Override public boolean execute() throws SQLException { try (Span ignored = SpanHelper.startSpan("Execute: " + getSql())) { return underlying.execute(); } } @Override public void addBatch() throws SQLException { underlying.addBatch(); } @Override public void setCharacterStream(int parameterIndex, Reader reader, int length) throws SQLException { underlying.setCharacterStream(parameterIndex, reader, length); } @Override public void setRef(int parameterIndex, Ref x) throws SQLException { underlying.setRef(parameterIndex, x); } @Override public void setBlob(int parameterIndex, Blob x) throws SQLException { underlying.setBlob(parameterIndex, x); } @Override public void setClob(int parameterIndex, Clob x) throws SQLException { underlying.setClob(parameterIndex, x); } @Override public void setArray(int parameterIndex, Array x) throws SQLException { underlying.setArray(parameterIndex, x); } @Override public ResultSetMetaData getMetaData() throws SQLException { return underlying.getMetaData(); } @Override public void setDate(int parameterIndex, Date x, Calendar cal) throws SQLException { underlying.setDate(parameterIndex, x, cal); } @Override public void setTime(int parameterIndex, Time x, Calendar cal) throws SQLException { underlying.setTime(parameterIndex, x, cal); } @Override public void setTimestamp(int parameterIndex, Timestamp x, Calendar cal) throws SQLException { underlying.setTimestamp(parameterIndex, x, cal); } @Override public void setNull(int parameterIndex, int sqlType, String typeName) throws SQLException { underlying.setNull(parameterIndex, sqlType, typeName); } @Override public void setURL(int parameterIndex, URL x) throws SQLException { underlying.setURL(parameterIndex, x); } @Override public ParameterMetaData getParameterMetaData() throws SQLException { return underlying.getParameterMetaData(); } @Override public void setRowId(int parameterIndex, RowId x) throws SQLException { underlying.setRowId(parameterIndex, x); } @Override public void setNString(int parameterIndex, String value) throws SQLException { underlying.setNString(parameterIndex, value); } @Override public void setNCharacterStream(int parameterIndex, Reader value, long length) throws SQLException { underlying.setNCharacterStream(parameterIndex, value, length); } @Override public void setNClob(int parameterIndex, NClob value) throws SQLException { underlying.setNClob(parameterIndex, value); } @Override public void setClob(int parameterIndex, Reader reader, long length) throws SQLException { underlying.setClob(parameterIndex, reader, length); } @Override public void setBlob(int parameterIndex, InputStream inputStream, long length) throws SQLException { underlying.setBlob(parameterIndex, inputStream, length); } @Override public void setNClob(int parameterIndex, Reader reader, long length) throws SQLException { underlying.setNClob(parameterIndex, reader, length); } @Override public void setSQLXML(int parameterIndex, SQLXML xmlObject) throws SQLException { underlying.setSQLXML(parameterIndex, xmlObject); } @Override public void setObject(int parameterIndex, Object x, int targetSqlType, int scaleOrLength) throws SQLException { underlying.setObject(parameterIndex, x, targetSqlType, scaleOrLength); } @Override public void setAsciiStream(int parameterIndex, InputStream x, long length) throws SQLException { underlying.setAsciiStream(parameterIndex, x, length); } @Override public void setBinaryStream(int parameterIndex, InputStream x, long length) throws SQLException { underlying.setBinaryStream(parameterIndex, x, length); } @Override public void setCharacterStream(int parameterIndex, Reader reader, long length) throws SQLException { underlying.setCharacterStream(parameterIndex, reader, length); } @Override public void setAsciiStream(int parameterIndex, InputStream x) throws SQLException { underlying.setAsciiStream(parameterIndex, x); } @Override public void setBinaryStream(int parameterIndex, InputStream x) throws SQLException { underlying.setBinaryStream(parameterIndex, x); } @Override public void setCharacterStream(int parameterIndex, Reader reader) throws SQLException { underlying.setCharacterStream(parameterIndex, reader); } @Override public void setNCharacterStream(int parameterIndex, Reader value) throws SQLException { underlying.setNCharacterStream(parameterIndex, value); } @Override public void setClob(int parameterIndex, Reader reader) throws SQLException { underlying.setClob(parameterIndex, reader); } @Override public void setBlob(int parameterIndex, InputStream inputStream) throws SQLException { underlying.setBlob(parameterIndex, inputStream); } @Override public void setNClob(int parameterIndex, Reader reader) throws SQLException { underlying.setNClob(parameterIndex, reader); } }