C# LockFreeStack类

C# LockFreeStack类

单链表实现的LockFreeStack

 

using System;
using System.Collections;
using System.Threading;

namespace Rocky.Stored
{
public class LockFreeStack : IEnumerable
{
private class SingleLinkedNode
{
public object Value;
public SingleLinkedNode Next;
}

private SingleLinkedNode m_head;
private int count;

public int Count
{
get { return count; }
}

public void Push( object item)
{
SingleLinkedNode node = new SingleLinkedNode();
node.Value = item;
SingleLinkedNode head;
do
{
head = m_head;
node.Next = head;
}
while (m_head != head || Interlocked.CompareExchange( ref m_head, node, head) != head);
Interlocked.Increment( ref count);
}

public object Peek()
{
SingleLinkedNode head = m_head;
if (head == null )
{
throw new InvalidOperationException( " Stack Empty! " );
}
return head.Value;
}

public object Pop()
{
object result;
if (TryPop( out result))
{
return result;
}
throw new InvalidOperationException( " Stack Empty! " );
}

/// <summary>
/// 考虑到在多线程环境中，这里不抛出异常。我们需要人为判断其是否为空，即 !TryPop() or result != null
/// </summary>
/// <returns></returns>
public bool TryPop( out object item)
{
SingleLinkedNode head;
SingleLinkedNode next;
do
{
head = m_head;
if (head == null )
{
item = null ;
return false ;
}
next = m_head.Next;
}
while (Interlocked.CompareExchange( ref m_head, next, head) != head);
Interlocked.Decrement( ref count);
item = head.Value;
return true ;
}

public object SpinPop()
{
SingleLinkedNode head;
SpinWait spin = new SpinWait();
while ( true )
{
SingleLinkedNode next;
do
{
head = m_head;
if (head == null )
{
goto emptySpin;
}
next = head.Next;
}
while (m_head != head || Interlocked.CompareExchange( ref m_head, next, head) != head);
break ;
emptySpin:
spin.Spin();
}
Interlocked.Decrement( ref count);
return head.Value;
}

public bool Contains( object item)
{
SingleLinkedNode head = m_head;
while (m_head != null && head != null )
{
if (head.Value.Equals(item))
{
return true ;
}
head = head.Next;
}
return false ;
}

public void Clear()
{
Interlocked.Exchange( ref m_head, null );
Interlocked.Exchange( ref count, 0 );
}

public IEnumerator GetEnumerator()
{
SingleLinkedNode head = m_head;
while (m_head != null && head != null )
{
yield return head.Value;
head = head.Next;
}
}
}
}

 

 

 

 

posted on 2010-06-09 09:23 RockyLOMO 阅读(...) 评论(...) 编辑 收藏