mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-23 10:15:41 +00:00
48517afae6
MozPromise most common use is to have an single or exclusive listener. By making the MozPromise generated by IPDL exclusive we can also use move semantics. While at it, we also use move semantics for the ResponseRejectReason and via the callback's reject method so that the lambda used with the MozPromise::Then can be identical to the one used by the IPDL callback. As it currently is, it provides no advantage over a copy as it's just an enum; however, this will facilitate future changes where it may not be. Differential Revision: https://phabricator.services.mozilla.com/D13906 --HG-- extra : moz-landing-system : lando
94 lines
2.4 KiB
C++
94 lines
2.4 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
* You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "nsCodeCoverage.h"
|
|
#include "mozilla/CodeCoverageHandler.h"
|
|
#include "mozilla/Unused.h"
|
|
#include "mozilla/dom/ContentParent.h"
|
|
#include "mozilla/dom/Promise.h"
|
|
|
|
using namespace mozilla;
|
|
using namespace mozilla::dom;
|
|
using namespace mozilla::ipc;
|
|
|
|
NS_IMPL_ISUPPORTS(nsCodeCoverage, nsICodeCoverage)
|
|
|
|
nsCodeCoverage::nsCodeCoverage() {}
|
|
|
|
nsCodeCoverage::~nsCodeCoverage() {}
|
|
|
|
enum RequestType { Flush };
|
|
|
|
class ProcessCount final {
|
|
NS_INLINE_DECL_REFCOUNTING(ProcessCount);
|
|
|
|
public:
|
|
explicit ProcessCount(uint32_t c) : mCount(c) {}
|
|
operator uint32_t() const { return mCount; }
|
|
ProcessCount& operator--() {
|
|
mCount--;
|
|
return *this;
|
|
}
|
|
|
|
private:
|
|
~ProcessCount() {}
|
|
uint32_t mCount;
|
|
};
|
|
|
|
nsresult Request(JSContext* cx, Promise** aPromise, RequestType requestType) {
|
|
MOZ_ASSERT(XRE_IsParentProcess());
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
nsIGlobalObject* global = xpc::CurrentNativeGlobal(cx);
|
|
if (NS_WARN_IF(!global)) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
ErrorResult result;
|
|
RefPtr<Promise> promise = Promise::Create(global, result);
|
|
if (NS_WARN_IF(result.Failed())) {
|
|
return result.StealNSResult();
|
|
}
|
|
|
|
uint32_t processCount = 0;
|
|
for (auto* cp : ContentParent::AllProcesses(ContentParent::eLive)) {
|
|
Unused << cp;
|
|
++processCount;
|
|
}
|
|
|
|
if (requestType == RequestType::Flush) {
|
|
CodeCoverageHandler::FlushCounters();
|
|
}
|
|
|
|
if (processCount == 0) {
|
|
promise->MaybeResolveWithUndefined();
|
|
} else {
|
|
RefPtr<ProcessCount> processCountHolder(new ProcessCount(processCount));
|
|
|
|
auto resolve = [processCountHolder, promise](bool unused) {
|
|
if (--(*processCountHolder) == 0) {
|
|
promise->MaybeResolveWithUndefined();
|
|
}
|
|
};
|
|
|
|
auto reject = [promise](ResponseRejectReason&& aReason) {
|
|
promise->MaybeReject(NS_ERROR_FAILURE);
|
|
};
|
|
|
|
for (auto* cp : ContentParent::AllProcesses(ContentParent::eLive)) {
|
|
if (requestType == RequestType::Flush) {
|
|
cp->SendFlushCodeCoverageCounters(resolve, reject);
|
|
}
|
|
}
|
|
}
|
|
|
|
promise.forget(aPromise);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP nsCodeCoverage::FlushCounters(JSContext* cx, Promise** aPromise) {
|
|
return Request(cx, aPromise, RequestType::Flush);
|
|
}
|