fix: Natural order for Int96 and Decimal96 (#28)

cube-js · Nov 15, 2023 · af17361 · af17361
1 parent 2f71eda
commit af17361
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/arrow/src/array/ord.rs b/arrow/src/array/ord.rs
@@ -126,6 +126,7 @@ pub fn build_compare(left: &Array, right: &Array) -> Result<DynComparator> {
         (Int16, Int16) => compare_primitives::<Int16Type>(left, right),
         (Int32, Int32) => compare_primitives::<Int32Type>(left, right),
         (Int64, Int64) => compare_primitives::<Int64Type>(left, right),
+        (Int96, Int96) => compare_primitives::<Int96Type>(left, right),
         (Int64Decimal(0), Int64Decimal(0)) => {
             compare_primitives::<Int64Decimal0Type>(left, right)
         }
@@ -147,6 +148,27 @@ pub fn build_compare(left: &Array, right: &Array) -> Result<DynComparator> {
         (Int64Decimal(10), Int64Decimal(10)) => {
             compare_primitives::<Int64Decimal10Type>(left, right)
         }
+        (Int96Decimal(0), Int96Decimal(0)) => {
+            compare_primitives::<Int96Decimal0Type>(left, right)
+        }
+        (Int96Decimal(1), Int96Decimal(1)) => {
+            compare_primitives::<Int96Decimal1Type>(left, right)
+        }
+        (Int96Decimal(2), Int96Decimal(2)) => {
+            compare_primitives::<Int96Decimal2Type>(left, right)
+        }
+        (Int96Decimal(3), Int96Decimal(3)) => {
+            compare_primitives::<Int96Decimal3Type>(left, right)
+        }
+        (Int96Decimal(4), Int96Decimal(4)) => {
+            compare_primitives::<Int96Decimal4Type>(left, right)
+        }
+        (Int96Decimal(5), Int96Decimal(5)) => {
+            compare_primitives::<Int96Decimal5Type>(left, right)
+        }
+        (Int96Decimal(10), Int96Decimal(10)) => {
+            compare_primitives::<Int96Decimal10Type>(left, right)
+        }
         (Float32, Float32) => compare_float::<Float32Type>(left, right),
         (Float64, Float64) => compare_float::<Float64Type>(left, right),
         (Date32, Date32) => compare_primitives::<Date32Type>(left, right),