Add type files

Author: wmoustafa

coral-common/src/main/java/com/linkedin/coral/common/types/ArrayType.java

@@ -0,0 +1,69 @@
+/**
+ * Copyright 2024-2025 LinkedIn Corporation. All rights reserved.
+ * Licensed under the BSD-2 Clause license.
+ * See LICENSE in the project root for license information.
+ */
+package com.linkedin.coral.common.types;
+
+import java.util.Objects;
+
+
+/**
+ * Represents an array data type in the Coral type system.
+ */
+public final class ArrayType implements CoralDataType {
+  private final CoralDataType elementType;
+  private final boolean nullable;
+
+  /**
+   * Creates a new array type.
+   * @param elementType the type of elements in the array
+   * @param nullable whether this type allows null values
+   */
+  public static ArrayType of(CoralDataType elementType, boolean nullable) {
+    return new ArrayType(elementType, nullable);
+  }
+
+  private ArrayType(CoralDataType elementType, boolean nullable) {
+    this.elementType = Objects.requireNonNull(elementType, "Element type cannot be null");
+    this.nullable = nullable;
+  }
+
+  /**
+   * Returns the type of elements in this array.
+   * @return the element type
+   */
+  public CoralDataType getElementType() {
+    return elementType;
+  }
+
+  @Override
+  public CoralTypeKind getKind() {
+    return CoralTypeKind.ARRAY;
+  }
+
+  @Override
+  public boolean isNullable() {
+    return nullable;
+  }
+
+  @Override
+  public boolean equals(Object o) {
+    if (this == o)
+      return true;
+    if (o == null || getClass() != o.getClass())
+      return false;
+    ArrayType that = (ArrayType) o;
+    return nullable == that.nullable && Objects.equals(elementType, that.elementType);
+  }
+
+  @Override
+  public int hashCode() {
+    return Objects.hash(elementType, nullable);
+  }
+
+  @Override
+  public String toString() {
+    return "ARRAY<" + elementType + ">" + (nullable ? " NULL" : " NOT NULL");
+  }
+}

coral-common/src/main/java/com/linkedin/coral/common/types/CharType.java

@@ -0,0 +1,72 @@
+/**
+ * Copyright 2024-2025 LinkedIn Corporation. All rights reserved.
+ * Licensed under the BSD-2 Clause license.
+ * See LICENSE in the project root for license information.
+ */
+package com.linkedin.coral.common.types;
+
+import java.util.Objects;
+
+
+/**
+ * Represents a fixed-length character data type in the Coral type system.
+ */
+public final class CharType implements CoralDataType {
+  private final int length;
+  private final boolean nullable;
+
+  /**
+   * Creates a new CHAR type.
+   * @param length the fixed length of the character string
+   * @param nullable whether this type allows null values
+   */
+  public static CharType of(int length, boolean nullable) {
+    if (length <= 0) {
+      throw new IllegalArgumentException("Length must be positive, got: " + length);
+    }
+    return new CharType(length, nullable);
+  }
+
+  private CharType(int length, boolean nullable) {
+    this.length = length;
+    this.nullable = nullable;
+  }
+
+  /**
+   * Returns the fixed length of this CHAR type.
+   * @return the length
+   */
+  public int getLength() {
+    return length;
+  }
+
+  @Override
+  public CoralTypeKind getKind() {
+    return CoralTypeKind.CHAR;
+  }
+
+  @Override
+  public boolean isNullable() {
+    return nullable;
+  }
+
+  @Override
+  public boolean equals(Object o) {
+    if (this == o)
+      return true;
+    if (o == null || getClass() != o.getClass())
+      return false;
+    CharType that = (CharType) o;
+    return length == that.length && nullable == that.nullable;
+  }
+
+  @Override
+  public int hashCode() {
+    return Objects.hash(length, nullable);
+  }
+
+  @Override
+  public String toString() {
+    return "CHAR(" + length + ")" + (nullable ? " NULL" : " NOT NULL");
+  }
+}

coral-common/src/main/java/com/linkedin/coral/common/types/DecimalType.java

@@ -0,0 +1,87 @@
+/**
+ * Copyright 2024-2025 LinkedIn Corporation. All rights reserved.
+ * Licensed under the BSD-2 Clause license.
+ * See LICENSE in the project root for license information.
+ */
+package com.linkedin.coral.common.types;
+
+import java.util.Objects;
+
+
+/**
+ * Represents a decimal data type with precision and scale in the Coral type system.
+ */
+public final class DecimalType implements CoralDataType {
+  private final int precision;
+  private final int scale;
+  private final boolean nullable;
+
+  /**
+   * Creates a new decimal type.
+   * @param precision the total number of digits
+   * @param scale the number of digits after the decimal point
+   * @param nullable whether this type allows null values
+   */
+  public static DecimalType of(int precision, int scale, boolean nullable) {
+    if (precision <= 0) {
+      throw new IllegalArgumentException("Precision must be positive, got: " + precision);
+    }
+    if (scale < 0 || scale > precision) {
+      throw new IllegalArgumentException(
+          "Scale must be non-negative and <= precision, got scale=" + scale + ", precision=" + precision);
+    }
+    return new DecimalType(precision, scale, nullable);
+  }
+
+  private DecimalType(int precision, int scale, boolean nullable) {
+    this.precision = precision;
+    this.scale = scale;
+    this.nullable = nullable;
+  }
+
+  /**
+   * Returns the precision (total number of digits).
+   * @return the precision
+   */
+  public int getPrecision() {
+    return precision;
+  }
+
+  /**
+   * Returns the scale (number of digits after decimal point).
+   * @return the scale
+   */
+  public int getScale() {
+    return scale;
+  }
+
+  @Override
+  public CoralTypeKind getKind() {
+    return CoralTypeKind.DECIMAL;
+  }
+
+  @Override
+  public boolean isNullable() {
+    return nullable;
+  }
+
+  @Override
+  public boolean equals(Object o) {
+    if (this == o)
+      return true;
+    if (o == null || getClass() != o.getClass())
+      return false;
+    DecimalType that = (DecimalType) o;
+    return precision == that.precision && scale == that.scale && nullable == that.nullable;
+  }
+
+  @Override
+  public int hashCode() {
+    return Objects.hash(precision, scale, nullable);
+  }
+
+  @Override
+  public String toString() {
+    return "DECIMAL(" + precision + "," + scale + ")" + (nullable ? " NULL" : " NOT NULL");
+  }
+}

