alamb commented on code in PR #4020:
URL: https://github.com/apache/arrow-rs/pull/4020#discussion_r1167249827
##########
arrow-cast/src/cast.rs:
##########
@@ -458,6 +460,120 @@ where
}
}
+/// Cast the array from interval to duration
+fn cast_interval_to_duration<D: ArrowTemporalType<Native = i64>>(
+ array: &dyn Array,
+ cast_options: &CastOptions,
+) -> Result<ArrayRef, ArrowError> {
+ let array = array
+ .as_any()
+ .downcast_ref::<IntervalMonthDayNanoArray>()
+ .ok_or_else(|| {
+ ArrowError::ComputeError(
+ "Internal Error: Cannot cast interval to IntervalArray of
expected type"
+ .to_string(),
+ )
+ })?;
+
+ let scale = match D::DATA_TYPE {
+ DataType::Duration(TimeUnit::Second) => 1_000_000_000,
+ DataType::Duration(TimeUnit::Millisecond) => 1_000_000,
+ DataType::Duration(TimeUnit::Microsecond) => 1_000,
+ DataType::Duration(TimeUnit::Nanosecond) => 1,
+ _ => unreachable!(),
+ };
+
+ if cast_options.safe {
+ let iter = array.iter().map(|v| {
+ v.and_then(|v| {
+ let v = v / scale;
+ if v > i64::MAX as i128 {
+ None
+ } else {
+ Some(v as i64)
+ }
+ })
+ });
+ Ok(Arc::new(unsafe {
+ PrimitiveArray::<D>::from_trusted_len_iter(iter)
+ }))
+ } else {
+ let vec = array
+ .iter()
+ .map(|v| {
+ v.map(|v| {
+ let v = v / scale;
+ if v > i64::MAX as i128 {
+ Err(ArrowError::ComputeError(format!(
+ "Cannot cast to {:?}. Overflowing on {:?}",
+ D::DATA_TYPE,
+ v
+ )))
+ } else {
+ Ok(v as i64)
+ }
+ })
+ .transpose()
+ })
+ .collect::<Result<Vec<_>, _>>()?;
+ Ok(Arc::new(unsafe {
+ PrimitiveArray::<D>::from_trusted_len_iter(vec.iter())
+ }))
+ }
+}
+
+/// Cast the array from duration and interval
+fn cast_duration_to_interval<D: ArrowTemporalType<Native = i64>>(
+ array: &dyn Array,
+ cast_options: &CastOptions,
+) -> Result<ArrayRef, ArrowError> {
+ let array = array
+ .as_any()
+ .downcast_ref::<PrimitiveArray<D>>()
+ .ok_or_else(|| {
+ ArrowError::ComputeError(
+ "Internal Error: Cannot cast duration to DurationArray of
expected type"
+ .to_string(),
+ )
+ })?;
+
+ let scale = match array.data_type() {
+ DataType::Duration(TimeUnit::Second) => 1_000_000_000,
+ DataType::Duration(TimeUnit::Millisecond) => 1_000_000,
+ DataType::Duration(TimeUnit::Microsecond) => 1_000,
+ DataType::Duration(TimeUnit::Nanosecond) => 1,
+ _ => unreachable!(),
+ };
+
+ if cast_options.safe {
+ let iter = array
+ .iter()
+ .map(|v| v.and_then(|v| ((v as i128).checked_mul(scale))));
Review Comment:
I am not sure this code correctly converts `v` into an
`IntervalMonthDayNanoType` -- the `IntervalMonthDayNanoType` is represented as
two fields but this is treating it as a single native type)
I think we should construct IntervalMonthDayNano with this method:
https://docs.rs/arrow/37.0.0/arrow/datatypes/struct.IntervalMonthDayNanoType.html#method.make_value
So something like
```something
let months = 0;
let days = 0;
let nanos: i64 = v.checked_mul(scale)?;
Ok(IntervalMonthDayNanoType::make_value(months, days, nanos))
```
##########
arrow-cast/src/cast.rs:
##########
@@ -8246,4 +8385,137 @@ mod tests {
);
assert_eq!("Invalid argument error: 1234567000 is too large to store
in a Decimal256 of precision 7. Max is 9999999", err.unwrap_err().to_string());
}
+
+ #[test]
+ fn test_cast_from_duration_to_interval() {
+ // from duration second to interval month day nano
+ let array = vec![1234567];
Review Comment:
Can you also please test a value if `i64::MAX` here and ensure an error is
returned?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
