352 lines
14 KiB
C#
352 lines
14 KiB
C#
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using ShardingCore.Core.VirtualRoutes.TableRoutes.RoutingRuleEngine;
|
|
using ShardingCore.DbContexts.VirtualDbContexts;
|
|
using ShardingCore.Extensions;
|
|
using ShardingCore.Test50.Domain.Entities;
|
|
using Xunit;
|
|
|
|
namespace ShardingCore.Test50
|
|
{
|
|
/*
|
|
* @Author: xjm
|
|
* @Description:
|
|
* @Date: Friday, 15 January 2021 17:22:10
|
|
* @Email: 326308290@qq.com
|
|
*/
|
|
public class ShardingTest
|
|
{
|
|
private readonly IVirtualDbContext _virtualDbContext;
|
|
private readonly IRoutingRuleEngineFactory _routingRuleEngineFactory;
|
|
|
|
public ShardingTest(IVirtualDbContext virtualDbContext,IRoutingRuleEngineFactory routingRuleEngineFactory)
|
|
{
|
|
_virtualDbContext = virtualDbContext;
|
|
_routingRuleEngineFactory = routingRuleEngineFactory;
|
|
}
|
|
|
|
//[Fact]
|
|
//public async Task Route_TEST()
|
|
//{
|
|
// var queryable1 = _virtualDbContext.Set<SysUserMod>().Where(o=>o.Id=="339");
|
|
// var routeResults1 = _routingRuleEngineFactory.Route(queryable1);
|
|
// Assert.Equal(1,routeResults1.Count());
|
|
// Assert.Equal(1,routeResults1.FirstOrDefault().ReplaceTables.Count());
|
|
// Assert.Equal("0",routeResults1.FirstOrDefault().ReplaceTables.FirstOrDefault().Tail);
|
|
// Assert.Equal(nameof(SysUserMod),routeResults1.FirstOrDefault().ReplaceTables.FirstOrDefault().OriginalName);
|
|
// var ids = new[] {"339", "124","142"};
|
|
// var queryable2= _virtualDbContext.Set<SysUserMod>().Where(o=>ids.Contains(o.Id));
|
|
// var routeResult2s = _routingRuleEngineFactory.Route(queryable2);
|
|
// Assert.Equal(2,routeResult2s.Count());
|
|
// Assert.Equal(1,routeResult2s.FirstOrDefault().ReplaceTables.Count());
|
|
// Assert.Equal(2,routeResult2s.SelectMany(o=>o.ReplaceTables).Count());
|
|
// Assert.Equal(true,routeResult2s.SelectMany(o=>o.ReplaceTables).All(o=>new[]{"0","1"}.Contains(o.Tail)));
|
|
//}
|
|
[Fact]
|
|
public async Task ToList_All_Test()
|
|
{
|
|
var mods = await _virtualDbContext.Set<SysUserMod>().ToShardingListAsync();
|
|
Assert.Equal(1000, mods.Count);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_Join_Test()
|
|
{
|
|
var list = await (from u in _virtualDbContext.Set<SysUserMod>()
|
|
join salary in _virtualDbContext.Set<SysUserSalary>()
|
|
on u.Id equals salary.UserId
|
|
select new
|
|
{
|
|
Salary = salary.Salary,
|
|
DateOfMonth = salary.DateOfMonth,
|
|
Name = u.Name
|
|
}).ToShardingListAsync();
|
|
Assert.Equal(24000, list.Count());
|
|
Assert.Equal(24, list.Count(o => o.Name == "name_200"));
|
|
|
|
|
|
var queryable = (from u in _virtualDbContext.Set<SysUserMod>().Where(o => o.Id == "300")
|
|
join salary in _virtualDbContext.Set<SysUserSalary>()
|
|
on u.Id equals salary.UserId
|
|
select new
|
|
{
|
|
Salary = salary.Salary,
|
|
DateOfMonth = salary.DateOfMonth,
|
|
Name = u.Name
|
|
});
|
|
var list1 = await queryable.ToShardingListAsync();
|
|
Assert.Equal(24, list1.Count());
|
|
Assert.DoesNotContain(list1, o => o.Name != "name_300");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_OrderBy_Asc_Desc_Test()
|
|
{
|
|
var modascs = await _virtualDbContext.Set<SysUserMod>().OrderBy(o => o.Age).ToShardingListAsync();
|
|
Assert.Equal(1000, modascs.Count);
|
|
var i = 1;
|
|
foreach (var age in modascs)
|
|
{
|
|
Assert.Equal(i, age.Age);
|
|
i++;
|
|
}
|
|
|
|
var moddescs = await _virtualDbContext.Set<SysUserMod>().OrderByDescending(o => o.Age).ToShardingListAsync();
|
|
Assert.Equal(1000, moddescs.Count);
|
|
var j = 1000;
|
|
foreach (var age in moddescs)
|
|
{
|
|
Assert.Equal(j, age.Age);
|
|
j--;
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_Id_In_Test()
|
|
{
|
|
var ids = new[] {"1", "2", "3", "4"};
|
|
var sysUserMods = await _virtualDbContext.Set<SysUserMod>().Where(o => ids.Contains(o.Id)).ToShardingListAsync();
|
|
foreach (var id in ids)
|
|
{
|
|
Assert.Contains(sysUserMods, o => o.Id == id);
|
|
}
|
|
|
|
Assert.DoesNotContain(sysUserMods, o => o.Age > 4);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_Id_Eq_Test()
|
|
{
|
|
var mods = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Id == "3").ToShardingListAsync();
|
|
Assert.Single(mods);
|
|
Assert.Equal("3", mods[0].Id);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_Id_Not_Eq_Test()
|
|
{
|
|
var mods = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Id != "3").ToShardingListAsync();
|
|
Assert.Equal(999, mods.Count);
|
|
Assert.DoesNotContain(mods, o => o.Id == "3");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_Id_Not_Eq_Skip_Test()
|
|
{
|
|
var mods = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Id != "3").OrderBy(o => o.Age).Skip(2).ToShardingListAsync();
|
|
Assert.Equal(997, mods.Count);
|
|
Assert.DoesNotContain(mods, o => o.Id == "3");
|
|
Assert.Equal(4, mods[0].Age);
|
|
Assert.Equal(5, mods[1].Age);
|
|
|
|
var modsDesc = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Id != "3").OrderByDescending(o => o.Age).Skip(13).ToShardingListAsync();
|
|
Assert.Equal(986, modsDesc.Count);
|
|
Assert.DoesNotContain(mods, o => o.Id == "3");
|
|
Assert.Equal(987, modsDesc[0].Age);
|
|
Assert.Equal(986, modsDesc[1].Age);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_Name_Eq_Test()
|
|
{
|
|
var mods = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name == "name_3").ToShardingListAsync();
|
|
Assert.Single(mods);
|
|
Assert.Equal("3", mods[0].Id);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_Id_Eq_Not_In_Db_Test()
|
|
{
|
|
var mods = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Id == "1001").ToShardingListAsync();
|
|
Assert.Empty(mods);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ToList_Name_Eq_Not_In_Db_Test()
|
|
{
|
|
var mods = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name == "name_1001").ToShardingListAsync();
|
|
Assert.Empty(mods);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task FirstOrDefault_Order_By_Id_Test()
|
|
{
|
|
var sysUserModAge = await _virtualDbContext.Set<SysUserMod>().OrderBy(o => o.Age).ShardingFirstOrDefaultAsync();
|
|
Assert.True(sysUserModAge != null && sysUserModAge.Id == "1");
|
|
var sysUserModAgeDesc = await _virtualDbContext.Set<SysUserMod>().OrderByDescending(o => o.Age).ShardingFirstOrDefaultAsync();
|
|
Assert.True(sysUserModAgeDesc != null && sysUserModAgeDesc.Id == "1000");
|
|
var sysUserMod = await _virtualDbContext.Set<SysUserMod>().OrderBy(o => o.Id).ShardingFirstOrDefaultAsync();
|
|
Assert.True(sysUserMod != null && sysUserMod.Id == "1");
|
|
|
|
var sysUserModDesc = await _virtualDbContext.Set<SysUserMod>().OrderByDescending(o => o.Id).ShardingFirstOrDefaultAsync();
|
|
Assert.True(sysUserModDesc != null && sysUserModDesc.Id == "999");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task FirstOrDefault2()
|
|
{
|
|
var sysUserMod = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Id == "1").ShardingFirstOrDefaultAsync();
|
|
Assert.NotNull(sysUserMod);
|
|
Assert.True(sysUserMod.Id == "1");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task FirstOrDefault3()
|
|
{
|
|
var sysUserMod = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name == "name_2").ShardingFirstOrDefaultAsync();
|
|
Assert.NotNull(sysUserMod);
|
|
Assert.Equal("2", sysUserMod.Id);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task FirstOrDefault4()
|
|
{
|
|
var sysUserMod = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Id != "1").ShardingFirstOrDefaultAsync();
|
|
Assert.NotNull(sysUserMod);
|
|
Assert.True(sysUserMod.Id != "1");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task FirstOrDefault5()
|
|
{
|
|
var sysUserMod = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name == "name_1001").ShardingFirstOrDefaultAsync();
|
|
Assert.Null(sysUserMod);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Count_Test()
|
|
{
|
|
var a = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name == "name_1000").ShardingCountAsync();
|
|
Assert.Equal(1, a);
|
|
var b = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name != "name_1000").ShardingCountAsync();
|
|
Assert.Equal(999, b);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Sum_Test()
|
|
{
|
|
var a = await _virtualDbContext.Set<SysUserMod>().ShardingSumAsync(o => o.Age);
|
|
var expected = 0;
|
|
for (int i = 1; i <= 1000; i++)
|
|
{
|
|
expected += i;
|
|
}
|
|
|
|
Assert.Equal(expected, a);
|
|
var b = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name != "name_1000").ShardingSumAsync(o => o.Age);
|
|
Assert.Equal(expected - 1000, b);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Max_Test()
|
|
{
|
|
var a = await _virtualDbContext.Set<SysUserMod>().ShardingMaxAsync(o => o.Age);
|
|
Assert.Equal(1000, a);
|
|
var b = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name != "name_1000").ShardingMaxAsync(o => o.Age);
|
|
Assert.Equal(999, b);
|
|
var c = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Age < 500).ShardingMaxAsync(o => o.Age);
|
|
Assert.Equal(499, c);
|
|
var e = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Age <= 500).ShardingMaxAsync(o => o.Age);
|
|
Assert.Equal(500, e);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Max_Join_Test()
|
|
{
|
|
var queryable = (from u in _virtualDbContext.Set<SysUserMod>().Where(o => o.Id == "300")
|
|
join salary in _virtualDbContext.Set<SysUserSalary>()
|
|
on u.Id equals salary.UserId
|
|
select new
|
|
{
|
|
Salary = salary.Salary,
|
|
DateOfMonth = salary.DateOfMonth,
|
|
Name = u.Name
|
|
});
|
|
var maxSalary = await queryable.ShardingMaxAsync(o => o.Salary);
|
|
Assert.Equal(1390000, maxSalary);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Min_Test()
|
|
{
|
|
var a = await _virtualDbContext.Set<SysUserMod>().ShardingMinAsync(o => o.Age);
|
|
Assert.Equal(1, a);
|
|
var b = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name != "name_1").ShardingMinAsync(o => o.Age);
|
|
Assert.Equal(2, b);
|
|
var c = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Age > 500).ShardingMinAsync(o => o.Age);
|
|
Assert.Equal(501, c);
|
|
var e = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Age >= 500).ShardingMinAsync(o => o.Age);
|
|
Assert.Equal(500, e);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Any_Test()
|
|
{
|
|
var a = await _virtualDbContext.Set<SysUserMod>().ShardingAnyAsync(o => o.Age == 100);
|
|
Assert.True(a);
|
|
var b = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Name != "name_1").ShardingAnyAsync(o => o.Age == 1);
|
|
Assert.False(b);
|
|
var c = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Age > 500).ShardingAnyAsync(o => o.Age <= 500);
|
|
Assert.False(c);
|
|
var e = await _virtualDbContext.Set<SysUserMod>().Where(o => o.Age >= 500).ShardingAnyAsync(o => o.Age <= 500);
|
|
Assert.True(e);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Group_Test()
|
|
{
|
|
var ids = new[] {"200", "300"};
|
|
var dateOfMonths = new[] {202111, 202110};
|
|
var group = await (from u in _virtualDbContext.Set<SysUserSalary>()
|
|
.Where(o => ids.Contains(o.UserId) && dateOfMonths.Contains(o.DateOfMonth))
|
|
group u by new
|
|
{
|
|
UId = u.UserId
|
|
}
|
|
into g
|
|
select new
|
|
{
|
|
GroupUserId = g.Key.UId,
|
|
Count = g.Count(),
|
|
TotalSalary = g.Sum(o => o.Salary),
|
|
AvgSalary = g.Average(o => o.Salary),
|
|
AvgSalaryDecimal = g.Average(o => o.SalaryDecimal),
|
|
MinSalary = g.Min(o => o.Salary),
|
|
MaxSalary = g.Max(o => o.Salary)
|
|
}).ToShardingListAsync();
|
|
Assert.Equal(2, group.Count);
|
|
Assert.Equal(2, group[0].Count);
|
|
Assert.Equal(2260000, group[0].TotalSalary);
|
|
Assert.Equal(1130000, group[0].AvgSalary);
|
|
Assert.Equal(11300, group[0].AvgSalaryDecimal);
|
|
Assert.Equal(1120000, group[0].MinSalary);
|
|
Assert.Equal(1140000, group[0].MaxSalary);
|
|
}
|
|
[Fact]
|
|
public async Task Group_API_Test()
|
|
{
|
|
var ids = new[] {"200", "300"};
|
|
var dateOfMonths = new[] {202111, 202110};
|
|
var group = await _virtualDbContext.Set<SysUserSalary>()
|
|
.Where(o => ids.Contains(o.UserId) && dateOfMonths.Contains(o.DateOfMonth))
|
|
.ShardingGroupByAsync(g => new {UId = g.UserId}, g => new
|
|
{
|
|
|
|
GroupUserId = g.Key.UId,
|
|
Count = g.Count(),
|
|
TotalSalary = g.Sum(o => o.Salary),
|
|
AvgSalary = g.Average(o => o.Salary),
|
|
AvgSalaryDecimal = g.Average(o => o.SalaryDecimal),
|
|
MinSalary = g.Min(o => o.Salary),
|
|
MaxSalary = g.Max(o => o.Salary)
|
|
});
|
|
Assert.Equal(2, group.Count);
|
|
Assert.Equal(2, group[0].Count);
|
|
Assert.Equal(2260000, group[0].TotalSalary);
|
|
Assert.Equal(1130000, group[0].AvgSalary);
|
|
Assert.Equal(11300, group[0].AvgSalaryDecimal);
|
|
Assert.Equal(1120000, group[0].MinSalary);
|
|
Assert.Equal(1140000, group[0].MaxSalary);
|
|
}
|
|
}
|
|
} |