Merge "logd: Add 'print_all_logs' to replay_messages"
This commit is contained in:
commit
3e967861da
1 changed files with 49 additions and 20 deletions
|
|
@ -117,6 +117,23 @@ static log_time GetFirstTimeStamp(const MappedFile& recorded_messages) {
|
|||
return meta->realtime;
|
||||
}
|
||||
|
||||
static LogMask BuffersToLogMask(const char* buffers) {
|
||||
if (buffers == nullptr || !strcmp(buffers, "all")) {
|
||||
return kLogMaskAll;
|
||||
}
|
||||
auto string_ids = Split(buffers, ",");
|
||||
LogMask log_mask = 0;
|
||||
for (const auto& string_id : string_ids) {
|
||||
int buffer_id;
|
||||
if (!ParseInt(string_id, &buffer_id, 0, 7)) {
|
||||
fprintf(stderr, "Could not parse buffer_id '%s'\n", string_id.c_str());
|
||||
exit(1);
|
||||
}
|
||||
log_mask |= 1 << buffer_id;
|
||||
}
|
||||
return log_mask;
|
||||
}
|
||||
|
||||
class StdoutWriter : public LogWriter {
|
||||
public:
|
||||
StdoutWriter() : LogWriter(0, true) {}
|
||||
|
|
@ -134,6 +151,11 @@ class StdoutWriter : public LogWriter {
|
|||
return true;
|
||||
}
|
||||
|
||||
void Shutdown() override {
|
||||
fprintf(stderr, "LogWriter::Shutdown() called\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
std::string name() const override { return "stdout writer"; }
|
||||
};
|
||||
|
||||
|
|
@ -291,24 +313,7 @@ class PrintLogs : public SingleBufferOperation {
|
|||
PrintLogs(log_time first_log_timestamp, const char* buffer, const char* buffers,
|
||||
const char* print_point)
|
||||
: SingleBufferOperation(first_log_timestamp, buffer) {
|
||||
if (buffers != nullptr) {
|
||||
if (strcmp(buffers, "all") != 0) {
|
||||
std::vector<int> buffer_ids;
|
||||
auto string_ids = Split(buffers, ",");
|
||||
for (const auto& string_id : string_ids) {
|
||||
int result;
|
||||
if (!ParseInt(string_id, &result, 0, 7)) {
|
||||
fprintf(stderr, "Could not parse buffer_id '%s'\n", string_id.c_str());
|
||||
exit(1);
|
||||
}
|
||||
buffer_ids.emplace_back(result);
|
||||
}
|
||||
mask_ = 0;
|
||||
for (const auto& buffer_id : buffer_ids) {
|
||||
mask_ |= 1 << buffer_id;
|
||||
}
|
||||
}
|
||||
}
|
||||
mask_ = BuffersToLogMask(buffers);
|
||||
if (print_point != nullptr) {
|
||||
uint64_t result = 0;
|
||||
if (!ParseUint(print_point, &result)) {
|
||||
|
|
@ -326,7 +331,7 @@ class PrintLogs : public SingleBufferOperation {
|
|||
}
|
||||
}
|
||||
|
||||
void End() {
|
||||
void End() override {
|
||||
std::unique_ptr<LogWriter> test_writer(new StdoutWriter());
|
||||
std::unique_ptr<FlushToState> flush_to_state = log_buffer_->CreateFlushToState(1, mask_);
|
||||
log_buffer_->FlushTo(test_writer.get(), *flush_to_state, nullptr);
|
||||
|
|
@ -353,7 +358,7 @@ class PrintLatency : public SingleBufferOperation {
|
|||
durations_.emplace_back(duration);
|
||||
}
|
||||
|
||||
void End() {
|
||||
void End() override {
|
||||
std::sort(durations_.begin(), durations_.end());
|
||||
auto q1 = durations_.size() / 4;
|
||||
auto q2 = durations_.size() / 2;
|
||||
|
|
@ -373,6 +378,27 @@ class PrintLatency : public SingleBufferOperation {
|
|||
std::vector<long long> durations_;
|
||||
};
|
||||
|
||||
class PrintAllLogs : public SingleBufferOperation {
|
||||
public:
|
||||
PrintAllLogs(log_time first_log_timestamp, const char* buffer, const char* buffers)
|
||||
: SingleBufferOperation(first_log_timestamp, buffer) {
|
||||
LogMask mask = BuffersToLogMask(buffers);
|
||||
auto lock = std::unique_lock{reader_list_.reader_threads_lock()};
|
||||
std::unique_ptr<LogWriter> stdout_writer(new StdoutWriter());
|
||||
std::unique_ptr<LogReaderThread> log_reader(
|
||||
new LogReaderThread(log_buffer_.get(), &reader_list_, std::move(stdout_writer),
|
||||
false, 0, mask, 0, {}, 1, {}));
|
||||
reader_list_.reader_threads().emplace_back(std::move(log_reader));
|
||||
}
|
||||
|
||||
void Operation() override {
|
||||
// If the rate of reading logs is slower than the rate of incoming logs, then the reader
|
||||
// thread is disconnected to not overflow log buffers, therefore we artificially slow down
|
||||
// the incoming log rate.
|
||||
usleep(100);
|
||||
}
|
||||
};
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
if (argc < 3) {
|
||||
fprintf(stderr, "Usage: %s FILE OPERATION [BUFFER] [OPTIONS]\n", argv[0]);
|
||||
|
|
@ -415,6 +441,9 @@ int main(int argc, char** argv) {
|
|||
} else if (!strcmp(argv[2], "print_logs")) {
|
||||
operation.reset(new PrintLogs(first_log_timestamp, argv[3], argc > 4 ? argv[4] : nullptr,
|
||||
argc > 5 ? argv[5] : nullptr));
|
||||
} else if (!strcmp(argv[2], "print_all_logs")) {
|
||||
operation.reset(
|
||||
new PrintAllLogs(first_log_timestamp, argv[3], argc > 4 ? argv[4] : nullptr));
|
||||
} else if (!strcmp(argv[2], "nothing")) {
|
||||
operation.reset(new SingleBufferOperation(first_log_timestamp, argv[3]));
|
||||
} else {
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue