Add GetValidateLevel & Fix

apache · Apr 17, 2023 · c7e7038 · c7e7038
1 parent 9b7f7f1
commit c7e7038
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/cpp/include/gar/writer/arrow_chunk_writer.h b/cpp/include/gar/writer/arrow_chunk_writer.h
@@ -222,6 +222,13 @@ class EdgeChunkWriter {
     validate_level_ = validate_level;
   }
 
+  /**
+   * @brief Get the validate level.
+   *
+   * @return The validate level of this writer.
+   */
+  inline ValidateLevel GetValidateLevel() const { return validate_level_; }
+
   /**
    * @brief Check if the writer operation for offset is allowed.
    *

diff --git a/cpp/src/arrow_chunk_writer.cc b/cpp/src/arrow_chunk_writer.cc
@@ -84,15 +84,15 @@ Status VertexPropertyWriter::Validate(
   if (validate_level == ValidateLevel::default_validate)
     validate_level = validate_level_;
   // no validate
-  if (validate_level_ == ValidateLevel::no_validate)
+  if (validate_level == ValidateLevel::no_validate)
     return Status::OK();
   // weak validate
   if (input_table->num_rows() > vertex_info_.GetChunkSize())
     return Status::OutOfRange();
   if (!vertex_info_.ContainPropertyGroup(property_group))
-    return Status::Invalid("invalid property group");
+    return Status::InvalidOperation("invalid property group");
   if (chunk_index < 0)
-    return Status::Invalid("invalid chunk index");
+    return Status::InvalidOperation("invalid chunk index");
   // strong validate
   if (validate_level == ValidateLevel::strong_validate) {
     auto schema = input_table->schema();