address review comment

Author: laurit

instrumentation/jdbc/library/src/main/java/io/opentelemetry/instrumentation/jdbc/internal/OpenTelemetryPreparedStatement.java

@@ -364,4 +364,14 @@ public void setSQLXML(int parameterIndex, SQLXML xmlObject) throws SQLException
   public void clearParameters() throws SQLException {
     delegate.clearParameters();
   }
+
+  @Override
+  public int[] executeBatch() throws SQLException {
+    return wrapBatchCall(delegate::executeBatch);
+  }
+
+  private <T, E extends Exception> T wrapBatchCall(ThrowingSupplier<T, E> callable) throws E {
+    DbRequest request = DbRequest.create(dbInfo, query, batchSize);
+    return wrapCall(request, callable);
+  }
 }

instrumentation/jdbc/library/src/main/java/io/opentelemetry/instrumentation/jdbc/internal/OpenTelemetryStatement.java

@@ -25,7 +25,6 @@
 import io.opentelemetry.instrumentation.api.instrumenter.Instrumenter;
 import io.opentelemetry.instrumentation.jdbc.internal.dbinfo.DbInfo;
 import java.sql.Connection;
-import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.sql.SQLWarning;
@@ -300,7 +299,7 @@ protected <T, E extends Exception> T wrapCall(String sql, ThrowingSupplier<T, E>
     return wrapCall(request, callable);
   }
 
-  private <T, E extends Exception> T wrapCall(DbRequest request, ThrowingSupplier<T, E> callable)
+  protected <T, E extends Exception> T wrapCall(DbRequest request, ThrowingSupplier<T, E> callable)
       throws E {
     Context parentContext = Context.current();
 
@@ -320,11 +319,8 @@ private <T, E extends Exception> T wrapCall(DbRequest request, ThrowingSupplier<
     return result;
   }
 
-  protected <T, E extends Exception> T wrapBatchCall(ThrowingSupplier<T, E> callable) throws E {
-    DbRequest request =
-        this instanceof PreparedStatement
-            ? DbRequest.create(dbInfo, query, batchSize)
-            : DbRequest.create(dbInfo, batchCommands, batchSize);
+  private <T, E extends Exception> T wrapBatchCall(ThrowingSupplier<T, E> callable) throws E {
+    DbRequest request = DbRequest.create(dbInfo, batchCommands, batchSize);
     return wrapCall(request, callable);
   }
 }