119 lines
3.5 KiB
C#
119 lines
3.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Data;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using Oqtane.Interfaces;
|
|
using Oqtane.Models;
|
|
// ReSharper disable ConvertToUsingDeclaration
|
|
// ReSharper disable InvertIf
|
|
// ReSharper disable BuiltInTypeReferenceStyleForMemberAccess
|
|
|
|
namespace Oqtane.Repository
|
|
{
|
|
public class SqlRepository : ISqlRepository
|
|
{
|
|
private IEnumerable<IOqtaneDatabase> _databases;
|
|
|
|
public SqlRepository(IEnumerable<IOqtaneDatabase> databases)
|
|
{
|
|
_databases = databases;
|
|
}
|
|
|
|
public void ExecuteScript(Tenant tenant, string script)
|
|
{
|
|
// execute script in current tenant
|
|
foreach (var query in script.Split("GO", StringSplitOptions.RemoveEmptyEntries))
|
|
{
|
|
ExecuteNonQuery(tenant, query);
|
|
}
|
|
}
|
|
|
|
public bool ExecuteScript(string connectionString, string databaseType, Assembly assembly, string fileName)
|
|
{
|
|
var success = true;
|
|
var script = GetScriptFromAssembly(assembly, fileName);
|
|
|
|
if (!string.IsNullOrEmpty(script))
|
|
{
|
|
try
|
|
{
|
|
foreach (var query in script.Split("GO", StringSplitOptions.RemoveEmptyEntries))
|
|
{
|
|
ExecuteNonQuery(connectionString, databaseType, query);
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
success = false;
|
|
}
|
|
}
|
|
|
|
return success;
|
|
}
|
|
|
|
public bool ExecuteScript(Tenant tenant, Assembly assembly, string fileName)
|
|
{
|
|
var success = true;
|
|
var script = GetScriptFromAssembly(assembly, fileName);
|
|
|
|
if (!string.IsNullOrEmpty(script))
|
|
{
|
|
try
|
|
{
|
|
ExecuteScript(tenant, script);
|
|
}
|
|
catch
|
|
{
|
|
success = false;
|
|
}
|
|
}
|
|
|
|
return success;
|
|
}
|
|
|
|
public int ExecuteNonQuery(Tenant tenant, string query)
|
|
{
|
|
return ExecuteNonQuery(tenant.DBConnectionString, tenant.DBType, query);
|
|
}
|
|
|
|
public IDataReader ExecuteReader(Tenant tenant, string query)
|
|
{
|
|
var db = _databases.Single(d => d.Name == tenant.DBType);
|
|
return db.ExecuteReader(tenant.DBConnectionString, query);
|
|
}
|
|
|
|
private int ExecuteNonQuery(string connectionString, string databaseType, string query)
|
|
{
|
|
var db = _databases.Single(d => d.Name == databaseType);
|
|
|
|
return db.ExecuteNonQuery(connectionString, query);
|
|
}
|
|
|
|
private string GetScriptFromAssembly(Assembly assembly, string fileName)
|
|
{
|
|
// script must be included as an Embedded Resource within an assembly
|
|
var script = "";
|
|
|
|
if (assembly != null)
|
|
{
|
|
var name = assembly.GetManifestResourceNames().FirstOrDefault(item => item.EndsWith("." + fileName));
|
|
if (name != null)
|
|
{
|
|
var resourceStream = assembly.GetManifestResourceStream(name);
|
|
if (resourceStream != null)
|
|
{
|
|
using (var reader = new StreamReader(resourceStream))
|
|
{
|
|
script = reader.ReadToEnd();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return script;
|
|
}
|
|
}
|
|
}
|