Method supportedNonAggWindowFuncOffsetBySpill

bolt/exec/Window.cpp:1091–1124

}

bool Window::supportedNonAggWindowFuncOffsetBySpill(
    const core::WindowNode::Function& windowNodeFunction) {
  auto numArgs = windowNodeFunction.functionCall->inputs().size();
  if (UNLIKELY(numArgs == 0)) {
    return false;
  }
  if (numArgs == 1) {
    return true;
  }
  const auto& inputType = windowNode_->sources()[0]->outputType();
  auto& arg = windowNodeFunction.functionCall->inputs()[1];
  auto channel = exprToChannel(arg.get(), inputType);
  if (channel == kConstantChannel) {
    const auto& constantArg = core::TypedExprs::asConstant(arg);
    const auto& constantVector = constantArg->toConstantVector(pool());
    if (constantVector->isNullAt(0)) {
      return true;
    }
    if (arg->type()->kind() == TypeKind::INTEGER) {
      if (constantVector->as<ConstantVector<int32_t>>()->valueAt(0) >= 2) {
        return false;
      }
    }
    if (arg->type()->kind() == TypeKind::BIGINT) {
      if (constantVector->as<ConstantVector<int64_t>>()->valueAt(0) >= 2) {
        return false;
      }
    }
  } else {
    // offset if not constant
    return false;
  }
  return true;
}

void Window::recordWindowStats() {
  auto spillStats = windowBuild_->spilledStats();