Skip to content

Commit a54b5d1

Browse files
chore: rustfmt octet_length
1 parent 7c89fa0 commit a54b5d1

File tree

1 file changed

+32
-32
lines changed

1 file changed

+32
-32
lines changed

datafusion/functions/src/string/octet_length.rs

Lines changed: 32 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@ use std::any::Any;
2020

2121
use crate::utils::utf8_to_int_type;
2222
use arrow::array::{
23-
Array, Int32Array, Int32Builder, Int64Builder, LargeStringArray,
24-
StringArray, StringViewArray,
23+
Array, Int32Array, Int32Builder, Int64Builder, LargeStringArray, StringArray,
24+
StringViewArray,
2525
};
2626
use datafusion_common::types::logical_string;
2727
use datafusion_common::utils::take_function_args;
@@ -95,37 +95,37 @@ impl ScalarUDFImpl for OctetLengthFunc {
9595

9696
match array {
9797
ColumnarValue::Array(v) => {
98-
if let Some(arr) = v.as_any().downcast_ref::<StringArray>() {
99-
let mut builder = Int32Builder::with_capacity(arr.len());
100-
for i in 0..arr.len() {
101-
if arr.is_null(i) {
102-
builder.append_null();
103-
} else {
104-
builder.append_value(arr.value_length(i) as i32);
105-
}
106-
}
107-
Ok(ColumnarValue::Array(Arc::new(builder.finish())))
108-
} else if let Some(arr) = v.as_any().downcast_ref::<LargeStringArray>() {
109-
let mut builder = Int64Builder::with_capacity(arr.len());
110-
for i in 0..arr.len() {
111-
if arr.is_null(i) {
112-
builder.append_null();
113-
} else {
114-
builder.append_value(arr.value_length(i) as i64);
115-
}
116-
}
117-
Ok(ColumnarValue::Array(Arc::new(builder.finish())))
118-
} else if let Some(arr) = v.as_any().downcast_ref::<StringViewArray>() {
119-
let result = arr
120-
.iter()
121-
.map(|s| s.map(|s| s.len() as i32))
122-
.collect::<Int32Array>();
98+
if let Some(arr) = v.as_any().downcast_ref::<StringArray>() {
99+
let mut builder = Int32Builder::with_capacity(arr.len());
100+
for i in 0..arr.len() {
101+
if arr.is_null(i) {
102+
builder.append_null();
103+
} else {
104+
builder.append_value(arr.value_length(i) as i32);
105+
}
106+
}
107+
Ok(ColumnarValue::Array(Arc::new(builder.finish())))
108+
} else if let Some(arr) = v.as_any().downcast_ref::<LargeStringArray>() {
109+
let mut builder = Int64Builder::with_capacity(arr.len());
110+
for i in 0..arr.len() {
111+
if arr.is_null(i) {
112+
builder.append_null();
113+
} else {
114+
builder.append_value(arr.value_length(i) as i64);
115+
}
116+
}
117+
Ok(ColumnarValue::Array(Arc::new(builder.finish())))
118+
} else if let Some(arr) = v.as_any().downcast_ref::<StringViewArray>() {
119+
let result = arr
120+
.iter()
121+
.map(|s| s.map(|s| s.len() as i32))
122+
.collect::<Int32Array>();
123123

124-
Ok(ColumnarValue::Array(Arc::new(result)))
125-
} else {
126-
unreachable!("octet_length expects string arrays")
127-
}
128-
}
124+
Ok(ColumnarValue::Array(Arc::new(result)))
125+
} else {
126+
unreachable!("octet_length expects string arrays")
127+
}
128+
}
129129

130130
ColumnarValue::Scalar(v) => match v {
131131
ScalarValue::Utf8(v) => Ok(ColumnarValue::Scalar(ScalarValue::Int32(

0 commit comments

Comments
 (0)