fix: Fix Column Type Name for PostgreSQL ARRAY types

Author: sakthivelmanii

src/main/java/com/google/cloud/spanner/jdbc/AbstractJdbcWrapper.java

@@ -77,6 +77,9 @@ static int extractColumnType(Type type) {
   }
 
   static String getSpannerTypeName(Type type, Dialect dialect) {
+    if (dialect == Dialect.POSTGRESQL && type.getCode() == Code.ARRAY) {
+      return "_" + getSpannerTypeName(type.getArrayElementType(), dialect);
+    }
     return Preconditions.checkNotNull(type).getSpannerTypeName(dialect);
   }
 

src/main/java/com/google/cloud/spanner/jdbc/JdbcResultSetMetaData.java

@@ -16,7 +16,9 @@
 
 package com.google.cloud.spanner.jdbc;
 
+import com.google.cloud.spanner.Dialect;
 import com.google.cloud.spanner.ResultSet;
+import com.google.cloud.spanner.Type;
 import com.google.cloud.spanner.connection.ConnectionProperties;
 import com.google.common.base.Preconditions;
 import java.sql.Connection;
@@ -192,7 +194,12 @@ public int getColumnType(int column) {
 
   @Override
   public String getColumnTypeName(int column) {
-    return spannerResultSet.getColumnType(column - 1).getCode().name();
+    Type columnType = spannerResultSet.getColumnType(column - 1);
+    if (statement instanceof JdbcStatement) {
+      Dialect dialect = ((JdbcStatement) statement).getConnection().getDialect();
+      return getSpannerTypeName(columnType, dialect);
+    }
+    return columnType.getCode().name();
   }
 
   @Override

src/test/java/com/google/cloud/spanner/jdbc/JdbcResultSetMetaDataTest.java

@@ -23,6 +23,7 @@
 import static org.mockito.Mockito.when;
 
 import com.google.cloud.ByteArray;
+import com.google.cloud.spanner.Dialect;
 import com.google.cloud.spanner.ResultSet;
 import com.google.cloud.spanner.ResultSets;
 import com.google.cloud.spanner.Struct;
@@ -540,6 +541,30 @@ public void getColumnTypeName() {
     }
   }
 
+  @Test
+  public void getColumnTypeNameForPostgreSQL() throws SQLException {
+    JdbcConnection connection = mock(JdbcConnection.class);
+    JdbcStatement statement = mock(JdbcStatement.class);
+    JdbcResultSet resultSet = getFooTestResultSet(statement);
+    when(connection.getSchema()).thenReturn("");
+    when(connection.getCatalog()).thenReturn("test-database");
+    when(statement.getConnection()).then(new Returns(connection));
+    when(connection.getDialect()).thenReturn(Dialect.POSTGRESQL);
+
+    JdbcResultSetMetaData sub = resultSet.getMetaData();
+
+    int index = 1;
+    for (TestColumn col : TEST_COLUMNS) {
+      if (col.type.getCode() == Type.Code.ARRAY
+          && col.type.getSpannerTypeName(Dialect.POSTGRESQL).contains("bool")) {
+        assertEquals("_boolean", sub.getColumnTypeName(index));
+      } else if (col.type.getCode() == Type.Code.BOOL) {
+        assertEquals("boolean", sub.getColumnTypeName(index));
+      }
+      index++;
+    }
+  }
+
   @Test
   public void testIsReadOnly() {
     for (int i = 0; i < TEST_COLUMNS.size(); i++) {