From 34b9c7449b4dc1a898ceddebc146e2b062ca02f7 Mon Sep 17 00:00:00 2001
From: Max Fitton <maxfitton@anyscale.com>
Date: Mon, 7 Dec 2020 21:40:49 -0800
Subject: [PATCH] [Dashboard] Fix object store memory display. (#12664)

---
 .../dashboard/node-info/features/ObjectStoreMemory.tsx | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/dashboard/client/src/pages/dashboard/node-info/features/ObjectStoreMemory.tsx b/dashboard/client/src/pages/dashboard/node-info/features/ObjectStoreMemory.tsx
index 0201040ff..95f4421b1 100644
--- a/dashboard/client/src/pages/dashboard/node-info/features/ObjectStoreMemory.tsx
+++ b/dashboard/client/src/pages/dashboard/node-info/features/ObjectStoreMemory.tsx
@@ -18,20 +18,22 @@ export const ClusterObjectStoreMemory: ClusterFeatureRenderFn = ({ nodes }) => {
     nodes.map((n) => n.raylet.objectStoreAvailableMemory),
   );
   const totalUsed = sum(nodes.map((n) => n.raylet.objectStoreUsedMemory));
+  const total = totalUsed + totalAvailable;
   return (
     <div style={{ minWidth: 60 }}>
       <UsageBar
-        percent={100 * (totalUsed / totalAvailable)}
-        text={formatUsage(totalUsed, totalAvailable, "mebibyte", false)}
+        percent={100 * (totalUsed / total)}
+        text={formatUsage(totalUsed, total, "mebibyte", false)}
       />
     </div>
   );
 };
 
 export const NodeObjectStoreMemory: NodeFeatureRenderFn = ({ node }) => {
-  const total = node.raylet.objectStoreAvailableMemory;
+  const totalAvailable = node.raylet.objectStoreAvailableMemory;
   const used = node.raylet.objectStoreUsedMemory;
-  if (used === undefined || total === undefined || total === 0) {
+  const total = totalAvailable + used;
+  if (used === undefined || totalAvailable === undefined || total === 0) {
     return (
       <Typography color="textSecondary" component="span" variant="inherit">
         N/A