mirrors/scylladb
1
0
Fork 0
mirror of https://github.com/scylladb/scylladb.git synced 2026-04-21 09:00:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

service/migration_manager: Migration listener hooks

Browse Source 
Signed-off-by: Pekka Enberg <penberg@cloudius-systems.com>
This commit is contained in:
Pekka Enberg
2015-08-03 10:41:35 +03:00
parent 5b3f7f8091
commit 12d99bd282
2 changed files with 69 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
service/migration_manager.cc
View File

@@ -24,6 +24,7 @@
#include "service/migration_manager.hh"
#include "service/migration_listener.hh"
#include "message/messaging_service.hh"
#include "service/storage_service.hh"
#include "service/migration_task.hh"
@@ -41,6 +42,7 @@ using namespace std::chrono_literals;
const std::chrono::milliseconds migration_manager::MIGRATION_DELAY_IN_MS = 60000ms;
migration_manager::migration_manager()
: _listeners{}
{
}
@@ -49,17 +51,15 @@ future<> migration_manager::stop()
return make_ready_future<>();
}
#if 0
public void register(IMigrationListener listener)
void migration_manager::register_listener(migration_listener* listener)
{
listeners.add(listener);
_listeners.emplace_back(listener);
}
public void unregister(IMigrationListener listener)
void migration_manager::unregister_listener(migration_listener* listener)
{
listeners.remove(listener);
_listeners.erase(std::remove(_listeners.begin(), _listeners.end(), listener), _listeners.end());
}
#endif
future<> migration_manager::schedule_schema_pull(const gms::inet_address& endpoint, const gms::endpoint_state& state)
{
@@ -135,19 +135,27 @@ public static boolean isReadyForBootstrap()
{
return ((ThreadPoolExecutor) StageManager.getStage(Stage.MIGRATION)).getActiveCount() == 0;
}
#endif
public void notifyCreateKeyspace(KSMetaData ksm)
future<> migration_manager::notify_create_keyspace(const lw_shared_ptr<keyspace_metadata>& ksm)
{
for (IMigrationListener listener : listeners)
listener.onCreateKeyspace(ksm.name);
return get_migration_manager().invoke_on_all([name = ksm->name()] (auto&& mm) {
for (auto&& listener : mm._listeners) {
listener->on_create_keyspace(name);
}
});
}
public void notifyCreateColumnFamily(CFMetaData cfm)
future<> migration_manager::notify_create_column_family(schema_ptr cfm)
{
for (IMigrationListener listener : listeners)
listener.onCreateColumnFamily(cfm.ksName, cfm.cfName);
return get_migration_manager().invoke_on_all([ks_name = cfm->ks_name(), cf_name = cfm->cf_name()] (auto&& mm) {
for (auto&& listener : mm._listeners) {
listener->on_create_column_family(ks_name, cf_name);
}
});
}
#if 0
public void notifyCreateUserType(UserType ut)
{
for (IMigrationListener listener : listeners)
@@ -165,19 +173,27 @@ public void notifyCreateAggregate(UDAggregate udf)
for (IMigrationListener listener : listeners)
listener.onCreateAggregate(udf.name().keyspace, udf.name().name);
}
#endif
public void notifyUpdateKeyspace(KSMetaData ksm)
future<> migration_manager::notify_update_keyspace(const lw_shared_ptr<keyspace_metadata>& ksm)
{
for (IMigrationListener listener : listeners)
listener.onUpdateKeyspace(ksm.name);
return get_migration_manager().invoke_on_all([name = ksm->name()] (auto&& mm) {
for (auto&& listener : mm._listeners) {
listener->on_update_keyspace(name);
}
});
}
public void notifyUpdateColumnFamily(CFMetaData cfm)
future<> migration_manager::notify_update_column_family(schema_ptr cfm)
{
for (IMigrationListener listener : listeners)
listener.onUpdateColumnFamily(cfm.ksName, cfm.cfName);
return get_migration_manager().invoke_on_all([ks_name = cfm->ks_name(), cf_name = cfm->cf_name()] (auto&& mm) {
for (auto&& listener : mm._listeners) {
listener->on_update_column_family(ks_name, cf_name);
}
});
}
#if 0
public void notifyUpdateUserType(UserType ut)
{
for (IMigrationListener listener : listeners)
@@ -195,19 +211,27 @@ public void notifyUpdateAggregate(UDAggregate udf)
for (IMigrationListener listener : listeners)
listener.onUpdateAggregate(udf.name().keyspace, udf.name().name);
}
#endif
public void notifyDropKeyspace(KSMetaData ksm)
future<> migration_manager::notify_drop_keyspace(const lw_shared_ptr<keyspace_metadata>& ksm)
{
for (IMigrationListener listener : listeners)
listener.onDropKeyspace(ksm.name);
return get_migration_manager().invoke_on_all([name = ksm->name()] (auto&& mm) {
for (auto&& listener : mm._listeners) {
listener->on_drop_keyspace(name);
}
});
}
public void notifyDropColumnFamily(CFMetaData cfm)
future<> migration_manager::notify_drop_column_family(schema_ptr cfm)
{
for (IMigrationListener listener : listeners)
listener.onDropColumnFamily(cfm.ksName, cfm.cfName);
return get_migration_manager().invoke_on_all([ks_name = cfm->ks_name(), cf_name = cfm->cf_name()] (auto&& mm) {
for (auto&& listener : mm._listeners) {
listener->on_drop_column_family(ks_name, cf_name);
}
});
}
#if 0
public void notifyDropUserType(UserType ut)
{
for (IMigrationListener listener : listeners)

26
service/migration_manager.hh
View File

@@ -24,8 +24,8 @@
#pragma once
#include "service/migration_listener.hh"
#include "db/legacy_schema_tables.hh"
#include "gms/endpoint_state.hh"
#include "core/distributed.hh"
#include "gms/inet_address.hh"
@@ -42,20 +42,36 @@ class migration_manager {
private final RuntimeMXBean runtimeMXBean = ManagementFactory.getRuntimeMXBean();
#endif
static const std::chrono::milliseconds MIGRATION_DELAY_IN_MS;
std::vector<migration_listener*> _listeners;
#if 0
private final List<IMigrationListener> listeners = new CopyOnWriteArrayList<>();
#endif
static const std::chrono::milliseconds MIGRATION_DELAY_IN_MS;
public:
migration_manager();
/// Register a migration listener on current shard.
void register_listener(migration_listener* listener);
/// Unregister a migration listener on current shard.
void unregister_listener(migration_listener* listener);
future<> schedule_schema_pull(const gms::inet_address& endpoint, const gms::endpoint_state& state);
future<> maybe_schedule_schema_pull(service::storage_proxy& proxy, const utils::UUID& their_version, const gms::inet_address& endpoint);
future<> submit_migration_task(service::storage_proxy& proxy, const gms::inet_address& endpoint);
static future<> notify_create_keyspace(const lw_shared_ptr<keyspace_metadata>& ksm);
static future<> notify_create_column_family(schema_ptr cfm);
static future<> notify_update_keyspace(const lw_shared_ptr<keyspace_metadata>& ksm);
static future<> notify_update_column_family(schema_ptr cfm);
static future<> notify_drop_keyspace(const lw_shared_ptr<keyspace_metadata>& ksm);
static future<> notify_drop_column_family(schema_ptr cfm);
bool should_pull_schema_from(const gms::inet_address& endpoint);
future<> announce_new_keyspace(distributed<service::storage_proxy>& proxy, lw_shared_ptr<keyspace_metadata> ksm, bool announce_locally = false);