ServiceMonitor/ServiceControl.cs at master · Kquinn/ServiceMonitor · 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
97
98
99
100
101
102
103
104
105
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.ServiceProcess;
using System.Threading;

namespace ServiceMonitor
{
    public class ServiceControl
    {
        public static void StopService(ServiceController service)
        {
            var timeout = new TimeSpan(0, 0, 30);
            Messages.LogMessage($"Stopping {service.ServiceName} service. Current status is {service.Status}");
            if (service.Status != ServiceControllerStatus.Stopped)
            {
                service.Stop();
                service.WaitForStatus(ServiceControllerStatus.Stopped, timeout);
                if (service.Status != ServiceControllerStatus.Stopped)
                {
                    throw new Exception($"Timed out waiting for {service.ServiceName} to stop.");
                }
            }
            Messages.LogMessage($"{service.ServiceName} service successfully stopped");
        }
        public static void StartService(ServiceController service)
        {
            var timeout = new TimeSpan(0, 0, 30);
            Messages.LogMessage($"Starting {service.ServiceName} service. Current status is {service.Status}");
            if (service.Status != ServiceControllerStatus.Running)
            {
                service.Start();
                service.WaitForStatus(ServiceControllerStatus.Running, timeout);
                if (service.Status != ServiceControllerStatus.Running)
                {
                    throw new Exception($"Timed out waiting for {service.ServiceName} to start.");
                }
            }
            Messages.LogMessage($"{service.ServiceName} service successfully started");
        }

        public static bool RestartService(ServiceController serviceController, bool memoryThresholdReached, bool clearMSMQ = false)
        {

            List<ServiceController> dependentServicesList = serviceController.DependentServices.ToList();
            foreach (ServiceController currentDependentService in dependentServicesList)
            {
                StopService(currentDependentService);
            }

            StopService(serviceController);
            if (clearMSMQ)
            {
                int filecount = 0;
                ServiceController msmqServiceController = new ServiceController("MSMQ");
                StopService(msmqServiceController);
                try
                {

                    Messages.LogMessage($"Cleaning MSMQ files...");
                    string mqFileDirectory = "C:\\Windows\\SysNative\\msmq\\storage\\";
                    string mqBackupDirectory = "C:\\msmq_backupMQ";

                    try
                    {
                        if (!Directory.Exists(mqBackupDirectory))
                        {
                            Directory.CreateDirectory(mqBackupDirectory);
                        }

                        string[] files = Directory.GetFiles(mqFileDirectory, "*.mq");

                        foreach (string file in files)
                        {
                            var filename = file.Split('\\').Last();
                            File.Move(file, mqBackupDirectory + '\\' + filename);
                            filecount++;
                        }
                    }
                    catch (Exception ex1)
                    {
                        Messages.LogMessage(ex1.Message, EventLogEntryType.Error, EventID.Generic);
                    }

                }
                catch (Exception ex)
                {

                    Messages.LogMessage(ex.Message, EventLogEntryType.Error, EventID.Generic);
                    return false;
                    //log exception and continue to ensure services start back up.
                }
                Thread.Sleep(500);
                StartService(msmqServiceController);
            }

            StartService(serviceController);
            return true;


        }
    }
}