# HG changeset patch
# User David Douard <david.douard@logilab.fr>
# Date 1329133264 -3600
# Node ID e35d4d4f7eb3d2ca7b064a964c669d3ecf9af559
# Parent  0a467663c3feac464e7614794c864cf26bdde83d
[repository] add a ``gc_stats`` method and 2 more cache size for ``stats`` (closes #2179735)

This method returns memory informations in a dictionnary (same informations
as the content displayed by the GCView).

diff -r 0a467663c3fe -r e35d4d4f7eb3 server/repository.py
--- a/server/repository.py	Fri Feb 10 17:20:38 2012 +0100
+++ b/server/repository.py	Mon Feb 13 12:41:04 2012 +0100
@@ -492,6 +492,8 @@
             results['%s_cache_hit' % title] =  hits
             results['%s_cache_miss' % title] = misses
             results['%s_cache_hit_percent' % title] = (hits * 100) / (hits + misses)
+        results['type_source_cache_size'] = len(self._type_source_cache)
+        results['extid_cache_size'] = len(self._extid_cache)
         results['sql_no_cache'] = self.system_source.no_cache
         results['nb_open_sessions'] = len(self._sessions)
         results['nb_active_threads'] = threading.activeCount()
@@ -500,6 +502,43 @@
         results['threads'] = ', '.join(sorted(str(t) for t in threading.enumerate()))
         return results
 
+    def gc_stats(self, nmax=20):
+        """Return a dictionary containing some statistics about the repository
+        memory usage.
+
+        This is a public method, not requiring a session id.
+
+        nmax is the max number of (most) referenced object returned as
+        the 'referenced' result
+        """
+
+        from cubicweb._gcdebug import gc_info
+        from cubicweb.appobject import AppObject
+        from cubicweb.rset import ResultSet
+        from cubicweb.dbapi import Connection, Cursor
+        from cubicweb.web.request import CubicWebRequestBase
+        from rql.stmts import Union
+
+        lookupclasses = (AppObject,
+                         Union, ResultSet,
+                         Connection, Cursor,
+                         CubicWebRequestBase)
+        try:
+            from cubicweb.server.session import Session, InternalSession
+            lookupclasses += (InternalSession, Session)
+        except ImportError:
+            pass # no server part installed
+
+        results = {}
+        counters, ocounters, garbage = gc_info(lookupclasses,
+                                               viewreferrersclasses=())
+        values = sorted(counters.iteritems(), key=lambda x: x[1], reverse=True)
+        results['lookupclasses'] = values
+        values = sorted(ocounters.iteritems(), key=lambda x: x[1], reverse=True)[:nmax]
+        results['referenced'] = values
+        results['unreachable'] = len(garbage)
+        return results
+
     def get_schema(self):
         """Return the instance schema.