Skip to content

Commit

Permalink
Introduce FuseboxTracer for DevTools tracing
Browse files Browse the repository at this point in the history
Summary:
Introduce a simplified and minimable tracing backend for Fusebox. This backend is sufficient to implement a pretty usable performance panel.

Although the more I see how easy this is and how annoying working with Perfetto is, the more I think we should just maintain this going forward. Anyways we can figure that out incrementally. For now the plan is still for this to be temporary.

Differential Revision: D57981944
  • Loading branch information
bgirard authored and facebook-github-bot committed Jun 14, 2024
1 parent 7061649 commit 39774fd
Show file tree
Hide file tree
Showing 5 changed files with 238 additions and 18 deletions.
50 changes: 33 additions & 17 deletions packages/react-native/ReactCommon/jsinspector-modern/HostAgent.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,8 @@

#include <chrono>

#include <reactperflogger/FuseboxTracer.h>

using namespace std::chrono;
using namespace std::literals::string_view_literals;

Expand Down Expand Up @@ -146,25 +148,39 @@ void HostAgent::handleRequest(const cdp::PreparsedRequest& req) {
shouldSendOKResponse = true;
isFinishedHandlingRequest = true;
} else if (req.method == "Tracing.start") {
// @cdp Tracing.start is implemented as a stub only.
frontendChannel_(cdp::jsonNotification(
// @cdp Tracing.bufferUsage is implemented as a stub only.
"Tracing.bufferUsage",
folly::dynamic::object("percentFull", 0)("eventCount", 0)("value", 0)));
shouldSendOKResponse = true;
// @cdp Tracing.start support is experimental.
if (!FuseboxTracer::isTracing()) {
FuseboxTracer::startTracing();
shouldSendOKResponse = true;
} else {
frontendChannel_(cdp::jsonError(
req.id,
cdp::ErrorCode::InternalError,
"Tracing session already started"));
return;
}
isFinishedHandlingRequest = true;
} else if (req.method == "Tracing.end") {
// @cdp Tracing.end is implemented as a stub only.
frontendChannel_(cdp::jsonNotification(
// @cdp Tracing.dataCollected is implemented as a stub only.
"Tracing.dataCollected",
folly::dynamic::object("value", folly::dynamic::array())));
frontendChannel_(cdp::jsonNotification(
// @cdp Tracing.tracingComplete is implemented as a stub only.
"Tracing.tracingComplete",
folly::dynamic::object("dataLossOccurred", false)));
shouldSendOKResponse = true;
isFinishedHandlingRequest = true;
// @cdp Tracing.end support is experimental.
if (FuseboxTracer::isTracing()) {
// Send ok manually before spending results
frontendChannel_(cdp::jsonResult(req.id));
FuseboxTracer::stopTracing([this](const folly::dynamic& eventsChunk) {
frontendChannel_(cdp::jsonNotification(
"Tracing.dataCollected",
folly::dynamic::object("value", eventsChunk)));
});
frontendChannel_(cdp::jsonNotification(
"Tracing.tracingComplete",
folly::dynamic::object("dataLossOccurred", false)));
return;
} else {
frontendChannel_(cdp::jsonError(
req.id,
cdp::ErrorCode::InternalError,
"Tracing session not started"));
return;
}
}

if (!isFinishedHandlingRequest && instanceAgent_ &&
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,8 @@
#include <cxxreact/ReactMarker.h>
#include <jsi/instrumentation.h>
#include <react/performance/timeline/PerformanceEntryReporter.h>

#include <reactperflogger/FuseboxTracer.h>
#include "NativePerformance.h"
#include "Plugins.h"

#ifdef WITH_PERFETTO
Expand Down Expand Up @@ -112,6 +113,17 @@ void NativePerformance::measure(
}
}
#endif
std::string trackName = "Web Performance";
const int TRACK_PREFIX = 6;
if (name.starts_with("Track:")) {
const auto trackNameDelimiter = name.find(':', TRACK_PREFIX);
if (trackNameDelimiter != std::string::npos) {
trackName = name.substr(TRACK_PREFIX, trackNameDelimiter - TRACK_PREFIX);
name = name.substr(trackNameDelimiter + 1);
}
}
FuseboxTracer::addEvent(
name, (uint64_t)startTime, (uint64_t)endTime, trackName);
PerformanceEntryReporter::getInstance()->measure(
name, startTime, endTime, duration, startMark, endMark);
}
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,87 @@
/*
* 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 <mutex>
#include <unordered_map>

#include "FuseboxTracer.h"

// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
static bool tracing = false;
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
static std::vector<BufferEvent> buffer;
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
static std::mutex mutex;

/* static */ bool FuseboxTracer::isTracing() {
return tracing;
}

/* static */ void FuseboxTracer::startTracing() {
tracing = true;
}

