80 lines
2.5 KiB
C#
80 lines
2.5 KiB
C#
|
using HashidsNet;
|
|||
|
|
using Yitter.IdGenerator;
|
|||
|
|
|
|||
|
|
namespace Easy.Core;
|
|||
|
|
|
|||
|
|
public static class YitIdHelperExtension
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 盐值
|
|||
|
|
/// </summary>
|
|||
|
|
private const string Salt = "dkjsDLVK1S7y5be8XYzoq0C2nHaEmBfA";
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 注册雪花ID服务（默认配置）
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="service"></param>
|
|||
|
|
public static void AddIdGenerator(this IServiceCollection service)
|
|||
|
|
{
|
|||
|
|
service.AddIdGenerator(new IdGeneratorOptions(0));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 注册雪花ID服务（自定义配置）
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="service"></param>
|
|||
|
|
/// <param name="sectionName"></param>
|
|||
|
|
public static void AddIdGenerator(this IServiceCollection service, string sectionName)
|
|||
|
|
{
|
|||
|
|
var options = App.GetConfig<IdGeneratorOptions>(sectionName) ?? new IdGeneratorOptions(0);
|
|||
|
|
options.BaseTime = options.BaseTime.ToUniversalTime();
|
|||
|
|
YitIdHelper.SetIdGenerator(options);
|
|||
|
|
service.AddSingleton<IIdGenerator>(new DefaultIdGenerator(options));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 注册雪花ID服务（自定义配置）
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="service"></param>
|
|||
|
|
/// <param name="options">配置</param>
|
|||
|
|
public static void AddIdGenerator(this IServiceCollection service, IdGeneratorOptions options)
|
|||
|
|
{
|
|||
|
|
options.BaseTime = options.BaseTime.ToUniversalTime();
|
|||
|
|
YitIdHelper.SetIdGenerator(options);
|
|||
|
|
service.AddSingleton<IIdGenerator>(new DefaultIdGenerator(options));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 生成ID
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="generator"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public static long NextId(this IIdGenerator generator)
|
|||
|
|
{
|
|||
|
|
return generator.NewLong();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 加密ID（生成短ID）
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="generator"></param>
|
|||
|
|
/// <param name="id">long类型ID</param>
|
|||
|
|
/// <param name="salt">盐值</param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public static string Encode(this IIdGenerator generator, long id, string salt = Salt)
|
|||
|
|
{
|
|||
|
|
return new Hashids(salt).EncodeLong(id);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 解密加密ID（解密短ID）
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="generator"></param>
|
|||
|
|
/// <param name="text">密文</param>
|
|||
|
|
/// <param name="salt">盐值</param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public static long Decode(this IIdGenerator generator, string text, string salt = Salt)
|
|||
|
|
{
|
|||
|
|
return new Hashids(salt).DecodeSingleLong(text);
|
|||
|
|
}
|
|||
|
|
}
|