.NET：可擴展的單據編號生成器 之 順序號（防止重復）

背景

我在上篇文章“.NET：可擴展的單據編號生成器 + 簡單的解釋器”中介紹了一個簡單的單據編號框架。有朋友留言問如何實現“順序號，且不能重復”，本篇文章就針對這個問題用上篇介紹的框架進行實現。

思路

順序號 = 上次順序號 + 步長

根據上面的公式，問題可以化解為：如何獲取上次順序號？獲取上次順序號有兩種方式：

1. 1. 掃描單據表，找出最新的一條記錄。
   2. 引入種子表，種子表記錄了最新的順序號。

因為生成的順序號不能重復，這里就有了並發的要求，為了最大限度的提高並發性，我選擇2（引入種子表）。

並發處理可以選擇：悲觀鎖或樂觀鎖，這里為了簡單，我選擇悲觀鎖。

實現

代碼下載：http://yunpan.cn/Q5KMUTA3qGPct。

種子表設計

CREATE TABLE [dbo].[CodeSeeds] (
    [Id]    UNIQUEIDENTIFIER NOT NULL,
    [Key]   NVARCHAR (500)   NOT NULL,
    [Value] INT              NOT NULL,
    PRIMARY KEY CLUSTERED ([Id] ASC)
);

SeedCodeRuleProvider.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

using System.Transactions;
using System.Text.RegularExpressions;
using System.Data.Entity.Infrastructure;

namespace EntityCodeRuleDemo
{
    public class SeedCodeRuleProvider : ICodeRuleProvider
    {
        private readonly int _width;

        public SeedCodeRuleProvider(int width)
        {
            _width = width;
        }

        public string Generate(object entity)
        {
            return GetSeedValue(entity).ToString().PadLeft(_width, '0');
        }

        protected virtual string GetKey(object entity)
        {
            return entity.GetType().FullName;
        }

        private int GetSeedValue(object entity)
        {
            try
            {
                var value = 0;
                var key = this.GetKey(entity);

                using (var ts = new TransactionScope(TransactionScopeOption.RequiresNew,
                    new TransactionOptions { IsolationLevel = IsolationLevel.RepeatableRead }))
                {
                    using (var context = new TestContext())
                    {
                        var seed = context.CodeSeeds.Where(x => x.Key == key).FirstOrDefault();
                        if (seed == null)
                        {
                            seed = new CodeSeed { Id = Guid.NewGuid(), Key = key, Value = -1 };
                            context.CodeSeeds.Add(seed);
                        }

                        seed.Value++;
                        value = seed.Value;
                        context.SaveChanges();
                    }

                    ts.Complete();
                }

                return value;
            }
            catch (DbUpdateException)
            {
                return this.GetSeedValue(entity);
            }
        }

        public static SeedCodeRuleProvider SeedCodeRuleProviderFactory(string literal)
        {
            var match = new Regex("^<種子(:(?<寬度>.*?))?>$").Match(literal);

            var width = match.Groups["寬度"].Value;

            return new SeedCodeRuleProvider(string.IsNullOrEmpty(width) ? 5 : int.Parse(width));
        }
    }
}

Program.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

using System.Text.RegularExpressions;

namespace EntityCodeRuleDemo
{
    class Program
    {
        static void Main(string[] args)
        {
            CodeRuleInterpreter.RegistProviderFactory(new Regex("^<種子(:(?<寬度>.*?))?>$"), SeedCodeRuleProvider.SeedCodeRuleProviderFactory);

            var employeeCode = CodeRuleInterpreter
                .Interpret("前綴_<日期:yyyy_MM_dd>_<屬性:NamePinYin>_<種子:6>")
                .Generate(new Employee { NamePinYin = "DUANGW" });

            Console.WriteLine(employeeCode);
        }
    }
}

運行結果

備注

有寫業務要求會強制編號的連續性或編號的隨機性，對於這兩種需求，還需要單獨開發Provider，有機會再寫文章介紹了。