FlatPostgresCollection Create Doc Impl

document-store/src/main/java/org/hypertrace/core/documentstore/CreateResult.java

@@ -1,16 +1,51 @@
 package org.hypertrace.core.documentstore;
 
+import java.util.Collections;
+import java.util.List;
+import lombok.Getter;
+
 /*
  * Represent the result object for CREATE operation of document store APIs.
  * */
+@Getter
 public class CreateResult {
-  private boolean succeed;
 
-  public CreateResult(boolean succeed) {
-    this.succeed = succeed;
+  private final CreateStatus status;
+  private final boolean onRetry;
+  private final List<String> skippedFields;
+
+  public CreateResult(CreateStatus status, boolean onRetry, List<String> skippedFields) {
+    this.status = status;
+    this.onRetry = onRetry;
+    this.skippedFields = skippedFields != null ? skippedFields : Collections.emptyList();
+  }
+
+  public CreateResult(boolean isSuccess) {
+    this(isSuccess ? CreateStatus.SUCCESS : CreateStatus.FAILED, false, Collections.emptyList());
+  }
+
+  public CreateResult(boolean isSucceed, boolean onRetry, List<String> skippedFields) {
+    this(determineStatus(isSucceed, skippedFields), onRetry, skippedFields);
+  }
+
+  private static CreateStatus determineStatus(boolean isSucceed, List<String> skippedFields) {
+    if (!isSucceed) {
+      return CreateStatus.FAILED;
+    }
+    return (skippedFields != null && !skippedFields.isEmpty())
+        ? CreateStatus.PARTIAL
+        : CreateStatus.SUCCESS;
   }
 
   public boolean isSucceed() {
-    return succeed;
+    return status == CreateStatus.SUCCESS || status == CreateStatus.PARTIAL;
+  }
+
+  public boolean isPartial() {
+    return status == CreateStatus.PARTIAL;
+  }
+
+  public boolean isDocumentIgnored() {
+    return status == CreateStatus.IGNORED;
   }
 }

document-store/src/main/java/org/hypertrace/core/documentstore/CreateStatus.java

@@ -0,0 +1,16 @@
+package org.hypertrace.core.documentstore;
+
+/** Status of a document create operation. */
+public enum CreateStatus {
+  /** Document created successfully with all fields. */
+  SUCCESS,
+
+  /** Document created but some fields were skipped (didn't match schema). */
+  PARTIAL,
+
+  /** Document was intentionally not created due to IGNORE_DOCUMENT strategy. */
+  IGNORED,
+
+  /** Operation failed (no valid columns found or other error). */
+  FAILED
+}

document-store/src/main/java/org/hypertrace/core/documentstore/commons/ColumnMetadata.java

@@ -18,4 +18,9 @@ public interface ColumnMetadata {
    * @return whether this column can be set to NULL
    */
   boolean isNullable();
+
+  /**
+   * @return whether this column is an array type
+   */
+  boolean isArray();
 }

document-store/src/main/java/org/hypertrace/core/documentstore/model/exception/SchemaMismatchException.java

@@ -0,0 +1,22 @@
+package org.hypertrace.core.documentstore.model.exception;
+
+import java.io.IOException;
+
+/**
+ * Exception thrown when a document field doesn't match the expected schema. This can occur when:
+ *
+ * <ul>
+ *   <li>A field in the document doesn't exist in the schema
+ *   <li>A field's value type doesn't match the expected schema type
+ * </ul>
+ */
+public class SchemaMismatchException extends IOException {
+
+  public SchemaMismatchException(String message) {
+    super(message);
+  }
+
+  public SchemaMismatchException(String message, Throwable cause) {
+    super(message, cause);
+  }
+}

document-store/src/main/java/org/hypertrace/core/documentstore/model/options/MissingColumnStrategy.java

@@ -0,0 +1,23 @@
+package org.hypertrace.core.documentstore.model.options;
+
+/**
+ * Strategy for handling document fields that don't match the schema during write operations. This
+ * is mostly applicable to flat collections as they conform to a schema
+ *
+ * <p>This enum defines how the system should behave when encountering fields that either don't
+ * exist in the schema or have incompatible types.
+ */
+public enum MissingColumnStrategy {
+  /**
+   * Skip unmatched fields and continue with the write operation. The document will be written with
+   * only the fields that match the schema.
+   */
+  SKIP,
+
+  /**
+   * Throw a {@link org.hypertrace.core.documentstore.model.exception.SchemaMismatchException} when
+   * a field doesn't match the schema. The write operation will fail.
+   */
+  THROW,
+  IGNORE_DOCUMENT
+}