public static IAsyncResult StartWork(string source, TableAliasType aliasType, AsyncCallback callback, object state)
{
SqlTableAliasesWorkItem result = new SqlTableAliasesWorkItem(source, aliasType, callback, state);
ThreadPool.QueueUserWorkItem(ThreadPool_WaitCallback, result);
return(result);
}
private static void HandleTableAliasesEndRegexWork(IAsyncResult ar)
{
SqlTableAliasesWorkItem item = AsyncSqlTableAliasesWorker.EndWork(ar);
if (item != null && item.IsCancelled)
{
switch (item.AliasType)
{
case TableAliasType.From:
HostServicesSingleton.HostServices.MsgService.LogWarning("SqlAnalyzer timed out for: Table aliases (From).");
break;
case TableAliasType.Join:
HostServicesSingleton.HostServices.MsgService.LogWarning("SqlAnalyzer timed out for: Table aliases (Join).");
break;
default:
break;
}
}
}
public static SqlAnalyzerResults AnalyzeSql(string sqlText, bool combine)
{
SqlAnalyzerResults result = new SqlAnalyzerResults();
SqlVariablesWorkItem varsItem1 = (SqlVariablesWorkItem)AsyncSqlVariablesWorker.StartWork(sqlText, false, HandleVariablesEndRegexWork);
SqlTableAliasesWorkItem tblAliasesItem1 = (SqlTableAliasesWorkItem)AsyncSqlTableAliasesWorker.StartWork(sqlText, TableAliasType.From, HandleTableAliasesEndRegexWork);
SqlTableAliasesWorkItem tblAliasesItem2 = (SqlTableAliasesWorkItem)AsyncSqlTableAliasesWorker.StartWork(sqlText, TableAliasType.Join, HandleTableAliasesEndRegexWork);
SqlTablesWorkItem tablesItem1 = (SqlTablesWorkItem)AsyncSqlTablesWorker.StartWork(sqlText, TableType.Temp, HandleTablesEndRegexWork);
SqlTablesWorkItem tablesItem2 = (SqlTablesWorkItem)AsyncSqlTablesWorker.StartWork(sqlText, TableType.Variable, HandleTablesEndRegexWork);
CancellableWorkItem[] jobs = new CancellableWorkItem[] {
varsItem1,
tblAliasesItem1,
tblAliasesItem2,
tablesItem1,
tablesItem2
};
// Wait for 2 seconds. After that we cancel any jobs that have not completed
WaitHandle[] waitHandles = new WaitHandle[] { varsItem1.AsyncWaitHandle,
tblAliasesItem1.AsyncWaitHandle,
tblAliasesItem2.AsyncWaitHandle,
tablesItem1.AsyncWaitHandle,
tablesItem2.AsyncWaitHandle };
WaitAll(waitHandles);
for (int i = 0; i < jobs.Length; i++)
{
if (jobs[i].IsCompleted)
{
if (jobs[i] is SqlVariablesWorkItem)
{
SqlVariablesWorkItem vars = jobs[i] as SqlVariablesWorkItem;
if (!vars.NamesOnly)
{
result.VariableNamesWithDataTypes = vars.Result;
}
else
{
result.Variables = vars.Result;
}
}
else if (jobs[i] is SqlTableAliasesWorkItem)
{
SqlTableAliasesWorkItem aliases = jobs[i] as SqlTableAliasesWorkItem;
switch (aliases.AliasType)
{
case TableAliasType.From:
result.TableAliasFrom = aliases.Result;
break;
case TableAliasType.Join:
result.TableAliasJoin = aliases.Result;
break;
default:
break;
}
}
else if (jobs[i] is SqlTablesWorkItem)
{
SqlTablesWorkItem tables = jobs[i] as SqlTablesWorkItem;
switch (tables.TableType)
{
case TableType.Temp:
result.TableAsTemp = tables.Result;
break;
case TableType.Variable:
result.TableAsVariable = tables.Result;
break;
default:
break;
}
}
}
else
{
jobs[i].Cancel();
}
}
if (combine)
{
result.CombineResults();
}
return(result);
}
