refactor: change anyhow error to custom error variants

Author: 7h3cyb3rm0nk

src/static_schema.rs

@@ -18,7 +18,6 @@
 
 use crate::event::DEFAULT_TIMESTAMP_KEY;
 use crate::utils::arrow::get_field;
-use anyhow::{anyhow, Error as AnyError};
 use serde::{Deserialize, Serialize};
 use std::str;
 
@@ -60,7 +59,7 @@ pub fn convert_static_schema_to_arrow_schema(
     static_schema: StaticSchema,
     time_partition: &str,
     custom_partition: Option<&String>,
-) -> Result<Arc<Schema>, AnyError> {
+) -> Result<Arc<Schema>, StaticSchemaError> {
     let mut parsed_schema = ParsedSchema {
         fields: Vec::new(),
         metadata: HashMap::new(),
@@ -83,7 +82,7 @@ pub fn convert_static_schema_to_arrow_schema(
 
         for partition in &custom_partition_list {
             if !custom_partition_exists.contains_key(*partition) {
-                return Err(anyhow!("custom partition field {partition} does not exist in the schema for the static schema logstream"));
+                return Err(StaticSchemaError::MissingCustomPartition(partition.to_string()));
             }
         }
     }
@@ -132,18 +131,14 @@ pub fn convert_static_schema_to_arrow_schema(
         parsed_schema.fields.push(parsed_field);
     }
     if !time_partition.is_empty() && !time_partition_exists {
-        return Err(anyhow! {
-            format!(
-                "time partition field {time_partition} does not exist in the schema for the static schema logstream"
-            ),
-        });
+        return Err(StaticSchemaError::MissingTimePartition(time_partition.to_string()));
     }
     add_parseable_fields_to_static_schema(parsed_schema)
 }
 
 fn add_parseable_fields_to_static_schema(
     parsed_schema: ParsedSchema,
-) -> Result<Arc<Schema>, AnyError> {
+) -> Result<Arc<Schema>, StaticSchemaError> {
 
     let mut schema: Vec<Arc<Field>> = Vec::new();
     for field in parsed_schema.fields.iter() {
@@ -152,10 +147,7 @@ fn add_parseable_fields_to_static_schema(
     }
 
     if get_field(&schema, DEFAULT_TIMESTAMP_KEY).is_some() {
-        return Err(anyhow!(
-            "field {} is a reserved field",
-            DEFAULT_TIMESTAMP_KEY
-        ));
+        return Err(StaticSchemaError::DefaultTime);
     };
 
     // add the p_timestamp field to the event schema to the 0th index
@@ -183,19 +175,43 @@ fn default_dict_is_ordered() -> bool {
     false
 }
 
-fn validate_field_names(field_name: &str, existing_fields: &mut HashSet<String>) -> Result<(), AnyError> {
+fn validate_field_names(field_name: &str, existing_fields: &mut HashSet<String>) -> Result<(), StaticSchemaError> {
 
     if field_name.is_empty() {
-        return Err(anyhow!("field names should not be empty"));
+        return Err(StaticSchemaError::EmptyFieldName);
     }
 
     if !existing_fields.insert(field_name.to_string()) {
-        return Err(anyhow!("duplicate field name: {}", field_name));
+        return Err(StaticSchemaError::DuplicateField(field_name.to_string()));
     }
 
     Ok(())
 }
 
+
+#[derive(Debug, thiserror::Error)]
+pub enum StaticSchemaError{
+
+    #[error(
+        "custom partition field {0} does not exist in the schema for the static schema logstream"
+    )]
+    MissingCustomPartition(String),
+
+    #[error(
+        "time partition field {0} does not exist in the schema for the static schema logstream"
+    )]
+    MissingTimePartition(String),
+
+    #[error("field {DEFAULT_TIMESTAMP_KEY:?} is a reserved field")]
+    DefaultTime,
+
+    #[error("field name cannot be empty")]
+    EmptyFieldName,
+
+    #[error("duplicate field name: {0}")]
+    DuplicateField(String),
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;