extract helper function

mapleFU · Aug 17, 2024 · 2fda2db · 2fda2db
1 parent fda14a4
commit 2fda2db
Showing 1 changed file with 15 additions and 17 deletions.
diff --git a/cpp/src/arrow/compute/kernels/row_encoder.cc b/cpp/src/arrow/compute/kernels/row_encoder.cc
@@ -75,33 +75,31 @@ void BooleanKeyEncoder::AddLengthNull(int32_t* length) {
 
 Status BooleanKeyEncoder::Encode(const ExecValue& data, int64_t batch_length,
                                  uint8_t** encoded_bytes) {
+  auto handle_next_valid_value = [encoded_bytes](bool value) mutable {
+    auto& encoded_ptr = *encoded_bytes++;
+    *encoded_ptr++ = kValidByte;
+    *encoded_ptr++ = value;
+  };
+  auto handle_next_null_value = [encoded_bytes]() mutable {
+    auto& encoded_ptr = *encoded_bytes++;
+    *encoded_ptr++ = kNullByte;
+    *encoded_ptr++ = 0;
+  };
+
   if (data.is_array()) {
     VisitArraySpanInline<BooleanType>(
-        data.array,
-        [&](bool value) {
-          auto& encoded_ptr = *encoded_bytes++;
-          *encoded_ptr++ = kValidByte;
-          *encoded_ptr++ = value;
-        },
-        [&] {
-          auto& encoded_ptr = *encoded_bytes++;
-          *encoded_ptr++ = kNullByte;
-          *encoded_ptr++ = 0;
-        });
+        data.array, [&](bool value) { handle_next_valid_value(value); },
+        [&] { handle_next_null_value(); });
   } else {
     const auto& scalar = data.scalar_as<BooleanScalar>();
     if (!scalar.is_valid) {
       for (int64_t i = 0; i < batch_length; i++) {
-        auto& encoded_ptr = *encoded_bytes++;
-        *encoded_ptr++ = kNullByte;
-        *encoded_ptr++ = 0;
+        handle_next_null_value();
       }
     } else {
       const bool value = scalar.value;
       for (int64_t i = 0; i < batch_length; i++) {
-        auto& encoded_ptr = *encoded_bytes++;
-        *encoded_ptr++ = kValidByte;
-        *encoded_ptr++ = value;
+        handle_next_valid_value(value);
       }
     }
   }