fix short conversion

Author: Neenu1995

java-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryParameterHandler.java

@@ -21,6 +21,7 @@
 import com.google.cloud.bigquery.StandardSQLTypeName;
 import com.google.cloud.bigquery.exception.BigQueryJdbcException;
 import com.google.cloud.bigquery.exception.BigQueryJdbcSqlFeatureNotSupportedException;
+import java.math.BigInteger;
 import java.sql.SQLException;
 import java.util.ArrayList;
 
@@ -59,6 +60,16 @@ QueryJobConfiguration.Builder configureParameters(
 
         Object parameterValue = getParameter(i);
         StandardSQLTypeName sqlType = getSqlType(i);
+        if (parameterValue != null) {
+          if (sqlType == StandardSQLTypeName.INT64
+              && (parameterValue instanceof Short
+                  || parameterValue instanceof Byte
+                  || parameterValue instanceof BigInteger)) {
+            parameterValue = ((Number) parameterValue).longValue();
+          } else if (sqlType == StandardSQLTypeName.FLOAT64 && parameterValue instanceof Float) {
+            parameterValue = ((Number) parameterValue).doubleValue();
+          }
+        }
         LOG.finest(
             "Parameter %s of type %s at index %s added to QueryJobConfiguration",
             parameterValue, sqlType, i);

java-bigquery-jdbc/src/test/java/com/google/cloud/bigquery/jdbc/BigQueryParameterHandlerTest.java

@@ -19,6 +19,7 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 
+import com.google.cloud.bigquery.QueryJobConfiguration;
 import com.google.cloud.bigquery.StandardSQLTypeName;
 import com.google.cloud.bigquery.jdbc.BigQueryParameterHandler.BigQueryStatementParameterType;
 import org.junit.jupiter.api.Test;
@@ -139,4 +140,20 @@ public void testGetSetParameterByIndex() throws Exception {
     assertEquals(String.class, paramHandler.getType(2));
     assertEquals(StandardSQLTypeName.STRING, paramHandler.getSqlType(2));
   }
+
+  @Test
+  public void testConfigureParametersWidenNumericTypes() throws Exception {
+    BigQueryParameterHandler paramHandler = new BigQueryParameterHandler(3);
+    paramHandler.setParameter(1, (short) 5, Short.class);
+    paramHandler.setParameter(2, (byte) 10, Byte.class);
+    paramHandler.setParameter(3, 3.14f, Float.class);
+
+    QueryJobConfiguration.Builder builder = QueryJobConfiguration.newBuilder("SELECT 1");
+    paramHandler.configureParameters(builder);
+
+    QueryJobConfiguration config = builder.build();
+    assertEquals(3, config.getPositionalParameters().size());
+    assertEquals("5", config.getPositionalParameters().get(0).getValue());
+    assertEquals("10", config.getPositionalParameters().get(1).getValue());
+  }
 }