aboutsummaryrefslogtreecommitdiffstats
path: root/Software/Visual_Studio/Utilities/Tango.Telemetry.Tester.IOT.CLI/Program.cs
blob: 0565ab00d712243d56b58ecda8831501f727a52f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Tango.BL;
using Tango.BL.Enumerations;
using Tango.Logging;
using Tango.Telemetry.Destinations;
using Tango.Telemetry.Mappers;
using Tango.Telemetry.Sources;

namespace Tango.Telemetry.Tester.IOT.CLI
{
    class Program
    {
        static void Main(string[] args)
        {
            LogManager.Default.RegisterLogger(new Tango.Logging.VSOutputLogger("IOT TESTER"));

            TelemetryPublisher publisher = new TelemetryPublisher(new TelemetryPublisherConfiguration()
            {
                Environment = "DEV",
                SerialNumber = "dev-machine",
                Organization = "Twine",
                MachineType = MachineTypes.TS1800,
                HistorySourcesRequestInterval = TimeSpan.FromSeconds(1),
                EnableBackoff = false,

            }, null);

            (publisher.StorageManager as TelemetryLiteDBStorageManager).EnableCheckPointsRecovery = false;

            publisher.RegisterSource(new JobRunsTestSource());
            publisher.RegisterDestination(new TelemetryAzureHubDestination("HostName=iot-twine-dev-weu.azure-devices.net;DeviceId=telemetry-dev-01;SharedAccessKey=cZhCMhiVL+TF7p13fpX+lFmyxoy8ZqCkbxUwumWw18Q="));

            publisher.PublishResultAvailable += Publisher_PublishResultAvailable;

            publisher.Start().GetAwaiter().GetResult();

            Console.Clear();
            Console.WriteLine("=== Telemetry IoT Hub Test Utility ===");
            Console.WriteLine($"Publisher started. Streaming every {publisher.Config.HistorySourcesRequestInterval.TotalSeconds} seconds.");
            Console.WriteLine("Press any key to stop streaming data...");

            Console.ReadKey();

            Console.WriteLine("Disposing publisher...");
            publisher.Dispose();
        }

        private static void Publisher_PublishResultAvailable(object sender, TelemetryPublishResultAvailableEventArgs e)
        {
            Console.ForegroundColor = ConsoleColor.DarkGray;
            Console.WriteLine($"Package publish result available:\n{e.PublishResult.ToString()}");

            if (e.PublishResult.DestinationsResults.Any(d => d.Status == TelemetryPublishResult.DestinationStatus.Failed))
            {
                Console.ForegroundColor = ConsoleColor.Red;
                Console.WriteLine("One or more destinations failed to receive the package.");
            }
            else
            {
                Console.ForegroundColor = ConsoleColor.Green;
                Console.WriteLine("Package successfully published.");
            }
            Console.ResetColor();

            Console.WriteLine("Press any key to stop streaming data...");
        }
    }

    public class JobRunsTestSource : ITelemetryHistorySource
    {
        private bool _busy;

        public string Name { get; } = "JobRuns Source";
        public bool RequiresTelemetryDuplicationTracking { get; } = false;
        public TelemetryHistorySourceDirection Direction { get => TelemetryHistorySourceDirection.Descending; }

        public Task<bool> CanRequestHistory(DateTime from)
        {
            return Task.FromResult(!_busy);
        }

        public void Dispose()
        {

        }

        public Task<IEnumerable<ITelemetry>> RequestHistory(DateTime from)
        {
            _busy = true;
            
            ObservablesContext.OverrideSettingsDataSource(new Core.DataSource()
            {
                Address = "localhost\\SQLPPC",
                IntegratedSecurity = true,
                Catalog = "Tango",
                Type = Core.DataSourceType.SQLServer,
            });

            using (ObservablesContext db = ObservablesContext.CreateDefault())
            {
                var runs = db.JobRuns.OrderByDescending(x => x.LastUpdated).Where(x => x.LastUpdated < from).Take(1).ToList();
                var runsT = runs.Select(x => JobRunMapper.MapJobRun(x)).ToList();

                _busy = false;
                return Task.FromResult<IEnumerable<ITelemetry>>(runsT);
            }
        }
    }
}