fix tests

Author: g31pranjal

fdb-relational-api/src/main/java/com/apple/foundationdb/relational/api/ArrayMetaData.java

@@ -21,7 +21,9 @@
 package com.apple.foundationdb.relational.api;
 
 import com.apple.foundationdb.relational.api.exceptions.ErrorCode;
+import com.apple.foundationdb.relational.api.metadata.DataType;
 
+import javax.annotation.Nonnull;
 import java.sql.SQLException;
 import java.sql.Wrapper;
 
@@ -59,4 +61,7 @@ default StructMetaData getElementStructMetaData() throws SQLException {
      * @throws SQLException if the type of the column is not an array, or if something else goes wrong.
      */
     ArrayMetaData getElementArrayMetaData() throws SQLException;
+
+    @Nonnull
+    DataType.ArrayType getRelationalDataType() throws SQLException;
 }

fdb-relational-api/src/main/java/com/apple/foundationdb/relational/api/FieldDescription.java

@@ -22,6 +22,7 @@
 
 import javax.annotation.Nonnull;
 import java.sql.SQLException;
+import java.util.UUID;
 
 /**
  * Builder for {@link RelationalArray}.
@@ -44,5 +45,7 @@ public interface RelationalArrayBuilder {
 
     RelationalArrayBuilder addLong(@Nonnull long value) throws SQLException;
 
+    RelationalArrayBuilder addUuid(@Nonnull UUID value) throws SQLException;
+
     RelationalArrayBuilder addStruct(RelationalStruct struct) throws SQLException;
 }

fdb-relational-api/src/main/java/com/apple/foundationdb/relational/api/RelationalArrayBuilder.java

@@ -21,15 +21,14 @@
 package com.apple.foundationdb.relational.api;
 
 import com.apple.foundationdb.annotation.API;
-
 import com.apple.foundationdb.relational.api.exceptions.ErrorCode;
 import com.apple.foundationdb.relational.api.exceptions.RelationalException;
-
+import com.apple.foundationdb.relational.api.metadata.DataType;
 import com.google.common.base.Suppliers;
 
 import javax.annotation.Nonnull;
+import java.sql.DatabaseMetaData;
 import java.sql.SQLException;
-import java.sql.Types;
 import java.util.Objects;
 import java.util.function.Supplier;
 
@@ -39,41 +38,37 @@
 @API(API.Status.EXPERIMENTAL)
 public final class RelationalArrayMetaData implements ArrayMetaData {
 
-    private final FieldDescription element;
+    private final DataType.ArrayType type;
 
     private final Supplier<Integer> hashCodeSupplier;
 
-    private RelationalArrayMetaData(@Nonnull FieldDescription element) {
-        this.element = element;
+    public RelationalArrayMetaData(@Nonnull DataType.ArrayType type) {
+        this.type = type;
         this.hashCodeSupplier = Suppliers.memoize(this::calculateHashCode);
     }
 
-    public static RelationalArrayMetaData ofPrimitive(int sqlType, int nullable) {
-        return new RelationalArrayMetaData(FieldDescription.primitive("VALUE", sqlType, nullable));
-    }
-
-    public static RelationalArrayMetaData ofStruct(@Nonnull StructMetaData metaData, int nullable) {
-        return new RelationalArrayMetaData(FieldDescription.struct("VALUE", nullable, metaData));
-    }
-
     @Override
-    public int isElementNullable() throws SQLException {
-        return element.isNullable();
+    public int isElementNullable() {
+        if (type.getElementType().isNullable()) {
+            return DatabaseMetaData.columnNullable;
+        } else {
+            return DatabaseMetaData.columnNoNulls;
+        }
     }
 
     @Override
     public String getElementName() throws SQLException {
-        return element.getName();
+        return "VALUE";
     }
 
     @Override
     public int getElementType() throws SQLException {
-        return element.getSqlTypeCode();
+        return type.getElementType().getJdbcSqlCode();
     }
 
     @Override
-    public String getElementTypeName() throws SQLException {
-        return SqlTypeNamesSupport.getSqlTypeName(element.getSqlTypeCode());
+    public String getElementTypeName() {
+        return SqlTypeNamesSupport.getSqlTypeName(type.getElementType().getJdbcSqlCode());
     }
 
     /**
@@ -85,11 +80,11 @@ public String getElementTypeName() throws SQLException {
      * @throws SQLException if the type of the column is not a struct, or if something else goes wrong.
      */
     @Override
