Implements equals method

opentracing-contrib · Sep 14, 2020 · 1f148dc · 1f148dc
1 parent 469cf22
commit 1f148dc
Show file tree

Hide file tree

Showing 5 changed files with 77 additions and 0 deletions.
diff --git a/src/main/java/io/opentracing/contrib/jdbc/TracingCallableStatement.java b/src/main/java/io/opentracing/contrib/jdbc/TracingCallableStatement.java
@@ -621,4 +621,19 @@ public <T> T getObject(int parameterIndex, Class<T> type) throws SQLException {
   public <T> T getObject(String parameterName, Class<T> type) throws SQLException {
     return statement.getObject(parameterName, type);
   }
+
+  @Override
+  public boolean equals(Object obj) {
+    if (this == obj) {
+        return true;
+    }
+    if (!(obj instanceof CallableStatement)) {
+        return false;
+    }
+    if (obj instanceof TracingCallableStatement) {
+        return statement.equals(((TracingCallableStatement) obj).statement);
+    } else {
+        return statement.equals(obj);
+    }
+  }
 }
diff --git a/src/main/java/io/opentracing/contrib/jdbc/TracingConnection.java b/src/main/java/io/opentracing/contrib/jdbc/TracingConnection.java
@@ -360,4 +360,20 @@ public <T> T unwrap(Class<T> iface) throws SQLException {
   public boolean isWrapperFor(Class<?> iface) throws SQLException {
     return connection.isWrapperFor(iface);
   }
+
+  @Override
+  public boolean equals(Object obj) {
+    if (this == obj) {
+        return true;
+    }
+    if (!(obj instanceof Connection)) {
+        return false;
+    }
+    if (obj instanceof TracingConnection) {
+        return connection.equals(((TracingConnection) obj).connection);
+    } else {
+        return connection.equals(obj);
+    }
+  }
+
 }
diff --git a/src/main/java/io/opentracing/contrib/jdbc/TracingDataSource.java b/src/main/java/io/opentracing/contrib/jdbc/TracingDataSource.java
@@ -134,6 +134,21 @@ public boolean isWrapperFor(final Class<?> iface) throws SQLException {
     return underlying.isWrapperFor(iface);
   }
 
+  @Override
+  public boolean equals(Object obj) {
+    if (this == obj) {
+        return true;
+    }
+    if (!(obj instanceof DataSource)) {
+        return false;
+    }
+    if (obj instanceof TracingDataSource) {
+        return underlying.equals(((TracingDataSource) obj).underlying);
+    } else {
+        return underlying.equals(obj);
+    }
+  }
+
   @Override
   public void close() throws Exception {
     if (underlying instanceof AutoCloseable) {

diff --git a/src/main/java/io/opentracing/contrib/jdbc/TracingPreparedStatement.java b/src/main/java/io/opentracing/contrib/jdbc/TracingPreparedStatement.java
@@ -362,4 +362,19 @@ public void setNClob(int parameterIndex, Reader reader) throws SQLException {
     preparedStatement.setNClob(parameterIndex, reader);
   }
 
+  @Override
+  public boolean equals(Object obj) {
+    if (this == obj) {
+        return true;
+    }
+    if (!(obj instanceof PreparedStatement)) {
+        return false;
+    }
+    if (obj instanceof TracingPreparedStatement) {
+        return preparedStatement.equals(((TracingPreparedStatement) obj).preparedStatement);
+    } else {
+        return preparedStatement.equals(obj);
+    }
+  }
+
 }
diff --git a/src/main/java/io/opentracing/contrib/jdbc/TracingStatement.java b/src/main/java/io/opentracing/contrib/jdbc/TracingStatement.java
@@ -19,6 +19,7 @@
 import io.opentracing.Scope;
 import io.opentracing.Span;
 import io.opentracing.Tracer;
+
 import java.sql.Connection;
 import java.sql.ResultSet;
 import java.sql.SQLException;
@@ -372,6 +373,21 @@ public String toString() {
     return getQuery();
   }
 
+  @Override
+  public boolean equals(Object obj) {
+    if (this == obj) {
+        return true;
+    }
+    if (!(obj instanceof Statement)) {
+        return false;
+    }
+    if (obj instanceof TracingStatement) {
+        return statement.equals(((TracingStatement) obj).statement);
+    } else {
+        return statement.equals(obj);
+    }
+  }
+
   private Span buildSpanForBatch() {
     StringBuilder sqlBuilder = new StringBuilder();
     if (query != null) {