多线程计数器 原子操

net_robber

我现在有一个需求，多线程并发循环执行一个操作，但是要求精确计数操作了多少次
由于i++操作不是原子操作，所以，从kernel中扣了一段代码，试图进行计数

多次试验后发现，该段代码仍然不是原子操作

atomic自增操作定义如下 （该定义不完整，修改后可以工作的代码在二楼）

#ifndef ATOMIC_32 #define ATOMIC_32 typedef struct {         volatile int counter; } atomic_t; static inline void atomic_inc(atomic_t *v) {         asm volatile( "incl %0"                      : "+m" (v->counter)); } #endif //ATOMIC_32

测试代码如下

#include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <string.h> #include <getopt.h> #include <pthread.h> #include "atomic.h" int thread_num=100; typedef struct _Count {         unsigned int m;         unsigned int n; } Count; typedef struct _Thread {         pthread_t thread_id;         Count count; } Thread; Thread* thread; int atomic_start( Count* count); atomic_t x={0}; int main(int argc, char** argv) {        thread=(Thread*)malloc(thread_num*sizeof(Thread));         memset(thread,0,thread_num*sizeof(Thread));         int i;         for(i=0;i<thread_num;i++)         {                 if(pthread_create(&((thread+i)->thread_id),NULL, atomic_start,&((thread+i)->count)))                 {                         printf("thread create faild: %d",i--);                 }         }         int count=0;         for(i=0;i<10;i++)         {                 printf("%d\n",x);                 sleep(1);         }         return 0; } int atomic_start( Count* count) {         pthread_t thread_id=pthread_self();         pthread_detach(thread_id);         int i;         for(i=1000;i>0;i--)         {                 atomic_inc(&x);         } }

[ 本帖最后由 net_robber 于 2009-9-24 22:42 编辑 ]

net_robber

搞定
自增定义少了一截
新的atomic自增定义如下

#ifndef ATOMIC_32 #define ATOMIC_32 #define CONFIG_SMP #ifdef CONFIG_SMP #define LOCK_PREFIX \                 ".section .smp_locks,\"a\"\n"   \                 "  .align 4\n"                  \                 "  .long 661f\n" /* address */  \                 ".previous\n"                   \                 "661:\n\tlock; " #else /* ! CONFIG_SMP */ #define LOCK_PREFIX "" #endif typedef struct {         volatile int counter; } atomic_t; static inline void atomic_inc(atomic_t *v) {         asm volatile( LOCK_PREFIX "incl %0"                      : "+m" (v->counter)); } #endif //ATOMIC_32

epegasus

难道是在smp环境?

drangon

原子操作可以去找glib、ACE等现成封装好的库来用，从kernel里面找代码有时候会有风险

net_robber

我代码一共才几百行，而且是专用的小工具，用glib或者ACE，成本过高

crazyhadoop

atomic。。

sonicling

不需要atomic。每个线程单独计数，print的时候把它们加起来，顶多volatile + memory barrier。理论上效率比atomic高，因为线程无需smp_lock。