Files
Umbraco-CMS/src/Umbraco.Core/Persistence/DbConnectionExtensions.cs

73 lines
2.4 KiB
C#
Raw Normal View History

using System;
using System.Data;
using System.Data.Common;
using System.Linq;
2016-04-12 19:55:50 +02:00
using NPoco;
using Umbraco.Core.Logging;
namespace Umbraco.Core.Persistence
{
internal static class DbConnectionExtensions
{
2016-04-12 19:55:50 +02:00
public static string DetectProviderNameFromConnectionString(string connectionString)
{
2016-04-12 19:55:50 +02:00
var builder = new DbConnectionStringBuilder { ConnectionString = connectionString };
var allKeys = builder.Keys.Cast<string>();
2016-04-12 19:55:50 +02:00
var mySql = new[] { "Server", "Database", "Uid", "Pwd" };
if (mySql.All(x => allKeys.InvariantContains(x)))
{
2016-04-12 19:55:50 +02:00
return Constants.DbProviderNames.MySql;
}
2016-04-12 19:55:50 +02:00
if (allKeys.InvariantContains("Data Source")
//this dictionary is case insensitive
&& builder["Data source"].ToString().InvariantContains(".sdf"))
{
2016-04-12 19:55:50 +02:00
return Constants.DbProviderNames.SqlCe;
}
2016-04-12 19:55:50 +02:00
return Constants.DbProviderNames.SqlServer;
}
2016-04-12 19:55:50 +02:00
public static bool IsConnectionAvailable(string connectionString, string providerName)
{
2016-04-12 19:55:50 +02:00
if (providerName != Constants.DbProviderNames.SqlCe
&& providerName != Constants.DbProviderNames.MySql
&& providerName != Constants.DbProviderNames.SqlServer)
throw new NotSupportedException($"Provider \"{providerName}\" is not supported.");
2016-04-12 19:55:50 +02:00
var factory = DbProviderFactories.GetFactory(providerName);
var conn = factory.CreateConnection();
2016-04-12 19:55:50 +02:00
if (conn == null)
2016-04-12 19:55:50 +02:00
throw new InvalidOperationException($"Could not create a connection for provider \"{providerName}\".");
conn.ConnectionString = connectionString;
using (var connection = conn)
{
return connection.IsAvailable();
}
}
public static bool IsAvailable(this IDbConnection connection)
{
try
{
connection.Open();
connection.Close();
}
2016-04-12 19:55:50 +02:00
catch (DbException e)
{
// Don't swallow this error, the exception is super handy for knowing "why" its not available
2016-04-12 19:55:50 +02:00
LogHelper.WarnWithException<IDbConnection>("Configured database is reporting as not being available!", e);
return false;
}
return true;
}
2016-04-12 19:55:50 +02:00
}
}