diff options
Diffstat (limited to 'src/arrow/cpp/src/gandiva/execution_context.h')
-rw-r--r-- | src/arrow/cpp/src/gandiva/execution_context.h | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/src/arrow/cpp/src/gandiva/execution_context.h b/src/arrow/cpp/src/gandiva/execution_context.h new file mode 100644 index 000000000..efa546874 --- /dev/null +++ b/src/arrow/cpp/src/gandiva/execution_context.h @@ -0,0 +1,54 @@ +// Licensed to the Apache Software Foundation (ASF) under one +// or more contributor license agreements. See the NOTICE file +// distributed with this work for additional information +// regarding copyright ownership. The ASF licenses this file +// to you 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. + +#pragma once + +#include <memory> +#include <string> +#include "gandiva/simple_arena.h" + +namespace gandiva { + +/// Execution context during llvm evaluation +class ExecutionContext { + public: + explicit ExecutionContext(arrow::MemoryPool* pool = arrow::default_memory_pool()) + : arena_(pool) {} + std::string get_error() const { return error_msg_; } + + void set_error_msg(const char* error_msg) { + // Remember the first error only. + if (error_msg_.empty()) { + error_msg_ = std::string(error_msg); + } + } + + bool has_error() const { return !error_msg_.empty(); } + + SimpleArena* arena() { return &arena_; } + + void Reset() { + error_msg_.clear(); + arena_.Reset(); + } + + private: + std::string error_msg_; + SimpleArena arena_; +}; + +} // namespace gandiva |