internal async Task <PrtgOrphan> GetTreeAsync(Either <PrtgObject, int> objectOrId)
{
if (IsRoot(objectOrId) && !RequestType.Contains(TreeRequestType.Lazy))
{
return(await GetTreeFastAsync().ConfigureAwait(false));
}
if (!objectOrId.IsLeft)
{
Token.ThrowIfCancellationRequested();
if (objectOrId.Right == WellKnownId.Root)
{
objectOrId = await client.GetGroupAsync(WellKnownId.Root).ConfigureAwait(false);
}
else
{
objectOrId = await client.GetObjectAsync(objectOrId.Right, true).ConfigureAwait(false);
}
}
//With each additional level we parse, a new TreeBuilderLevel will be constructed and recursed
var level = new TreeBuilderLevel(objectOrId.Left, this);
return(await level.ProcessObjectAsync().ConfigureAwait(false));
}
/// <summary>
/// Retrieves <see cref="ITreeValue"/> values objects from a set of object factories, retrieves the children
/// of each object from the next <see cref="TreeBuilderLevel"/> and encapsulates the object and its children in a <see cref="PrtgOrphan"/>.
/// </summary>
/// <param name="factories">The factories to retrieve objects from.</param>
/// <returns>A list of <see cref="PrtgOrphan"/> objects encapsulating the values returnd from the factories and their respective children.</returns>
private async Task <List <PrtgOrphan> > GetOrphansAsync(params ObjectFactory[] factories)
{
List <Tuple <ITreeValue, ObjectFactory> > results = new List <Tuple <ITreeValue, ObjectFactory> >();
foreach (var factory in factories)
{
Token.ThrowIfCancellationRequested();
var objs = await factory.ObjectsAsync(Value.Id.Value, Token).ConfigureAwait(false);
results.AddRange(objs.Select(o => Tuple.Create(o, factory)));
}
ProgressManager.OnLevelWidthKnown(Value, ValueType, results.Count);
var orphans = new List <PrtgOrphan>();
foreach (var item in results)
{
ProgressManager.OnProcessValue(item.Item1);
var level = new TreeBuilderLevel(item.Item1, item.Item2.Type, item.Item2.Orphan, builder);
orphans.Add(await level.ProcessObjectAsync().ConfigureAwait(false));
}
return(orphans);
}
private async Task <PrtgOrphan> GetTreeFastAsync()
{
var config = new FastAsyncHelper();
if (Options.Contains(TreeParseOption.Probes))
{
config.Probe.CreateTask = GetFastObjectsAsync(PrtgNodeType.Probe, client.GetProbesAsync());
}
if (Options.Contains(TreeParseOption.Groups))
{
config.Group.CreateTask = GetFastObjectsAsync(PrtgNodeType.Group, client.GetGroupsAsync());
}
else
{
config.Group.CreateTask = GetFastObjectsAsync(PrtgNodeType.Group, client.GetGroupsAsync(Property.Id, WellKnownId.Root), true);
}
if (Options.Contains(TreeParseOption.Devices))
{
config.Device.CreateTask = GetFastObjectsAsync(PrtgNodeType.Device, client.GetDevicesAsync());
}
if (Options.Contains(TreeParseOption.Sensors))
{
config.Sensor.CreateTask = GetFastObjectsAsync(PrtgNodeType.Sensor, client.GetSensorsAsync());
}
await config.Stage1Async().ConfigureAwait(false);
//Await sensors separately so we don't kill PRTG
await config.Stage2Async().ConfigureAwait(false);
ObjectManager = new CachedObjectManager(
client,
await config.Sensor.Task.ConfigureAwait(false),
await config.Device.Task.ConfigureAwait(false),
await config.Group.Task.ConfigureAwait(false),
await config.Probe.Task.ConfigureAwait(false)
);
var root = (await config.Group.Task.ConfigureAwait(false)).First(g => g.Id == WellKnownId.Root);
var level = new TreeBuilderLevel(root, this);
return(await level.ProcessObjectAsync().ConfigureAwait(false));
}
