@@ -1,38 +0,0 @@ | |||||
// Copyright (c) .NET Core Community. All rights reserved. | |||||
// Licensed under the MIT License. See License.txt in the project root for license information. | |||||
namespace DotNetCore.CAP | |||||
{ | |||||
public class DiscoveryOptions | |||||
{ | |||||
public const string DefaultDiscoveryServerHost = "localhost"; | |||||
public const int DefaultDiscoveryServerPort = 8500; | |||||
public const string DefaultCurrentNodeHostName = "localhost"; | |||||
public const int DefaultCurrentNodePort = 5000; | |||||
public const string DefaultMatchPath = "/cap"; | |||||
public DiscoveryOptions() | |||||
{ | |||||
DiscoveryServerHostName = DefaultDiscoveryServerHost; | |||||
DiscoveryServerPort = DefaultDiscoveryServerPort; | |||||
CurrentNodeHostName = DefaultCurrentNodeHostName; | |||||
CurrentNodePort = DefaultCurrentNodePort; | |||||
MatchPath = DefaultMatchPath; | |||||
} | |||||
public string DiscoveryServerHostName { get; set; } | |||||
public int DiscoveryServerPort { get; set; } | |||||
public string CurrentNodeHostName { get; set; } | |||||
public int CurrentNodePort { get; set; } | |||||
public string NodeId { get; set; } | |||||
public string NodeName { get; set; } | |||||
public string MatchPath { get; set; } | |||||
} | |||||
} |
@@ -1,60 +0,0 @@ | |||||
// Copyright (c) .NET Core Community. All rights reserved. | |||||
// Licensed under the MIT License. See License.txt in the project root for license information. | |||||
using System; | |||||
using DotNetCore.CAP; | |||||
using DotNetCore.CAP.NodeDiscovery; | |||||
using Microsoft.Extensions.DependencyInjection; | |||||
namespace DotNetCore.CAP | |||||
{ | |||||
internal sealed class DiscoveryOptionsExtension : ICapOptionsExtension | |||||
{ | |||||
private readonly Action<DiscoveryOptions> _options; | |||||
public DiscoveryOptionsExtension(Action<DiscoveryOptions> option) | |||||
{ | |||||
_options = option; | |||||
} | |||||
public void AddServices(IServiceCollection services) | |||||
{ | |||||
var discoveryOptions = new DiscoveryOptions(); | |||||
_options?.Invoke(discoveryOptions); | |||||
services.AddSingleton(discoveryOptions); | |||||
services.AddSingleton<IDiscoveryProviderFactory, DiscoveryProviderFactory>(); | |||||
services.AddSingleton<IProcessingServer, ConsulProcessingNodeServer>(); | |||||
services.AddSingleton(x => | |||||
{ | |||||
var configOptions = x.GetService<DiscoveryOptions>(); | |||||
var factory = x.GetService<IDiscoveryProviderFactory>(); | |||||
return factory.Create(configOptions); | |||||
}); | |||||
} | |||||
} | |||||
} | |||||
namespace Microsoft.Extensions.DependencyInjection | |||||
{ | |||||
public static class CapDiscoveryOptionsExtensions | |||||
{ | |||||
public static CapOptions UseDiscovery(this CapOptions capOptions) | |||||
{ | |||||
return capOptions.UseDiscovery(opt => { }); | |||||
} | |||||
public static CapOptions UseDiscovery(this CapOptions capOptions, Action<DiscoveryOptions> options) | |||||
{ | |||||
if (options == null) | |||||
{ | |||||
throw new ArgumentNullException(nameof(options)); | |||||
} | |||||
capOptions.RegisterExtension(new DiscoveryOptionsExtension(options)); | |||||
return capOptions; | |||||
} | |||||
} | |||||
} |
@@ -1,28 +0,0 @@ | |||||
// Copyright (c) .NET Core Community. All rights reserved. | |||||
// Licensed under the MIT License. See License.txt in the project root for license information. | |||||
using System; | |||||
using Microsoft.Extensions.Logging; | |||||
namespace DotNetCore.CAP.NodeDiscovery | |||||
{ | |||||
internal class DiscoveryProviderFactory : IDiscoveryProviderFactory | |||||
{ | |||||
private readonly ILoggerFactory _loggerFactory; | |||||
public DiscoveryProviderFactory(ILoggerFactory loggerFactory) | |||||
{ | |||||
_loggerFactory = loggerFactory; | |||||
} | |||||
public INodeDiscoveryProvider Create(DiscoveryOptions options) | |||||
{ | |||||
if (options == null) | |||||
{ | |||||
throw new ArgumentNullException(nameof(options)); | |||||
} | |||||
return new ConsulNodeDiscoveryProvider(_loggerFactory, options); | |||||
} | |||||
} | |||||
} |
@@ -1,10 +0,0 @@ | |||||
// Copyright (c) .NET Core Community. All rights reserved. | |||||
// Licensed under the MIT License. See License.txt in the project root for license information. | |||||
namespace DotNetCore.CAP.NodeDiscovery | |||||
{ | |||||
internal interface IDiscoveryProviderFactory | |||||
{ | |||||
INodeDiscoveryProvider Create(DiscoveryOptions options); | |||||
} | |||||
} |
@@ -1,107 +0,0 @@ | |||||
// Copyright (c) .NET Core Community. All rights reserved. | |||||
// Licensed under the MIT License. See License.txt in the project root for license information. | |||||
using System; | |||||
using System.Collections.Generic; | |||||
using System.Linq; | |||||
using System.Threading.Tasks; | |||||
using Consul; | |||||
using DotNetCore.CAP.Internal; | |||||
using Microsoft.Extensions.Logging; | |||||
namespace DotNetCore.CAP.NodeDiscovery | |||||
{ | |||||
public class ConsulNodeDiscoveryProvider : INodeDiscoveryProvider, IDisposable | |||||
{ | |||||
private readonly ILogger<ConsulNodeDiscoveryProvider> _logger; | |||||
private readonly DiscoveryOptions _options; | |||||
private ConsulClient _consul; | |||||
public ConsulNodeDiscoveryProvider(ILoggerFactory logger, DiscoveryOptions options) | |||||
{ | |||||
_logger = logger.CreateLogger<ConsulNodeDiscoveryProvider>(); | |||||
_options = options; | |||||
InitClient(); | |||||
} | |||||
public void Dispose() | |||||
{ | |||||
_consul.Dispose(); | |||||
} | |||||
public async Task<IList<Node>> GetNodes() | |||||
{ | |||||
try | |||||
{ | |||||
var nodes = new List<Node>(); | |||||
var services = await _consul.Catalog.Services(); | |||||
foreach (var service in services.Response) | |||||
{ | |||||
var serviceInfo = await _consul.Catalog.Service(service.Key); | |||||
var node = serviceInfo.Response.SkipWhile(x => !x.ServiceTags.Contains("CAP")) | |||||
.Select(info => new Node | |||||
{ | |||||
Id = info.ServiceID, | |||||
Name = info.ServiceName, | |||||
Address = info.ServiceAddress, | |||||
Port = info.ServicePort, | |||||
Tags = string.Join(", ", info.ServiceTags) | |||||
}).ToList(); | |||||
nodes.AddRange(node); | |||||
} | |||||
CapCache.Global.AddOrUpdate("cap.nodes.count", nodes.Count, TimeSpan.FromSeconds(60), true); | |||||
return nodes; | |||||
} | |||||
catch (Exception ex) | |||||
{ | |||||
CapCache.Global.AddOrUpdate("cap.nodes.count", 0, TimeSpan.FromSeconds(20)); | |||||
_logger.LogError( | |||||
$"Get consul nodes raised an exception. Exception:{ex.Message},{ex.InnerException.Message}"); | |||||
return null; | |||||
} | |||||
} | |||||
public Task RegisterNode() | |||||
{ | |||||
try | |||||
{ | |||||
return _consul.Agent.ServiceRegister(new AgentServiceRegistration | |||||
{ | |||||
ID = _options.NodeId, | |||||
Name = _options.NodeName, | |||||
Address = _options.CurrentNodeHostName, | |||||
Port = _options.CurrentNodePort, | |||||
Tags = new[] {"CAP", "Client", "Dashboard"}, | |||||
Check = new AgentServiceCheck | |||||
{ | |||||
DeregisterCriticalServiceAfter = TimeSpan.FromSeconds(30), | |||||
Interval = TimeSpan.FromSeconds(10), | |||||
Status = HealthStatus.Passing, | |||||
HTTP = | |||||
$"http://{_options.CurrentNodeHostName}:{_options.CurrentNodePort}{_options.MatchPath}/health" | |||||
} | |||||
}); | |||||
} | |||||
catch (Exception ex) | |||||
{ | |||||
_logger.LogError( | |||||
$"Get consul nodes raised an exception. Exception:{ex.Message},{ex.InnerException.Message}"); | |||||
return null; | |||||
} | |||||
} | |||||
private void InitClient() | |||||
{ | |||||
_consul = new ConsulClient(config => | |||||
{ | |||||
config.WaitTime = TimeSpan.FromSeconds(5); | |||||
config.Address = new Uri($"http://{_options.DiscoveryServerHostName}:{_options.DiscoveryServerPort}"); | |||||
}); | |||||
} | |||||
} | |||||
} |
@@ -1,15 +0,0 @@ | |||||
// Copyright (c) .NET Core Community. All rights reserved. | |||||
// Licensed under the MIT License. See License.txt in the project root for license information. | |||||
using System.Collections.Generic; | |||||
using System.Threading.Tasks; | |||||
namespace DotNetCore.CAP.NodeDiscovery | |||||
{ | |||||
public interface INodeDiscoveryProvider | |||||
{ | |||||
Task<IList<Node>> GetNodes(); | |||||
Task RegisterNode(); | |||||
} | |||||
} |
@@ -1,35 +0,0 @@ | |||||
// Copyright (c) .NET Core Community. All rights reserved. | |||||
// Licensed under the MIT License. See License.txt in the project root for license information. | |||||
namespace DotNetCore.CAP.NodeDiscovery | |||||
{ | |||||
internal class ConsulProcessingNodeServer : IProcessingServer | |||||
{ | |||||
private readonly DiscoveryOptions _dashboardOptions; | |||||
private readonly IDiscoveryProviderFactory _discoveryProviderFactory; | |||||
public ConsulProcessingNodeServer( | |||||
DiscoveryOptions dashboardOptions, | |||||
IDiscoveryProviderFactory discoveryProviderFactory) | |||||
{ | |||||
_dashboardOptions = dashboardOptions; | |||||
_discoveryProviderFactory = discoveryProviderFactory; | |||||
} | |||||
public void Start() | |||||
{ | |||||
var discoveryProvider = _discoveryProviderFactory.Create(_dashboardOptions); | |||||
discoveryProvider.RegisterNode(); | |||||
} | |||||
public void Pulse() | |||||
{ | |||||
//ignore | |||||
} | |||||
public void Dispose() | |||||
{ | |||||
} | |||||
} | |||||
} |
@@ -1,18 +0,0 @@ | |||||
// Copyright (c) .NET Core Community. All rights reserved. | |||||
// Licensed under the MIT License. See License.txt in the project root for license information. | |||||
namespace DotNetCore.CAP.NodeDiscovery | |||||
{ | |||||
public class Node | |||||
{ | |||||
public string Id { get; set; } | |||||
public string Name { get; set; } | |||||
public string Address { get; set; } | |||||
public int Port { get; set; } | |||||
public string Tags { get; set; } | |||||
} | |||||
} |