.NET MasaFramework 锁与分布式锁

csharp 投稿 1500 0 评论

前言

什么是锁

加锁(lock)是2018年公布的计算机科学技术名词,是指将控制变量置位,控制共享资源不能被其他线程访问。通过加锁,可以确保在同一时刻只有一个线程在访问被锁住的代码片段,我们在单机部署时可使用最简单的加锁完成资源的独享,如:

public class Program
{
    private static readonly object Obj = new { };

    public static void Main()
    {
        lock (obj)
        {
            //同一时刻只有一个线程可以访问
        }
    }
}

什么是分布式锁

但随着业务发展的需要,原单体单机部署的系统被部署成分布式集群系统后,原来的并发控制策略失效,为了解决这个问题就需要引入分布式锁,那分布式锁应该具备哪些条件?

  • 原子性:在分布式环境下,一个方法在同一个时间点只能被一台机器下的一个线程所执行,防止数据资源的并发访问,避免数据不一致情况

  • 高可用:具备自动失效机制,防止死锁,获取锁后如果出现错误,并且无法释放锁,则使用租约一段时间后自动释放锁

  • 阻塞性:具备非阻塞锁特性(没有获取到锁时直接返回获取锁失败,不会长时间因等待锁导致阻塞)

  • 高性能:高性能的获取锁与释放锁

  • 可重入性:具备可重入特性,在同一线程外层函数获得锁之后,内层方法会自动获取锁

.NET MasaFramework  锁与分布式锁

实现

Local、Medallion,下面会介绍如何配置并使用它们

本地锁

SemaphoreSlim实现的,它不是真正的分布式锁,我们建议你在开发和测试环境中使用它,不需要联网也不会与其他人冲突

Medallion

DistributedLock实现的分布式锁,它提供了很多种技术的实现,包括Microsoft SQL ServerPostgresqlMySQL 或 MariaDBOracleRedisAzure blobApache ZooKeeper锁文件操作系统全局WaitHandles(Windows),我们只需要任选一种实现即可,目前Medallion提供的分布式锁并不支持可重入性,点击了解原因

快速入门

  • 安装

  • .NET 6.0

本地锁单应用锁为例:

  1. 新建ASP.NET Core 空项目

  2. Assignment.DistributedLock.Local
  3. ,并安装

  4. Masa.Contrib.Data.DistributedLock.Local
dotnet new web -o Assignment.DistributedLock.Local
cd Assignment.DistributedLock.Local
dotnet add package Masa.Contrib.Data.DistributedLock.Local --version 0.6.0-preview.10
  1. 注册锁,修改类Program

builder.Services.AddLocalDistributedLock();//注册本地锁
  1. 如何使用锁?修改类Program

app.MapGet("lock", (IDistributedLock distributedLock) =>
{
    using var @lock = distributedLock.TryGet("test");//获取锁
    if (@lock != null)
    {
        //todo: 获取锁成功
        return "success";
    }
    return "获取超时";
});

通过DI获取IDistributedLock,并通过TryGet方法获取锁,如果获取锁失败,则返回null,如果返回到的对象不为null,则表明获取锁成功,最后在获取锁成功后写自己的业务代码即可

TryGet方法拥有以下参数

    key
  • (string, 必须): 锁的唯一名称,可通过key来访问不同的资源,执行不同的业务

  • timeout (TimeSpan): 等待获取锁的最大超时时间. 默认值为: TimeSpan.Zero(代表如果锁已经被另一个应用程序拥有, 它不会等待.)

TryGetAsync方法除了拥有TryGet的所有参数之外,还拥有以下参数

    cancellationToken
  • : 取消令牌可在触发后取消操作

Medallion,只需要选择一种技术实现,并根据Readme注册锁即可,在使用锁上是没有区别的

如何扩展其它的分布式锁

    Masa.Contrib.Data.DistributedLock.{分布式锁名},并添加引用Masa.BuildingBlocks.Data.csproj

  1. DefaultDistributedLock,并实现IDistributedLock

public class DefaultDistributedLock : IDistributedLock
{
    public IDisposable? TryGet(string key, TimeSpan timeout = default)
    {
        // 获取锁失败则返回null,当资源被释放时,主动释放锁, 无需人为手动释放
        throw new NotImplementedException();
    }

    public Task<IAsyncDisposable?> TryGetAsync(string key, TimeSpan timeout = default, CancellationToken cancellationToken = default)
    {
        //获取锁失败则返回null,当资源被释放时,主动释放锁, 无需人为手动释放
        throw new NotImplementedException();
    }
}
  1. 新建类ServiceCollectionExtensions,注册分布式锁到服务集合

public static class ServiceCollectionExtensions
{
    public static IServiceCollection AddDistributedLock(this IServiceCollection services, Action<MedallionBuilder> builder)
    {
        services.TryAddSingleton<IDistributedLock, DefaultDistributedLock>();
        return services;
    }
}

小知识

为什么TryGetTryGetAsync方法的返回类型分别是IDisposableIAsyncDisposable

IDisposable、IAsyncDisposable时,使用完毕后会触发DisposeDisposeAsync,这样一来就可以使得开发者可以忽略释放锁的逻辑

public class DefaultLocalDistributedLock : IDistributedLock
{
    private readonly MemoryCache<string, SemaphoreSlim> _localObjects = new();

    public IDisposable? TryGet(string key, TimeSpan timeout = default)
    {
        var semaphore = GetSemaphoreSlim(key);

        if (!semaphore.Wait(timeout))
        {
            return null;
        }

        return new DisposeAction(semaphore);
    }

    //todo: 以下省略 TryGetAsync 方法

    private SemaphoreSlim GetSemaphoreSlim(string key)
    {
        ArgumentNullOrWhiteSpaceException.ThrowIfNullOrWhiteSpace(key);
        return _localObjects.GetOrAdd(key, _ => new SemaphoreSlim(1, 1));
    }
}

internal class DisposeAction : IDisposable, IAsyncDisposable
{
    private readonly SemaphoreSlim _semaphore;

    public DisposeAction(SemaphoreSlim semaphore) => _semaphore = semaphore;

    public ValueTask DisposeAsync()
    {
        _semaphore.Release();
        return ValueTask.CompletedTask;
    }

    public void Dispose() => _semaphore.Release();
}

本章源码

Assignment09

https://github.com/zhenlei520/MasaFramework.Practice

开源地址

https://github.com/masastack/MASA.Framework

https://github.com/masalabs/MASA.EShop

https://github.com/BlazorComponent/MASA.Blazor

.NET MasaFramework  锁与分布式锁

编程学习分享 » .NET MasaFramework 锁与分布式锁

赞 (0) or 分享 (0)
游客 发表我的评论   换个身份
取消评论

表情
(0)个小伙伴在吐槽

高效,专业,符合SEO

联系我们