NpgsqlMigrationsSqlGenerator issue after upgrading to EF Core 5.0.1

I’ve just upgraded to 5.0.1 and my custom migration SQL generator failed.

I replace it like below

  .ReplaceService<IMigrationsSqlGenerator, MyMigrationSqlGenerator>()

All tables and constraints are created successfully but, while starting to seeding the cities table for a specific schema, I got the exception;

“There is no entity type mapped to the table ‘sample_tenant.cities’ used in a data operation. Either add the corresponding entity type to the model or specify the column types in the data operation.”

I got an exception on the line below of Generate( ) method for an InsertDataOperation

  base.Generate(operation, model, builder);

Whole MyMigrationSqlGenerator class;

    public class PmsMigrationSqlGenerator : NpgsqlMigrationsSqlGenerator
    {
        private readonly string _schema;

        public PmsMigrationSqlGenerator(MigrationsSqlGeneratorDependencies dependencies, INpgsqlOptions npgsqlOptions)
            : base(dependencies, npgsqlOptions)
        {
            _schema = UserLocalContext.Instance.GetTenantCode();
        }
        protected override void Generate(MigrationOperation operation, IModel model, MigrationCommandListBuilder builder)
        {
            Action action;
            switch (operation)
            {
                case EnsureSchemaOperation op:
                    action = ProcessEnsureSchemaOperation(op);
                    break;

                case CreateTableOperation op:
                    action = ProcessCreateTableOperation(op);
                    break;

                case SqlOperation op:
                    action = ProcessQueryOperation(op);
                    break;

                default:
                    action = ProcessBaseMigrationOperation(operation);
                    break;
            }

            if (action == Action.Skip)
                return;

            base.Generate(operation, model, builder);
        }

        private Action ProcessQueryOperation(SqlOperation op)
        {
            op.Sql = op.Sql.Replace("#schema", _schema);
            return Action.Apply;
        }

        private Action ProcessEnsureSchemaOperation(EnsureSchemaOperation op)
        {
            op.Name = _schema;
            return Action.Apply;
        }

        private Action ProcessCreateTableOperation(CreateTableOperation op)
        {
            if (ProcessBaseMigrationOperation(op) == Action.Skip)
                return Action.Skip;

            foreach (var item in op.UniqueConstraints)
                item.Schema = _schema;

            foreach (var foreignKey in op.ForeignKeys)
            {
                foreignKey.Schema = _schema;
                foreignKey.PrincipalSchema = _schema;
            }

            return Action.Apply;
        }

        private Action ProcessBaseMigrationOperation(MigrationOperation op)
        {
            var schemaProperty = op.GetType().GetProperties().FirstOrDefault(p => p.Name == "Schema");

            if (schemaProperty != null)
                schemaProperty.SetValue(op, _schema);

            var principleSchemaProperty = op.GetType().GetProperties().FirstOrDefault(p => p.Name == "PrincipalSchema");

            if (principleSchemaProperty == null)
                return Action.Apply;

            principleSchemaProperty.SetValue(op, _schema);

            return Action.Apply;
        }

        private enum Action
        {
            Apply,
            Skip
        }
    }

[AndriySvyryd]

op.KeyColumnTypes = table.PrimaryKey.Columns.Select(s => s.Name).ToArray();

I think you meant op.KeyColumnTypes = table.PrimaryKey.Columns.Select(s => s.StoreType).ToArray();

[AndriySvyryd]

@suadev As the exception message states “Either add the corresponding entity type to the model or specify the column types in the data operation.” So if you can’t alter the model appropriately then specify the column types in the InsertDataOperation. Though there’s currently an issue when it doesn’t actually flow through, so be sure to vote for it to be fixed soon: https://github.com/dotnet/efcore/issues/23503