60 lines
1.7 KiB
C++
60 lines
1.7 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#include "TracingAgent.h"
|
|
|
|
#include <reactperflogger/fusebox/FuseboxTracer.h>
|
|
|
|
namespace facebook::react::jsinspector_modern {
|
|
|
|
bool TracingAgent::handleRequest(const cdp::PreparsedRequest& req) {
|
|
if (req.method == "Tracing.start") {
|
|
// @cdp Tracing.start support is experimental.
|
|
if (FuseboxTracer::getFuseboxTracer().startTracing()) {
|
|
frontendChannel_(cdp::jsonResult(req.id));
|
|
} else {
|
|
frontendChannel_(cdp::jsonError(
|
|
req.id,
|
|
cdp::ErrorCode::InternalError,
|
|
"Tracing session already started"));
|
|
}
|
|
|
|
return true;
|
|
} else if (req.method == "Tracing.end") {
|
|
// @cdp Tracing.end support is experimental.
|
|
bool firstChunk = true;
|
|
auto id = req.id;
|
|
bool wasStopped = FuseboxTracer::getFuseboxTracer().stopTracing(
|
|
[this, firstChunk, id](const folly::dynamic& eventsChunk) {
|
|
if (firstChunk) {
|
|
frontendChannel_(cdp::jsonResult(id));
|
|
}
|
|
frontendChannel_(cdp::jsonNotification(
|
|
"Tracing.dataCollected",
|
|
folly::dynamic::object("value", eventsChunk)));
|
|
});
|
|
|
|
if (!wasStopped) {
|
|
frontendChannel_(cdp::jsonError(
|
|
req.id,
|
|
cdp::ErrorCode::InternalError,
|
|
"Tracing session not started"));
|
|
} else {
|
|
frontendChannel_(cdp::jsonNotification(
|
|
"Tracing.tracingComplete",
|
|
folly::dynamic::object("dataLossOccurred", false)));
|
|
}
|
|
|
|
frontendChannel_(cdp::jsonResult(req.id));
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
} // namespace facebook::react::jsinspector_modern
|