随笔-149  评论-223  文章-30  trackbacks-0
主类模板
   gcc从4.1.2版本开始提供了__sync_*系列的内置API,用于加减和逻辑运算,可以对1,2,4,8字节长度的数值或指针类型进行原子操作,为方便使用,笔者对这些API作了简单的封装。
 1template<typename T>
 2class atomic_base 
 3{
 4public:
 5    typedef T int_type;
 6    typedef atomic_base<T> self_type;
 7
 8public:
 9    atomic_base(int_type val = int_type())
10    : val_(val)
11    {
12    }

13
14    ~atomic_base()
15    {
16        BOOST_STATIC_ASSERT(sizeof(int_type)==1||sizeof(int_type)==2||sizeof(int_type)==4||sizeof(int_type)==8);
17    }

18
19    atomic_base(const self_type& other)
20    {
21        int_type val = other;
22        __sync_lock_test_and_set(&val_, val);
23    }

24
25    self_type& operator = (const self_type& other)
26    {
27        if (this!=&other)
28        {
29            int_type val = other;
30            __sync_lock_test_and_set(&val_, val);
31        }

32        return *this;
33    }

34    
35    operator int_type() const
36    {
37        return __sync_val_compare_and_swap(const_cast<volatile int_type*>(&val_),0,0);
38    }

39    
40    self_type operator++(int)
41    {
42        return __sync_fetch_and_add(&val_,1);
43    }

44
45    self_type& operator++()
46    {
47        __sync_add_and_fetch(&val_,1);
48        return *this;
49    }

50
51    self_type operator--(int)
52    {
53        return __sync_fetch_and_sub(&val_,1);
54    }

55
56    self_type& operator--()
57    {
58        __sync_sub_and_fetch(&val_,1);
59        return *this;
60    }

61
62    self_type& operator+=(int_type val)
63    {
64        __sync_add_and_fetch(&val_,val);
65        return *this;
66    }

67
68    self_type& operator-=(int_type val)
69    {
70        __sync_sub_and_fetch(&val_,val);
71        return *this;
72    }

73
74    self_type& operator&=(int_type val)
75    
76        __sync_and_and_fetch(&val_, val); 
77        return *this;
78    }

79
80    self_type& operator|=(int_type val)
81    {
82        __sync_or_and_fetch(&val_,val); 
83        return *this;
84    }

85    
86    self_type& operator^=(int_type val)
87    {
88        __sync_xor_and_fetch(&val_,val); 
89        return *this;
90    }

91
92private:
93    T val_;
94}
;
   上面析构函数中,使用了BOOST_STATIC_ASSERT宏来作编译期约束,确保类型T大小为1,2,4,8字节长度。由于提供了T到atomic_base的隐式转换和转型操作,因此没有必要再定义operato +,operator - ,opeator ^,operator |,operator &运算符成员或友元重载操作。

指针类型偏特化
   由于主模板定义了operator+=,operator-=,operator&=,operator|=,operator^=运算符重载操作,但当T为指针类型时,指针的加减和逻辑运算没有意义,但对于指针加减一个数值是有意义的,因此前两者操作保留且重定义,但没有后三者操作,需要特化实现:
 1template<typename T>
 2class atomic_base<T*> 
 3{
 4public:
 5    typedef T* pointer_type;
 6    typedef atomic_base<T*> self_type;
 7
 8public:
 9    atomic_base(pointer_type val = NULL)
10        : val_(val)
11    {
12    }

13
14    atomic_base(const self_type& other)
15    {
16        pointer_type val = other;
17        __sync_lock_test_and_set(&val_, val);
18    }

19
20    ~atomic_base()
21    {
22        BOOST_STATIC_ASSERT(sizeof(pointer_type)==1||sizeof(pointer_type)==2||sizeof(pointer_type)==4||sizeof(pointer_type)==8);
23    }

24
25    self_type& operator=(const self_type& other)
26    {
27        if (this!=&other)
28        {
29            pointer_type val = other;
30            __sync_lock_test_and_set(&val_, val);
31        }

32        return *this;
33    }

34
35    operator pointer_type() const
36    {
37        return __sync_val_compare_and_swap(const_cast<volatile pointer_type*>(&val_),0,0);
38    }

39
40    self_type operator++(int)
41    {
42        return __sync_fetch_and_add(&val_,1);
43    }

44
45    self_type& operator++()
46    {
47        __sync_add_and_fetch(&val_,1);
48        return *this;
49    }

50
51    self_type operator--(int)
52    {
53        return __sync_fetch_and_sub(&val_,1);
54    }

55
56    self_type& operator--()
57    {
58        __sync_sub_and_fetch(&val_,1);
59        return *this;
60    }

61
62    self_type& operator+=(std::ptrdiff_t d)
63    {
64        __sync_add_and_fetch(&val_,d);
65        return *this;
66    }

67
68    self_type& operator-=(std::ptrdiff_t d)
69    {
70        __sync_sub_and_fetch(&val_,d);
71        return *this;
72    }

73
74private:
75    T* val_;
76}
;
   同理,上面析构函数中,使用了BOOST_STATIC_ASSERT宏来作编译期约束,确保类型T的指针大小为1,2,4,8字节长度。
   
bool类型完全特化
   由于大小为1字节的类型包括bool布尔类型,它的取值只有true或false两种可能,不存在加减运算操作,也没有意义,但逻辑运算是有意义的,因此需要特化实现:
 1template<>
 2class atomic_base<bool>
 3{
 4public:
 5    typedef atomic_base<bool> self_type;
 6public:
 7    atomic_base(bool val = false)
 8        : val_(val)
 9    {
10    }

11    
12    atomic_base(const self_type& other)
13    {
14        __sync_lock_test_and_set(&val_, other.val_);
15    }

16    
17    self_type& operator=(const self_type& other)
18    {
19        if (this!=&other)
20            __sync_lock_test_and_set(&val_, other.val_);
21        return *this;
22    }

23    
24    operator bool() const
25    {
26        return __sync_val_compare_and_swap(const_cast<volatile bool*>(&val_),0,0);
27    }

28
29    self_type& operator&=(bool val)
30    
31        __sync_and_and_fetch(&val_, val); 
32        return *this;
33    }

34
35    self_type& operator|=(bool val)
36    {
37        __sync_or_and_fetch(&val_,val); 
38        return *this;
39    }

40
41    self_type& operator^=(bool val)
42    {
43        __sync_xor_and_fetch(&val_,val); 
44        return *this;
45    }

46
47private:
48    bool val_;
49}
;
  
整数类型别名   
   针对常见的基本数值类型,定义了其类型别名:
 1typedef atomic_base<bool>                     atomic_bool;
 2typedef atomic_base<signed char>           atomic_char;
 3typedef atomic_base<unsigned char>        atomic_uchar;
 4typedef atomic_base<short>                    atomic_short;
 5typedef atomic_base<unsigned short>       atomic_ushort;
 6typedef atomic_base<int>                       atomic_int;
 7typedef atomic_base<unsigned int>          atomic_uint;
 8typedef atomic_base<long>                     atomic_long;
 9typedef atomic_base<unsigned long>        atomic_ulong;
10typedef atomic_base<long long>              atomic_llong;
11typedef atomic_base<unsigned long long>  atomic_ullong;
posted on 2012-06-08 00:19 春秋十二月 阅读(4513) 评论(1)  编辑 收藏 引用 所属分类: C/C++

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理