128 lines
4.3 KiB
C++
128 lines
4.3 KiB
C++
// 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.
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <iostream>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
#include "common/config.h"
|
|
#include "exprs/expr.h"
|
|
#include "gen_cpp/DorisExternalService_types.h"
|
|
#include "gen_cpp/Types_types.h"
|
|
#include "gen_cpp/Exprs_types.h"
|
|
#include "olap/options.h"
|
|
#include "runtime/exec_env.h"
|
|
#include "runtime/memory_scratch_sink.h"
|
|
#include "runtime/mem_tracker.h"
|
|
#include "runtime/primitive_type.h"
|
|
#include "runtime/result_queue_mgr.h"
|
|
#include "runtime/row_batch.h"
|
|
#include "runtime/runtime_state.h"
|
|
#include "runtime/thread_resource_mgr.h"
|
|
#include "runtime/tuple_row.h"
|
|
#include "util/blocking_queue.hpp"
|
|
#include "util/logging.h"
|
|
#include "testutil/desc_tbl_builder.h"
|
|
|
|
namespace doris {
|
|
|
|
class MemoryScratchSinkTest : public testing::Test {
|
|
public:
|
|
MemoryScratchSinkTest() {
|
|
// all below is just only for test MemoryScratchSink
|
|
ResultQueueMgr* result_queue_mgr = new ResultQueueMgr();
|
|
ThreadResourceMgr* thread_mgr = new ThreadResourceMgr();
|
|
_exec_env._result_queue_mgr = result_queue_mgr;
|
|
_exec_env._thread_mgr = thread_mgr;
|
|
TQueryOptions query_options;
|
|
query_options.batch_size = 1024;
|
|
TUniqueId query_id;
|
|
query_id.lo = 10;
|
|
query_id.hi = 100;
|
|
_runtime_state = new RuntimeState(query_id, query_options, TQueryGlobals(), &_exec_env);
|
|
_runtime_state->init_instance_mem_tracker();
|
|
{
|
|
TExpr expr;
|
|
{
|
|
TExprNode node;
|
|
|
|
node.node_type = TExprNodeType::INT_LITERAL;
|
|
node.type = gen_type_desc(TPrimitiveType::TINYINT);
|
|
node.num_children = 0;
|
|
TIntLiteral data;
|
|
data.value = 1;
|
|
node.__set_int_literal(data);
|
|
expr.nodes.push_back(node);
|
|
}
|
|
_exprs.push_back(expr);
|
|
}
|
|
_mem_tracker = new MemTracker(-1, "MemoryScratchSinkTest", _runtime_state->instance_mem_tracker());
|
|
|
|
// init _row_desc
|
|
vector<bool> nullable_tuples(1, false);
|
|
vector<TTupleId> tuple_ids(1, static_cast<TTupleId>(0));
|
|
DescriptorTblBuilder int_builder(&_pool);
|
|
int_builder.declare_tuple() << TYPE_INT;
|
|
_row_desc = _pool.add(new RowDescriptor(*int_builder.build(), tuple_ids, nullable_tuples));
|
|
}
|
|
virtual ~MemoryScratchSinkTest() {
|
|
delete _runtime_state;
|
|
delete _mem_tracker;
|
|
}
|
|
|
|
protected:
|
|
virtual void SetUp() {
|
|
}
|
|
|
|
private:
|
|
ObjectPool _pool;
|
|
ExecEnv _exec_env;
|
|
std::vector<TExpr> _exprs;
|
|
RuntimeState* _runtime_state;
|
|
RowDescriptor* _row_desc;
|
|
TMemoryScratchSink _tsink;
|
|
MemTracker *_mem_tracker;
|
|
};
|
|
|
|
TEST_F(MemoryScratchSinkTest, work_flow_normal) {
|
|
MemoryScratchSink sink(*_row_desc, _exprs, _tsink);
|
|
TDataSink data_sink;
|
|
data_sink.memory_scratch_sink = _tsink;
|
|
ASSERT_TRUE(sink.init(data_sink).ok());
|
|
ASSERT_TRUE(sink.prepare(_runtime_state).ok());
|
|
RowBatch row_batch(*_row_desc, 1024, _mem_tracker);
|
|
row_batch.add_row();
|
|
row_batch.commit_last_row();
|
|
ASSERT_TRUE(sink.send(_runtime_state, &row_batch).ok());
|
|
ASSERT_TRUE(sink.close(_runtime_state, Status::OK()).ok());
|
|
}
|
|
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
std::string conffile = std::string(getenv("DORIS_HOME")) + "/conf/be.conf";
|
|
if (!doris::config::init(conffile.c_str(), false)) {
|
|
fprintf(stderr, "error read config file. \n");
|
|
return -1;
|
|
}
|
|
doris::init_glog("be-test");
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
doris::CpuInfo::init();
|
|
return RUN_ALL_TESTS();
|
|
}
|