From e35968cda166399722eecf343846f1edcdbea21c Mon Sep 17 00:00:00 2001
From: Gabriel Ebner <gebner@gebner.org>
Date: Tue, 14 Mar 2017 08:10:17 +0100
Subject: [PATCH] feat(util/memory): implement get_current_rss using the
 jemalloc API

---
 src/util/memory.cpp | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/src/util/memory.cpp b/src/util/memory.cpp
index 7bd19614c1..1aa02ef122 100644
--- a/src/util/memory.cpp
+++ b/src/util/memory.cpp
@@ -15,7 +15,27 @@ Author: Leonardo de Moura
 #define LEAN_CHECK_MEM_THRESHOLD 200
 #endif
 
-#if defined(LEAN_WINDOWS)
+#if defined(HAS_JEMALLOC)
+#include <jemalloc/jemalloc.h>
+
+namespace lean {
+
+size_t get_peak_rss() {
+    size_t allocated;
+    size_t sz = sizeof(size_t);
+    if (mallctl("stats.allocated", &allocated, &sz, NULL, 0) == 0) {
+        return allocated;
+    } else {
+        return 0;
+    }
+}
+
+size_t get_current_rss() {
+    return get_peak_rss();
+}
+
+}
+#elif defined(LEAN_WINDOWS)
 /* ----------------------------------------------------
    Windows version for get_peak_rss and get_current_rss
    --------------------------------------------------- */