mirror of
https://github.com/scylladb/scylladb.git
synced 2026-05-30 19:46:48 +00:00
b7fe191e3d
The storage service needs migration manager to sync schema on lifecycle notifiers and to stop the guy on drain. So this patch just pushes the migration manager reference all the way through the storage service constructor. Few words about tests. Since now storage service needs the migration manager in constructor, some tests should take it from somewhere. The cql_test_env already has (and uses) it, all the others can just provide a not-started sharded one, it won't be in use in _those_ tests anyway. Signed-off-by: Pavel Emelyanov <xemul@scylladb.com>
113 lines
4.7 KiB
C++
113 lines
4.7 KiB
C++
|
|
/*
|
|
* Copyright (C) 2015 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 <boost/test/unit_test.hpp>
|
|
|
|
#include <seastar/util/defer.hh>
|
|
|
|
#include <seastar/testing/test_case.hh>
|
|
#include "message/messaging_service.hh"
|
|
#include "gms/failure_detector.hh"
|
|
#include "gms/gossiper.hh"
|
|
#include "gms/feature_service.hh"
|
|
#include <seastar/core/seastar.hh>
|
|
#include "service/storage_service.hh"
|
|
#include <seastar/core/distributed.hh>
|
|
#include <seastar/core/abort_source.hh>
|
|
#include "cdc/generation_service.hh"
|
|
#include "database.hh"
|
|
#include "db/system_distributed_keyspace.hh"
|
|
#include "db/config.hh"
|
|
#include "sstables/compaction_manager.hh"
|
|
|
|
namespace db::view {
|
|
class view_update_generator;
|
|
}
|
|
|
|
SEASTAR_TEST_CASE(test_boot_shutdown){
|
|
return seastar::async([] {
|
|
distributed<database> db;
|
|
database_config dbcfg;
|
|
dbcfg.available_memory = memory::stats().total_memory();
|
|
auto cfg = std::make_unique<db::config>();
|
|
sharded<service::migration_notifier> mm_notif;
|
|
sharded<abort_source> abort_sources;
|
|
sharded<db::system_distributed_keyspace> sys_dist_ks;
|
|
sharded<db::view::view_update_generator> view_update_generator;
|
|
utils::fb_utilities::set_broadcast_address(gms::inet_address("127.0.0.1"));
|
|
sharded<gms::feature_service> feature_service;
|
|
sharded<locator::shared_token_metadata> token_metadata;
|
|
sharded<netw::messaging_service> _messaging;
|
|
sharded<cdc::generation_service> cdc_generation_service;
|
|
sharded<service::migration_manager> migration_manager;
|
|
|
|
token_metadata.start().get();
|
|
auto stop_token_mgr = defer([&token_metadata] { token_metadata.stop().get(); });
|
|
|
|
mm_notif.start().get();
|
|
auto stop_mm_notif = defer([&mm_notif] { mm_notif.stop().get(); });
|
|
|
|
abort_sources.start().get();
|
|
auto stop_abort_sources = defer([&] { abort_sources.stop().get(); });
|
|
|
|
feature_service.start(gms::feature_config_from_db_config(*cfg)).get();
|
|
auto stop_feature_service = defer([&] { feature_service.stop().get(); });
|
|
|
|
locator::i_endpoint_snitch::create_snitch("SimpleSnitch").get();
|
|
auto stop_snitch = defer([&] { locator::i_endpoint_snitch::stop_snitch().get(); });
|
|
|
|
_messaging.start(gms::inet_address("127.0.0.1"), 7000).get();
|
|
auto stop_messaging_service = defer([&] { _messaging.stop().get(); });
|
|
|
|
gms::get_gossiper().start(std::ref(abort_sources), std::ref(feature_service), std::ref(token_metadata), std::ref(_messaging), std::ref(*cfg)).get();
|
|
auto stop_gossiper = defer([&] { gms::get_gossiper().stop().get(); });
|
|
|
|
service::storage_service_config sscfg;
|
|
sscfg.available_memory = memory::stats().total_memory();
|
|
|
|
service::get_storage_service().start(std::ref(abort_sources), std::ref(db), std::ref(gms::get_gossiper()), std::ref(sys_dist_ks), std::ref(view_update_generator), std::ref(feature_service), sscfg, std::ref(mm_notif), std::ref(migration_manager), std::ref(token_metadata), std::ref(_messaging), std::ref(cdc_generation_service), true).get();
|
|
auto stop_ss = defer([&] { service::get_storage_service().stop().get(); });
|
|
|
|
sharded<semaphore> sst_dir_semaphore;
|
|
sst_dir_semaphore.start(cfg->initial_sstable_loading_concurrency()).get();
|
|
auto stop_sst_dir_sem = defer([&sst_dir_semaphore] {
|
|
sst_dir_semaphore.stop().get();
|
|
});
|
|
|
|
db.start(std::ref(*cfg), dbcfg, std::ref(mm_notif), std::ref(feature_service), std::ref(token_metadata), std::ref(abort_sources), std::ref(sst_dir_semaphore)).get();
|
|
db.invoke_on_all([] (database& db) {
|
|
db.get_compaction_manager().enable();
|
|
}).get();
|
|
|
|
auto stop_db = defer([&] { db.stop().get(); });
|
|
auto stop_database_d = defer([&db] {
|
|
stop_database(db).get();
|
|
});
|
|
|
|
cdc_generation_service.start(std::ref(*cfg), std::ref(gms::get_gossiper()), std::ref(sys_dist_ks), std::ref(abort_sources), std::ref(token_metadata)).get();
|
|
auto stop_cdc_generation_service = defer([&cdc_generation_service] {
|
|
cdc_generation_service.stop().get();
|
|
});
|
|
});
|
|
}
|