/// <summary> All calls pass through this method. </summary>
/// <remarks> This handles enforcement of SLA and other cross-cutting concerns. </remarks>
/// <param name="options"> Optional option overrides. </param>
private void AdHocCall(Action<QueryOptions, CancellationToken> makeCall, QueryOptions options)
{
var actualOptions = options ?? Settings.DefaultOptions;
EnforceLimits(actualOptions);
var token = new CancellationTokenSource();
var task = Task.Run
(
(Action)(() =>
{
makeCall(actualOptions, token.Token);
}),
token.Token
);
try
{
var timeout = actualOptions.QueryLimits.MaximumTime;
if (!task.Wait(timeout))
{
token.Cancel();
task.Wait();
}
}
finally
{
task.Dispose();
}
}
public object Evaluate(string text, IDictionary<string, object> args = null, QueryOptions options = null)
{
object result = null;
AdHocCall
(
(ao, ct) =>
{
result = InternalExecute(text, args, ao, ct);
},
options
);
return result;
}
public EvaluateResult Evaluate(string text, IDictionary<string, object> args = null, QueryOptions options = null)
{
// TODO: overload that returns the type
EvaluateResult result = new EvaluateResult();
AdHocCall
(
(ao, ct) =>
{
result = InternalExecute(text, args, ao, ct);
},
options
);
return result;
}
public Guid Prepare(string text, QueryOptions options = null)
{
throw new NotImplementedException("Not implemented.");
}
public void Execute(string text, IDictionary<string, object> args = null, QueryOptions options = null)
{
AdHocCall((ao, ct) => { InternalExecute(text, args, ao, ct); }, options);
}
private EvaluateResult InternalExecute(string text, IDictionary<string, object> args, QueryOptions actualOptions, CancellationToken cancelToken)
{
// Create assembly and source file names
var assemblyName = "DotQL" + DateTime.Now.Ticks;
var sourceFileName = System.IO.Path.Combine(System.IO.Path.GetTempPath(), System.IO.Path.ChangeExtension(assemblyName, "dql"));
// Save the file if we're debugging
if (Settings.DebugOn)
System.IO.File.WriteAllText(sourceFileName, text);
// Parse
var parser = new Parser() { ErrorIfNotEOF = true };
var script = Parser.ParseFrom(parser.Script, text);
// Compile
var result =
Compiler.CreateExecutable
(
new CompilerOptions
{
DefaultUsings = actualOptions.DefaultUsings,
DebugOn = Settings.DebugOn,
AssemblyName = assemblyName,
SourceFileName = sourceFileName,
Factory = Settings.RepositoryFactory,
},
script
);
// Run
return new EvaluateResult { Result = result.Execute(args, Settings.RepositoryFactory, cancelToken), Type = result.Type };
}
/// <summary> Adjusts the limits according to policy. </summary>
private void EnforceLimits(QueryOptions actualOptions)
{
var limits = actualOptions.QueryLimits ?? new QueryLimits();
limits.MaximumTime = Math.Min(limits.MaximumTime, Settings.AdHocMaximumTime);
limits.MaximumRows = Math.Min(limits.MaximumRows, Settings.AdHocMaximumRows);
actualOptions.QueryLimits = limits;
}
