xpcom/base/nsMemoryReporterManager.cpp

   1
   2 #include "nsCOMPtr.h"
   3 #include "nsServiceManagerUtils.h"
   4 #include "nsMemoryReporterManager.h"
   5
   6 #include "nsArrayEnumerator.h"
   7
   8 NS_IMPL_ISUPPORTS1(nsMemoryReporterManager, nsIMemoryReporterManager)
   9
  10 NS_IMETHODIMP
  11 nsMemoryReporterManager::EnumerateReporters(nsISimpleEnumerator **result)
  12 {
  13     return NS_NewArrayEnumerator(result, mReporters);
  14 }
  15
  16 NS_IMETHODIMP
  17 nsMemoryReporterManager::RegisterReporter(nsIMemoryReporter *reporter)
  18 {
  19     if (mReporters.IndexOf(reporter) != -1)
  20         return NS_ERROR_FAILURE;
  21
  22     mReporters.AppendObject(reporter);
  23     return NS_OK;
  24 }
  25
  26 NS_IMETHODIMP
  27 nsMemoryReporterManager::UnregisterReporter(nsIMemoryReporter *reporter)
  28 {
  29     if (!mReporters.RemoveObject(reporter))
  30         return NS_ERROR_FAILURE;
  31
  32     return NS_OK;
  33 }
  34
  35 NS_COM nsresult
  36 NS_RegisterMemoryReporter (nsIMemoryReporter *reporter)
  37 {
  38     nsCOMPtr<nsIMemoryReporterManager> mgr = do_GetService("@mozilla.org/memory-reporter-manager;1");
  39     if (mgr == nsnull)
  40         return NS_ERROR_FAILURE;
  41     return mgr->RegisterReporter(reporter);
  42 }
  43
  44 NS_COM nsresult
  45 NS_UnregisterMemoryReporter (nsIMemoryReporter *reporter)
  46 {
  47     nsCOMPtr<nsIMemoryReporterManager> mgr = do_GetService("@mozilla.org/memory-reporter-manager;1");
  48     if (mgr == nsnull)
  49         return NS_ERROR_FAILURE;
  50     return mgr->UnregisterReporter(reporter);
  51 }
  52