dan/Cosmos
1
0
Fork 0
mirror of https://github.com/danbulant/Cosmos synced 2026-05-27 05:52:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix additional error conditions for issue #5628

Browse source
This commit is contained in:
BlueSkeye_cp 2012-09-19 17:15:19 +00:00
parent 4acc61602c
commit 6a91eeb14f
1 changed files with 60 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

108
source2/Debug/Cosmos.Debug.Common/DebugInfo.cs
View file

@ -9,6 +9,7 @@ using System.Data.Common;
using System.Data.Objects; using System.Data.Objects;
using System.Data.Objects.DataClasses; using System.Data.Objects.DataClasses;
using System.Reflection; using System.Reflection;
using System.Text;
using Microsoft.Win32; using Microsoft.Win32;
using Microsoft.Samples.Debugging.CorSymbolStore; using Microsoft.Samples.Debugging.CorSymbolStore;
using System.Diagnostics.SymbolStore; using System.Diagnostics.SymbolStore;
@ -40,57 +41,68 @@ namespace Cosmos.Debug.Common {
int databaseId = 0; int databaseId = 0;
bool xExists = false; bool xExists = false;
using (var xCmd = xConn.CreateCommand()) { using (var xCmd = xConn.CreateCommand())
xCmd.CommandText = "select database_id from sys.databases where name = '" + aDbName + "'"; {
object rawId = xCmd.ExecuteScalar(); xCmd.CommandText = "select database_id from sys.databases where name = '" + aDbName + "'";
if (null != rawId) { object rawId = xCmd.ExecuteScalar();
databaseId = (int)rawId; if (null != rawId)
xExists = true; {
} databaseId = (int)rawId;
} xExists = true;
if (xExists) {
List<string> databaseFiles = new List<string>();
bool damagedDatabase = false;
using (var xCmd = xConn.CreateCommand()) {
xCmd.CommandText = "SELECT [physical_name] FROM [master].[sys].[master_files] WHERE [database_id] = " + databaseId;
using (SqlDataReader reader = xCmd.ExecuteReader()) {
if (reader.NextResult()) {
while(reader.Read()) {
string filePath = reader.GetString(0);
if (!File.Exists(filePath)) { damagedDatabase = true; }
else { databaseFiles.Add(filePath); }
} }
}
} }
}
if (!damagedDatabase) { if (xExists)
// Necessary to because of SQL pooled connections etc, even if all our connections are closed. {
using (var xCmd = xConn.CreateCommand()) { List<string> databaseFiles = new List<string>();
xCmd.CommandText = "ALTER DATABASE " + aDbName + " SET SINGLE_USER WITH ROLLBACK IMMEDIATE"; bool damagedDatabase = false;
xCmd.ExecuteNonQuery();
using (var xCmd = xConn.CreateCommand())
{
xCmd.CommandText = "SELECT [physical_name] FROM [master].[sys].[master_files] WHERE [database_id] = " + databaseId.ToString();
using (SqlDataReader reader = xCmd.ExecuteReader())
{
while (reader.Read())
{
string filePath = reader.GetString(0);
if (!File.Exists(filePath)) { damagedDatabase = true; }
else { databaseFiles.Add(filePath); }
}
}
}
if (!damagedDatabase)
{
// Necessary to because of SQL pooled connections etc, even if all our connections are closed.
using (var xCmd = xConn.CreateCommand())
{
xCmd.CommandText = "ALTER DATABASE " + aDbName + " SET SINGLE_USER WITH ROLLBACK IMMEDIATE";
xCmd.ExecuteNonQuery();
}
}
// Yes this throws an exception if the database doesnt exist, so we have to run it only if we
// know it exists. This will detach and also delete the physical files.
using (var xCmd = xConn.CreateCommand())
{
xCmd.CommandText = "DROP DATABASE " + aDbName;
xCmd.ExecuteNonQuery();
}
if (damagedDatabase)
{
//// Just detach database.
//using (var xCmd = xConn.CreateCommand())
//{
// xCmd.CommandText = string.Format("sp_detach_db '{0}', 'true'", aDbName);
// xCmd.ExecuteNonQuery();
//}
// And try to cleanup remaining files.
foreach (string filePath in databaseFiles)
{
try { File.Delete(filePath); }
catch { }
}
}
} }
// Yes this throws an exception if the database doesnt exist, so we have to run it only if we
// know it exists. This will detach and also delete the physical files.
using (var xCmd = xConn.CreateCommand()) {
xCmd.CommandText = "DROP DATABASE " + aDbName;
xCmd.ExecuteNonQuery();
}
}
else {
// Just detach database.
using (var xCmd = xConn.CreateCommand()) {
xCmd.CommandText = string.Format("sp_detach_db '{0}', 'true'" + aDbName);
xCmd.ExecuteNonQuery();
}
// And try to cleanup remaining files.
foreach (string filePath in databaseFiles) {
try { File.Delete(filePath); } catch { }
}
}
}
} }
// By now DB file should be gone. But sometimes when there are errors, it wont be in the catalog // By now DB file should be gone. But sometimes when there are errors, it wont be in the catalog