public async Task <StreamAnalyticsJobModel> AddAlertingAsync(string tenantId)
{
// Attempt to create the necessary database if it does not already exist
await this.cosmosDb.CreateDatabaseIfNotExistsAsync(SaCosmosDbDatabaseName);
StreamAnalyticsJobModel saJobModel = await this.GetAlertingAsync(tenantId);
if (this.SaJobExists(saJobModel))
{
throw new Exception("The given tenant already has a deployed stream analytics job.");
}
TenantModel tenant = await this.GetTenantFromContainerAsync(tenantId);
string saJobName = string.Format(SaNameFormat, tenantId.Substring(0, 8));
await this.runbookHelper.CreateAlerting(tenantId, saJobName, tenant.IotHubName);
return(new StreamAnalyticsJobModel
{
TenantId = tenant.TenantId,
StreamAnalyticsJobName = saJobName,
IsActive = false,
JobState = "Creating",
});
}
public async Task GetAlertingAsyncTest(bool createIfNotExists, bool expectCreateNew)
{
// Arrange
StreamAnalyticsJobModel streamAnalyticsModel = new StreamAnalyticsJobModel()
{
TenantId = TenantId,
StreamAnalyticsJobName = "StreamAnalyticsJob",
JobState = "Job State",
IsActive = true,
};
this.mockAlertingContainer.Setup(x => x.AddAlertingAsync(It.IsAny <string>()))
.ReturnsAsync(streamAnalyticsModel);
this.mockAlertingContainer.Setup(x => x.GetAlertingAsync(It.IsAny <string>()))
.ReturnsAsync(streamAnalyticsModel);
// Act
var result = await this.controller.GetAlertingAsync(createIfNotExists);
// Assert
if (createIfNotExists)
{
Assert.True(expectCreateNew);
}
else
{
Assert.False(expectCreateNew);
}
Assert.Equal(result.TenantId, TenantId);
}
public async Task <StreamAnalyticsJobModel> StopAlertingAsync(string tenantId)
{
StreamAnalyticsJobModel saJobModel = await this.GetAlertingAsync(tenantId);
if (!this.SaJobExists(saJobModel))
{
throw new Exception("There is no StreamAnalyticsJob is available to stop for this tenant.");
}
await this.streamAnalyticsHelper.StopAsync(saJobModel.StreamAnalyticsJobName);
return(saJobModel);
}
public async Task <StreamAnalyticsJobModel> AddAlertingAsync(string tenantId)
{
string alarmsCollectionAppConfigKey = string.Format(AlarmsCollectionAppConfigKeyFormat, tenantId);
string alarmsCollectionId;
try
{
alarmsCollectionId = this.appConfig.GetValue(alarmsCollectionAppConfigKey);
}
catch (Exception)
{
alarmsCollectionId = string.Format(AlarmsCollectionNameFormat, tenantId);
}
try
{
await this.cosmosDb.CreateCollectionIfNotExistsAsync(
AlarmsDatabaseName,
alarmsCollectionId,
AlarmsCollectionPartitionKey);
await this.appConfig.SetValueAsync(alarmsCollectionAppConfigKey, alarmsCollectionId);
}
catch (Exception e)
{
throw new Exception("Unable to ensure the necessary AppConfig configuration and CosmosDb collection exist before creating the stream analytics job.", e);
}
StreamAnalyticsJobModel saJobModel = await this.GetAlertingAsync(tenantId);
if (this.SaJobExists(saJobModel))
{
throw new Exception("The given tenant already has a deployed stream analytics job.");
}
TenantModel tenant = await this.GetTenantFromContainerAsync(tenantId);
string saJobName = string.Format(SaNameFormat, tenantId.Substring(0, 8));
await this.tableStorageClient.InsertAsync(
"tenantOperations",
new TenantOperationModel(tenantId, TenantOperation.SaJobCreation, saJobName));
return(new StreamAnalyticsJobModel
{
TenantId = tenant.TenantId,
StreamAnalyticsJobName = saJobName,
IsActive = false,
JobState = "Creating",
});
}
public async Task <StreamAnalyticsJobModel> GetAlertingAsync([FromQuery] bool createIfNotExists = false)
{
string tenantId = this.GetTenantId();
StreamAnalyticsJobModel model = await this.alertingContainer.GetAlertingAsync(tenantId);
if (!this.alertingContainer.SaJobExists(model) && createIfNotExists)
{
// If the tenant does not have an sa job, start creating it
this.logger.LogInformation("The tenant does not already have alerting enabled and the createIfNotExists parameter was set to true. Creating a stream analytics job now. TenantId: {tenantId}", tenantId);
return(await this.alertingContainer.AddAlertingAsync(tenantId));
}
else
{
return(model);
}
}
public async Task RemoveAlertingAsyncTest()
{
// Arrange
StreamAnalyticsJobModel streamAnalyticsModel = new StreamAnalyticsJobModel()
{
TenantId = TenantId,
StreamAnalyticsJobName = "StreamAnalyticsJob",
JobState = "Job State",
IsActive = false,
};
this.mockAlertingContainer.Setup(x => x.RemoveAlertingAsync(It.IsAny <string>()))
.ReturnsAsync(streamAnalyticsModel);
// Act
var result = await this.controller.RemoveAlertingAsync();
// Assert
Assert.False(result.IsActive);
Assert.Equal(result.TenantId, TenantId);
}
public bool SaJobExists(StreamAnalyticsJobModel saJobModel)
{
return(!string.IsNullOrEmpty(saJobModel.StreamAnalyticsJobName) && !string.IsNullOrEmpty(saJobModel.JobState));
}
