InsertObjects.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Waher.Persistence;
using Waher.Persistence.Serialization;
using Waher.Runtime.Collections;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Abstraction.Sets;
using Waher.Script.Model;
using Waher.Script.Objects;
using Waher.Script.Objects.VectorSpaces;
using Waher.Script.Operators;
 
namespace Waher.Script.Persistence.SQL
{
    public class InsertObjects : ScriptNode, IEvaluateAsync
    {
        private SourceDefinition source;
        private ElementList objects;
        private readonly bool lazy;
 
        public InsertObjects(SourceDefinition Source, ElementList Objects, bool Lazy, int Start, int Length, Expression Expression)
            : base(Start, Length, Expression)
        {
            this.source = Source;
            this.source?.SetParent(this);
 
            this.objects = Objects;
            this.objects?.SetParent(this);
            
            this.lazy = Lazy;
        }
 
        public override bool IsAsynchronous => true;
 
        public override IElement Evaluate(Variables Variables)
        {
            return this.EvaluateAsync(Variables).Result;
        }
 
        public override async Task<IElement> EvaluateAsync(Variables Variables)
        {
            IDataSource Source = await this.source.GetSource(Variables);
            ChunkedList<IElement> Result = new ChunkedList<IElement>();
            IEnumerable<IElement> Objects;
            IElement E;
            long Count = 0;
            object Item;
 
            await Database.StartBulk();
            try
            {
                foreach (ScriptNode Object in this.objects.Elements)
                {
                    E = await Object.EvaluateAsync(Variables);
                    if (E is IVector V)
                        Objects = V.ChildElements;
                    else if (E is ISet S)
                        Objects = S.ChildElements;
                    else
                        Objects = new IElement[] { E };
 
                    foreach (IElement E2 in Objects)
                    {
                        Item = E2.AssociatedObjectValue;
 
                        if (Item is Dictionary<string, IElement> ObjExNihilo)
                        {
                            GenericObject Obj2 = new GenericObject(Source.CollectionName, Source.TypeName, Guid.Empty);
 
                            foreach (KeyValuePair<string, IElement> P in ObjExNihilo)
                                Obj2[P.Key] = P.Value.AssociatedObjectValue;
 
                            Item = Obj2;
                        }
                        else if (Item is Dictionary<string, object> ObjExNihilo2)
                        {
                            GenericObject Obj2 = new GenericObject(Source.CollectionName, Source.TypeName, Guid.Empty);
 
                            foreach (KeyValuePair<string, object> P in ObjExNihilo2)
                                Obj2[P.Key] = P.Value;
 
                            Item = Obj2;
                        }
 
                        await Source.Insert(this.lazy, Item);
                        Count++;
 
                        Result.Add(new ObjectValue(Item));
                    }
                }
            }
            finally
            {
                await Database.EndBulk();
            }
 
            if (Result.Count == 1)
                return Result[0];
            else
                return new ObjectVector(Result.ToArray());
        }
 
        public override bool ForAllChildNodes(ScriptNodeEventHandler Callback, object State, SearchMethod Order)
        {
            if (Order == SearchMethod.DepthFirst)
            {
                if (!(this.source?.ForAllChildNodes(Callback, State, Order) ?? true))
                    return false;
 
                if (!(this.objects?.ForAllChildNodes(Callback, State, Order) ?? true))
                    return false;
            }
 
            ScriptNode NewNode;
            bool b;
 
            if (!(this.source is null))
            {
                b = !Callback(this.source, out NewNode, State);
                if (!(NewNode is null) && NewNode is SourceDefinition Source2)
                {
                    this.source = Source2;
                    this.source.SetParent(this);
                }
 
                if (b || (Order == SearchMethod.TreeOrder && !this.source.ForAllChildNodes(Callback, State, Order)))
                    return false;
            }
 
            if (!(this.objects is null))
            {
                b = !Callback(this.objects, out NewNode, State);
                if (!(NewNode is null) && NewNode is ElementList NewObjects)
                {
                    this.objects = NewObjects;
                    this.objects.SetParent(this);
                }
 
                if (b || (Order == SearchMethod.TreeOrder && !this.objects.ForAllChildNodes(Callback, State, Order)))
                    return false;
            }
 
            if (Order == SearchMethod.BreadthFirst)
            {
                if (!(this.source?.ForAllChildNodes(Callback, State, Order) ?? true))
                    return false;
 
                if (!(this.objects?.ForAllChildNodes(Callback, State, Order) ?? true))
                    return false;
            }
 
            return true;
        }
 
        public override bool Equals(object obj)
        {
            return (obj is InsertObjects O &&
                AreEqual(this.source, O.source) &&
                AreEqual(this.objects, O.objects) &&
                base.Equals(obj));
        }
 
        public override int GetHashCode()
        {
            int Result = base.GetHashCode();
            Result ^= Result << 5 ^ GetHashCode(this.source);
            Result ^= Result << 5 ^ GetHashCode(this.objects);
            return Result;
        }
 
    }
}