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 +}