Method GetIntValInterpreted

be/src/exprs/kudu-partition-expr.cc:89–121

}

IntVal KuduPartitionExpr::GetIntValInterpreted(
    ScalarExprEvaluator* eval, const TupleRow* row) const {
  FunctionContext* fn_ctx = eval->fn_context(fn_ctx_idx_);
  KuduPartitionExprCtx* ctx = reinterpret_cast<KuduPartitionExprCtx*>(
      fn_ctx->GetFunctionState(FunctionContext::THREAD_LOCAL));
  DCHECK(ctx != nullptr);
  kudu::KuduPartialRow* kudu_row = ctx->row.get();
  for (int i = 0; i < children_.size(); ++i) {
    void* val = eval->GetValue(*GetChild(i), row);
    if (val == NULL) {
      // We don't currently support nullable partition columns, but pass it along and let
      // the KuduTableSink generate the error message.
      return IntVal(-1);
    }
    int col = tkudu_partition_expr_.referenced_columns[i];
    const ColumnDescriptor& col_desc = table_desc_->col_descs()[col];
    const ColumnType& type = col_desc.type();
    DCHECK_EQ(GetChild(i)->type().type, type.type);
    Status s = WriteKuduValue(col, type, val, false, kudu_row);
    // This can only fail if we set a col to an incorect type, which would be a bug in
    // planning, so we can DCHECK.
    DCHECK(s.ok()) << "WriteKuduValue failed for col = " << col_desc.name()
                   << " and type = " << col_desc.type() << ": " << s.GetDetail();
  }

  int32_t kudu_partition = -1;
  kudu::Status s = ctx->partitioner->PartitionRow(*kudu_row, &kudu_partition);
  // This can only fail if we fail to supply some of the partition cols, which would be a
  // bug in planning, so we can DCHECK.
  DCHECK(s.ok()) << "KuduPartitioner::PartitionRow failed on row = '"
                 << kudu_row->ToString() << "': " << s.ToString();
  return IntVal(kudu_partition);
}

/// Codegens code that retrieves the KuduPartialRow and KuduPartitioner pointers stored in
/// the function context. Sets *kudu_row_ptr' and '*kudu_partitioner_ptr' to point to the