代码的线程安全替换？

Question

在开发过程中，我经常会遇到下一个问题：如果某个方法已经被一个线程执行，那么方法一定不能被另一个线程执行。另一个线程必须什么都不做 - 简单地退出方法，因为它我不能使用“锁”。通常，我是这样解决这个问题的：

private bool _isSomeMethodExecuted = false;

public void SomeMethod ()
{
 if (!this._isSomeMethodExecuted) //check if method is already executed
 {
        this._isSomeMethodExecuted = true;

        //Main code of method

        this._isSomeMethodExecuted = false;
 }
}

但是这段代码不是线程安全的：如果一个线程执行条件语句但它在设置标志为真之前停止并且另一个线程可以执行条件 - 那么两个线程都在方法代码中。

是否有任何线程安全的替换？

Answer 1

以下是线程安全的，如果方法已经在执行，则不会阻塞 - 即使它已经在同一个线程上执行......这为所有场景提供了重入保护。

private long _isSomeMethodExecuted = 0;

public void SomeMethod ()
{
 if (Interlocked.Increment (ref this._isSomeMethodExecuted) == 1) //check if method is already executed
 {
        //Main code of method

 }
Interlocked.Decrement (ref this._isSomeMethodExecuted);
}

参考见http://msdn.microsoft.com/en-us/library/zs86dyzy.aspx

Answer 2

Monitor 为您完成这项工作，但锁是线程范围的（因此对递归调用开放！）。 lock 语句也使用Monitor（使用阻塞Enter 方法），但您可以使用TryEnter 方法代替：

    if(Monitor.TryEnter(myLockObject))
    {
       try
       {
           DoSomething(); // main code
       }
       finally
       {
           Monitor.Exit(myLockObject);
       }
    }

TryEnter 不会阻塞，而是返回一个bool，表示锁是否成功获取。

如果您希望递归调用不再进入主代码块，则应使用信号量。信号量使用计数器而不是锁定对象，因此即使从同一个线程也无法重新进入：

class Program
{
    private static Semaphore sem = new Semaphore(1, 1);

    static void Main(string[] args)
    {
        MyMethod();
        MyMethod();
    }

    private static void MyMethod()
    {
        if(sem.WaitOne(0))
        {
            try
            {
                Console.WriteLine("Entered.");
                MyMethod(); // recursive calls won't re-enter
            }
            finally
            {
                sem.Release();
            }
        }
        else
        {
            Console.WriteLine("Not entered.");
        }
    }
}