jackwener commented on code in PR #6433:
URL: https://github.com/apache/arrow-datafusion/pull/6433#discussion_r1204162324
##########
datafusion/physical-expr/src/expressions/binary.rs:
##########
@@ -1689,7 +1874,524 @@ pub fn interval_scalar_interval_op(
scalar.get_datatype(),
)))?,
};
- Ok(ColumnarValue::Array(ret))
+ Ok(ret)
+}
+
+/// This function handles interval +/- interval operations where the former is
+/// a scalar and the latter is an array, resulting in an interval array.
+pub fn scalar_interval_op_interval(
+ scalar: &ScalarValue,
+ sign: i32,
+ array: &ArrayRef,
+) -> Result<ArrayRef> {
+ let ret = match (scalar, array.data_type()) {
+ // YearMonth op YearMonth
+ (
+ ScalarValue::IntervalYearMonth(Some(lhs)),
+ DataType::Interval(IntervalUnit::YearMonth),
+ ) => {
+ let array = as_interval_ym_array(&array)?;
+ let ret: PrimitiveArray<IntervalYearMonthType> =
+ unary(array, |rhs| op_ym(*lhs, rhs, sign));
+ Arc::new(ret) as _
Review Comment:
`Arc::new(ret) as _` occurs many times, is there any way we can avoid it?
--
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]
