/// <summary>
/// Register tenant shard
/// </summary>
/// <param name="tenantServerConfig">The tenant server configuration.</param>
/// <param name="databaseConfig">The database configuration.</param>
/// <param name="catalogConfig">The catalog configuration.</param>
/// <param name="resetEventDate">If set to true, the events dates for all tenants will be reset </param>
public async void RegisterTenantShard(TenantServerConfig tenantServerConfig, DatabaseConfig databaseConfig, CatalogConfig catalogConfig, bool resetEventDate)
{
//get all database in devtenantserver
var tenants = GetAllTenantNames(tenantServerConfig, databaseConfig);
var connectionString = new SqlConnectionStringBuilder
{
UserID = databaseConfig.DatabaseUser,
Password = databaseConfig.DatabasePassword,
ApplicationName = "EntityFramework",
ConnectTimeout = databaseConfig.ConnectionTimeOut
};
foreach (var tenant in tenants)
{
var tenantId = GetTenantKey(tenant);
var result = await Sharding.RegisterNewShard(tenant, tenantId, tenantServerConfig.TenantServer, databaseConfig.DatabaseServerPort, catalogConfig.ServicePlan);
if (result)
{
// resets all tenants' event dates
if (resetEventDate)
{
#region EF6
try
{
//use EF6 since execution of Stored Procedure in EF Core for anonymous return type is not supported yet
using (var context = new TenantContext(Sharding.ShardMap, tenantId, connectionString.ConnectionString))
{
context.Database.ExecuteSqlCommand("sp_ResetEventDates");
}
}
catch (Microsoft.Azure.SqlDatabase.ElasticScale.ShardManagement.ShardManagementException ex)
{
string errorText;
if (ex.ErrorCode == Microsoft.Azure.SqlDatabase.ElasticScale.ShardManagement.ShardManagementErrorCode.MappingIsOffline)
{
errorText = "Tenant '" + tenant + "' is offline. Could not reset event dates:" + ex.ToString();
}
else
{
errorText = ex.ToString();
}
Console.WriteLine(errorText);
}
catch (Exception ex)
{
Console.WriteLine(ex.ToString());
}
#endregion
#region EF core
//https://github.com/aspnet/EntityFramework/issues/7032
//using (var context = new TenantDbContext(Sharding.ShardMap, tenantId, connectionString))
//{
// context.Database.ExecuteSqlCommand("sp_ResetEventDates");
//}
#endregion
}
}
}
}
/// <summary>
/// Registers the new shard.
/// Verify if shard exists for the tenant. If not then create new shard and add tenant details to Tenants table in catalog
/// </summary>
/// <param name="tenantName">Name of the tenant.</param>
/// <param name="tenantId">The tenant identifier.</param>
/// <param name="tenantServerConfig">The tenant server configuration.</param>
/// <param name="databaseConfig">The database configuration.</param>
/// <param name="catalogConfig">The catalog configuration.</param>
public static bool RegisterNewShard(string tenantName, int tenantId, TenantServerConfig tenantServerConfig, DatabaseConfig databaseConfig, CatalogConfig catalogConfig)
{
try
{
Shard shard;
ShardLocation shardLocation = new ShardLocation(tenantServerConfig.TenantServer, tenantName, databaseConfig.SqlProtocol, databaseConfig.DatabaseServerPort);
if (!ShardMap.TryGetShard(shardLocation, out shard))
{
//create shard if it does not exist
shard = ShardMap.CreateShard(shardLocation);
}
// Register the mapping of the tenant to the shard in the shard map.
// After this step, DDR on the shard map can be used
PointMapping <int> mapping;
if (!ShardMap.TryGetMappingForKey(tenantId, out mapping))
{
var pointMapping = ShardMap.CreatePointMapping(tenantId, shard);
//convert from int to byte[] as tenantId has been set as byte[] in Tenants entity
var key = _helper.ConvertIntKeyToBytesArray(pointMapping.Value);
//add tenant to Tenants table
var tenant = new Tenants
{
ServicePlan = catalogConfig.ServicePlan,
TenantId = key,
TenantName = tenantName
};
_tenantsRepository.Add(tenant);
}
return(true);
}
catch (Exception ex)
{
// _logger.LogError(0, ex, "Error in registering new shard.");
//throw new ApplicationException("Error in registering new shard.");
return(false);
}
}
/// <summary>
/// Gets all tenant names from tenant server
/// </summary>
/// <param name="tenantServerConfig">The tenant server configuration.</param>
/// <param name="databaseConfig">The database configuration.</param>
/// <returns></returns>
private List <string> GetAllTenantNames(TenantServerConfig tenantServerConfig, DatabaseConfig databaseConfig)
{
List <string> list = new List <string>();
string conString = $"Server={databaseConfig.SqlProtocol}:{tenantServerConfig.TenantServer},{databaseConfig.DatabaseServerPort};Database={""};User ID={databaseConfig.DatabaseUser};Password={databaseConfig.DatabasePassword};Trusted_Connection=False;Encrypt=True;Connection Timeout={databaseConfig.ConnectionTimeOut};";
using (SqlConnection con = new SqlConnection(conString))
{
con.Open();
using (SqlCommand cmd = new SqlCommand("SELECT name from sys.databases WHERE name NOT IN ('master')", con))
{
using (IDataReader dr = cmd.ExecuteReader())
{
while (dr.Read())
{
list.Add(dr[0].ToString());
}
}
}
}
return(list);
}