Function PerfTest

test/brpc_load_balancer_unittest.cpp:248–304

template<typename DBD>
void PerfTest(int thread_num, bool modify_during_reading) {
    g_started = false;
    g_stopped = false;
    DBD dbd;
    for (int i = 0; i < 1024; ++i) {
        dbd.Modify(AddMapN, i);
    }
    pthread_t threads[thread_num];
    std::vector<PerfArgs<DBD>> args(thread_num);
    for (int i = 0; i < thread_num; ++i) {
        args[i].dbd = &dbd;
        ASSERT_EQ(0, pthread_create(&threads[i], NULL, read_dbd<DBD>, &args[i]));
    }
    while (true) {
        bool all_ready = true;
        for (int i = 0; i < thread_num; ++i) {
            if (!args[i].ready) {
                all_ready = false;
                break;
            }
        }
        if (all_ready) {
            break;
        }
        usleep(1000);
    }
    g_started = true;
    char prof_name[32];
    snprintf(prof_name, sizeof(prof_name), "doubly_buffered_data_%d.prof", ++g_prof_name_counter);
    ProfilerStart(prof_name);
    int64_t run_ms = 5 * 1000;
    if (modify_during_reading) {
        int64_t start = butil::cpuwide_time_ms();
        int i = 1;
        while (butil::cpuwide_time_ms() - start < run_ms) {
            ASSERT_TRUE(dbd.Modify(AddMapN, i++));
            usleep(1000);
        }
    } else {
        usleep(run_ms * 1000);
    }
    ProfilerStop();
    g_stopped = true;
    int64_t wait_time = 0;
    int64_t count = 0;
    for (int i = 0; i < thread_num; ++i) {
        pthread_join(threads[i], NULL);
        wait_time += args[i].elapse_ns;
        count += args[i].counter;
    }
    LOG(INFO) << butil::class_name<DBD>()
              << " thread_num=" << thread_num
              << " modify_during_reading=" << modify_during_reading
              << " count=" << count
              << " average_time=" << wait_time / (double)count
              << " qps=" << (double)count / wait_time * (1000 * 1000 * 1000);
}