Skip to content

Commit

Permalink
Merge a50cd6f into b7eb2d0
Browse files Browse the repository at this point in the history
  • Loading branch information
@RutgerLubbers
RutgerLubbers committed Feb 17, 2020
2 parents b7eb2d0 + a50cd6f commit 4e7d611
Show file tree
Hide file tree
Showing 4 changed files with 370 additions and 345 deletions.
348 changes: 6 additions & 342 deletions src/main/java/io/opentracing/contrib/jdbc/TracingConnection.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,349 +13,13 @@
*/
package io.opentracing.contrib.jdbc;

import java.sql.Array;
import java.sql.Blob;
import java.sql.CallableStatement;
import java.sql.Clob;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.NClob;
import java.sql.PreparedStatement;
import java.sql.SQLClientInfoException;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.sql.SQLXML;
import java.sql.Savepoint;
import java.sql.Statement;
import java.sql.Struct;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.concurrent.Executor;

import io.opentracing.Tracer;
import io.opentracing.contrib.common.WrapperProxy;

public class TracingConnection implements Connection {
private final Connection connection;
private final ConnectionInfo connectionInfo;
private final boolean withActiveSpanOnly;
private final Set<String> ignoredStatements;
private final Tracer tracer;

public TracingConnection(Connection connection, ConnectionInfo connectionInfo,
boolean withActiveSpanOnly, Set<String> ignoredStatements, Tracer tracer) {
this.connection = connection;
this.connectionInfo = connectionInfo;
this.withActiveSpanOnly = withActiveSpanOnly;
this.ignoredStatements = ignoredStatements;
this.tracer = tracer;
}

@Override
public Statement createStatement() throws SQLException {
final Statement statement = connection.createStatement();
return WrapperProxy
.wrap(statement, new TracingStatement(statement, connectionInfo, withActiveSpanOnly,
ignoredStatements, tracer));
}

@Override
public PreparedStatement prepareStatement(String sql) throws SQLException {
final PreparedStatement statement = connection.prepareStatement(sql);
return WrapperProxy.wrap(statement, new TracingPreparedStatement(statement, sql, connectionInfo,
withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public CallableStatement prepareCall(String sql) throws SQLException {
final CallableStatement statement = connection.prepareCall(sql);
return WrapperProxy.wrap(statement,
new TracingCallableStatement(connection.prepareCall(sql), sql, connectionInfo,
withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public String nativeSQL(String sql) throws SQLException {
return connection.nativeSQL(sql);
}

@Override
public void setAutoCommit(boolean autoCommit) throws SQLException {
connection.setAutoCommit(autoCommit);
}

@Override
public boolean getAutoCommit() throws SQLException {
return connection.getAutoCommit();
}

@Override
public void commit() throws SQLException {
connection.commit();
}

@Override
public void rollback() throws SQLException {
connection.rollback();
}

@Override
public void close() throws SQLException {
connection.close();
}

@Override
public boolean isClosed() throws SQLException {
return connection.isClosed();
}

@Override
public DatabaseMetaData getMetaData() throws SQLException {
return connection.getMetaData();
}

@Override
public void setReadOnly(boolean readOnly) throws SQLException {
connection.setReadOnly(readOnly);
}

@Override
public boolean isReadOnly() throws SQLException {
return connection.isReadOnly();
}

@Override
public void setCatalog(String catalog) throws SQLException {
connection.setCatalog(catalog);
}

@Override
public String getCatalog() throws SQLException {
return connection.getCatalog();
}

@Override
public void setTransactionIsolation(int level) throws SQLException {
connection.setTransactionIsolation(level);
}

@Override
public int getTransactionIsolation() throws SQLException {
return connection.getTransactionIsolation();
}

@Override
public SQLWarning getWarnings() throws SQLException {
return connection.getWarnings();
}

@Override
public void clearWarnings() throws SQLException {
connection.clearWarnings();
}

@Override
public Statement createStatement(int resultSetType, int resultSetConcurrency)
throws SQLException {
final Statement statement = connection.createStatement(resultSetType, resultSetConcurrency);
return WrapperProxy.wrap(statement, new TracingStatement(statement,
connectionInfo, withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public PreparedStatement prepareStatement(String sql, int resultSetType, int resultSetConcurrency)
throws SQLException {
final PreparedStatement statement = connection
.prepareStatement(sql, resultSetType, resultSetConcurrency);
return WrapperProxy.wrap(statement, new TracingPreparedStatement(statement, sql, connectionInfo,
withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public CallableStatement prepareCall(String sql, int resultSetType, int resultSetConcurrency)
throws SQLException {
final CallableStatement statement = connection
.prepareCall(sql, resultSetType, resultSetConcurrency);
return WrapperProxy.wrap(statement, new TracingCallableStatement(statement, sql, connectionInfo,
withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public Map<String, Class<?>> getTypeMap() throws SQLException {
return connection.getTypeMap();
}

@Override
public void setTypeMap(Map<String, Class<?>> map) throws SQLException {
connection.setTypeMap(map);
}

@Override
public void setHoldability(int holdability) throws SQLException {
connection.setHoldability(holdability);
}

@Override
public int getHoldability() throws SQLException {
return connection.getHoldability();
}

@Override
public Savepoint setSavepoint() throws SQLException {
return connection.setSavepoint();
}

@Override
public Savepoint setSavepoint(String name) throws SQLException {
return connection.setSavepoint(name);
}

@Override
public void rollback(Savepoint savepoint) throws SQLException {
connection.rollback(savepoint);
}

@Override
public void releaseSavepoint(Savepoint savepoint) throws SQLException {
connection.releaseSavepoint(savepoint);
}

@Override
public Statement createStatement(int resultSetType, int resultSetConcurrency,
int resultSetHoldability) throws SQLException {
final Statement statement = connection
.createStatement(resultSetType, resultSetConcurrency, resultSetHoldability);
return WrapperProxy.wrap(statement, new TracingStatement(statement,
connectionInfo, withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public PreparedStatement prepareStatement(String sql, int resultSetType, int resultSetConcurrency,
int resultSetHoldability) throws SQLException {
final PreparedStatement statement = connection
.prepareStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability);
return WrapperProxy.wrap(statement, new TracingPreparedStatement(statement,
sql, connectionInfo, withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public CallableStatement prepareCall(String sql, int resultSetType, int resultSetConcurrency,
int resultSetHoldability) throws SQLException {
final CallableStatement statement = connection
.prepareCall(sql, resultSetType, resultSetConcurrency, resultSetHoldability);
return WrapperProxy.wrap(statement, new TracingCallableStatement(statement, sql,
connectionInfo, withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public PreparedStatement prepareStatement(String sql, int autoGeneratedKeys) throws SQLException {
final PreparedStatement statement = connection.prepareStatement(sql, autoGeneratedKeys);
return WrapperProxy.wrap(statement, new TracingPreparedStatement(statement, sql,
connectionInfo, withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public PreparedStatement prepareStatement(String sql, int[] columnIndexes) throws SQLException {
final PreparedStatement statement = connection.prepareStatement(sql, columnIndexes);
return WrapperProxy.wrap(statement, new TracingPreparedStatement(statement, sql,
connectionInfo, withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public PreparedStatement prepareStatement(String sql, String[] columnNames) throws SQLException {
final PreparedStatement statement = connection.prepareStatement(sql, columnNames);
return WrapperProxy.wrap(statement, new TracingPreparedStatement(statement, sql, connectionInfo,
withActiveSpanOnly, ignoredStatements, tracer));
}

@Override
public Clob createClob() throws SQLException {
return connection.createClob();
}

@Override
public Blob createBlob() throws SQLException {
return connection.createBlob();
}

@Override
public NClob createNClob() throws SQLException {
return connection.createNClob();
}

@Override
public SQLXML createSQLXML() throws SQLException {
return connection.createSQLXML();
}

@Override
public boolean isValid(int timeout) throws SQLException {
return connection.isValid(timeout);
}

@Override
public void setClientInfo(String name, String value) throws SQLClientInfoException {
connection.setClientInfo(name, value);
}

@Override
public void setClientInfo(Properties properties) throws SQLClientInfoException {
connection.setClientInfo(properties);
}

@Override
public String getClientInfo(String name) throws SQLException {
return connection.getClientInfo(name);
}

@Override
public Properties getClientInfo() throws SQLException {
return connection.getClientInfo();
}

@Override
public Array createArrayOf(String typeName, Object[] elements) throws SQLException {
return connection.createArrayOf(typeName, elements);
}

@Override
public Struct createStruct(String typeName, Object[] attributes) throws SQLException {
return connection.createStruct(typeName, attributes);
}

@Override
public void setSchema(String schema) throws SQLException {
connection.setSchema(schema);
}

@Override
public String getSchema() throws SQLException {
return connection.getSchema();
}

@Override
public void abort(Executor executor) throws SQLException {
connection.abort(executor);
}

@Override
public void setNetworkTimeout(Executor executor, int milliseconds) throws SQLException {
connection.setNetworkTimeout(executor, milliseconds);
}

@Override
public int getNetworkTimeout() throws SQLException {
return connection.getNetworkTimeout();
}

@Override
public <T> T unwrap(Class<T> iface) throws SQLException {
return connection.unwrap(iface);
}
/**
* Marker interface for the tracing connection wrapper.
* <p>
* Used, amongst others, by the JdbcAspect to find out if the connection is already wrapped with tracing capabilities.
*/
public interface TracingConnection extends Connection {

@Override
public boolean isWrapperFor(Class<?> iface) throws SQLException {
return connection.isWrapperFor(iface);
}
}

0 comments on commit 4e7d611

Please sign in to comment.