Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Entity Framework CTP 4 - Code First Custom Database Initializer

I would like to implement a custom database initialization strategy so that I can generate the database schema and apply it to an EXISTING EMPTY SQL database using a supplied User ID and Password.

Unfortunately the built-in strategies don’t provide what I’m looking for:

// The default strategy creates the DB only if it doesn't exist - but it does  // exist so this does nothing Database.SetInitializer(new CreateDatabaseOnlyIfNotExists<DataContext>());  // Drops and re-creates the database but then this breaks my security mapping and  // only works if using a “Trusted" connection Database.SetInitializer(new RecreateDatabaseIfModelChanges<DataContext>());  // Strategy for always recreating the DB every time the app is run. – no good for  // what I want Database.SetInitializer(new AlwaysRecreateDatabase<DataContext>()); 

I have worked out the following but this does not create the ModelHash so I’m unable to use "context.Database.ModelMatchesDatabase()" to validate that the database schema has been created and prevent multiple initializations:

public class Initializer : IDatabaseInitializer<DataContext>   {      Public void InitializeDatabase(DataContext context)       {                 // this generates the SQL script from my POCO Classes          var sql = context.ObjectContext.CreateDatabaseScript();           // As expected - when run the second time it bombs out here with "there is already an          // object named xxxxx in the database"          context.ObjectContext.ExecuteStoreCommand(sql);            this.seed(context)          context.SaveChanges();     } }   

Questions:

Does anyone know how I can get/create the model hash? (which is an EdmMetadata Entity)

-Or-

Is there a better way of doing this in general using the Code First CTP?

like image 769
Mark Avatar asked Oct 27 '10 08:10

Mark


1 Answers

I ran into the same problem. I didn't really solve it, but I managed to get a little nasty workaround running, so i can deploy my solution to AppHarbor ;)

Its a IDatabaseInitializer implementation, that doesn't delete the db, but just nukes all the constraints and tables, and then uses the ObjectContext.CreateDatabaseScript() method to generate the sql, and then I execute it as a storecommand. A lot like the above implementation in the question.

But i also added functionality to create a hash from the model and save it in db, and when it runs again it checks if the current model-hash matches the one i db. Just like the real code-first implementation.

I couldn't make it work with the build in context.Database.CompatibleWithModel(true) - but this should work just as well, and seeing as its a temporary workaround it should be fine.

using System; using System.Data.Entity; using System.Data.Entity.Database; using System.Data.Entity.Design; using System.Data.Entity.Infrastructure; using System.Data.Metadata.Edm; using System.Data.Objects; using System.Globalization; using System.Security.Cryptography; using System.Text; using System.Xml; using System.Linq;  namespace Devtalk {     public class DontDropDbJustCreateTablesIfModelChanged<T> : IDatabaseInitializer<T> where T : DbContext     {         private EdmMetadata _edmMetaData;          public void InitializeDatabase(T context)         {             ObjectContext objectContext = ((IObjectContextAdapter)context).ObjectContext;             string modelHash = GetModelHash(objectContext);              if (CompatibleWithModel(modelHash, context, objectContext)) return;              DeleteExistingTables(objectContext);             CreateTables(objectContext);              SaveModelHashToDatabase(context, modelHash, objectContext);         }          private void SaveModelHashToDatabase(T context, string modelHash, ObjectContext objectContext)         {             if (_edmMetaData != null) objectContext.Detach(_edmMetaData);              _edmMetaData = new EdmMetadata();             context.Set<EdmMetadata>().Add(_edmMetaData);              _edmMetaData.ModelHash = modelHash;             context.SaveChanges();         }          private void CreateTables(ObjectContext objectContext)         {             string dataBaseCreateScript = objectContext.CreateDatabaseScript();             objectContext.ExecuteStoreCommand(dataBaseCreateScript);         }          private void DeleteExistingTables(ObjectContext objectContext)         {             objectContext.ExecuteStoreCommand(Dropallconstraintsscript);             objectContext.ExecuteStoreCommand(Deletealltablesscript);         }          private string GetModelHash(ObjectContext context)         {             var csdlXmlString = GetCsdlXmlString(context).ToString();             return ComputeSha256Hash(csdlXmlString);         }          private bool CompatibleWithModel(string modelHash, DbContext context, ObjectContext objectContext)         {             var isEdmMetaDataInStore = objectContext.ExecuteStoreQuery<int>(LookupEdmMetaDataTable).FirstOrDefault();             if (isEdmMetaDataInStore == 1)             {                             _edmMetaData = context.Set<EdmMetadata>().FirstOrDefault();                 if (_edmMetaData != null)                 {                     return modelHash == _edmMetaData.ModelHash;                 }             }             return false;         }          private string GetCsdlXmlString(ObjectContext context)         {             if (context != null)             {                 var entityContainerList = context.MetadataWorkspace.GetItems<EntityContainer>(DataSpace.SSpace);                 if (entityContainerList != null)                 {                     EntityContainer entityContainer = entityContainerList.FirstOrDefault();                     var generator = new EntityModelSchemaGenerator(entityContainer);                     var stringBuilder = new StringBuilder();                     var xmlWRiter = XmlWriter.Create(stringBuilder);                     generator.GenerateMetadata();                     generator.WriteModelSchema(xmlWRiter);                     xmlWRiter.Flush();                     return stringBuilder.ToString();                 }             }             return string.Empty;         }          private static string ComputeSha256Hash(string input)         {             byte[] buffer = new SHA256Managed().ComputeHash(Encoding.ASCII.GetBytes(input));             var builder = new StringBuilder(buffer.Length * 2);             foreach (byte num in buffer)             {                 builder.Append(num.ToString("X2", CultureInfo.InvariantCulture));             }             return builder.ToString();         }          private const string Dropallconstraintsscript =             @"select                   'ALTER TABLE ' + so.table_name + ' DROP CONSTRAINT ' + so.constraint_name                   from INFORMATION_SCHEMA.TABLE_CONSTRAINTS so";          private const string Deletealltablesscript =             @"declare @cmd varchar(4000)                 declare cmds cursor for                  Select                     'drop table [' + Table_Name + ']'                 From                     INFORMATION_SCHEMA.TABLES                  open cmds                 while 1=1                 begin                     fetch cmds into @cmd                     if @@fetch_status != 0 break                     print @cmd                     exec(@cmd)                 end                 close cmds                 deallocate cmds";          private const string LookupEdmMetaDataTable =             @"Select COUNT(*)                FROM INFORMATION_SCHEMA.TABLES T                Where T.TABLE_NAME = 'EdmMetaData'";     } } 
like image 94
Luhmann Avatar answered Sep 25 '22 21:09

Luhmann