On the first day of your new job, your boss asks you to find all primes between 1 and 10^10(never mind why), using a parallel machine that supports ten concurrent threads. This machine is rented by the minute, so the longer your program takes, the more it costs. You want to make a good impression. What do you do?
我对多线程编程不是很熟，所以想用这个例子学习一下多线程编程。
单线程编程
如果不会用线程，那么就用最简单的单线程解决这个问题。
- //example1.c
- #include <stdio.h>
- #include <stdlib.h>
- #include <math.h>
- long long isPrime(int num)
- {
- long long i, s;
- s = (long long)sqrt(double(num));
- for(i = 2; i < s; i++)
- if(num % i == 0)
- return 0;
- return 1;
- }
- int main()
- {
- long long i,primes_count = 0;
- for(i = 1 ; i <= 10000000000L; i++)
- primes_count += isPrime(i);
- printf("%lld primes);
- return 0;
- }
注意：编译时加上-lm，gcc -lm example1.c ; 打印long long整数要用%lld
如果真用这段代码跑10000000000L，估计几个小时跑不完，几天不知道能跑完吗。我打算在16核的机器上跑一下，不过16核和单核一样，因为是单线程。
多线程方法1
如果会使用线程，一个简单的方法是10个线程平分这些数。
-------------------------------------------------------------------------------------------
pthread相关数据结构和函数
pthread_t 调用pthread_t返回的线程号，就是一个unsigned int;
int pthread_create(pthread_t *, pthread_attr_t *, void* (*) (void*) , void*);
第一个参数是一个pthread_t，第二个是pthread_attr_t，具体看下面的用法，第三个是线程要运行的主函数，第四个是要给这个函数的参数。
(void*) (*) (void*) 是一个函数类型，就是类似于 void * run (void*)的函数。
void pthread_exit() 线程退出
int pthread_join(pthread_t *, void **) 主线程等待子线程退出，第二个参数直接用NULL就行
----------------------------------------------------------------------------------------------------------
- //example2.c
- #include <stdio.h>
- #include <stdlib.h>
- #include <pthread.h>
- #include <sys/types.h>
- #include <unistd.h>
- #include <math.h>
- long long isPrime(long long num)
- {
- long long i, s;
- s = (long long)sqrt((double)num);
- for(i = 2; i < s; i++)
- if(num % i == 0)
- return 0;
- return 1;
- }
- void * run(void * param)
- {
- long long start,end;
- long long i,primes_count = 0;
- start =( (long long*)param)[0];
- end =( (long long*)param)[1];
- for(i = start ; i < end; i++)
- primes_count += isPrime(i);
- printf("Process %d Thread %u gets %lld primes from %lld to %lld\n",getpid(),(unsigned)pthread_self(),primes_count,start,end);
- pthread_exit(0);
- }
- int main()
- {
- pthread_t tid[10];
- pthread_attr_t attr;
- long long start_end[10][2];
- long long i;
- for(i = 0; i < 10; i++)
- {
- start_end[i][0] = i*10000;
- start_end[i][1] = (i+1)*10000;
- pthread_attr_init(&attr);
- pthread_create(&tid[i],&attr,run,(void*)start_end[i]);
- }
- for(i= 0; i < 10; i++)
- pthread_join(tid[i],NULL);
- return 0;
- }
注意：使用pthread编译要加上-lpthread。 gcc -lm -lpthread example2.c
运行结果：（不是10^10,只计算到100000)
------------------------------------------------------------------------------------------------
遇到的问题：
最开始我的主函数写成这样：
- int main()
- {
- pthread_t tid[10];
- pthread_attr_t attr;
- long long start_end[2];
- long long i;
- for(i = 0; i < 10; i++)
- {
- start_end[0] = i*10000;
- start_end[1] = (i+1)*10000;
- pthread_attr_init(&attr);
- pthread_create(&tid[i],&attr,run,(void*)start_end);
- pthread_join(tid[i],NULL);
- }
- return 0;
- }
很明显pthread_join地方写错了，这样第一个进程运行完了才会创建第二个，但是由于每个线程平分，所以我没发现结果有异常。
写第三个程序时pthread_join也写错了位置，这才发现。但是我将pthread_join放到后面时，发现每个线程处理的数据段都是90000-100000，没找到原因，因此我的start_end改成了二维数组，这样各个线程就不会干扰了，具体的原因以后再研究。
--------------------------------------------------------------------------------------
这个程序应该比第一个快，但是判断第一段数据的线程使用的时间肯定没有最后一段使用的时间的千分之一，最后9个人看一个人干活，我突然想到了什么。。。
多线程方法2
多线程的目的就是压榨计算资源。我们可以使用下面的方法。
每个线程取一个数，然后判断，判断完了继续取下一个，让每个线程都不停的干活
线程取了1，判断，线程2取2判断，线程3取3，这时线程1干完了，接着取4，线程2又取5。。。。
- #include <stdio.h>
- #include <stdlib.h>
- #include <pthread.h>
- #include <sys/types.h>
- #include <unistd.h>
- #include <math.h>
- long long number = 1; //每个线程先取数，再把这个数加1
- long long isPrime(long long num)
- {
- long long i, s;
- s = (long long)sqrt((double)num);
- for(i = 2; i < s; i++)
- if(num % i == 0)
- return 0;
- return 1;
- }
- static pthread_mutex_t mutex;
- void * run(void * param)
- {
- long long primes_count=0;
- long long i = 1;
- while(i < 100000)
- {
- //要使用锁
- pthread_mutex_lock(&mutex);
- i = number;
- number ++;
- pthread_mutex_unlock(&mutex);
- primes_count += isPrime(i);
- }
- printf("Thread %u gets %lld primes\n",(unsigned int)pthread_self(),primes_count);
- pthread_exit(0);
- }
- int main()
- {
- pthread_t tid[10];
- pthread_attr_t attr;
- pthread_mutex_init(&mutex,NULL);
- int i;
- for(i = 0; i < 10; i++)
- {
- pthread_attr_init(&attr);
- pthread_attr_setscope (&attr, PTHREAD_SCOPE_SYSTEM);
- pthread_create(&tid[i],&attr,run,NULL);
- //pthread_join(tid[i],NULL); 一开始把join放这，结果只有第一个线程干活，找了好久才发现位置错了
- }
- for(i = 0; i < 10; i++)
- pthread_join(tid[i],NULL);
- return 0;
- }
运行结果：
多进程实现
用多线程能完成的，多进程也可以。
- #include <stdio.h>
- #include <stdlib.h>
- #include <pthread.h>
- #include <sys/types.h>
- #include <unistd.h>
- #include <math.h>
- #include <sys/ipc.h>
- #include <sys/shm.h>
- #include <errno.h>
- #include <string.h>
- long long isPrime(long long num)
- {
- long long i, s;
- s = (long long)sqrt((double)num);
- for(i = 2; i < s; i++)
- if(num % i == 0)
- return 0;
- return 1;
- }
- int main()
- {
- pid_t pid[10];
- int shmid;
- long long i,primes_count=0;
- long long * addr;
- shmid = shmget(IPC_PRIVATE,4096,IPC_CREAT|0600);
- if(shmid < 0)
- {
- printf("get shared memory failed\n");
- exit(1);
- }
- addr = (long long*) shmat(shmid,0,0);
- *addr = 1;
- for(i = 0; i < 3; i++)
- fork();
- while(i < 100000)
- {
- i = (*addr);
- (*addr) ++;
- primes_count += isPrime(i);
- }
- printf("pid %d get %lld primes\n",getpid(),primes_count);
- return 0;
- }
注意：(1)使用循环fork，那么子进程还会fork，因此fork三次就得到了8个进程，在fork就16个进程了。要想产生10个进程，可以在主进程里写10个fork语句，还得写很多if(fork()==0)语句，懒得写，就直接用循环了。加上主进程一个8个，先生完了孩子再干活，生完之后都成了兄弟，一起干。
(2)这个程序有个严重的问题，使用了共享内存，但是没有使用同步机制，没有锁，挺麻烦的，不会用，就没写。结果可能不正确。
运行结果：
小结
我就截几个图
这是单线程的，CPU使用了100%，看来也不错，但是我的CPU有200%，有两个核（我用的虚拟机，可以分16个核哦亲），因此单线程不能充分利用计算资源。
多线程的把我的CPU都用完了。哈哈
多进程的呢？算一算，也是200%。
其实在Linux里面线程也是进程，称为LWP(Light Weighted Process)，只不过在生成的时候参数不同。在以后的文章里会深入研究。
本文转自nxlhero 51CTO博客，原文链接：http://blog.51cto.com/nxlhero/1075979，如需转载请自行联系原作者
