Skip to content

Commit ed745da

Browse files
committed
Set up resource-usage plugin api so that serverside emits host available memory.
1 parent 97f8708 commit ed745da

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

jupyter_resource_usage/api.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@ async def get(self):
4646
except (psutil.NoSuchProcess, psutil.AccessDenied) as e:
4747
pass
4848

49+
available = psutil.virtual_memory().available
50+
4951
if callable(config.mem_limit):
5052
mem_limit = config.mem_limit(rss=rss, pss=pss)
5153
else: # mem_limit is an Int
@@ -58,6 +60,7 @@ async def get(self):
5860
)
5961

6062
metrics = {"rss": rss, "limits": limits}
63+
metrics["mem_avail"] = available
6164
if pss is not None:
6265
metrics["pss"] = pss
6366

jupyter_resource_usage/config.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,10 @@ class ResourceUseDisplay(Configurable):
5454

5555
system_memory_metrics = List(
5656
trait=PSUtilMetric(),
57-
default_value=[{"name": "virtual_memory", "attribute": "total"}],
57+
default_value=[
58+
{"name": "virtual_memory", "attribute": "total"},
59+
{"name": "virtual_memory", "attribute": "available"}
60+
],
5861
)
5962

6063
process_cpu_metrics = List(

0 commit comments

Comments
 (0)