84 lines
1.9 KiB
C++
84 lines
1.9 KiB
C++
#include <inttypes.h>
|
||
#include <vector>
|
||
#include "idgen/IdGenerator.h"
|
||
|
||
const int GenIdCount = 500000;
|
||
const bool multiThread = true;
|
||
const int threadCount = 50;
|
||
const int method = 1;
|
||
|
||
void RunMultiThread()
|
||
{
|
||
int64_t start = idgen::GetCurrentMicroTime();
|
||
for (int i = 0; i < GenIdCount / threadCount; i++)
|
||
{
|
||
int64_t id = idgen::IdGenerator::NextId();
|
||
printf("ID: %" PRId64 "\n", id);
|
||
}
|
||
|
||
int64_t end = idgen::GetCurrentMicroTime();
|
||
printf("%s，total：%" PRId64 " μs\n", method == 1 ? "1" : "2", (end - start));
|
||
}
|
||
|
||
void RunSingle()
|
||
{
|
||
int64_t start = idgen::GetCurrentMicroTime();
|
||
for (int i = 0; i < GenIdCount; i++)
|
||
{
|
||
int64_t id = idgen::IdGenerator::NextId();
|
||
printf("ID: %" PRId64 "\n", id);
|
||
}
|
||
|
||
int64_t end = idgen::GetCurrentMicroTime();
|
||
|
||
printf("%s, total: %" PRId64 " us\n", method == 1 ? "1" : "2", (end - start));
|
||
}
|
||
|
||
void create_instance_fun()
|
||
{
|
||
idgen::IdGeneratorOptions options;
|
||
options.Method = method;
|
||
options.WorkerId = 1;
|
||
options.SeqBitLength = 10;
|
||
idgen::IdGenerator::CreateInstance(options);
|
||
}
|
||
|
||
void test_multithread_run_create_instance()
|
||
{
|
||
std::vector<std::thread> threads;
|
||
for (int i = 0; i < threadCount; i++)
|
||
{
|
||
threads.emplace_back(create_instance_fun);
|
||
}
|
||
for (auto &t : threads)
|
||
{
|
||
t.join();
|
||
}
|
||
}
|
||
|
||
int main()
|
||
{
|
||
if (multiThread) {
|
||
test_multithread_run_create_instance(); // 测试多线程调用同时初始化，也确保只会安全初始化一次，确保单例唯一性
|
||
} else {
|
||
create_instance_fun(); // 单线程调用初始化
|
||
}
|
||
|
||
if (multiThread)
|
||
{
|
||
std::vector<std::thread> threads;
|
||
for (int i = 0; i < threadCount; i++)
|
||
{
|
||
threads.emplace_back(RunMultiThread);
|
||
}
|
||
for (auto &t : threads)
|
||
{
|
||
t.join();
|
||
}
|
||
}
|
||
else
|
||
{
|
||
RunSingle();
|
||
}
|
||
}
|