feat: Implement TRIM_ARRAY array function

AmrDeveloper · Sep 25, 2024 · 8c6f813 · 8c6f813
1 parent b8f6b87
commit 8c6f813
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/crates/gitql-std/src/array/mod.rs b/crates/gitql-std/src/array/mod.rs
@@ -23,6 +23,7 @@ pub fn register_std_array_functions(map: &mut HashMap<&'static str, Function>) {
     map.insert("array_positions", array_positions);
     map.insert("array_dims", array_dims);
     map.insert("array_replace", array_replace);
+    map.insert("trim_array", array_trim);
 }
 
 #[inline(always)]
@@ -109,6 +110,13 @@ pub fn register_std_array_function_signatures(map: &mut HashMap<&'static str, Si
             return_type: DataType::Dynamic(first_element_type),
         },
     );
+    map.insert(
+        "trim_array",
+        Signature {
+            parameters: vec![DataType::Array(Box::new(DataType::Any)), DataType::Integer],
+            return_type: DataType::Dynamic(first_element_type),
+        },
+    );
 }
 
 pub fn array_append(inputs: &[Value]) -> Value {
@@ -198,3 +206,12 @@ pub fn array_replace(inputs: &[Value]) -> Value {
     }
     Value::Array(array_type, array_values)
 }
+
+pub fn array_trim(inputs: &[Value]) -> Value {
+    let mut array = inputs[0].as_array();
+    let array_type = inputs[0].data_type();
+    let array_len = array.len();
+    let n = i64::min(array.len().try_into().unwrap(), inputs[1].as_int());
+    array.truncate(array_len - n as usize);
+    Value::Array(array_type, array)
+}
diff --git a/docs/functions/array.md b/docs/functions/array.md
@@ -18,4 +18,5 @@
 | ARRAY_POSITION  | Array, Any      | Integer        | Return the position of element in array or NULL if not found.                     |
 | ARRAY_POSITIONS | Array, Any      | Array<Integer> | Return the an array of positions of element in array.                             |
 | ARRAY_DIMS      | Array           | Text           | Returns a text representation of the array's dimensions.                          |
-| ARRAY_REPLACE   | Array, Any, Any | Array          | Replaces each array element equal to the second argument with the third argument. |
+| ARRAY_REPLACE   | Array, Any, Any | Array          | Replaces each array element equal to the second argument with the third argument. |
+| TRIM_ARRAY      | Array, Integer  | Array          | Remove the last n elements from the array.                                        |