gecko-dev/tools/profiler/gecko/ChildProfilerController.cpp

103 строки
2.9 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 "nsThreadUtils.h"
#include "ProfilerChild.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(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>&&>(
this, &ChildProfilerController::SetupProfilerChild, 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*>(
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 = 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