NdbThread.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:4k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #include <ndb_global.h>
  14. #include <NdbThread.h>
  15. #include <my_pthread.h>
  16. #include <NdbMem.h>
  17. #define MAX_THREAD_NAME 16
  18. /*#define USE_PTHREAD_EXTRAS*/
  19. #ifdef NDB_SHM_TRANSPORTER
  20. int g_ndb_shm_signum= 0;
  21. #endif
  22. struct NdbThread 
  23.   pthread_t thread;
  24.   char thread_name[MAX_THREAD_NAME];
  25.   NDB_THREAD_FUNC * func;
  26.   void * object;
  27. };
  28. static
  29. void*
  30. ndb_thread_wrapper(void* _ss){
  31.   my_thread_init();
  32.   {
  33.     DBUG_ENTER("ndb_thread_wrapper");
  34. #ifdef NDB_SHM_TRANSPORTER
  35.     if (g_ndb_shm_signum)
  36.     {
  37.       sigset_t mask;
  38.       DBUG_PRINT("info",("Block signum %d",g_ndb_shm_signum));
  39.       sigemptyset(&mask);
  40.       sigaddset(&mask, g_ndb_shm_signum);
  41.       pthread_sigmask(SIG_BLOCK, &mask, 0);
  42.     }
  43. #endif
  44.     {
  45.       /**
  46.        * Block all signals to thread by default
  47.        *   let them go to main process instead
  48.        */
  49.       sigset_t mask;
  50.       sigfillset(&mask);
  51.       pthread_sigmask(SIG_BLOCK, &mask, 0);
  52.     }      
  53.     
  54.     {
  55.       void *ret;
  56.       struct NdbThread * ss = (struct NdbThread *)_ss;
  57.       ret= (* ss->func)(ss->object);
  58.       DBUG_POP();
  59.       NdbThread_Exit(ret);
  60.     }
  61.   /* will never be reached */
  62.     DBUG_RETURN(0);
  63.   }
  64. }
  65. struct NdbThread* NdbThread_Create(NDB_THREAD_FUNC *p_thread_func,
  66.                       NDB_THREAD_ARG *p_thread_arg,
  67.          const NDB_THREAD_STACKSIZE thread_stack_size,
  68.       const char* p_thread_name,
  69.                       NDB_THREAD_PRIO thread_prio)
  70. {
  71.   struct NdbThread* tmpThread;
  72.   int result;
  73.   pthread_attr_t thread_attr;
  74.   DBUG_ENTER("NdbThread_Create");
  75.   (void)thread_prio; /* remove warning for unused parameter */
  76.   if (p_thread_func == NULL)
  77.     DBUG_RETURN(NULL);
  78.   tmpThread = (struct NdbThread*)NdbMem_Allocate(sizeof(struct NdbThread));
  79.   if (tmpThread == NULL)
  80.     DBUG_RETURN(NULL);
  81.   DBUG_PRINT("info",("thread_name: %s", p_thread_name));
  82.   strnmov(tmpThread->thread_name,p_thread_name,sizeof(tmpThread->thread_name));
  83.   pthread_attr_init(&thread_attr);
  84. #if (SIZEOF_CHARP == 8)
  85.   pthread_attr_setstacksize(&thread_attr, 2*thread_stack_size);
  86. #else
  87.   pthread_attr_setstacksize(&thread_attr, thread_stack_size);
  88. #endif
  89. #ifdef USE_PTHREAD_EXTRAS
  90.   /* Guard stack overflow with a 2k databuffer */
  91.   pthread_attr_setguardsize(&thread_attr, 2048);
  92. #endif
  93. #ifdef PTHREAD_CREATE_JOINABLE /* needed on SCO */
  94.   pthread_attr_setdetachstate(&thread_attr, PTHREAD_CREATE_JOINABLE);
  95. #endif
  96.   tmpThread->func= p_thread_func;
  97.   tmpThread->object= p_thread_arg;
  98.   result = pthread_create(&tmpThread->thread, 
  99.   &thread_attr,
  100.              ndb_thread_wrapper,
  101.              tmpThread);
  102.   assert(result==0);
  103.   pthread_attr_destroy(&thread_attr);
  104.   DBUG_PRINT("exit",("ret: %lx", tmpThread));
  105.   DBUG_RETURN(tmpThread);
  106. }
  107. void NdbThread_Destroy(struct NdbThread** p_thread)
  108. {
  109.   DBUG_ENTER("NdbThread_Destroy");
  110.   if (*p_thread != NULL){
  111.     DBUG_PRINT("enter",("*p_thread: %lx", * p_thread));
  112.     free(* p_thread); 
  113.     * p_thread = 0;
  114.   }
  115.   DBUG_VOID_RETURN;
  116. }
  117. int NdbThread_WaitFor(struct NdbThread* p_wait_thread, void** status)
  118. {
  119.   int result;
  120.   if (p_wait_thread == NULL)
  121.     return 0;
  122.   if (p_wait_thread->thread == 0)
  123.     return 0;
  124.   result = pthread_join(p_wait_thread->thread, status);
  125.   
  126.   return result;
  127. }
  128. void NdbThread_Exit(void *status)
  129. {
  130.   my_thread_end();
  131.   pthread_exit(status);
  132. }
  133. int NdbThread_SetConcurrencyLevel(int level)
  134. {
  135. #ifdef USE_PTHREAD_EXTRAS
  136.   return pthread_setconcurrency(level);
  137. #else
  138.   (void)level; /* remove warning for unused parameter */
  139.   return 0;
  140. #endif
  141. }