33 lines
1.4 KiB
C#
33 lines
1.4 KiB
C#
using System.Linq;
|
|
using Microsoft.EntityFrameworkCore.Metadata;
|
|
using Microsoft.EntityFrameworkCore.Migrations;
|
|
using Microsoft.EntityFrameworkCore.Migrations.Operations;
|
|
using Microsoft.EntityFrameworkCore.Update;
|
|
using ShardingCore.Core.RuntimeContexts;
|
|
using ShardingCore.Helpers;
|
|
|
|
namespace Sample.SqlServer.Shardings
|
|
{
|
|
public class ShardingSqlServerMigrationsSqlGenerator:SqlServerMigrationsSqlGenerator
|
|
{
|
|
private readonly IShardingRuntimeContext _shardingRuntimeContext;
|
|
|
|
public ShardingSqlServerMigrationsSqlGenerator(IShardingRuntimeContext shardingRuntimeContext,MigrationsSqlGeneratorDependencies dependencies, ICommandBatchPreparer commandBatchPreparer) : base(dependencies, commandBatchPreparer)
|
|
{
|
|
_shardingRuntimeContext = shardingRuntimeContext;
|
|
}
|
|
protected override void Generate(
|
|
MigrationOperation operation,
|
|
IModel model,
|
|
MigrationCommandListBuilder builder)
|
|
{
|
|
var oldCmds = builder.GetCommandList().ToList();
|
|
base.Generate(operation, model, builder);
|
|
var newCmds = builder.GetCommandList().ToList();
|
|
var addCmds = newCmds.Where(x => !oldCmds.Contains(x)).ToList();
|
|
|
|
MigrationHelper.Generate(_shardingRuntimeContext, operation, builder, Dependencies.SqlGenerationHelper, addCmds);
|
|
}
|
|
}
|
|
}
|