Windows与Linux下TLS实现

Under Windows：

#include <stdio.h> #include <windows.h> #include <process.h> // 利用TLS记录线程的运行时间 DWORD g_tlsUsedTime; void InitStartTime(); DWORD GetUsedTime(); UINT __stdcall ThreadFunc(LPVOID) { int i; // 初始化开始时间 InitStartTime(); // 模拟长时间工作 i = 10000*10000; while(i--) { } // 打印出本线程运行的时间 printf(" This thread is coming to end. Thread ID: %-5d, Used Time: %d /n", ::GetCurrentThreadId(), GetUsedTime()); return 0; } int main(int argc, char* argv[]) { UINT uId; int i; HANDLE h[10]; // 通过在进程位数组中申请一个索引，初始化线程运行时间记录系统 g_tlsUsedTime = ::TlsAlloc(); // 令十个线程同时运行，并等待它们各自的输出结果 for(i=0; i<10; i++) { h[i] = (HANDLE)::_beginthreadex(NULL, 0, ThreadFunc, NULL, 0, &uId); } for(i=0; i<10; i++) { ::WaitForSingleObject(h[i], INFINITE); ::CloseHandle(h[i]); } // 通过释放线程局部存储索引，释放时间记录系统占用的资源 ::TlsFree(g_tlsUsedTime); return 0; } // 初始化线程的开始时间 void InitStartTime() { // 获得当前时间，将线程的创建时间与线程对象相关联 DWORD dwStart = ::GetTickCount(); ::TlsSetValue(g_tlsUsedTime, (LPVOID)dwStart); } // 取得一个线程已经运行的时间 DWORD GetUsedTime() { // 获得当前时间，返回当前时间和线程创建时间的差值 DWORD dwElapsed = ::GetTickCount(); dwElapsed = dwElapsed - (DWORD)::TlsGetValue(g_tlsUsedTime); return dwElapsed; }  

 

Under Linux：

#include <stdio.h> #include <pthread.h> #include <malloc.h> static pthread_key_t thread_log_key; void write_to_thread_log( char const * message ) { FILE * thread_log = ( FILE * )pthread_getspecific( thread_log_key ); fprintf( thread_log, "%s/n", message ); } void close_thread_log( void * thread_log ) { fclose( ( FILE * )thread_log ); } void * thread_function( void * args ) { char thread_log_filename[ 20 ]; FILE* thread_log; sprintf( thread_log_filename, "thread%d.log", ( int )pthread_self() ); thread_log = fopen( thread_log_filename, "w" ); pthread_setspecific( thread_log_key, thread_log ); write_to_thread_log( "Thread starting." ); return NULL; } int main() { int i; pthread_t threads[5]; pthread_key_create( &thread_log_key, close_thread_log ); for ( i = 0; i < 5; ++i ) { pthread_create( &(threads[i]), NULL, thread_function, NULL ); } for ( i = 0; i < 5; ++i ) { pthread_join( threads[i], NULL ); } return 0; }