squidex/backend/extensions/Squidex.Extensions/Text/ElasticSearch/ElasticSearchClient.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using Elasticsearch.Net;

namespace Squidex.Extensions.Text.ElasticSearch;

public sealed class ElasticSearchClient : IElasticSearchClient
{
    private readonly ElasticLowLevelClient elasticSearch;

    public ElasticSearchClient(string configurationString)
    {
        var config = new ConnectionConfiguration(new Uri(configurationString));

        elasticSearch = new ElasticLowLevelClient(config);
    }

    public async Task CreateIndexAsync<T>(string indexName, T request,
        CancellationToken ct)
    {
        var result = await elasticSearch.Indices.PutMappingAsync<StringResponse>(indexName, CreatePost(request), ctx: ct);

        if (!result.Success)
        {
            throw new InvalidOperationException($"Failed with ${result.Body}", result.OriginalException);
        }
    }

    public async Task BulkAsync<T>(List<T> requests,
        CancellationToken ct)
    {
        var result = await elasticSearch.BulkAsync<StringResponse>(CreatePost(requests), ctx: ct);

        if (!result.Success)
        {
            throw new InvalidOperationException($"Failed with ${result.Body}", result.OriginalException);
        }
    }

    public async Task<List<dynamic>> SearchAsync<T>(string indexName, T request,
        CancellationToken ct)
    {
        var result = await elasticSearch.SearchAsync<DynamicResponse>(indexName, CreatePost(request), ctx: ct);

        if (!result.Success)
        {
            throw result.OriginalException;
        }

        var hits = new List<dynamic>();

        foreach (var item in result.Body.hits.hits)
        {
            if (item != null)
            {
                hits.Add(item);
            }
        }

        return hits;
    }

    private static PostData CreatePost<T>(List<T> requests)
    {
        return PostData.MultiJson(requests.OfType<object>());
    }

    private static PostData CreatePost<T>(T data)
    {
        return new SerializableData<T>(data);
    }
}