From 2b4b27c08b80250145470713115ffd8b2ade0031 Mon Sep 17 00:00:00 2001
From: surisfang <surisfang@tencent.com>
Date: Sat, 31 Jul 2021 18:00:16 +0800
Subject: [PATCH] fix keys lost bug when split sparse keys by ranges of servers

---
 src/postoffice.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/postoffice.cc b/src/postoffice.cc
index e9f4386d..edc84cbd 100644
--- a/src/postoffice.cc
+++ b/src/postoffice.cc
@@ -172,7 +172,9 @@ const std::vector<Range>& Postoffice::GetServerKeyRanges() {
     for (int i = 0; i < num_servers_; ++i) {
       server_key_ranges_.push_back(Range(
           kMaxKey / num_servers_ * i,
-          kMaxKey / num_servers_ * (i+1)));
+          // kMaxKey maynot be divided by num_servers_ exactly
+          // the end of the last range is kMaxKey
+          i + 1 == num_servers_ ? kMaxKey : kMaxKey / num_servers_ * (i+1)));
     }
   }
   server_key_ranges_mu_.unlock();