/* static */ void FuseboxTracer::stopTracing(
const std::function<void(const folly::dynamic& eventsChunk)>&
resultCallback) {
std::unique_lock<std::mutex> lock(mutex);
auto traceEvents = folly::dynamic::array();
if (buffer.empty()) {
return;
}
auto savedBuffer = std::move(buffer);
buffer.clear();
tracing = false;

std::unordered_map<std::string, uint64_t> trackIdMap;
uint64_t nextTrack = 1000;

// Name the main process. Only one process is supported currently.
traceEvents.push_back(folly::dynamic::object(
"args", folly::dynamic::object("name", "Main App"))("cat", "__metadata")(
"name", "process_name")("ph", "M")("pid", 1000)("tid", 0)("ts", 0));

for (auto& event : savedBuffer) {
if (!trackIdMap.contains(event.track)) {
auto trackId = nextTrack++;
trackIdMap[event.track] = trackId;
// New track
traceEvents.push_back(folly::dynamic::object(
"args", folly::dynamic::object("name", event.track))(
"cat", "__metadata")("name", "thread_name")("ph", "M")("pid", 1000)(
"tid", trackId)("ts", 0));
}
auto trackId = trackIdMap[event.track];

// New event
traceEvents.push_back(folly::dynamic::object(
"args", folly::dynamic::object())("cat", "react.native")(
"dur", (event.end - event.start) * 1000)("name", event.name)("ph", "X")(
"ts", event.start * 1000)("pid", 1000)("tid", trackId));

if (traceEvents.size() >= 1000) {
resultCallback(traceEvents);
traceEvents = folly::dynamic::array();
}
}

if (traceEvents.size() >= 1) {
resultCallback(traceEvents);
}
}

/* static */ void FuseboxTracer::addEvent(
const std::string& name,
uint64_t start,
uint64_t end,
const std::string& track) {
if (!isTracing()) {
return;
}
std::unique_lock<std::mutex> lock(mutex);
buffer.push_back(BufferEvent{start, end, name, track});
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
/*
* 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.
*/

#pragma once

#include <functional>
#include <vector>
#include "folly/json/dynamic.h"

struct BufferEvent {
uint64_t start;
uint64_t end;
std::string name;
std::string track;
};

class FuseboxTracer {
public:
static bool isTracing();
static void startTracing();
static void stopTracing(
const std::function<void(const folly::dynamic& eventsChunk)>&
resultCallback);
static void addEvent(
const std::string& name,
uint64_t start,
uint64_t end,
const std::string& track);
};
Original file line number Diff line number Diff line change
@@ -0,0 +1,72 @@
/*
* 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 <gtest/gtest.h>

#include "reactperflogger/FuseboxTracer.h"

folly::dynamic stopTracingAndCollect() {
folly::dynamic trace = folly::dynamic::array;
FuseboxTracer::stopTracing([&trace](const folly::dynamic& eventsChunk) {
for (const auto& event : eventsChunk) {
trace.push_back(event);
}
});
return trace;
}

class FuseboxTracerTest : public ::testing::Test {
protected:
FuseboxTracerTest() = default;

~FuseboxTracerTest() override = default;

void SetUp() override {
stopTracingAndCollect();
}

void TearDown() override {
stopTracingAndCollect();
}
};

TEST_F(FuseboxTracerTest, TracingOffByDefault) {
EXPECT_FALSE(FuseboxTracer::isTracing());
}

TEST_F(FuseboxTracerTest, TracingOn) {
FuseboxTracer::startTracing();
EXPECT_TRUE(FuseboxTracer::isTracing());
}

TEST_F(FuseboxTracerTest, DiscardEventWhenNotOn) {
EXPECT_FALSE(FuseboxTracer::isTracing());
EXPECT_EQ(stopTracingAndCollect().size(), 0);
FuseboxTracer::addEvent("test", 0, 0, "default track");
FuseboxTracer::addEvent("test", 0, 0, "default track");
EXPECT_EQ(stopTracingAndCollect().size(), 0);
}

TEST_F(FuseboxTracerTest, NoDefaultEvents) {
FuseboxTracer::startTracing();
EXPECT_EQ(stopTracingAndCollect().size(), 0);
}

TEST_F(FuseboxTracerTest, SimpleEvent) {
FuseboxTracer::startTracing();
FuseboxTracer::addEvent("test", 0, 0, "default track");
EXPECT_GE(stopTracingAndCollect().size(), 1);
}

TEST_F(FuseboxTracerTest, MultiEvents) {
FuseboxTracer::startTracing();
for (int i = 0; i < 10; i++) {
FuseboxTracer::addEvent("test", 0, 0, "default track");
}
EXPECT_GE(stopTracingAndCollect().size(), 10);
EXPECT_EQ(stopTracingAndCollect().size(), 0);
}

0 comments on commit 39774fd

Please sign in to comment.