squidex/src/Squidex.Domain.Apps.Entities/Contents/Text/GrainTextIndexer.cs

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

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Orleans;
using Squidex.Domain.Apps.Core.Contents;
using Squidex.Domain.Apps.Entities.Apps;
using Squidex.Domain.Apps.Entities.Schemas;
using Squidex.Infrastructure;
using Squidex.Infrastructure.Log;

namespace Squidex.Domain.Apps.Entities.Contents.Text
{
    public sealed class GrainTextIndexer : ITextIndexer
    {
        private readonly IGrainFactory grainFactory;
        private readonly ISemanticLog log;

        public GrainTextIndexer(IGrainFactory grainFactory, ISemanticLog log)
        {
            Guard.NotNull(grainFactory, nameof(grainFactory));
            Guard.NotNull(log, nameof(log));

            this.grainFactory = grainFactory;

            this.log = log;
        }

        public async Task DeleteAsync(Guid schemaId, Guid id)
        {
            var index = grainFactory.GetGrain<ITextIndexerGrain>(schemaId);

            using (Profiler.TraceMethod<GrainTextIndexer>())
            {
                try
                {
                    await index.DeleteAsync(id);
                }
                catch (Exception ex)
                {
                    log.LogError(ex, w => w
                        .WriteProperty("action", "DeleteTextEntry")
                        .WriteProperty("status", "Failed"));
                }
            }
        }

        public async Task IndexAsync(Guid schemaId, Guid id, NamedContentData data, NamedContentData dataDraft)
        {
            var index = grainFactory.GetGrain<ITextIndexerGrain>(schemaId);

            using (Profiler.TraceMethod<GrainTextIndexer>())
            {
                try
                {
                    if (data != null)
                    {
                        await index.IndexAsync(id, new IndexData { Data = data });
                    }

                    if (dataDraft != null)
                    {
                        await index.IndexAsync(id, new IndexData { Data = dataDraft, IsDraft = true });
                    }
                }
                catch (Exception ex)
                {
                    log.LogError(ex, w => w
                        .WriteProperty("action", "UpdateTextEntry")
                        .WriteProperty("status", "Failed"));
                }
            }
        }

        public async Task<List<Guid>> SearchAsync(string queryText, IAppEntity app, Guid schemaId, bool useDraft = false)
        {
            if (string.IsNullOrWhiteSpace(queryText))
            {
                return null;
            }

            var index = grainFactory.GetGrain<ITextIndexerGrain>(schemaId);

            using (Profiler.TraceMethod<GrainTextIndexer>())
            {
                var context = CreateContext(app, useDraft);

                return await index.SearchAsync(queryText, context);
            }
        }

        private static SearchContext CreateContext(IAppEntity app, bool useDraft)
        {
            var languages = new HashSet<string>(app.LanguagesConfig.Select(x => x.Key));

            return new SearchContext { Languages = languages, IsDraft = useDraft };
        }
    }
}