Merge branch 'webwatcher' of https://github.com/ElfoCompany/Elfo.Wardein

into webwatcher
ElfoCompany · Mar 20, 2020 · 8978e8c · 8978e8c
2 parents e175ece + fe4e82c
commit 8978e8c
Show file tree

Hide file tree

Showing 4 changed files with 91 additions and 104 deletions.
diff --git a/Elfo.Wardein.Abstractions/WebWatcher/IAmUrlResponseManager.cs b/Elfo.Wardein.Abstractions/WebWatcher/IAmUrlResponseManager.cs
@@ -0,0 +1,10 @@
+using System.Threading.Tasks;
+
+namespace Elfo.Firmenich.Wardein.Abstractions.WebWatcher
+{
+    public interface IAmUrlResponseManager
+    {
+        Task<bool> IsHealty(bool assertWithStatusCode, string assertWithRegex);
+        void RestartPool(string poolName);
+    }
+}
diff --git a/Elfo.Wardein.Core/ServiceManager/HttpClientUrlResponseManager.cs b/Elfo.Wardein.Core/ServiceManager/HttpClientUrlResponseManager.cs
@@ -0,0 +1,78 @@
+using Elfo.Firmenich.Wardein.Abstractions.WebWatcher;
+using Elfo.Wardein.Core.ServiceManager;
+using System;
+using System.Net;
+using System.Net.Http;
+using System.Net.Http.Headers;
+using System.Threading.Tasks;
+
+namespace Elfo.Firmenich.Wardein.Core.ServiceManager
+{
+    public class HttpClientUrlResponseManager : IAmUrlResponseManager
+    {
+
+        public Task<bool> IsHealty(bool assertWithStatusCode, string assertWithRegex)
+        {
+            var endPoint = "";
+            using (var handler = new HttpClientHandler())
+            using (var client = new HttpClient(handler))
+            {
+                client.BaseAddress = new Uri("http://localhost:55587/");
+                client.DefaultRequestHeaders.Accept.Clear();
+                client.DefaultRequestHeaders.Accept.Add(new MediaTypeWithQualityHeaderValue("Keep-Alive"));
+
+                var response = client.GetAsync(endPoint);
+                HttpResponseMessage httpWebResponse = response.Result;
+
+                if (!assertWithStatusCode)
+                {
+                    if (!string.IsNullOrEmpty(assertWithRegex))
+                    {
+                        var isMatch = false;
+                        //check if html content matches the regex
+                        if (isMatch)
+                        {
+                            return Task.FromResult(false);
+                        }
+                        else
+                        {
+                            return Task.FromResult(true);
+                        }
+                    }
+                    else
+                    {
+                        return Task.FromResult(true);
+                    }
+                }
+                else
+                {
+                    if (httpWebResponse.StatusCode != HttpStatusCode.OK)
+                    {
+                        return Task.FromResult(false);
+                    }
+                    else
+                    {
+                        if (!string.IsNullOrEmpty(assertWithRegex))
+                        {
+                            var isMatch = false;
+                            //check if html content matches the regex
+                            if (isMatch)
+                            {
+                                return Task.FromResult(false);
+                            }
+                            else
+                            {
+                                return Task.FromResult(true);
+                            }
+                        }
+                    }
+
+                }
+                return Task.FromResult(false);
+            }
+        }
+
+        public void RestartPool(string poolName)
+            => new IISPoolManager(poolName).Restart();
+    }
+}
diff --git a/Elfo.Wardein.Core/ServiceManager/WebWatcherManager.cs b/Elfo.Wardein.Core/ServiceManager/WebWatcherManager.cs
diff --git a/Elfo.Wardein.Watchers/WebWatcher/WebWatcher.cs b/Elfo.Wardein.Watchers/WebWatcher/WebWatcher.cs
@@ -149,7 +149,7 @@ internal virtual async Task RunCheck()
             {
                 using (var persistenceService = ServicesContainer.PersistenceService(Const.DB_PATH))
                 {
-                    IAmServiceManager serviceManager = GetServiceManager();
+                    IAmResponceManager serviceManager = GetServiceManager();
                     if (serviceManager == null)
                         continue; // If the service doesn't exist, skip the check 
 
@@ -168,9 +168,9 @@ internal virtual async Task RunCheck()
 
                     #region Local Functions
 
-                    IAmServiceManager GetServiceManager()
+                    IAmResponceManager GetServiceManager()
                     {
-                        IAmServiceManager svc = null;
+                        IAmResponceManager svc = null;
                         try
                         {
                             svc = ServicesContainer.ServiceManager(service.ServiceName, service.ServiceManagerType);