118 lines
3.1 KiB
C++
118 lines
3.1 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 <folly/json.h>
|
||
|
|
#include <fstream>
|
||
|
|
#include <mutex>
|
||
|
|
#include <unordered_map>
|
||
|
|
|
||
|
|
#include "FuseboxTracer.h"
|
||
|
|
|
||
|
|
namespace facebook::react {
|
||
|
|
|
||
|
|
bool FuseboxTracer::isTracing() {
|
||
|
|
std::lock_guard lock(mutex_);
|
||
|
|
return tracing_;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool FuseboxTracer::startTracing() {
|
||
|
|
std::lock_guard lock(mutex_);
|
||
|
|
if (tracing_) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
tracing_ = true;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool FuseboxTracer::stopTracing(
|
||
|
|
const std::function<void(const folly::dynamic& eventsChunk)>&
|
||
|
|
resultCallback) {
|
||
|
|
std::lock_guard lock(mutex_);
|
||
|
|
|
||
|
|
if (!tracing_) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
tracing_ = false;
|
||
|
|
if (buffer_.empty()) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
auto traceEvents = folly::dynamic::array();
|
||
|
|
auto savedBuffer = std::move(buffer_);
|
||
|
|
buffer_.clear();
|
||
|
|
|
||
|
|
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);
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
void FuseboxTracer::addEvent(
|
||
|
|
const std::string_view& name,
|
||
|
|
uint64_t start,
|
||
|
|
uint64_t end,
|
||
|
|
const std::string_view& track) {
|
||
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
||
|
|
if (!tracing_) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
buffer_.push_back(
|
||
|
|
BufferEvent{start, end, std::string(name), std::string(track)});
|
||
|
|
}
|
||
|
|
|
||
|
|
bool FuseboxTracer::stopTracingAndWriteToFile(const std::string& path) {
|
||
|
|
auto file = std::ofstream(path);
|
||
|
|
folly::dynamic traceEvents = folly::dynamic::array();
|
||
|
|
bool result = stopTracing([&traceEvents](const folly::dynamic& eventsChunk) {
|
||
|
|
traceEvents.insert(
|
||
|
|
traceEvents.end(), eventsChunk.begin(), eventsChunk.end());
|
||
|
|
});
|
||
|
|
file << folly::toJson(traceEvents) << std::endl;
|
||
|
|
file.close();
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* static */ FuseboxTracer& FuseboxTracer::getFuseboxTracer() {
|
||
|
|
static FuseboxTracer tracer;
|
||
|
|
return tracer;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace facebook::react
|