53 lines
2.0 KiB
C#
53 lines
2.0 KiB
C#
|
// 大名科技（天津）有限公司版权所有 电话：18020030720 QQ：515096995
|
|||
|
|
//
|
|||
|
|
// 此源代码遵循位于源代码树根目录中的 LICENSE 文件的许可证
|
|||
|
|
|
|||
|
|
using Elasticsearch.Net;
|
|||
|
|
using Nest;
|
|||
|
|
|
|||
|
|
namespace Admin.NET.Core;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// ES服务注册
|
|||
|
|
/// </summary>
|
|||
|
|
public static class ElasticSearchSetup
|
|||
|
|
{
|
|||
|
|
public static void AddElasticSearch(this IServiceCollection services)
|
|||
|
|
{
|
|||
|
|
var option = App.GetConfig<ElasticSearchOptions>("Logging:ElasticSearch");
|
|||
|
|
if (!option.Enabled) return;
|
|||
|
|
|
|||
|
|
var uris = option.ServerUris.Select(u => new Uri(u));
|
|||
|
|
// 集群
|
|||
|
|
var connectionPool = new SniffingConnectionPool(uris);
|
|||
|
|
var connectionSettings = new ConnectionSettings(connectionPool).DefaultIndex(option.DefaultIndex);
|
|||
|
|
// 单连接
|
|||
|
|
//var connectionSettings = new ConnectionSettings(new SingleNodeConnectionPool(uris.FirstOrDefault())).DefaultIndex(option.DefaultIndex);
|
|||
|
|
|
|||
|
|
// 认证类型
|
|||
|
|
if (option.AuthType == ElasticSearchAuthTypeEnum.Basic)// Basic 认证
|
|||
|
|
{
|
|||
|
|
connectionSettings.BasicAuthentication(option.User, option.Password);
|
|||
|
|
}
|
|||
|
|
else if (option.AuthType == ElasticSearchAuthTypeEnum.ApiKey) //ApiKey 认证
|
|||
|
|
{
|
|||
|
|
connectionSettings.ApiKeyAuthentication(option.ApiId, option.ApiKey);
|
|||
|
|
}
|
|||
|
|
else if (option.AuthType == ElasticSearchAuthTypeEnum.Base64ApiKey)// Base64ApiKey 认证
|
|||
|
|
{
|
|||
|
|
connectionSettings.ApiKeyAuthentication(new ApiKeyAuthenticationCredentials(option.Base64ApiKey));
|
|||
|
|
}
|
|||
|
|
else return;
|
|||
|
|
|
|||
|
|
// ES使用Https时的证书指纹
|
|||
|
|
if (!string.IsNullOrEmpty(option.Fingerprint))
|
|||
|
|
{
|
|||
|
|
connectionSettings.CertificateFingerprint(option.Fingerprint);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var client = new ElasticClient(connectionSettings);
|
|||
|
|
client.Indices.Create(option.DefaultIndex, u => u.Map<SysLogOp>(m => m.AutoMap()));
|
|||
|
|
|
|||
|
|
services.AddSingleton(client); // 单例注册
|
|||
|
|
}
|
|||
|
|
}
|