зеркало из https://github.com/mozilla/gecko-dev.git
100 строки
3.2 KiB
C++
100 строки
3.2 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* 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 "ChildProfilerController.h"
|
|
|
|
#include "ProfilerChild.h"
|
|
|
|
#include "nsThreadUtils.h"
|
|
|
|
using namespace mozilla::ipc;
|
|
|
|
namespace mozilla {
|
|
|
|
/* static */
|
|
already_AddRefed<ChildProfilerController> ChildProfilerController::Create(
|
|
mozilla::ipc::Endpoint<PProfilerChild>&& aEndpoint) {
|
|
MOZ_RELEASE_ASSERT(NS_IsMainThread());
|
|
RefPtr<ChildProfilerController> cpc = new ChildProfilerController();
|
|
cpc->Init(std::move(aEndpoint));
|
|
return cpc.forget();
|
|
}
|
|
|
|
ChildProfilerController::ChildProfilerController() {
|
|
MOZ_COUNT_CTOR(ChildProfilerController);
|
|
}
|
|
|
|
void ChildProfilerController::Init(Endpoint<PProfilerChild>&& aEndpoint) {
|
|
if (NS_SUCCEEDED(
|
|
NS_NewNamedThread("ProfilerChild", getter_AddRefs(mThread)))) {
|
|
// Now that mThread has been set, run SetupProfilerChild on the thread.
|
|
mThread->Dispatch(
|
|
NewRunnableMethod<Endpoint<PProfilerChild>&&>(
|
|
"ChildProfilerController::SetupProfilerChild", this,
|
|
&ChildProfilerController::SetupProfilerChild, std::move(aEndpoint)),
|
|
NS_DISPATCH_NORMAL);
|
|
}
|
|
}
|
|
|
|
nsCString ChildProfilerController::GrabShutdownProfileAndShutdown() {
|
|
nsCString shutdownProfile;
|
|
ShutdownAndMaybeGrabShutdownProfileFirst(&shutdownProfile);
|
|
return shutdownProfile;
|
|
}
|
|
|
|
void ChildProfilerController::Shutdown() {
|
|
ShutdownAndMaybeGrabShutdownProfileFirst(nullptr);
|
|
}
|
|
|
|
void ChildProfilerController::ShutdownAndMaybeGrabShutdownProfileFirst(
|
|
nsCString* aOutShutdownProfile) {
|
|
if (mThread) {
|
|
mThread->Dispatch(NewRunnableMethod<nsCString*>(
|
|
"ChildProfilerController::ShutdownProfilerChild",
|
|
this, &ChildProfilerController::ShutdownProfilerChild,
|
|
aOutShutdownProfile),
|
|
NS_DISPATCH_NORMAL);
|
|
// Shut down the thread. This call will spin until all runnables (including
|
|
// the ShutdownProfilerChild runnable) have been processed.
|
|
mThread->Shutdown();
|
|
mThread = nullptr;
|
|
}
|
|
}
|
|
|
|
ChildProfilerController::~ChildProfilerController() {
|
|
MOZ_COUNT_DTOR(ChildProfilerController);
|
|
|
|
MOZ_ASSERT(!mThread,
|
|
"Please call Shutdown before destroying ChildProfilerController");
|
|
MOZ_ASSERT(!mProfilerChild);
|
|
}
|
|
|
|
void ChildProfilerController::SetupProfilerChild(
|
|
Endpoint<PProfilerChild>&& aEndpoint) {
|
|
MOZ_RELEASE_ASSERT(mThread == NS_GetCurrentThread());
|
|
MOZ_ASSERT(aEndpoint.IsValid());
|
|
|
|
mProfilerChild = new ProfilerChild();
|
|
Endpoint<PProfilerChild> endpoint = std::move(aEndpoint);
|
|
|
|
if (!endpoint.Bind(mProfilerChild)) {
|
|
MOZ_CRASH("Failed to bind ProfilerChild!");
|
|
}
|
|
}
|
|
|
|
void ChildProfilerController::ShutdownProfilerChild(
|
|
nsCString* aOutShutdownProfile) {
|
|
MOZ_RELEASE_ASSERT(mThread == NS_GetCurrentThread());
|
|
|
|
if (aOutShutdownProfile) {
|
|
*aOutShutdownProfile = mProfilerChild->GrabShutdownProfile();
|
|
}
|
|
mProfilerChild->Destroy();
|
|
mProfilerChild = nullptr;
|
|
}
|
|
|
|
} // namespace mozilla
|