tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bench full logging time in log test.

Browse Source 
GitOrigin-RevId: e224260d4f81f622af09c2df8f529dcced2ed0c2
This commit is contained in:
levlam 2020-08-03 23:08:03 +03:00
parent 4bf3b013c7
commit e5fa1c3c7f
1 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tdutils/test/log.cpp
View File

@ -24,11 +24,15 @@ char disable_linker_warning_about_empty_file_tdutils_test_log_cpp TD_UNUSED;
template <class Log> template <class Log>
class LogBenchmark : public td::Benchmark { class LogBenchmark : public td::Benchmark {
public: public:
LogBenchmark(std::string name, int threads_n, std::function<td::unique_ptr<Log>()> creator) LogBenchmark(std::string name, int threads_n, bool test_full_logging, std::function<td::unique_ptr<Log>()> creator)
: name_(std::move(name)), threads_n_(threads_n), creator_(std::move(creator)) { : name_(std::move(name))
, threads_n_(threads_n)
, test_full_logging_(test_full_logging)
, creator_(std::move(creator)) {
} }
std::string get_description() const override { std::string get_description() const override {
return PSTRING() << name_ << " " << td::tag("threads_n", threads_n_); return PSTRING() << name_ << " " << (test_full_logging_ ? "ERROR" : "PLAIN") << " "
<< td::tag("threads_n", threads_n_);
} }
void start_up() override { void start_up() override {
log_ = creator_(); log_ = creator_();
@ -41,12 +45,17 @@ class LogBenchmark : public td::Benchmark {
log_.reset(); log_.reset();
} }
void run(int n) override { void run(int n) override {
auto old_log_interface = td::log_interface;
td::log_interface = log_.get();
for (auto &thread : threads_) { for (auto &thread : threads_) {
thread = td::thread([this, n] { this->run_thread(n); }); thread = td::thread([this, n] { this->run_thread(n); });
} }
for (auto &thread : threads_) { for (auto &thread : threads_) {
thread.join(); thread.join();
} }
td::log_interface = old_log_interface;
} }
void run_thread(int n) { void run_thread(int n) {
@ -55,7 +64,11 @@ class LogBenchmark : public td::Benchmark {
if (i % 10000 == 0) { if (i % 10000 == 0) {
log_->rotate(); log_->rotate();
} }
log_->append(str, 1); if (test_full_logging_) {
LOG(ERROR) << str;
} else {
LOG(PLAIN) << str;
}
} }
} }
@ -63,14 +76,17 @@ class LogBenchmark : public td::Benchmark {
std::string name_; std::string name_;
td::unique_ptr<td::LogInterface> log_; td::unique_ptr<td::LogInterface> log_;
int threads_n_{0}; int threads_n_{0};
bool test_full_logging_{false};
std::function<td::unique_ptr<Log>()> creator_; std::function<td::unique_ptr<Log>()> creator_;
std::vector<td::thread> threads_; std::vector<td::thread> threads_;
}; };
template <class F> template <class F>
static void bench_log(std::string name, F &&f) { static void bench_log(std::string name, F &&f) {
for (auto threads_n : {1, 4, 8}) { for (auto test_full_logging : {false, true}) {
bench(LogBenchmark<typename decltype(f())::element_type>(name, threads_n, f)); for (auto threads_n : {1, 4, 8}) {
bench(LogBenchmark<typename decltype(f())::element_type>(name, threads_n, test_full_logging, f));
}
} }
}; };