Koda-specific wrapper around MakeBoundOperator.

Also support functions that return `absl::Status` in `arolla::MakeBoundOperator`. The goal is to have a single entry point for all Koda QExpr operators defined via `OparatorFamily`. The current motivation is to avoid direct usages of `OperatorEvalError` in their bodies (see the child CL).

PiperOrigin-RevId: 716550906
Change-Id: I907e9a09cb6af83ddc7bacbe1a0e81994aafbb62

Author: timofey-stepanov

koladata/internal/op_utils/BUILD

@@ -917,3 +917,28 @@ cc_test(
         "@com_google_googletest//:gtest_main",
     ],
 )
+
+cc_library(
+    name = "qexpr",
+    hdrs = ["qexpr.h"],
+    deps = [
+        ":error",
+        "@com_google_absl//absl/status",
+        "@com_google_arolla//arolla/memory",
+        "@com_google_arolla//arolla/qexpr",
+    ],
+)
+
+cc_test(
+    name = "qexpr_test",
+    srcs = ["qexpr_test.cc"],
+    deps = [
+        ":qexpr",
+        "//koladata/internal:error_cc_proto",
+        "//koladata/internal:error_utils",
+        "@com_google_absl//absl/status",
+        "@com_google_arolla//arolla/memory",
+        "@com_google_arolla//arolla/qexpr",
+        "@com_google_googletest//:gtest_main",
+    ],
+)

koladata/internal/op_utils/qexpr.h

@@ -0,0 +1,48 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+#ifndef KOLADATA_INTERNAL_OP_UTILS_QEXPR_H_
+#define KOLADATA_INTERNAL_OP_UTILS_QEXPR_H_
+
+#include <memory>
+#include <string>
+#include <utility>
+
+#include "absl/status/status.h"
+#include "koladata/internal/op_utils/error.h"
+#include "arolla/memory/frame.h"
+#include "arolla/qexpr/bound_operators.h"
+#include "arolla/qexpr/eval_context.h"
+#include "arolla/qexpr/operators.h"
+
+namespace koladata {
+
+// Creates a bound operator implemented by the provided functor. All the errors
+// returned by the functor are wrapped into OperatorEvalError with the given
+// name.
+template <typename Functor>
+std::unique_ptr<arolla::BoundOperator> MakeBoundOperator(std::string name,
+                                                         Functor functor) {
+  static_assert(std::is_same_v<decltype(functor(
+                                   std::declval<arolla::EvaluationContext*>(),
+                                   std::declval<arolla::FramePtr>())),
+                               absl::Status>,
+                "functor(ctx, frame) must return absl::Status");
+  return arolla::MakeBoundOperator(
+      internal::ReturnsOperatorEvalError(std::move(name), std::move(functor)));
+}
+
+}  // namespace koladata
+
+#endif  // KOLADATA_INTERNAL_OP_UTILS_QEXPR_H_

koladata/internal/op_utils/qexpr_test.cc

@@ -0,0 +1,54 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+#include "koladata/internal/op_utils/qexpr.h"
+
+#include <optional>
+
+#include "gmock/gmock.h"
+#include "gtest/gtest.h"
+#include "absl/status/status.h"
+#include "koladata/internal/error.pb.h"
+#include "koladata/internal/error_utils.h"
+#include "arolla/memory/frame.h"
+#include "arolla/memory/memory_allocation.h"
+#include "arolla/qexpr/eval_context.h"
+
+namespace koladata {
+namespace {
+
+using ::testing::Eq;
+using ::testing::status::StatusIs;
+
+TEST(QExpr, MakeBoundOperator) {
+  auto bound_op = MakeBoundOperator(
+      "op_name", [](arolla::EvaluationContext* ctx, arolla::FramePtr frame) {
+        return absl::InvalidArgumentError("test error");
+      });
+  arolla::EvaluationContext ctx;
+  arolla::FrameLayout memory_layout = arolla::FrameLayout::Builder().Build();
+  arolla::MemoryAllocation alloc(&memory_layout);
+
+  bound_op->Run(&ctx, alloc.frame());
+
+  EXPECT_THAT(ctx.status(),
+              StatusIs(absl::StatusCode::kInvalidArgument, "test error"));
+  std::optional<internal::Error> payload =
+      internal::GetErrorPayload(ctx.status());
+  ASSERT_TRUE(payload.has_value());
+  EXPECT_THAT(payload->error_message(), Eq("op_name: test error"));
+}
+
+}  // namespace
+}  // namespace koladata