NucuCar/NucuCar.Telemetry/Publishers/BasePublisher.cs

using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.Extensions.Logging;
using Newtonsoft.Json.Linq;
using NucuCar.Telemetry.Abstractions;

namespace NucuCar.Telemetry.Publishers
{
    /// <summary>
    /// The TelemetryPublisher is an abstract class, which provides a base for implementing telemetry publishers.
    /// </summary>
    public abstract class BasePublisher : IDisposable, ITelemetryPublisher
    {
        /// <summary>
        /// Raw connection string that is used to connect to the cloud service. Should be parsed if required.
        /// </summary>
        protected string ConnectionString { get; set; }

        /// <summary>
        /// Telemetry source that indicates the source of the telemetry data.
        /// </summary>
        protected string TelemetrySource { get; set; }

        /// <summary>
        /// A list containing entries to the telemeters that want to publish data to the cloud.
        /// </summary>
        protected readonly List<ITelemeter> RegisteredTelemeters;

        /// <summary>
        /// The logging instance, if it's null then the module won't log anything.
        /// </summary>
        // ReSharper disable once UnassignedField.Global
        protected readonly ILogger Logger;

        /// <summary>
        /// Parameter less constructor, mainly used for testing.
        /// </summary>
        public BasePublisher()
        {
            RegisteredTelemeters = new List<ITelemeter>(10);
        }
        
        /// <summary>
        /// Constructor for <see cref="BasePublisher"/>.
        /// </summary>
        /// <param name="opts">TelemetryPublisher options, see: <see cref="PublisherOptions"/></param>
        protected BasePublisher(PublisherOptions opts)
        {
            ConnectionString = opts.ConnectionString;
            TelemetrySource = opts.TelemetrySource;
            Logger = opts.Logger;
            RegisteredTelemeters = new List<ITelemeter>(10);
        }


        /// <summary>
        /// Method that sends all data from the (<see cref="RegisteredTelemeters"/>) to the cloud.
        /// </summary>
        /// <param name="cancellationToken">A cancellation token.</param>
        /// <returns>A task</returns>
        public abstract Task PublishAsync(CancellationToken cancellationToken);

        /// <summary>
        /// Method that releases all managed resources.
        /// </summary>
        public abstract void Dispose();


        /// <summary>
        /// Method that adds a telemeter to the <see cref="RegisteredTelemeters"/> collection.
        /// The telemeter can register only once.
        /// </summary>
        /// <param name="t">The <see cref="ITelemeter"/></param>
        /// <returns>Returns true if the telemeter has registered successfully and false otherwise.</returns>
        public bool RegisterTelemeter(ITelemeter t)
        {
            if (RegisteredTelemeters.Contains(t) || !t.IsTelemetryEnabled()) return false;
            Logger?.LogDebug("Registering telemeter {Identifier}", t.GetIdentifier());
            RegisteredTelemeters.Add(t);
            return true;
        }

        /// <summary>
        /// Method that deletes a telemeter from the <see cref="RegisteredTelemeters"/> collection.
        /// </summary>
        /// <param name="t">The <see cref="ITelemeter"/></param>
        /// <returns>Returns true if the telemeter has unregistered successfully and false otherwise.</returns>
        public bool UnRegisterTelemeter(ITelemeter t)
        {
            if (!RegisteredTelemeters.Contains(t)) return false;
            Logger?.LogDebug("UnRegistering telemeter {Identifier}", t.GetIdentifier());
            RegisteredTelemeters.Remove(t);
            return true;
        }
        
        /// <summary>
        /// Iterates through the registered telemeters and returns the telemetry data as dictionary.
        /// It also adds metadata information such as: source and timestamp.
        /// </summary>
        /// <returns>A dictionary containing all telemetry data. <see cref="DataAggregate"/></returns>
        protected virtual DataAggregate GetTelemetry()
        {
            var source = TelemetrySource ?? nameof(BasePublisher);
            var allTelemetryData = new List<JObject>();
            foreach (var telemeter in RegisteredTelemeters)
            {
                var telemetryData = telemeter.GetTelemetryJson();
                if (telemetryData == null)
                {
                    Logger?.LogWarning("Warning! Data for {Identifier} is null!", telemeter.GetIdentifier());
                    continue;
                }

                telemetryData["sensor_name"] = telemeter.GetIdentifier();
                allTelemetryData.Add(telemetryData);
            }
            return new DataAggregate(source, allTelemetryData);
        }
    }
}