public void KillWebConsole(string userId, string webConsoleId)
{
_logger.LogInformation($"Kill web console {webConsoleId}");
try {
Sessions.Instance.EnsureValidUser(userId);
if (_userWebConsoles.Contains(userId, webConsoleId))
{
_userWebConsoles.RemoveData(userId, webConsoleId);
}
_runspacesStatsMonitor.Unregister(webConsoleId);
_runspaceProvider.KillWebConsole(webConsoleId);
if (_userWebConsoles.List(userId) == null)
{
_userWebConsoles.RemoveUser(userId);
}
} catch (Exception ex) {
throw new RunspaceProviderException(
string.Format(
APIGatewayResources.MultiTenantRunspaceProvider_KillFailed,
userId,
ex.Message),
ex);
}
}
public void RemoveUser()
{
// Arrange
const string UserID1 = "user-id1";
const string UserID2 = "user-id2";
const string RunspaceID1 = "runspace-id1";
const string RunspaceID2 = "runspace-id2";
var userRunspaces = new UserToIdentifiableData <string>();
userRunspaces.Add(UserID1, RunspaceID1, RunspaceID1);
userRunspaces.Add(UserID2, RunspaceID2, RunspaceID2);
// Act
userRunspaces.RemoveUser(UserID2);
// Assert
Assert.IsFalse(userRunspaces.Contains(UserID2));
}
public void RemoveUserWhenNoUser()
{
// Arrange
const string UserID1 = "user-id1";
const string UserID2 = "user-id2";
const string RunspaceID1 = "runspace-id1";
const string RunspaceID2 = "runspace-id2";
var userRunspaces = new UserToIdentifiableData <string>();
userRunspaces.Add(UserID1, RunspaceID1, RunspaceID1);
userRunspaces.Add(UserID1, RunspaceID2, RunspaceID2);
// Act && Assert
Assert.DoesNotThrow(() => userRunspaces.RemoveUser(UserID2));
Assert.IsTrue(userRunspaces.Contains(UserID1));
Assert.IsTrue(userRunspaces.Contains(UserID1, RunspaceID1));
Assert.IsTrue(userRunspaces.Contains(UserID1, RunspaceID2));
}
public void Kill(string userId, string runspaceId)
{
_logger.LogInformation($"Kill runspace {runspaceId}");
try {
Sessions.Instance.EnsureValidUser(userId);
if (!_userRunspaces.Contains(userId))
{
throw new RunspaceProviderException(
string.Format(
APIGatewayResources.MultiTenantRunspaceProvider_UserHasNoRunspaces,
userId));
}
if (!_userRunspaces.Contains(userId, runspaceId))
{
throw new RunspaceProviderException(
string.Format(
APIGatewayResources.MultiTenantRunspaceProvider_UserHasNoRunspaceWithId,
userId,
runspaceId));
}
_userRunspaces.RemoveData(userId, runspaceId);
_runspaceProvider.Kill(runspaceId);
_runspacesStatsMonitor.Unregister(runspaceId);
if (_userRunspaces.List(userId) == null)
{
_userRunspaces.RemoveUser(userId);
}
} catch (Exception ex) {
throw new RunspaceProviderException(
string.Format(
APIGatewayResources.MultiTenantRunspaceProvider_KillFailed,
userId,
ex.Message),
ex);
}
}
public void Kill(string userId, string runspaceId)
{
_logger.LogInformation($"Kill runspace {runspaceId}");
try {
Sessions.Instance.EnsureValidUser(userId);
if (!_userRunspaces.Contains(userId))
{
throw new RunspaceProviderException(
string.Format(
APIGatewayResources.MultiTenantRunspaceProvider_UserHasNoRunspaces,
userId));
}
if (!_userRunspaces.Contains(userId, runspaceId))
{
throw new RunspaceProviderException(
string.Format(
APIGatewayResources.MultiTenantRunspaceProvider_UserHasNoRunspaceWithId,
userId,
runspaceId));
}
var runspaceInfo = _runspaceProvider.Get(runspaceId);
var runspaceData = _userRunspaces.GetData(userId, runspaceId);
_userRunspaces.RemoveData(userId, runspaceId);
if (runspaceData.RunVcConnectionScript)
{
// Run Disconnect script to close VC server sessions first and then kill the runspace
Task.Run(() => {
_logger.LogDebug("RunspaceProvider -> Run disconnect VI server ");
try {
var scriptExecutionRequest = new DataTypes.ScriptExecution {
OutputObjectsFormat = OutputObjectsFormat.Json,
Name = "disconnectallservers",
Script = PCLIScriptsReader.DisconnectAllServers
};
_logger.LogDebug($"Start Disconnect All Servers script");
var scriptResult = ScriptExecutionMediatorSingleton.
Instance.
ScriptExecutionMediator.
StartScriptExecution(userId, runspaceInfo, scriptExecutionRequest).Result;
_logger.LogDebug($"Wait Disconnect All Servers script to complete");
while (scriptResult.State == ScriptState.Running)
{
var intermediateResult = ScriptExecutionMediatorSingleton.Instance.ScriptExecutionMediator.GetScriptExecution(
userId,
scriptResult.Id);
if (intermediateResult != null)
{
scriptResult = intermediateResult;
}
Thread.Sleep(100);
}
} catch (RunspaceEndpointException runspaceEndointException) {
_logger.LogError(runspaceEndointException, "Runspace endpoint exception while waiting connect VC script");
} catch (Exception exc) {
_logger.LogError(exc, "Wait Disconnect All Servers script failed");
}
_runspaceProvider.Kill(runspaceId);
});
}
else
{
_runspaceProvider.Kill(runspaceId);
}
_runspacesStatsMonitor.Unregister(runspaceId);
if (_userRunspaces.List(userId) == null)
{
_userRunspaces.RemoveUser(userId);
}
} catch (Exception ex) {
throw new RunspaceProviderException(
string.Format(
APIGatewayResources.MultiTenantRunspaceProvider_KillFailed,
userId,
ex.Message),
ex);
}
}
