sharding/samples/Sample.SqlServer3x/Startup.cs

106 lines
3.5 KiB
C#
Raw Normal View History

2021-03-31 15:55:27 +08:00
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
2021-08-15 07:32:50 +08:00
using Microsoft.EntityFrameworkCore;
2021-03-31 15:55:27 +08:00
using Sample.SqlServer3x.Domain.Entities;
using ShardingCore;
using ShardingCore.Bootstrapers;
2021-03-31 15:55:27 +08:00
namespace Sample.SqlServer3x
{
public class Startup
{
public Startup(IConfiguration configuration)
{
Configuration = configuration;
}
public IConfiguration Configuration { get; }
// This method gets called by the runtime. Use this method to add services to the container.
public void ConfigureServices(IServiceCollection services)
{
services.AddControllers();
2021-08-20 23:10:34 +08:00
// services.AddShardingSqlServer(o =>
// {
// o.EnsureCreatedWithOutShardingTable = true;
// o.CreateShardingTableOnStart = true;
// o.UseShardingDbContext<DefaultDbContext>( dbConfig =>
// {
// dbConfig.AddShardingTableRoute<SysUserModVirtualTableRoute>();
// });
// //o.AddDataSourceVirtualRoute<>();
//
// });
// services.AddDbContext<DefaultDbContext>(o => o.UseSqlServer("Data Source=localhost;Initial Catalog=ShardingCoreDB3x;Integrated Security=True")
// .UseShardingSqlServerUpdateSqlGenerator());
2021-03-31 15:55:27 +08:00
}
// This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
if (env.IsDevelopment())
{
app.UseDeveloperExceptionPage();
}
var shardingBootstrapper = app.ApplicationServices.GetService<IShardingBootstrapper>();
shardingBootstrapper.Start();
app.UseRouting();
app.UseAuthorization();
app.UseEndpoints(endpoints =>
{
endpoints.MapControllers();
});
InitData(app).GetAwaiter().GetResult();
}
/// <summary>
2021-08-15 07:32:50 +08:00
/// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
2021-03-31 15:55:27 +08:00
/// </summary>
/// <param name="serviceProvider"></param>
/// <returns></returns>
private async Task InitData(IApplicationBuilder app)
{
var serviceProvider = app.ApplicationServices;
using (var scope = serviceProvider.CreateScope())
{
2021-08-15 07:32:50 +08:00
var virtualDbContext = scope.ServiceProvider.GetService<DefaultDbContext>();
2021-08-20 23:10:34 +08:00
if (!await virtualDbContext.Set<SysUserMod>().AnyAsync(o => true))
2021-03-31 15:55:27 +08:00
{
var ids = Enumerable.Range(1, 1000);
var userMods = new List<SysUserMod>();
var beginTime = new DateTime(2020, 1, 1);
var endTime = new DateTime(2021, 12, 1);
foreach (var id in ids)
{
userMods.Add(new SysUserMod()
{
Id = id.ToString(),
Age = id,
Name = $"name_{id}",
AgeGroup = Math.Abs(id % 10)
});
}
2021-08-15 07:32:50 +08:00
await virtualDbContext.AddRangeAsync(userMods);
2021-03-31 15:55:27 +08:00
await virtualDbContext.SaveChangesAsync();
}
}
}
}
}