From d124df17b4bfb9a2423ba6fe1fc341dc943c0edb Mon Sep 17 00:00:00 2001
From: Asias He <asias@scylladb.com>
Date: Thu, 5 Nov 2015 16:53:19 +0800
Subject: [PATCH] storage_service: Use get_local_tokens in set_tokens

We are using _bootstrap_tokens since get_local_tokens was not available.
---
 service/storage_service.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/service/storage_service.cc b/service/storage_service.cc
index 57efe1e68f..6d3abe681b 100644
--- a/service/storage_service.cc
+++ b/service/storage_service.cc
@@ -847,8 +847,7 @@ void storage_service::set_tokens(std::unordered_set<token> tokens) {
     logger.debug("Setting tokens to {}", tokens);
     db::system_keyspace::update_tokens(tokens).get();
     _token_metadata.update_normal_tokens(tokens, get_broadcast_address());
-    // Collection<Token> localTokens = getLocalTokens();
-    auto local_tokens = _bootstrap_tokens;
+    auto local_tokens = get_local_tokens();
     auto& gossiper = gms::get_local_gossiper();
     gossiper.add_local_application_state(gms::application_state::TOKENS, value_factory.tokens(local_tokens)).get();
     gossiper.add_local_application_state(gms::application_state::STATUS, value_factory.normal(local_tokens)).get();