[native] Add row expression optimizer

presto-native-execution/presto_cpp/main/CMakeLists.txt

@@ -13,6 +13,7 @@ add_subdirectory(operators)
 add_subdirectory(types)
 add_subdirectory(http)
 add_subdirectory(common)
+add_subdirectory(expression)
 add_subdirectory(thrift)
 
 add_library(
@@ -51,6 +52,7 @@ target_link_libraries(
   presto_http
   presto_operators
   presto_velox_conversion
+  presto_expression_optimizer
   velox_aggregates
   velox_caching
   velox_common_base

presto-native-execution/presto_cpp/main/PrestoServer.cpp

@@ -1521,6 +1521,26 @@ void PrestoServer::registerSidecarEndpoints() {
          proxygen::ResponseHandler* downstream) {
         http::sendOkResponse(downstream, getFunctionsMetadata());
       });
+  rowExpressionOptimizer_ =
+      std::make_unique<expression::RowExpressionOptimizer>();
+  httpServer_->registerPost(
+      "/v1/expressions",
+      [&](proxygen::HTTPMessage* message,
+          const std::vector<std::unique_ptr<folly::IOBuf>>& body,
+          proxygen::ResponseHandler* downstream) {
+        json::array_t inputRowExpressions =
+            json::parse(util::extractMessageBody(body));
+        auto result =
+            rowExpressionOptimizer_->optimize(message, inputRowExpressions);
+        if (result.second) {
+          VELOX_USER_CHECK(
+              result.first.is_array(),
+              "Output json should be an array of row expressions");
+          http::sendOkResponse(downstream, result.first);
+        } else {
+          http::sendErrorResponse(downstream, result.first);
+        }
+      });
   httpServer_->registerPost(
       "/v1/velox/plan",
       [server = this](

presto-native-execution/presto_cpp/main/PrestoServer.h

@@ -25,6 +25,7 @@
 #include "presto_cpp/main/PeriodicHeartbeatManager.h"
 #include "presto_cpp/main/PrestoExchangeSource.h"
 #include "presto_cpp/main/PrestoServerOperations.h"
+#include "presto_cpp/main/expression/RowExpressionOptimizer.h"
 #include "presto_cpp/main/types/VeloxPlanValidator.h"
 #include "velox/common/caching/AsyncDataCache.h"
 #include "velox/common/memory/MemoryAllocator.h"
@@ -288,6 +289,7 @@ class PrestoServer {
   std::string address_;
   std::string nodeLocation_;
   folly::SSLContextPtr sslContext_;
+  std::unique_ptr<expression::RowExpressionOptimizer> rowExpressionOptimizer_;
 };
 
 } // namespace facebook::presto

presto-native-execution/presto_cpp/main/expression/CMakeLists.txt

@@ -0,0 +1,32 @@
+# Copyright (c) Facebook, Inc. and its affiliates.
+#
+# 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.
+add_library(presto_expression_optimizer RowExpressionOptimizer.cpp)
+
+target_link_libraries(
+  presto_expression_optimizer
+  presto_type_converter
+  presto_types
+  presto_protocol
+  presto_http
+  velox_coverage_util
+  velox_parse_expression
+  velox_parse_parser
+  velox_presto_serializer
+  velox_serialization
+  velox_type_parser
+  ${FOLLY_WITH_DEPENDENCIES})
+
+if(PRESTO_ENABLE_TESTING)
+  add_subdirectory(tests)
+endif()