From 9e799cadd7a95aed350bcce4dc77deccdb515f3d Mon Sep 17 00:00:00 2001
From: Pekka Enberg <penberg@cloudius-systems.com>
Date: Thu, 16 Jul 2015 15:28:39 +0300
Subject: [PATCH] message/messaging_service: MIGRATION_REQUEST wrappers

Signed-off-by: Pekka Enberg <penberg@cloudius-systems.com>
---
 message/messaging_service.cc | 7 +++++++
 message/messaging_service.hh | 4 ++++
 2 files changed, 11 insertions(+)

diff --git a/message/messaging_service.cc b/message/messaging_service.cc
index 06e73272b0..44e5737117 100644
--- a/message/messaging_service.cc
+++ b/message/messaging_service.cc
@@ -397,6 +397,13 @@ future<> messaging_service::send_definitions_update(shard_id id, std::vector<fro
     return send_message_oneway(this, messaging_verb::DEFINITIONS_UPDATE, std::move(id), std::move(fm));
 }
 
+void messaging_service::register_migration_request(std::function<future<std::vector<frozen_mutation>> (gms::inet_address reply_to, unsigned shard)>&& func) {
+    register_handler(this, net::messaging_verb::MIGRATION_REQUEST, std::move(func));
+}
+future<std::vector<frozen_mutation>> messaging_service::send_migration_request(shard_id id, gms::inet_address reply_to, unsigned shard) {
+    return send_message<std::vector<frozen_mutation>>(this, messaging_verb::MIGRATION_REQUEST, std::move(id), std::move(reply_to), std::move(shard));
+}
+
 void messaging_service::register_mutation(std::function<rpc::no_wait_type (frozen_mutation fm, std::vector<inet_address> forward,
     inet_address reply_to, unsigned shard, response_id_type response_id)>&& func) {
     register_handler(this, net::messaging_verb::MUTATION, std::move(func));
diff --git a/message/messaging_service.hh b/message/messaging_service.hh
index b89b6277d4..73446cb6ae 100644
--- a/message/messaging_service.hh
+++ b/message/messaging_service.hh
@@ -466,6 +466,10 @@ public:
     void register_definitions_update(std::function<rpc::no_wait_type (std::vector<frozen_mutation> fm)>&& func);
     future<> send_definitions_update(shard_id id, std::vector<frozen_mutation> fm);
 
+    // Wrapper for MIGRATION_REQUEST
+    void register_migration_request(std::function<future<std::vector<frozen_mutation>> (gms::inet_address reply_to, unsigned shard)>&& func);
+    future<std::vector<frozen_mutation>> send_migration_request(shard_id id, gms::inet_address reply_to, unsigned shard);
+
     // FIXME: response_id_type is an alias in service::storage_proxy::response_id_type
     using response_id_type = uint64_t;
     // Wrapper for MUTATION