问题解决:如何管理线程私有(thread

    技术2022-07-16  73

    引言

    问题出现在实现项目的一个功能的时候,我需要一个thread_local的指针,因为整个项目已经差不多实现了,但在最后发现了内存泄露的问题。因为实现的是一个偏向于底层的库,且内存的使用基本都在掌握之内,所以在项目实现之初为了效率并没有使用智能指针,但在后来在这里出现了内存泄露问题。

    原本代码差不多是这样:

    __thread Co_Rountinue_Env* CurrentThread_CoEnv = nullptr;

    这样在线程结束以后很难去把它释放掉,所以需要想一个方法做到这样,且不想大规模改动原有代码,最终的解决方案是这样的:

    __thread Co_Rountinue_Env* CurrentThread_CoEnv = nullptr; thread_local std::unique_ptr<Co_Rountinue_Env, decltype(Free_Co_Rountinue_Env)*> DeleteThread_CoEnv(CurrentThread_CoEnv, Free_Co_Rountinue_Env);

    不必在意变量名,基本思路还是使用智能指针去管理内存,我们来看一个模拟的小demo:

    #include <bits/stdc++.h> #include <unistd.h> using namespace std; void Delete_(int* para){ if(para == nullptr) return; cout << "nihao\n"; delete para; return; } thread_local int* env = nullptr; thread_local std::unique_ptr<int, decltype(Delete_)*> Env(env, Delete_); void* Routine(void* para){ Env.reset(); env = new int(5); Env.reset(env); cout << "函数结束\n"; return 0; } int main(){ pthread_t tid[5]; for(size_t i = 0; i < 2; i++){ pthread_create(tid + i, nullptr, Routine, 0); } for(size_t i = 0; i < 2; i++){ pthread_join(tid[i], nullptr); } sleep(1); return 0; }

    输出:

    函数结束 nihao 函数结束 nihao

    这样就达到了预期。


    在解决问题的过程中还发现了boost库中也有一个玩意儿可以做到这样,即boost::thread_specific_ptr,它的作用就是用于每线程存储,且会在线程结束的时候执行预先设置的回调,就类似于一个特殊的智能指针,我们来看看其如何应用:

    #include <bits/stdc++.h> #include <boost/thread/tss.hpp> #include <unistd.h> using namespace std; void Delete_(int* para){ if(para == nullptr) return; cout << "nihao\n"; delete para; return; } boost::thread_specific_ptr<int> env {&Delete_}; void* Routine(void* para){ if(!env.get()){ env.reset(new int(5)); } cout << *env << endl; cout << "函数结束\n"; return 0; } int main(){ pthread_t tid[5]; for(size_t i = 0; i < 2; i++){ pthread_create(tid + i, nullptr, Routine, 0); } for(size_t i = 0; i < 2; i++){ pthread_join(tid[i], nullptr); } sleep(1); return 0; }

    执行以下命令:

    g++ XXX.cpp -pthread -lboost_system -lboost_thread ./a.out

    我们可以在使用时把thread_specific_ptr当做普通指针来用,因为其本身的成员函数并不多,相比与智能指针来说可谓一个天上一个地下。除去对于指针值的输出,我们可以看到其他输出都是一样的:

    5 函数结束 nihao 5 函数结束 nihao

    对thread_specific_ptr有兴趣的朋友可以去看看其文档,链接在文末。

    参考:

    文档《boost::thread_specific_ptr》
    Processed: 0.010, SQL: 9