xpcom/ds/nsObserverList.cpp

changeset 0
6474c204b198
     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/xpcom/ds/nsObserverList.cpp	Wed Dec 31 06:09:35 2014 +0100
     1.3 @@ -0,0 +1,136 @@
     1.4 +/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
     1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public
     1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this
     1.7 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
     1.8 +
     1.9 +#include "nsObserverList.h"
    1.10 +
    1.11 +#include "nsAutoPtr.h"
    1.12 +#include "nsCOMArray.h"
    1.13 +#include "nsISimpleEnumerator.h"
    1.14 +#include "xpcpublic.h"
    1.15 +
    1.16 +nsresult
    1.17 +nsObserverList::AddObserver(nsIObserver* anObserver, bool ownsWeak)
    1.18 +{
    1.19 +    NS_ASSERTION(anObserver, "Null input");
    1.20 +
    1.21 +    if (!ownsWeak) {
    1.22 +        ObserverRef* o = mObservers.AppendElement(anObserver);
    1.23 +        if (!o)
    1.24 +            return NS_ERROR_OUT_OF_MEMORY;
    1.25 +
    1.26 +        return NS_OK;
    1.27 +    }
    1.28 +        
    1.29 +    nsCOMPtr<nsIWeakReference> weak = do_GetWeakReference(anObserver);
    1.30 +    if (!weak)
    1.31 +        return NS_NOINTERFACE;
    1.32 +
    1.33 +    ObserverRef *o = mObservers.AppendElement(weak);
    1.34 +    if (!o)
    1.35 +        return NS_ERROR_OUT_OF_MEMORY;
    1.36 +
    1.37 +    return NS_OK;
    1.38 +}
    1.39 +
    1.40 +nsresult
    1.41 +nsObserverList::RemoveObserver(nsIObserver* anObserver)
    1.42 +{
    1.43 +    NS_ASSERTION(anObserver, "Null input");
    1.44 +
    1.45 +    if (mObservers.RemoveElement(static_cast<nsISupports*>(anObserver)))
    1.46 +        return NS_OK;
    1.47 +
    1.48 +    nsCOMPtr<nsIWeakReference> observerRef = do_GetWeakReference(anObserver);
    1.49 +    if (!observerRef)
    1.50 +        return NS_ERROR_FAILURE;
    1.51 +
    1.52 +    if (!mObservers.RemoveElement(observerRef))
    1.53 +        return NS_ERROR_FAILURE;
    1.54 +
    1.55 +    return NS_OK;
    1.56 +}
    1.57 +
    1.58 +nsresult
    1.59 +nsObserverList::GetObserverList(nsISimpleEnumerator** anEnumerator)
    1.60 +{
    1.61 +    nsRefPtr<nsObserverEnumerator> e(new nsObserverEnumerator(this));
    1.62 +    e.forget(anEnumerator);
    1.63 +    return NS_OK;
    1.64 +}
    1.65 +
    1.66 +void
    1.67 +nsObserverList::FillObserverArray(nsCOMArray<nsIObserver> &aArray)
    1.68 +{
    1.69 +    aArray.SetCapacity(mObservers.Length());
    1.70 +
    1.71 +    nsTArray<ObserverRef> observers(mObservers);
    1.72 +
    1.73 +    for (int32_t i = observers.Length() - 1; i >= 0; --i) {
    1.74 +        if (observers[i].isWeakRef) {
    1.75 +            nsCOMPtr<nsIObserver> o(do_QueryReferent(observers[i].asWeak()));
    1.76 +            if (o) {
    1.77 +                aArray.AppendObject(o);
    1.78 +            }
    1.79 +            else {
    1.80 +                // the object has gone away, remove the weakref
    1.81 +                mObservers.RemoveElement(observers[i].asWeak());
    1.82 +            }
    1.83 +        }
    1.84 +        else {
    1.85 +            aArray.AppendObject(observers[i].asObserver());
    1.86 +        }
    1.87 +    }
    1.88 +}
    1.89 +
    1.90 +void
    1.91 +nsObserverList::NotifyObservers(nsISupports *aSubject,
    1.92 +                                const char *aTopic,
    1.93 +                                const char16_t *someData)
    1.94 +{
    1.95 +    nsCOMArray<nsIObserver> observers;
    1.96 +    FillObserverArray(observers);
    1.97 +
    1.98 +    for (int32_t i = 0; i < observers.Count(); ++i) {
    1.99 +        observers[i]->Observe(aSubject, aTopic, someData);
   1.100 +    }
   1.101 +}
   1.102 +
   1.103 +void
   1.104 +nsObserverList::UnmarkGrayStrongObservers()
   1.105 +{
   1.106 +    for (uint32_t i = 0; i < mObservers.Length(); ++i) {
   1.107 +        if (!mObservers[i].isWeakRef) {
   1.108 +            xpc_TryUnmarkWrappedGrayObject(mObservers[i].asObserver());
   1.109 +        }
   1.110 +    }
   1.111 +}
   1.112 +
   1.113 +NS_IMPL_ISUPPORTS(nsObserverEnumerator, nsISimpleEnumerator)
   1.114 +
   1.115 +nsObserverEnumerator::nsObserverEnumerator(nsObserverList* aObserverList)
   1.116 +    : mIndex(0)
   1.117 +{
   1.118 +    aObserverList->FillObserverArray(mObservers);
   1.119 +}
   1.120 +
   1.121 +NS_IMETHODIMP
   1.122 +nsObserverEnumerator::HasMoreElements(bool *aResult)
   1.123 +{
   1.124 +    *aResult = (mIndex < mObservers.Count());
   1.125 +    return NS_OK;
   1.126 +}
   1.127 +
   1.128 +NS_IMETHODIMP
   1.129 +nsObserverEnumerator::GetNext(nsISupports* *aResult)
   1.130 +{
   1.131 +    if (mIndex == mObservers.Count()) {
   1.132 +        NS_ERROR("Enumerating after HasMoreElements returned false.");
   1.133 +        return NS_ERROR_UNEXPECTED;
   1.134 +    }
   1.135 +
   1.136 +    NS_ADDREF(*aResult = mObservers[mIndex]);
   1.137 +    ++mIndex;
   1.138 +    return NS_OK;
   1.139 +}

mercurial