-    public StructMetaData getElementStructMetaData() throws SQLException {
-        if (element.getSqlTypeCode() != Types.STRUCT) {
+    public RelationalStructMetaData getElementStructMetaData() throws SQLException {
+        if (type.getElementType().getCode() != DataType.Code.STRUCT) {
             throw new RelationalException("Element is not of STRUCT type", ErrorCode.CANNOT_CONVERT_TYPE).toSqlException();
         }
-        return element.getFieldMetaData();
+        return new RelationalStructMetaData((DataType.StructType) type.getElementType());
     }
 
     /**
@@ -102,15 +97,21 @@ public StructMetaData getElementStructMetaData() throws SQLException {
      */
     @Override
     public ArrayMetaData getElementArrayMetaData() throws SQLException {
-        if (element.getSqlTypeCode() != Types.ARRAY) {
+        if (type.getElementType().getCode() != DataType.Code.ARRAY) {
             throw new RelationalException("Element is not of ARRAY type", ErrorCode.CANNOT_CONVERT_TYPE).toSqlException();
         }
-        return element.getArrayMetaData();
+        return new RelationalArrayMetaData((DataType.ArrayType) type.getElementType());
+    }
+
+    @Nonnull
+    @Override
+    public DataType.ArrayType getRelationalDataType() throws SQLException {
+        return type;
     }
 
     @Nonnull
-    public FieldDescription getElementField() {
-        return element;
+    public DataType getElementDataType() {
+        return type.getElementType();
     }
 
     @Override
@@ -132,7 +133,7 @@ public boolean equals(Object other) {
         if (otherMetadata == this) {
             return true;
         }
-        return element.equals(otherMetadata.element);
+        return type.equals(otherMetadata.type);
     }
 
     @Override
@@ -141,7 +142,7 @@ public int hashCode() {
     }
 
     private int calculateHashCode() {
-        return Objects.hash(element);
+        return Objects.hash(type);
     }
 
 }

fdb-relational-api/src/main/java/com/apple/foundationdb/relational/api/RelationalArrayMetaData.java

@@ -23,6 +23,7 @@
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
 import java.sql.SQLException;
+import java.util.UUID;
 
 /**
  * For implementation by {@link RelationalStruct} <a href="https://refactoring.guru/design-patterns/builder">Builder</a>.
@@ -40,8 +41,6 @@ public interface RelationalStructBuilder {
 
     RelationalStructBuilder addBoolean(String fieldName, boolean b) throws SQLException;
 
-    RelationalStructBuilder addShort(String fieldName, short b) throws SQLException;
-
     RelationalStructBuilder addLong(String fieldName, long l) throws SQLException;
 
     RelationalStructBuilder addFloat(String fieldName, float f) throws SQLException;
@@ -52,7 +51,9 @@ public interface RelationalStructBuilder {
 
     RelationalStructBuilder addString(String fieldName, @Nullable String s) throws SQLException;
 
-    RelationalStructBuilder addObject(String fieldName, @Nullable Object obj, int targetSqlType) throws SQLException;
+    RelationalStructBuilder addUuid(String fieldName, @Nullable UUID uuid) throws SQLException;
+
+    RelationalStructBuilder addObject(String fieldName, @Nullable Object obj) throws SQLException;
 
     RelationalStructBuilder addStruct(String fieldName, @Nonnull RelationalStruct struct) throws SQLException;