milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix:reduce thread pool test time (#31893) 

#31877

Signed-off-by: luzhang <luzhang@zilliz.com>
Co-authored-by: luzhang <luzhang@zilliz.com>
pull/31926/head
zhagnlu 2024-04-05 10:05:12 +08:00 committed by GitHub
parent a2502bde75
commit b2669e26dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/core/unittest/test_disk_file_manager_test.cpp
View File

@ -120,7 +120,7 @@ TEST_F(DiskAnnFileManagerTest, AddFilePositiveParallel) {
int
test_worker(string s) {
std::cout << s << std::endl;
std::this_thread::sleep_for(std::chrono::seconds(4));
std::this_thread::sleep_for(std::chrono::seconds(1));
std::cout << s << std::endl;
return 1;
}
@ -161,10 +161,10 @@ TEST_F(DiskAnnFileManagerTest, TestThreadPoolBase) {
}
TEST_F(DiskAnnFileManagerTest, TestThreadPool) {
auto thread_pool = std::make_shared<milvus::ThreadPool>(50, "test");
auto thread_pool = std::make_shared<milvus::ThreadPool>(10, "test");
std::vector<std::future<int>> futures;
auto start = chrono::system_clock::now();
for (int i = 0; i < 100; i++) {
for (int i = 0; i < 10; i++) {
futures.push_back(
thread_pool->Submit(test_worker, "test_id" + std::to_string(i)));
}
@ -188,9 +188,9 @@ test_exception(string s) {
TEST_F(DiskAnnFileManagerTest, TestThreadPoolException) {
try {
auto thread_pool = std::make_shared<milvus::ThreadPool>(50, "test");
auto thread_pool = std::make_shared<milvus::ThreadPool>(10, "test");
std::vector<std::future<int>> futures;
for (int i = 0; i < 100; i++) {
for (int i = 0; i < 10; i++) {
futures.push_back(thread_pool->Submit(
test_exception, "test_id" + std::to_string(i)));
}