如何通過命令行使用pthread接受多少線程？

Question

我正在寫一個應用程序，它在使用C或python的Linux環境中有多個線程。我正在使用pthread。但是如何通過命令行接受多少個線程。

Answer 1

在C你通過具有main兩個參數處理命令行參數，其中

int main(int argc, char** argv) 

argc是命令行參數（包括程序本身）和argv數是指向的存儲器位置其中argc-1指向具有實際參數的字符串指針。例如：

int main(int argc, char** argv) 
{ 
    printf("The program was executed as %s.\n", argv[0]); 
    printf("The arguments were:\n"); 
    for (int i = 1; i < argc; i++) 
    printf("%s\n", argv[i]); 
    return 0; 
} 

現在讓我們假設你的程序需要一個命令行參數，一個整數告訴你有多少個線程產卵。整數給出一個字符串，所以我們必須用atoi將其轉換：

if (argc != 2) 
{ 
    printf("Need exactly one argument!\n"); 
    return 1; 
} 
int num_threads = atoi(argv[1]); // Convert first argument to integer. 
if (num_threads < 1) 
{ 
    printf("I'll spawn no less than 1 thread!\n"); 
    return 2; 
} 

現在你要做的就是簡單地創建線程句柄的數組，

pthread_t* threads = malloc(num_threads*sizeof(pthread_t)); 

，並用它來存儲當你開始使用線程句柄num_threads使用pthread_create的線程數。

如果您根本不熟悉pthreads，我推薦this short tutorial。

Answer 2

如果您使用的是線程框架，如OpenMP，那麼只需設置環境變量OMP_NUM_THREADS即可自動處理。

但是，如果您要「手動」實現線程，則需要按照大多數運行時配置完成的方式執行：通過解析argv[]或設置環境變量並使用getenv()。

Answer 3

通常情況下，您可以像其他任何參數一樣傳遞它。我在之前的項目中使用類似於以下代碼來指定固定線程數。這非常簡單，但適用於不需要線程池的全部功能的情況（儘管您可以用同樣的方法輕鬆設置最小和最大線程數）。

#include <stdio.h> 

#define THRD_DFLT 5 
#define THRD_MIN 2 
#define THRD_MAX 20 

static int numThreads = 0; 
int main (int argCount, char *argVal[]) { 
    if (argCount > 1) 
     numThreads = atoi (argVal[1]); 
    if ((numThreads < 5) || (numThreads > THRD_MAX)) { 
     printf ("Number of threads outside range %d-%d, using %d\n", 
      THRD_MIN, THRD_MAX, THRD_DFLT); 
     numThreads = THRD_DFLT; 
    } 
    : 
    :