Implements equals method

Author: quaff

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);
+    }
+  }
 }

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);
+    }
+  }
+
 }

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) {

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);
+    }
+  }
+
 }

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

@@ -372,6 +372,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) {