think_ya
3 years ago
15 changed files with 191 additions and 18 deletions
@ -0,0 +1,18 @@ |
|||
<Project Sdk="Microsoft.NET.Sdk"> |
|||
|
|||
<PropertyGroup> |
|||
<TargetFramework>net6.0</TargetFramework> |
|||
<RootNamespace /> |
|||
<IsPackable>false</IsPackable> |
|||
</PropertyGroup> |
|||
|
|||
<ItemGroup> |
|||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.5.0" /> |
|||
</ItemGroup> |
|||
|
|||
<ItemGroup> |
|||
<ProjectReference Include="..\..\modules\common\LINGYUN.Abp.IdGenerator\LINGYUN.Abp.IdGenerator.csproj" /> |
|||
<ProjectReference Include="..\LINGYUN.Abp.TestBase\LINGYUN.Abp.TestsBase.csproj" /> |
|||
</ItemGroup> |
|||
|
|||
</Project> |
|||
@ -0,0 +1,42 @@ |
|||
using LINGYUN.Abp.Tests; |
|||
using System.Collections.Generic; |
|||
using System.Threading.Tasks; |
|||
|
|||
namespace LINGYUN.Abp.IdGenerator; |
|||
|
|||
public class AbpIdGeneratorModuleTestBase : AbpTestsBase<AbpIdGeneratorModuleTestModule> |
|||
{ |
|||
protected IDistributedIdGenerator DistributedIdGenerator { get; set; } |
|||
|
|||
private void GenerateId(long[] idArray, int startIndex, int endIndex) |
|||
{ |
|||
for (int i = startIndex; i < endIndex; i++) |
|||
{ |
|||
idArray[i] = DistributedIdGenerator.Create(); |
|||
} |
|||
} |
|||
|
|||
protected long[] GenerateIdInSingleThread(int countIds) |
|||
{ |
|||
long[] ids = new long[countIds]; |
|||
|
|||
GenerateId(ids, 0, ids.Length); |
|||
|
|||
return ids; |
|||
} |
|||
|
|||
protected long[] GenerateIdInMutiThread(int countThreads, int countIdsPerThread) |
|||
{ |
|||
List<Task> tasks = new(); |
|||
long[] idArray = new long[countThreads * countIdsPerThread]; |
|||
|
|||
for (int i = 0; i < countThreads; i++) |
|||
{ |
|||
int threadId = i; |
|||
tasks.Add(Task.Run(() => GenerateId(idArray, threadId * countIdsPerThread, (threadId + 1) * countIdsPerThread))); |
|||
} |
|||
Task.WaitAll(tasks.ToArray()); |
|||
|
|||
return idArray; |
|||
} |
|||
} |
|||
@ -0,0 +1,21 @@ |
|||
using LINGYUN.Abp.IdGenerator.Snowflake; |
|||
using LINGYUN.Abp.Tests; |
|||
using Volo.Abp.Modularity; |
|||
|
|||
namespace LINGYUN.Abp.IdGenerator; |
|||
|
|||
[DependsOn( |
|||
typeof(AbpIdGeneratorModule), |
|||
typeof(AbpTestsBaseModule))] |
|||
public class AbpIdGeneratorModuleTestModule : AbpModule |
|||
{ |
|||
public override void PreConfigureServices(ServiceConfigurationContext context) |
|||
{ |
|||
PreConfigure<SnowflakeIdOptions>(options => |
|||
{ |
|||
options.WorkerId = 30010 - 30000; |
|||
options.WorkerIdBits = 5; |
|||
options.DatacenterId = 1; |
|||
}); |
|||
} |
|||
} |
|||
@ -0,0 +1,34 @@ |
|||
using Shouldly; |
|||
using System.Linq; |
|||
using Xunit; |
|||
|
|||
namespace LINGYUN.Abp.IdGenerator; |
|||
|
|||
public class IdGeneratorTests : AbpIdGeneratorModuleTestBase |
|||
{ |
|||
public IdGeneratorTests() |
|||
{ |
|||
DistributedIdGenerator = GetRequiredService<IDistributedIdGenerator>(); |
|||
} |
|||
|
|||
[Fact] |
|||
public void IdShouldBeUnique_SingleThread() |
|||
{ |
|||
const int countIds = 50000; |
|||
long[] ids = GenerateIdInSingleThread(countIds); |
|||
|
|||
var finalIds = ids.GroupBy(id => id).Select(d => d.Key).ToArray(); |
|||
finalIds.Length.ShouldBe(countIds); |
|||
} |
|||
|
|||
[Fact] |
|||
public void IdShouldBeUnique_MultiThread() |
|||
{ |
|||
const int countIdsPerThread = 50000; |
|||
const int countThreads = 8; |
|||
long[] ids = GenerateIdInMutiThread(countThreads, countIdsPerThread); |
|||
|
|||
var finalIds = ids.GroupBy(id => id).Select(d => d.Key).ToArray(); |
|||
finalIds.Length.ShouldBe(countThreads * countIdsPerThread); |
|||
} |
|||
} |
|||
@ -0,0 +1,51 @@ |
|||
using LINGYUN.Abp.IdGenerator.Snowflake; |
|||
using System; |
|||
using System.Linq; |
|||
using Shouldly; |
|||
using Xunit; |
|||
|
|||
namespace LINGYUN.Abp.IdGenerator; |
|||
|
|||
public class SnowflakeIdGeneratorTests : AbpIdGeneratorModuleTestBase |
|||
{ |
|||
public SnowflakeIdGeneratorTests() |
|||
{ |
|||
DistributedIdGenerator = null; |
|||
} |
|||
|
|||
[Theory] |
|||
[InlineData(-1, 5)] |
|||
[InlineData(0, 5)] |
|||
[InlineData(10, 5)] |
|||
[InlineData(31, 5)] |
|||
[InlineData(32, 5)] |
|||
[InlineData(30010, 5)] |
|||
public void WorkerIdShouldInRange(int workerId, int workerIdBits) |
|||
{ |
|||
var snowflakeIdGenerator = SnowflakeIdGenerator.Create(new SnowflakeIdOptions |
|||
{ |
|||
WorkerId = workerId, |
|||
WorkerIdBits = workerIdBits |
|||
}); |
|||
|
|||
snowflakeIdGenerator.WorkerId.ShouldBeInRange(0, (long)Math.Pow(2, workerIdBits) - 1); |
|||
} |
|||
|
|||
[Fact] |
|||
public void BiggerWorkerIdMakeIdRepeated() |
|||
{ |
|||
var snowflakeIdGenerator = SnowflakeIdGenerator.Create(new SnowflakeIdOptions |
|||
{ |
|||
WorkerId = 30010, |
|||
WorkerIdBits = 5 |
|||
}); |
|||
snowflakeIdGenerator.GetType().GetProperty("WorkerId").SetValue(snowflakeIdGenerator, 30010); |
|||
DistributedIdGenerator = snowflakeIdGenerator; |
|||
|
|||
const int countIds = 50000; |
|||
long[] ids = GenerateIdInSingleThread(countIds); |
|||
|
|||
var finalIds = ids.GroupBy(id => id).Select(d => d.Key).ToArray(); |
|||
finalIds.Length.ShouldBeLessThan(countIds); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue