Skip to content

Commit 5096e22

Browse files
committed
Implements equals method
1 parent 469cf22 commit 5096e22

File tree

5 files changed

+76
-0
lines changed

5 files changed

+76
-0
lines changed

src/main/java/io/opentracing/contrib/jdbc/TracingCallableStatement.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -621,4 +621,19 @@ public <T> T getObject(int parameterIndex, Class<T> type) throws SQLException {
621621
public <T> T getObject(String parameterName, Class<T> type) throws SQLException {
622622
return statement.getObject(parameterName, type);
623623
}
624+
625+
@Override
626+
public boolean equals(Object obj) {
627+
if (this == obj) {
628+
return true;
629+
}
630+
if (!(obj instanceof CallableStatement)) {
631+
return false;
632+
}
633+
if (obj instanceof TracingCallableStatement) {
634+
return statement.equals(((TracingCallableStatement) obj).statement);
635+
} else {
636+
return statement.equals(obj);
637+
}
638+
}
624639
}

src/main/java/io/opentracing/contrib/jdbc/TracingConnection.java

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -360,4 +360,20 @@ public <T> T unwrap(Class<T> iface) throws SQLException {
360360
public boolean isWrapperFor(Class<?> iface) throws SQLException {
361361
return connection.isWrapperFor(iface);
362362
}
363+
364+
@Override
365+
public boolean equals(Object obj) {
366+
if (this == obj) {
367+
return true;
368+
}
369+
if (!(obj instanceof Connection)) {
370+
return false;
371+
}
372+
if (obj instanceof TracingConnection) {
373+
return connection.equals(((TracingConnection) obj).connection);
374+
} else {
375+
return connection.equals(obj);
376+
}
377+
}
378+
363379
}

src/main/java/io/opentracing/contrib/jdbc/TracingDataSource.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,21 @@ public boolean isWrapperFor(final Class<?> iface) throws SQLException {
134134
return underlying.isWrapperFor(iface);
135135
}
136136

137+
@Override
138+
public boolean equals(Object obj) {
139+
if (this == obj) {
140+
return true;
141+
}
142+
if (!(obj instanceof DataSource)) {
143+
return false;
144+
}
145+
if (obj instanceof TracingDataSource) {
146+
return underlying.equals(((TracingDataSource) obj).underlying);
147+
} else {
148+
return underlying.equals(obj);
149+
}
150+
}
151+
137152
@Override
138153
public void close() throws Exception {
139154
if (underlying instanceof AutoCloseable) {

src/main/java/io/opentracing/contrib/jdbc/TracingPreparedStatement.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -362,4 +362,19 @@ public void setNClob(int parameterIndex, Reader reader) throws SQLException {
362362
preparedStatement.setNClob(parameterIndex, reader);
363363
}
364364

365+
@Override
366+
public boolean equals(Object obj) {
367+
if (this == obj) {
368+
return true;
369+
}
370+
if (!(obj instanceof PreparedStatement)) {
371+
return false;
372+
}
373+
if (obj instanceof TracingPreparedStatement) {
374+
return preparedStatement.equals(((TracingPreparedStatement) obj).preparedStatement);
375+
} else {
376+
return preparedStatement.equals(obj);
377+
}
378+
}
379+
365380
}

src/main/java/io/opentracing/contrib/jdbc/TracingStatement.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -372,6 +372,21 @@ public String toString() {
372372
return getQuery();
373373
}
374374

375+
@Override
376+
public boolean equals(Object obj) {
377+
if (this == obj) {
378+
return true;
379+
}
380+
if (!(obj instanceof Statement)) {
381+
return false;
382+
}
383+
if (obj instanceof TracingStatement) {
384+
return statement.equals(((TracingStatement) obj).statement);
385+
} else {
386+
return statement.equals(obj);
387+
}
388+
}
389+
375390
private Span buildSpanForBatch() {
376391
StringBuilder sqlBuilder = new StringBuilder();
377392
if (query != null) {

0 commit comments

Comments
 (0)