From 7779637fa4886a0dedc035214631ee9f7c5797d4 Mon Sep 17 00:00:00 2001
From: Stephanie Wang <swang@cs.berkeley.edu>
Date: Wed, 10 Feb 2021 12:22:08 -0800
Subject: [PATCH] Subtract from num bytes in use (#13944)

---
 src/ray/object_manager/plasma/store.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/ray/object_manager/plasma/store.cc b/src/ray/object_manager/plasma/store.cc
index 9bae68b3a..d12a8096a 100644
--- a/src/ray/object_manager/plasma/store.cc
+++ b/src/ray/object_manager/plasma/store.cc
@@ -562,6 +562,10 @@ void PlasmaStore::EraseFromObjectTable(const ObjectID &object_id) {
   if (object->device_num == 0) {
     PlasmaAllocator::Free(object->pointer, buff_size);
   }
+  if (object->ref_count > 0) {
+    // A client was using this object.
+    num_bytes_in_use_ -= object->data_size + object->metadata_size;
+  }
   store_info_.objects.erase(object_id);
 }