memcached 高并发，流水号生成方案

2023年8月23日 315次阅读 来源: DAOSR

流水号作为唯一编号，在银行以及财务业务系统中尤为常见。
在设计中一般采用年、月、日等递增方式，列如：2018012200001，表示2018年1月22日的第一个编号。

传统的处理方式

常见数据库递增不能按年、月、日格式生成递增编号，不过可以通过其他方式实现，如在 MSSQL 数据库中可以采用触发、自定义函数实现。

新建一个表 Order，必须拥有以下三个字段

名称类型说明
Codenvarchar(32)用于存储完整的流水号
Seedbigint种子值，考虑大数据量的情况采用 bigint 类型
Datedatetime当前时间

触发器代码创建

IF (OBJECT_ID('TGR_Order_INSERT', 'TR') IS NOT NULL) DROP TRIGGER TGR_Order_INSERT
GO
CREATE TRIGGER TGR_Order_INSERT ON Order FOR INSERT
AS
BEGIN
    BEGIN tran
    DECLARE @OrderID UNIQUEIDENTIFIER, @Seed BIGINT
    SET @Seed = ISNULL((SELECT MAX(Seed) FROM Order WITH (TABLOCK, HOLDLOCK) WHERE DATEDIFF(DAY, Date, GETDATE()) = 0 AND Seed > 0), CAST(CONVERT(NVARCHAR(8), GETDATE(), 112) + N'0000000000' AS BIGINT)) -- 取得当前最大一个流水号种子，如果不存在使用默认值
    DECLARE OrderCursor CURSOR FOR SELECT OrderID FROM INSERTED
    OPEN OrderCursor
    FETCH NEXT FROM OrderCursor INTO @OrderID
    WHILE @@FETCH_STATUS = 0
    BEGIN
        SET @Seed = @Seed + 1 --递增种子号
        UPDATE Order SET Seed = @Seed, Code = CAST(@Seed AS NVARCHAR(18)) WHERE OrderID = @OrderID  
        FETCH NEXT FROM OrderCursor INTO @OrderID
    END  
    CLOSE OrderCursor    
    DEALLOCATE OrderCursor
    COMMIT tran
END
GO
  1. 注意该触发使用了游标，为了保证数据计算的准确性进行锁表操作。
  2. 稍作修改即可改为自定义函数，这里我就不再写出自定义函数的创建。
  3. 可以见得这种方式在海量与高并发的数据环境中，就显得先天不足。

memcached

memcached 本身就是为高并发和快速存取诞生的，并且他还自带了增量方法。

以下示例开发语言是 C# 所以采用 EnyimMemcached 作为客户端，数据框架 Entity Framework。

在程序启动时初始化种子

var date = DateTime.Now;
using (var context = new DbContext())
{
    var seed = context.Order.Where(m => DbFunctions.DiffDays(m.Date, date) == 0).Max(m => (long?)m.Seed) ?? 0;
    new MemcachedClient().Store(StoreMode.Set, "Test_" + date.ToString("yyyyMMdd"), seed.ToString(), TimeSpan.FromHours(25));
}
  1. ASP.NET 可以在 Application_Start 中初始化。
  2. 因为按天递增，所以保存过期时间大于24小时即可。

插入数据的实体赋值

var date = DateTime.Now;
var seed = new MemcachedClient().Increment("Test_" + date.ToString("yyyyMMdd"), 1, 1, TimeSpan.FromHours(25));
var entity = new Order {
  Code = date.ToString("yyyyMMdd") + seed .ToString().PadLeft(14, '0'),
  Seed = seed,
  Date = date
};
/* 这里将实体保存到数据库即可 */

文章就写到这里，这只是 memcached 的一种应用，在网站访问统计、防刷新机制等方面你都可以使用它。

    原文作者：DAOSR
    原文地址: https://www.jianshu.com/p/4094b07a0270
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