coral-common/src/main/java/com/linkedin/coral/common/types/MapType.java

@@ -0,0 +1,81 @@
+/**
+ * Copyright 2024-2025 LinkedIn Corporation. All rights reserved.
+ * Licensed under the BSD-2 Clause license.
+ * See LICENSE in the project root for license information.
+ */
+package com.linkedin.coral.common.types;
+
+import java.util.Objects;
+
+
+/**
+ * Represents a map data type in the Coral type system.
+ */
+public final class MapType implements CoralDataType {
+  private final CoralDataType keyType;
+  private final CoralDataType valueType;
+  private final boolean nullable;
+
+  /**
+   * Creates a new map type.
+   * @param keyType the type of keys in the map
+   * @param valueType the type of values in the map
+   * @param nullable whether this type allows null values
+   */
+  public static MapType of(CoralDataType keyType, CoralDataType valueType, boolean nullable) {
+    return new MapType(keyType, valueType, nullable);
+  }
+
+  private MapType(CoralDataType keyType, CoralDataType valueType, boolean nullable) {
+    this.keyType = Objects.requireNonNull(keyType, "Key type cannot be null");
+    this.valueType = Objects.requireNonNull(valueType, "Value type cannot be null");
+    this.nullable = nullable;
+  }
+
+  /**
+   * Returns the type of keys in this map.
+   * @return the key type
+   */
+  public CoralDataType getKeyType() {
+    return keyType;
+  }
+
+  /**
+   * Returns the type of values in this map.
+   * @return the value type
+   */
+  public CoralDataType getValueType() {
+    return valueType;
+  }
+
+  @Override
+  public CoralTypeKind getKind() {
+    return CoralTypeKind.MAP;
+  }
+
+  @Override
+  public boolean isNullable() {
+    return nullable;
+  }
+
+  @Override
+  public boolean equals(Object o) {
+    if (this == o)
+      return true;
+    if (o == null || getClass() != o.getClass())
+      return false;
+    MapType that = (MapType) o;
+    return nullable == that.nullable && Objects.equals(keyType, that.keyType)
+        && Objects.equals(valueType, that.valueType);
+  }
+
+  @Override
+  public int hashCode() {
+    return Objects.hash(keyType, valueType, nullable);
+  }
+
+  @Override
+  public String toString() {
+    return "MAP<" + keyType + "," + valueType + ">" + (nullable ? " NULL" : " NOT NULL");
+  }
+}

coral-common/src/main/java/com/linkedin/coral/common/types/PrimitiveType.java

@@ -0,0 +1,62 @@
+/**
+ * Copyright 2024-2025 LinkedIn Corporation. All rights reserved.
+ * Licensed under the BSD-2 Clause license.
+ * See LICENSE in the project root for license information.
+ */
+package com.linkedin.coral.common.types;
+
+import java.util.Objects;
+
+
+/**
+ * Represents a primitive data type in the Coral type system.
+ * This includes basic types like BOOLEAN, INT, DOUBLE, STRING, etc.
+ */
+public final class PrimitiveType implements CoralDataType {
+  private final CoralTypeKind kind;
+  private final boolean nullable;
+
+  /**
+   * Creates a new primitive type.
+   * @param kind the type kind (must be a primitive type)
+   * @param nullable whether this type allows null values
+   */
+  public static PrimitiveType of(CoralTypeKind kind, boolean nullable) {
+    return new PrimitiveType(kind, nullable);
+  }
+
+  private PrimitiveType(CoralTypeKind kind, boolean nullable) {
+    this.kind = Objects.requireNonNull(kind, "Type kind cannot be null");
+    this.nullable = nullable;
+  }
+
+  @Override
+  public CoralTypeKind getKind() {
+    return kind;
+  }
+
+  @Override
+  public boolean isNullable() {
+    return nullable;
+  }
+
+  @Override
+  public boolean equals(Object o) {
+    if (this == o)
+      return true;
+    if (o == null || getClass() != o.getClass())
+      return false;
+    PrimitiveType that = (PrimitiveType) o;
+    return nullable == that.nullable && kind == that.kind;
+  }
+
+  @Override
+  public int hashCode() {
+    return Objects.hash(kind, nullable);
+  }
+
+  @Override
+  public String toString() {
+    return kind.name() + (nullable ? " NULL" : " NOT NULL");
+  }
+}