public async Task Run(string queryName, CancellationToken cancellationToken, Action <int> init, Action progress)
{
var sqls = _queryReader.Read();
var sql = sqls.Single(x => x.Name == queryName);
var template = Handlebars.Compile(sql.Query);
var servers = _credentialReader.Read();
var serversFiltered = servers.Where(x => string.IsNullOrEmpty(sql.Scope) || x.Scope == sql.Scope || x.Scope.StartsWith(sql.Scope + "/")).ToList();
init(serversFiltered.Count);
var sessionRecord = _sessionRepository.Insert(new SessionRecord
{
QueryName = queryName
});
foreach (var server in serversFiltered)
{
var serverPublic = new SqlServerInfoPublic();
serverPublic.InjectFrom(server);
if (cancellationToken.IsCancellationRequested)
{
throw new UserCancelledException();
}
var processed = template(serverPublic);
string dataJson;
try
{
var driver = GetDriver(server);
var connectionString = driver.BuildConnectionString(server);
var data = await driver.Run(connectionString, processed, serverPublic, cancellationToken);
dataJson = JsonConvert.SerializeObject(data);
}
catch (Exception e)
{
Log.Error(e, "Error while polling from server " + server.ServerId);
dataJson = "[{Error: " + JsonConvert.ToString(e.Message) + "}]";
}
_resultRepository.Insert(new ResultRecord
{
SessionRowId = sessionRecord.RowId,
ServerId = server.ServerId,
DataJson = dataJson
});
progress();
}
_sessionRepository.Finish(sessionRecord);
}
public async Task <IEnumerable <dynamic> > Run(string connectionString, string sql, SqlServerInfoPublic serverInfo,
CancellationToken cancellationToken)
{
var conn = new SqlConnection(connectionString);
var data = await conn.QueryAsync(
new CommandDefinition(sql, serverInfo, cancellationToken : cancellationToken,
commandTimeout : _config.CommandTimeoutInSeconds));
return(data);
}
