1779c3b7f6
There are two reasons for the move. First is that cql server lifetime is shorter than storage_proxy one and the later stores references to the semaphore in each service_permit it holds. Second - we want thrift (and in the future other user APIs) to share the same admission control memory pool. Fixes #4844 Message-Id: <20190814142614.GT17984@scylladb.com>
120 lines
4.5 KiB
C++
120 lines
4.5 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"
|
|
|
|
|
|
class storage_service_for_tests::impl {
|
|
sharded<gms::feature_service> _feature_service;
|
|
sharded<gms::gossiper> _gossiper;
|
|
distributed<database> _db;
|
|
db::config _cfg;
|
|
sharded<auth::service> _auth_service;
|
|
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"));
|
|
_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();
|
|
service::get_storage_service().start(std::ref(_db), std::ref(_gossiper), std::ref(_auth_service), 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();
|
|
netw::get_messaging_service().stop().get();
|
|
_db.stop().get();
|
|
_gossiper.stop().get();
|
|
_feature_service.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();
|
|
}
|