承接上一篇文章的业务场景[C#使用读写锁三行代码简单解决多线程并发写入文件时线程同步的问题]。
随着服务进程的增多,光凭进程内的线程同步已经不能满足现在的需求,导致多进程同时写入同一个文件时,一样提示文件被占用的问题。
在这种场景下,跨进程级的锁是不可避免的。在.NET提供的参考中,进程锁都继承了System.Threading.WaitHandle类。
而在本文中针对单个文件同一时间仅允许单个进程(线程)操作的场景,System.Threading.Mutex类无疑是最简单也是最合适的选择。
该类型的对象可以使用命名(字符串)互斥量实现当前会话级或操作系统级的同步需求。我选择了操作系统级别的同步编写示例,因为覆盖面更广。
namespace WaitHandleExample
{
class Program
{
static void Main(string[] args)
{
#region 简单使用
//var mutexKey = MutexExample.GetFilePathMutexKey("文件路径");
//MutexExample.MutexExec(mutexKey, () =>
//{
// Console.WriteLine("需要进程同步执行的代码");
//});
#endregion
#region 测试代码
var filePath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "test.log").ToUpper();
var mutexKey = MutexExample.GetFilePathMutexKey(filePath);
//同时开启N个写入线程
Parallel.For(0, LogCount, e =>
{
//没使用互斥锁操作写入,大量写入错误;FileStream包含FileShare的构造函数也仅实现了进程内的线程同步,多进程同时写入时也会出错
//WriteLog(filePath);
//使用互斥锁操作写入,由于同一时间仅有一个线程操作,所以不会出错
MutexExample.MutexExec(mutexKey, () =>
{
WriteLog(filePath);
});
});
Console.WriteLine(string.Format("Log Count:{0}.\t\tWrited Count:{1}.\tFailed Count:{2}.", LogCount.ToString(), WritedCount.ToString(), FailedCount.ToString()));
Console.Read();
#endregion
}
/// <summary>
/// C#互斥量使用示例代码
/// </summary>
/// <remarks>已在经过测试并上线运行,可直接使用</remarks>
public static class MutexExample
{
/// <summary>
/// 进程间同步执行的简单例子
/// </summary>
/// <param name="action">同步处理代码</param>
/// <param name="mutexKey">操作系统级的同步键
/// (如果将 name 指定为 null 或空字符串,则创建一个局部互斥体。
/// 如果名称以前缀“Global\”开头,则 mutex 在所有终端服务器会话中均为可见。
/// 如果名称以前缀“Local\”开头,则 mutex 仅在创建它的终端服务器会话中可见。
/// 如果创建已命名 mutex 时不指定前缀,则它将采用前缀“Local\”。)</param>
/// <remarks>不重试且不考虑异常情况处理的简单例子</remarks>
[Obsolete(error: false, message: "请使用MutexExec")]
public static void MutexExecEasy(string mutexKey, Action action)
{
//声明一个已命名的互斥体,实现进程间同步;该命名互斥体不存在则自动创建,已存在则直接获取
using (Mutex mut = new Mutex(false, mutexKey))
{
try
{
//上锁,其他线程需等待释放锁之后才能执行处理;若其他线程已经上锁或优先上锁,则先等待其他线程执行完毕
mut.WaitOne();
//执行处理代码(在调用WaitHandle.WaitOne至WaitHandle.ReleaseMutex的时间段里,只有一个线程处理,其他线程都得等待释放锁后才能执行该代码段)
action();
}
finally
{
//释放锁,让其他进程(或线程)得以继续执行
mut.ReleaseMutex();
}
}
}
/// <summary>
/// 获取文件名对应的进程同步键
/// </summary>
/// <param name="filePath">文件路径(请注意大小写及空格)</param>
/// <returns>进程同步键(互斥体名称)</returns>
public static string GetFilePathMutexKey(string filePath)
{
//生成文件对应的同步键,可自定义格式(互斥体名称对特殊字符支持不友好,遂转换为BASE64格式字符串)
var fileKey = Convert.ToBase64String(Encoding.Default.GetBytes(string.Format(@"FILE\{0}", filePath)));
//转换为操作系统级的同步键
var mutexKey = string.Format(@"Global\{0}", fileKey);
return mutexKey;
}
/// <summary>
/// 进程间同步执行
/// </summary>
/// <param name="mutexKey">操作系统级的同步键
/// (如果将 name 指定为 null 或空字符串,则创建一个局部互斥体。
/// 如果名称以前缀“Global\”开头,则 mutex 在所有终端服务器会话中均为可见。
/// 如果名称以前缀“Local\”开头,则 mutex 仅在创建它的终端服务器会话中可见。
/// 如果创建已命名 mutex 时不指定前缀,则它将采用前缀“Local\”。)</param>
/// <param name="action">同步处理操作</param>
public static void MutexExec(string mutexKey, Action action)
{
MutexExec(mutexKey: mutexKey, action: action, recursive: false);
}
/// <summary>
/// 进程间同步执行
/// </summary>
/// <param name="mutexKey">操作系统级的同步键
/// (如果将 name 指定为 null 或空字符串,则创建一个局部互斥体。
/// 如果名称以前缀“Global\”开头,则 mutex 在所有终端服务器会话中均为可见。
/// 如果名称以前缀“Local\”开头,则 mutex 仅在创建它的终端服务器会话中可见。
/// 如果创建已命名 mutex 时不指定前缀,则它将采用前缀“Local\”。)</param>
/// <param name="action">同步处理操作</param>
/// <param name="recursive">指示当前调用是否为递归处理,递归处理时检测到异常则抛出异常,避免进入无限递归</param>
private static void MutexExec(string mutexKey, Action action, bool recursive)
{
//声明一个已命名的互斥体,实现进程间同步;该命名互斥体不存在则自动创建,已存在则直接获取
//initiallyOwned: false:默认当前线程并不拥有已存在互斥体的所属权,即默认本线程并非为首次创建该命名互斥体的线程
//注意:并发声明同名的命名互斥体时,若间隔时间过短,则可能同时声明了多个名称相同的互斥体,并且同名的多个互斥体之间并不同步,高并发用户请另行处理
using (Mutex mut = new Mutex(initiallyOwned: false, name: mutexKey))
{
try
{
//上锁,其他线程需等待释放锁之后才能执行处理;若其他线程已经上锁或优先上锁,则先等待其他线程执行完毕
mut.WaitOne();
//执行处理代码(在调用WaitHandle.WaitOne至WaitHandle.ReleaseMutex的时间段里,只有一个线程处理,其他线程都得等待释放锁后才能执行该代码段)
action();
}
//当其他进程已上锁且没有正常释放互斥锁时(譬如进程忽然关闭或退出),则会抛出AbandonedMutexException异常
catch (AbandonedMutexException ex)
{
//避免进入无限递归
if (recursive)
throw ex;
//非递归调用,由其他进程抛出互斥锁解锁异常时,重试执行
MutexExec(mutexKey: mutexKey, action: action, recursive: true);
}
finally
{
//释放锁,让其他进程(或线程)得以继续执行
mut.ReleaseMutex();
}
}
}
}
#region 测试写文件的代码
static int LogCount = 500;
static int WritedCount = 0;
static int FailedCount = 0;
static void WriteLog(string logFilePath)
{
try
{
var now = DateTime.Now;
var logContent = string.Format("Tid: {0}{1} {2}.{3}\r\n", Thread.CurrentThread.ManagedThreadId.ToString().PadRight(4), now.ToLongDateString(), now.ToLongTimeString(), now.Millisecond.ToString());
File.AppendAllText(logFilePath, logContent);
WritedCount++;
}
catch (Exception ex)
{
Console.WriteLine(ex.Message);
FailedCount++;
}
}
#endregion
}
}
测试结果:6个进程同时进行3000次写入请求,仅成功写入277次
测试结果:6个进程同时进行3000次写入请求,全部成功写入
进程同步的资源消耗及效率比线程同步要差得多,请根据实际场景合理使用。
本文虽然是用写入文件作为示例,但进程同步的代码使用场景与文件操作无关。
Semaphore类(信号灯)虽然可以限制同时操作的线程数,甚至把最大同时操作数设置为1时,行为与Mutex类(互斥量)类似;但是由于信号灯在其他进程中出现异常退出时并不能接收到异常通知,只能通过等待超时触发异常,并不适合现在的场景,所以并没讲述。
关于进程同步的其他深入了解及应用,请参阅其他资料。