Method createWindowFrame

bolt/exec/Window.cpp:291–336

}; // namespace

Window::WindowFrame Window::createWindowFrame(
    const std::shared_ptr<const core::WindowNode>& windowNode,
    const core::WindowNode::Frame& frame,
    const RowTypePtr& inputType) {
  if (frame.type == core::WindowNode::WindowType::kRows) {
    checkRowFrameBounds(frame);
  }

  if (frame.type == core::WindowNode::WindowType::kRange &&
      (frame.startValue || frame.endValue)) {
    checkKRangeFrameBounds(windowNode, frame, inputType);
  }

  auto createFrameChannelArg =
      [&](const core::TypedExprPtr& frame) -> std::optional<FrameChannelArg> {
    // frame is nullptr for non (kPreceding or kFollowing) frames.
    if (frame == nullptr) {
      return std::nullopt;
    }
    auto frameChannel = exprToChannel(frame.get(), inputType);
    if (frameChannel == kConstantChannel) {
      auto constant = core::TypedExprs::asConstant(frame)->value();
      BOLT_CHECK(!constant.isNull(), "Window frame offset must not be null");
      auto value = VariantConverter::convert(constant, TypeKind::BIGINT)
                       .value<int64_t>();
      BOLT_USER_CHECK_GE(
          value, 0, "Window frame {} offset must not be negative", value);
      return std::make_optional(
          FrameChannelArg{kConstantChannel, nullptr, value});
    } else {
      return std::make_optional(FrameChannelArg{
          frameChannel,
          BaseVector::create(frame->type(), 0, pool()),
          std::nullopt});
    }
  };

  auto startFrameArg = createFrameChannelArg(frame.startValue);
  auto endFrameArg = createFrameChannelArg(frame.endValue);
  return WindowFrame(
      {frame.type,
       frame.startType,
       frame.endType,
       std::move(startFrameArg),
       std::move(endFrameArg)});
}

void Window::createWindowFunctions() {
  BOLT_CHECK_NOT_NULL(windowNode_);