public DbCommand(
DbContext dbContext,
System.Data.IDbCommand innerCommand)
{
Data = new DbCommandData(dbContext, innerCommand);
Data.ExecuteQueryHandler = new ExecuteQueryHandler(this);
}
public static string GetDefaultConnectionName()
{
var tableFeature = new string[] { "sys_button", "sys_code", "sys_codeType", "sys_log", "sys_loginHistory", "sys_roleMenuColumnMap", "sys_user", "sys_menu" };
IDbContext db = null;
Configuration config = WebConfigurationManager.OpenWebConfiguration("~/");
foreach (ConnectionStringSettings item in config.ConnectionStrings.ConnectionStrings)
{
var provider = Db.Provider(item.ProviderName);
if (provider == null)
continue;
//特征检测
db = new DbContext().ConnectionString(item.ConnectionString, provider);
var sql = string.Empty;
switch (item.ProviderName)
{
case "SqlServer":
sql = string.Format("select name from dbo.sysobjects where name in ('{0}')", string.Join("','", tableFeature));
break;
case "MySql":
sql = string.Format("select TABLE_NAME as TableName from information_schema.TABLES where TABLE_SCHEMA= database() and TABLE_NAME in ('{0}')", string.Join("','", tableFeature));
break;
}
if (string.IsNullOrEmpty(sql))
continue;
var result = db.Sql(sql).QueryMany<dynamic>();
if (result.Count == tableFeature.Length)
return item.Name;
}
return string.Empty;
}
public DbCommandData(DbContext context, System.Data.IDbCommand innerCommand)
{
Context = context;
InnerCommand = innerCommand;
InnerCommand.CommandType = (System.Data.CommandType)DbCommandTypes.Text;
Sql = new StringBuilder();
}
public IDbContext CreateDbContext()
{
var providers = new Dictionary<string, IDbProvider>(){
{"DB2",new DB2Provider()},
{"MySql",new MySqlProvider()},
{"Oracle",new OracleProvider()},
{"PostgreSql",new PostgreSqlProvider()},
{"SqlAzure",new SqlAzureProvider()},
{"Sqlite",new SqliteProvider()},
{"SqlServerCompact",new SqlServerCompactProvider()},
{"SqlServer",new SqlServerProvider()}
};
var connectionString = AppConfig["connectionString"];
var providerName = AppConfig["providerName"];
var provider = providers[providerName] != null ? providers[providerName] : new SqlServerProvider();
var dbcontext = new DbContext().ConnectionString(connectionString, provider);
return dbcontext;
}
public string Execute()
{
var provider = Db.Provider(providerName);
var dbexist = TestConnectionString();
var db = new DbContext().ConnectionString(GetDefaultConnectionString(), provider);
//如果数据库不存在，则创建数据库
if (!dbexist)
{
try
{
db.Sql(GetScriptCreateDatabase()).Execute();
}
catch (Exception e)
{
db.Dispose();
return e.Message;
}
}
try
{
//开启事务
db.UseTransaction(true);
db.CommandTimeout(60);
//创建表结构
var createScript = GetScriptCreateTable();
if (!string.IsNullOrEmpty(createScript))
db.Sql(GetScriptUseDatabase() + createScript).Execute();
//初始化数据
var dataScript = GetScriptInsertData();
if (!string.IsNullOrEmpty(createScript))
db.Sql(GetScriptUseDatabase() + dataScript).Execute();
db.Commit();
}
catch (Exception e)
{
try
{
db.Rollback();
db.UseTransaction(false);
}
catch{ }
//处理失败，删除刚刚创建的数据库
if (!dbexist)
db.Sql(GetScriptDropDatabase()).Execute();
return e.Message;
}
finally
{
db.Dispose();
}
return string.Empty;
}
