sort-exec-exprs.h

Go to the documentation of this file.

// Copyright 2012 Cloudera Inc.
//
// 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 IMPALA_EXEC_SORT_EXEC_EXPRS_H
#define IMPALA_EXEC_SORT_EXEC_EXPRS_H

#include "exprs/expr.h"
#include "runtime/runtime-state.h"

namespace impala {

class SortExecExprs {
 public:
  Status Init(const TSortInfo& sort_info, ObjectPool* pool);

  Status Init(const std::vector<TExpr>& ordering_exprs,
    const std::vector<TExpr>* sort_tuple_slot_exprs, ObjectPool* pool);

  Status Prepare(RuntimeState* state, const RowDescriptor& child_row_desc,
    const RowDescriptor& output_row_desc, MemTracker* expr_mem_tracker);

  Status Open(RuntimeState* state);

  void Close(RuntimeState* state);

  const std::vector<ExprContext*>& sort_tuple_slot_expr_ctxs() const {
    return sort_tuple_slot_expr_ctxs_;
  }

  const std::vector<ExprContext*>& lhs_ordering_expr_ctxs() const {
    return lhs_ordering_expr_ctxs_;
  }
  const std::vector<ExprContext*>& rhs_ordering_expr_ctxs() const {
    return rhs_ordering_expr_ctxs_;
  }

 private:
  std::vector<ExprContext*> lhs_ordering_expr_ctxs_;
  std::vector<ExprContext*> rhs_ordering_expr_ctxs_;

  bool materialize_tuple_;

  std::vector<ExprContext*> sort_tuple_slot_expr_ctxs_;
};

}

#endif