SniffableDatabase.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Waher.Content;
using Waher.Networking;
using Waher.Networking.Sniffers;
using Waher.Persistence;
using Waher.Persistence.FullTextSearch;
using Waher.Persistence.Serialization;
 
namespace Waher.IoTGateway.Setup.Databases.Sniffing
{
    public class SniffableDatabase : CommunicationLayer
    {
        private bool prevHasSniffers = false;
 
        public SniffableDatabase()
            : base(false)
        {
        }
 
        public override void Add(ISniffer Sniffer)
        {
            base.Add(Sniffer);
            this.CheckEventHandlers();
        }
 
        public override void AddRange(IEnumerable<ISniffer> Sniffers)
        {
            base.AddRange(Sniffers);
            this.CheckEventHandlers();
        }
 
        public override bool Remove(ISniffer Sniffer)
        {
            bool Result = base.Remove(Sniffer);
            this.CheckEventHandlers();
            return Result;
        }
 
        private void CheckEventHandlers()
        {
            bool b = this.HasSniffers;
 
            if (this.prevHasSniffers != b)
            {
                if (b)
                {
                    Database.CollectionCleared += this.Database_CollectionCleared;
                    Database.CollectionRepaired += this.Database_CollectionRepaired;
                    Database.ObjectDeleted += this.Database_ObjectDeleted;
                    Database.ObjectInserted += this.Database_ObjectInserted;
                    Database.ObjectUpdated += this.Database_ObjectUpdated;
 
                    Search.ObjectAddedToIndex += this.Search_ObjectAddedToIndex;
                    Search.ObjectUpdatedInIndex += this.Search_ObjectUpdatedInIndex;
                    Search.ObjectRemovedFromIndex += this.Search_ObjectRemovedFromIndex;
                }
                else
                {
                    Database.CollectionCleared -= this.Database_CollectionCleared;
                    Database.CollectionRepaired -= this.Database_CollectionRepaired;
                    Database.ObjectDeleted -= this.Database_ObjectDeleted;
                    Database.ObjectInserted -= this.Database_ObjectInserted;
                    Database.ObjectUpdated -= this.Database_ObjectUpdated;
 
                    Search.ObjectAddedToIndex -= this.Search_ObjectAddedToIndex;
                    Search.ObjectUpdatedInIndex -= this.Search_ObjectUpdatedInIndex;
                    Search.ObjectRemovedFromIndex -= this.Search_ObjectRemovedFromIndex;
                }
 
                this.prevHasSniffers = b;
            }
        }
 
        internal static async Task<string> GetJSON(object Object)
        {
            GenericObject Obj = await Database.Generalize(Object);
            Dictionary<string, object> Obj2 = new Dictionary<string, object>()
            {
                { "ObjectId", Obj.ObjectId },
                { "TypeName", Obj.TypeName },
                { "CollectionName", Obj.CollectionName }
            };
 
            foreach (KeyValuePair<string, object> P in Obj)
                Obj2[P.Key] = P.Value;
 
            return JSON.Encode(Obj2, true);
        }
 
        private async void Database_ObjectInserted(object Sender, ObjectEventArgs e)
        {
            if (this.HasSniffers)
            {
                try
                {
                    this.TransmitText(await GetJSON(e.Object));
                }
                catch (Exception)
                {
                    this.TransmitText("Object of type " + e.Object.GetType().FullName + " inserted.");
                }
            }
        }
 
        private async void Database_ObjectUpdated(object Sender, ObjectEventArgs e)
        {
            if (this.HasSniffers)
            {
                try
                {
                    this.ReceiveText(await GetJSON(e.Object));
                }
                catch (Exception)
                {
                    this.ReceiveText("Object of type " + e.Object.GetType().FullName + " updated.");
                }
            }
        }
 
        private async void Database_ObjectDeleted(object Sender, ObjectEventArgs e)
        {
            if (this.HasSniffers)
            {
                try
                {
                    this.Error(await GetJSON(e.Object));
                }
                catch (Exception)
                {
                    this.Error("Object of type " + e.Object.GetType().FullName + " deleted.");
                }
            }
        }
 
        private Task Database_CollectionRepaired(object Sender, CollectionRepairedEventArgs e)
        {
            if (this.HasSniffers)
            {
                try
                {
                    this.Warning("Collection has been repaired: " + e.Collection);
 
                    if (!(e.Flagged is null))
                    {
                        foreach (FlagSource Source in e.Flagged)
                            this.Exception("Flagged " + Source.Count.ToString() + " time(s) for:" + Source.Reason + "\r\n\r\n" + Source.StackTrace);
                    }
                }
                catch (Exception)
                {
                    // Ignore.
                }
            }
 
            return Task.CompletedTask;
        }
 
        private Task Database_CollectionCleared(object Sender, CollectionEventArgs e)
        {
            if (this.HasSniffers)
                this.Information("Collection has been cleared: " + e.Collection);
 
            return Task.CompletedTask;
        }
 
        private Task Search_ObjectAddedToIndex(object Sender, ObjectReferenceEventArgs e)
        {
            if (this.HasSniffers)
                this.Information("Object added to full-text-search index: " + e.Reference.IndexCollection);
        
            return Task.CompletedTask;
        }
 
        private Task Search_ObjectUpdatedInIndex(object Sender, ObjectReferenceEventArgs e)
        {
            if (this.HasSniffers)
                this.Information("Object updated in full-text-search index: " + e.Reference.IndexCollection);
        
            return Task.CompletedTask;
        }
 
        private Task Search_ObjectRemovedFromIndex(object Sender, ObjectReferenceEventArgs e)
        {
            if (this.HasSniffers)
                this.Information("Object removed from full-text-search index: " + e.Reference.IndexCollection);
        
            return Task.CompletedTask;
        }
    }
}