fergalmoran/DnsServer
1
0
Fork 0
mirror of https://github.com/fergalmoran/DnsServer.git synced 2026-01-02 06:47:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Failover: removed type specific monitoring code.

Browse Source
This commit is contained in:
Shreyas Zare
2021-08-21 12:29:48 +05:30
parent c70a2365a2
commit e6c8fd322a
2 changed files with 0 additions and 291 deletions
Download Patch File Download Diff File Expand all files Collapse all files

143
Apps/FailoverApp/AddressMonitoring.cs
View File

@@ -1,143 +0,0 @@
/*
Technitium DNS Server
Copyright (C) 2021 Shreyas Zare (shreyas@technitium.com)
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Net;
namespace Failover
{
class AddressMonitoring : IDisposable
{
#region variables
readonly HealthMonitoringService _service;
readonly IPAddress _address;
readonly ConcurrentDictionary<string, HealthMonitor> _healthMonitors = new ConcurrentDictionary<string, HealthMonitor>(1, 1);
#endregion
#region constructor
public AddressMonitoring(HealthMonitoringService service, IPAddress address, string healthCheck, Uri healthCheckUrl)
{
_service = service;
_address = address;
if (_service.HealthChecks.TryGetValue(healthCheck, out HealthCheck existingHealthCheck))
_healthMonitors.TryAdd(GetHealthMonitorKey(healthCheck, healthCheckUrl), new HealthMonitor(_service.DnsServer, _address, existingHealthCheck, healthCheckUrl));
}
#endregion
#region IDisposable
bool _disposed;
protected virtual void Dispose(bool disposing)
{
if (_disposed)
return;
if (disposing)
{
foreach (KeyValuePair<string, HealthMonitor> healthMonitor in _healthMonitors)
healthMonitor.Value.Dispose();
_healthMonitors.Clear();
}
_disposed = true;
}
public void Dispose()
{
Dispose(true);
GC.SuppressFinalize(this);
}
#endregion
#region private
private static string GetHealthMonitorKey(string healthCheck, Uri healthCheckUrl)
{
string healthMonitorKey = healthCheck;
if (healthCheckUrl is not null)
healthMonitorKey += "|" + healthCheckUrl.AbsoluteUri;
return healthMonitorKey;
}
#endregion
#region public
public HealthCheckStatus QueryStatus(string healthCheck, Uri healthCheckUrl)
{
string healthMonitorKey = GetHealthMonitorKey(healthCheck, healthCheckUrl);
if (_healthMonitors.TryGetValue(healthMonitorKey, out HealthMonitor monitor))
return monitor.HealthCheckStatus;
if (_service.HealthChecks.TryGetValue(healthCheck, out HealthCheck existingHealthCheck))
_healthMonitors.TryAdd(healthMonitorKey, new HealthMonitor(_service.DnsServer, _address, existingHealthCheck, healthCheckUrl));
return null;
}
public void RemoveHealthMonitor(string healthCheck)
{
foreach (KeyValuePair<string, HealthMonitor> healthMonitor in _healthMonitors)
{
if (healthMonitor.Key.StartsWith(healthCheck + "|"))
{
if (_healthMonitors.TryRemove(healthMonitor.Key, out HealthMonitor removedMonitor))
removedMonitor.Dispose();
}
}
}
public bool IsExpired()
{
foreach (KeyValuePair<string, HealthMonitor> healthMonitor in _healthMonitors)
{
if (healthMonitor.Value.IsExpired())
{
if (_healthMonitors.TryRemove(healthMonitor.Key, out HealthMonitor removedMonitor))
removedMonitor.Dispose();
}
}
return _healthMonitors.IsEmpty;
}
#endregion
#region property
public IPAddress Address
{ get { return _address; } }
#endregion
}
}

148
Apps/FailoverApp/DomainMonitoring.cs
View File

@@ -1,148 +0,0 @@
/*
Technitium DNS Server
Copyright (C) 2021 Shreyas Zare (shreyas@technitium.com)
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using TechnitiumLibrary.Net.Dns;
namespace Failover
{
class DomainMonitoring : IDisposable
{
#region variables
readonly HealthMonitoringService _service;
readonly string _domain;
readonly DnsResourceRecordType _type;
readonly ConcurrentDictionary<string, HealthMonitor> _healthMonitors = new ConcurrentDictionary<string, HealthMonitor>(1, 1);
#endregion
#region constructor
public DomainMonitoring(HealthMonitoringService service, string domain, DnsResourceRecordType type, string healthCheck, Uri healthCheckUrl)
{
_service = service;
_domain = domain;
_type = type;
if (_service.HealthChecks.TryGetValue(healthCheck, out HealthCheck existingHealthCheck))
_healthMonitors.TryAdd(GetHealthMonitorKey(healthCheck, healthCheckUrl), new HealthMonitor(_service.DnsServer, domain, type, existingHealthCheck, healthCheckUrl));
}
#endregion
#region IDisposable
bool _disposed;
protected virtual void Dispose(bool disposing)
{
if (_disposed)
return;
if (disposing)
{
foreach (KeyValuePair<string, HealthMonitor> healthMonitor in _healthMonitors)
healthMonitor.Value.Dispose();
_healthMonitors.Clear();
}
_disposed = true;
}
public void Dispose()
{
Dispose(true);
GC.SuppressFinalize(this);
}
#endregion
#region private
private static string GetHealthMonitorKey(string healthCheck, Uri healthCheckUrl)
{
string healthMonitorKey = healthCheck;
if (healthCheckUrl is not null)
healthMonitorKey += "|" + healthCheckUrl.AbsoluteUri;
return healthMonitorKey;
}
#endregion
#region public
public HealthCheckStatus QueryStatus(string healthCheck, Uri healthCheckUrl)
{
string healthMonitorKey = GetHealthMonitorKey(healthCheck, healthCheckUrl);
if (_healthMonitors.TryGetValue(healthMonitorKey, out HealthMonitor monitor))
return monitor.HealthCheckStatus;
if (_service.HealthChecks.TryGetValue(healthCheck, out HealthCheck existingHealthCheck))
_healthMonitors.TryAdd(healthMonitorKey, new HealthMonitor(_service.DnsServer, _domain, _type, existingHealthCheck, healthCheckUrl));
return null;
}
public void RemoveHealthMonitor(string healthCheck)
{
foreach (KeyValuePair<string, HealthMonitor> healthMonitor in _healthMonitors)
{
if (healthMonitor.Key.StartsWith(healthCheck + "|"))
{
if (_healthMonitors.TryRemove(healthMonitor.Key, out HealthMonitor removedMonitor))
removedMonitor.Dispose();
}
}
}
public bool IsExpired()
{
foreach (KeyValuePair<string, HealthMonitor> healthMonitor in _healthMonitors)
{
if (healthMonitor.Value.IsExpired())
{
if (_healthMonitors.TryRemove(healthMonitor.Key, out HealthMonitor removedMonitor))
removedMonitor.Dispose();
}
}
return _healthMonitors.IsEmpty;
}
#endregion
#region property
public string Domain
{ get { return _domain; } }
public DnsResourceRecordType Type
{ get { return _type; } }
#endregion
}
}