Make locality hashes work with DEL

src/StateMachine.cc

@@ -1007,6 +1007,19 @@ rocksdb::Status StateMachine::del(StagingArea &stagingArea, const VecIterator &s
       remove_all_with_prefix(locator.toSlice(), count, stagingArea);
       if(count != keyInfo.getSize()) qdb_throw("mismatch between keyInfo counter and number of elements deleted by remove_all_with_prefix: " << count << " vs " << keyInfo.getSize());
     }
+    else if(keyInfo.getKeyType() == KeyType::kLocalityHash) {
+      // wipe out fields
+      LocalityFieldLocator fieldLocator(*it);
+      int64_t count = 0;
+      remove_all_with_prefix(fieldLocator.toSlice(), count, stagingArea);
+      if(count != keyInfo.getSize()) qdb_throw("mismatch between keyInfo counter and number of elements deleted by remove_all_with_prefix: " << count << " vs " << keyInfo.getSize());
+
+      // wipe out indexes
+      LocalityIndexLocator indexLocator(*it);
+      count = 0;
+      remove_all_with_prefix(indexLocator.toSlice(), count, stagingArea);
+      if(count != keyInfo.getSize()) qdb_throw("mismatch between keyInfo counter and number of elements deleted by remove_all_with_prefix: " << count << " vs " << keyInfo.getSize());
+    }
     else {
       qdb_throw("should never happen");
     }

test/e2e.cc

@@ -1000,4 +1000,10 @@ TEST_F(Raft_e2e, LocalityHash) {
   ASSERT_REPLY(tunnel(leaderID)->exec("lhget", "mykey", "f1"), "v2");
   ASSERT_REPLY(tunnel(leaderID)->exec("lhget", "mykey", "f1", "hint2"), "v2");
   ASSERT_REPLY(tunnel(leaderID)->exec("lhget", "mykey", "f1", "hint1"), "v2");
+
+  // Delete key.
+  ASSERT_REPLY(tunnel(leaderID)->exec("del", "mykey"), 1);
+  ASSERT_REPLY(tunnel(leaderID)->exec("lhlen", "mykey"), 0);
+  ASSERT_REPLY(tunnel(leaderID)->exec("del", "mykey"), 0);
+  ASSERT_REPLY(tunnel(leaderID)->exec("lhget", "mykey", "f3", "aaaaa"), "");
 }