sync

Author: trask

instrumentation/jdbc/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/jdbc/PreparedStatementInstrumentation.java

@@ -60,6 +60,7 @@ public void transform(TypeTransformer transformer) {
     transformer.applyAdviceToMethod(
         namedOneOf(
                 "setBoolean",
+                "setByte",
                 "setShort",
                 "setInt",
                 "setLong",

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

@@ -107,6 +107,7 @@ public void setBoolean(int parameterIndex, boolean x) throws SQLException {
   @Override
   public void setByte(int parameterIndex, byte x) throws SQLException {
     delegate.setByte(parameterIndex, x);
+    putParameter(parameterIndex, String.valueOf(x));
   }
 
   @Override

instrumentation/jdbc/testing/src/main/java/io/opentelemetry/instrumentation/jdbc/testing/AbstractPreparedStatementParametersTest.java

@@ -134,6 +134,31 @@ void testBooleanPreparedStatementParameter(
         "true");
   }
 
+  @ParameterizedTest
+  @MethodSource("preparedStatementStream")
+  void testBytePreparedStatementParameter(
+      String system,
+      Connection connection,
+      String username,
+      String query,
+      String sanitizedQuery,
+      String spanName,
+      String url,
+      String table)
+      throws SQLException {
+    test(
+        system,
+        connection,
+        username,
+        query,
+        sanitizedQuery,
+        spanName,
+        url,
+        table,
+        statement -> statement.setByte(1, (byte) 42),
+        "42");
+  }
+
   @ParameterizedTest
   @MethodSource("preparedStatementStream")
   void testShortPreparedStatementParameter(