3a44fa9988
We need a way to configure the cql interpreter and runtime. So far we relied on accessing the configuration class via various backdoors, but that causes its own problems around initialization order and testability. To avoid that, this patch adds an empty cql_config class and propagates it from main.cc (and from tests) to the cql interpreter via the query_options class, which is already passed everywhere. Later patches will fill it with contents.
127 lines
4.8 KiB
C++
127 lines
4.8 KiB
C++
/*
|
|
* Copyright (C) 2019 ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* This file is part of Scylla.
|
|
*
|
|
* Scylla is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Scylla is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Scylla. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "tests/test_services.hh"
|
|
#include "auth/service.hh"
|
|
#include "db/config.hh"
|
|
#include "db/system_distributed_keyspace.hh"
|
|
#include "db/view/view_update_generator.hh"
|
|
#include "dht/i_partitioner.hh"
|
|
#include "dht/murmur3_partitioner.hh"
|
|
#include "gms/feature_service.hh"
|
|
#include "gms/gossiper.hh"
|
|
#include "message/messaging_service.hh"
|
|
#include "service/storage_service.hh"
|
|
#include "cql3/cql_config.hh"
|
|
|
|
|
|
class storage_service_for_tests::impl {
|
|
sharded<abort_source> _abort_source;
|
|
sharded<gms::feature_service> _feature_service;
|
|
sharded<gms::gossiper> _gossiper;
|
|
distributed<database> _db;
|
|
db::config _cfg;
|
|
sharded<auth::service> _auth_service;
|
|
sharded<cql3::cql_config> _cql_config;
|
|
sharded<db::system_distributed_keyspace> _sys_dist_ks;
|
|
sharded<db::view::view_update_generator> _view_update_generator;
|
|
public:
|
|
impl() {
|
|
auto thread = seastar::thread_impl::get();
|
|
assert(thread);
|
|
_cfg.broadcast_to_all_shards().get();
|
|
utils::fb_utilities::set_broadcast_address(gms::inet_address("localhost"));
|
|
utils::fb_utilities::set_broadcast_rpc_address(gms::inet_address("localhost"));
|
|
_abort_source.start().get();
|
|
_feature_service.start().get();
|
|
_gossiper.start(std::ref(_feature_service), std::ref(_cfg)).get();
|
|
netw::get_messaging_service().start(gms::inet_address("127.0.0.1"), 7000, false).get();
|
|
service::storage_service_config sscfg;
|
|
sscfg.available_memory = memory::stats().total_memory();
|
|
_cql_config.start().get();
|
|
service::get_storage_service().start(std::ref(_abort_source), std::ref(_db), std::ref(_gossiper), std::ref(_auth_service), std::ref(_cql_config), std::ref(_sys_dist_ks), std::ref(_view_update_generator), std::ref(_feature_service), sscfg, true).get();
|
|
service::get_storage_service().invoke_on_all([] (auto& ss) {
|
|
ss.enable_all_features();
|
|
}).get();
|
|
}
|
|
~impl() {
|
|
service::get_storage_service().stop().get();
|
|
_cql_config.stop().get();
|
|
netw::get_messaging_service().stop().get();
|
|
_db.stop().get();
|
|
_gossiper.stop().get();
|
|
_feature_service.stop().get();
|
|
_abort_source.stop().get();
|
|
}
|
|
};
|
|
|
|
storage_service_for_tests::storage_service_for_tests() : _impl(std::make_unique<impl>()) {
|
|
}
|
|
|
|
storage_service_for_tests::~storage_service_for_tests() = default;
|
|
|
|
dht::token create_token_from_key(sstring key) {
|
|
sstables::key_view key_view = sstables::key_view(bytes_view(reinterpret_cast<const signed char*>(key.c_str()), key.size()));
|
|
dht::token token = dht::global_partitioner().get_token(key_view);
|
|
assert(token == dht::global_partitioner().get_token(key_view));
|
|
return token;
|
|
}
|
|
|
|
range<dht::token> create_token_range_from_keys(sstring start_key, sstring end_key) {
|
|
dht::token start = create_token_from_key(start_key);
|
|
assert(engine().cpu_id() == dht::global_partitioner().shard_of(start));
|
|
dht::token end = create_token_from_key(end_key);
|
|
assert(engine().cpu_id() == dht::global_partitioner().shard_of(end));
|
|
assert(end >= start);
|
|
return range<dht::token>::make(start, end);
|
|
}
|
|
|
|
static const sstring some_keyspace("ks");
|
|
static const sstring some_column_family("cf");
|
|
|
|
db::nop_large_data_handler nop_lp_handler;
|
|
thread_local sstables::sstables_manager test_sstables_manager(nop_lp_handler);
|
|
|
|
column_family::config column_family_test_config() {
|
|
column_family::config cfg;
|
|
cfg.sstables_manager = &test_sstables_manager;
|
|
return cfg;
|
|
}
|
|
|
|
column_family_for_tests::column_family_for_tests()
|
|
: column_family_for_tests(
|
|
schema_builder(some_keyspace, some_column_family)
|
|
.with_column(utf8_type->decompose("p1"), utf8_type, column_kind::partition_key)
|
|
.build()
|
|
)
|
|
{ }
|
|
|
|
column_family_for_tests::column_family_for_tests(schema_ptr s)
|
|
: _data(make_lw_shared<data>())
|
|
{
|
|
_data->s = s;
|
|
_data->cfg = column_family_test_config();
|
|
_data->cfg.enable_disk_writes = false;
|
|
_data->cfg.enable_commitlog = false;
|
|
_data->cf = make_lw_shared<column_family>(_data->s, _data->cfg, column_family::no_commitlog(), _data->cm, _data->cl_stats, _data->tracker);
|
|
_data->cf->mark_ready_for_writes();
|
|
}
|