PowerShellLoggingModule/ApplicationInsightsOutputSubscriber.cs at 2a88a77b48d12485483d190f3a41fb0d15764e33 · dlwyatt/PowerShellLoggingModule · GitHub

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
// ReSharper disable MemberCanBePrivate.Global
// ReSharper disable UnusedMember.Global

using Microsoft.ApplicationInsights;
using Microsoft.ApplicationInsights.DataContracts;

namespace PSLogging
{
    using System;
    using System.IO;
    using System.Management.Automation;

    public class ApplicationInsightsOutputSubscriber : HostIOSubscriberBase
    {
        private readonly TelemetryClient client;

        public ApplicationInsightsOutputSubscriber(string key, // Azure Application Insights "InstrumentationKey"
                                                   string dateTimeFormat = "r")
        {
            client = new TelemetryClient();
            client.InstrumentationKey = key;
            DateTimeFormat = dateTimeFormat;
        }


        #region Properties

        public string DateTimeFormat { get; set; }

        #endregion


        public override void WriteDebug(string message)
        {
            if (string.IsNullOrEmpty(message))
            {
                return;
            }

            if (message.Trim() != String.Empty)
            {
                message = String.Format("{0,-29} - [D] {1}", DateTime.UtcNow.ToString(DateTimeFormat), message);
            }

            client.TrackTrace(message, SeverityLevel.Verbose);
        }

        public override void WriteError(string message)
        {
            if (string.IsNullOrEmpty(message))
            {
                return;
            }

            if (message.Trim() != String.Empty)
            {
                message = String.Format("{0,-29} - [E] {1}", DateTime.UtcNow.ToString(DateTimeFormat), message);
            }

            client.TrackTrace(message, SeverityLevel.Error);
        }

        public override void WriteVerbose(string message)
        {
            if (string.IsNullOrEmpty(message))
            {
                return;
            }

            if (message.Trim() != String.Empty)
            {
                message = String.Format("{0,-29} - [V] {1}", DateTime.UtcNow.ToString(DateTimeFormat), message);
            }

            client.TrackTrace(message, SeverityLevel.Verbose);
        }

        public override void WriteWarning(string message)
        {
            if (string.IsNullOrEmpty(message))
            {
                return;
            }

            if (message.Trim() != String.Empty)
            {
                message = String.Format("{0,-29} - [W] {1}", DateTime.UtcNow.ToString(DateTimeFormat), message);
            }

            client.TrackTrace(message, SeverityLevel.Warning);
        }
    }
}

// ReSharper restore MemberCanBePrivate.Global
// ReSharper restore UnusedMember.Global