Masuit.LuceneEFCore.SearchEngine/Masuit.LuceneEFCore.SearchEngine/LuceneIndexableBaseEntity.cs

using Lucene.Net.Documents;
using Masuit.LuceneEFCore.SearchEngine.Extensions;
using Masuit.LuceneEFCore.SearchEngine.Interfaces;
using Newtonsoft.Json;
using System;
using System.ComponentModel.DataAnnotations;
using System.ComponentModel.DataAnnotations.Schema;
using System.Reflection;

namespace Masuit.LuceneEFCore.SearchEngine
{
    /// <summary>
    /// 需要被索引的实体基类
    /// </summary>
    public abstract class LuceneIndexableBaseEntity : ILuceneIndexable
    {
        /// <summary>
        /// 主键id
        /// </summary>
        [LuceneIndex(Name = "Id", Store = Field.Store.YES), Key]
#if Int
        [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
        public int Id { get; set; }
#endif
#if Long
        [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
        public long Id { get; set; } 
#endif
#if String
        public string Id { get; set; } 
#endif
#if Guid
        public Guid Id { get; set; }
#endif

        /// <summary>
        /// 索引唯一id
        /// </summary>
        [LuceneIndex(Name = "IndexId", Store = Field.Store.YES)]
        [NotMapped, JsonIgnore]
        public string IndexId
        {
            get => GetType().Name + ":" + Id;
            set
            {
            }
        }

        /// <summary>
        /// 转换成Lucene文档
        /// </summary>
        /// <returns></returns>
        public virtual Document ToDocument()
        {
            var doc = new Document();
            var type = GetType();
            if (type.Assembly.IsDynamic && type.FullName.Contains("Prox"))
            {
                type = type.BaseType;
            }

            var classProperties = type.GetProperties();
            doc.Add(new StringField("Type", type.AssemblyQualifiedName, Field.Store.YES));
            foreach (var propertyInfo in classProperties)
            {
                var propertyValue = propertyInfo.GetValue(this);
                if (propertyValue == null)
                {
                    continue;
                }

                var attrs = propertyInfo.GetCustomAttributes<LuceneIndexAttribute>();
                foreach (var attr in attrs)
                {
                    string name = !string.IsNullOrEmpty(attr.Name) ? attr.Name : propertyInfo.Name;
                    string value = attr.IsHtml ? propertyValue.ToString().RemoveHtmlTag() : propertyValue.ToString();
                    switch (propertyValue)
                    {
                        case string _ when value.Length < 8191:
                            doc.Add(new TextField(name, value, attr.Store));
                            break;
                        case string _:
                            doc.Add(new TextField(name, value, attr.Store));
                            break;
                        case int num:
                            doc.Add(new Int32Field(name, num, attr.Store));
                            break;
                        case long num:
                            doc.Add(new Int64Field(name, num, attr.Store));
                            break;
                        case float num:
                            doc.Add(new SingleField(name, num, attr.Store));
                            break;
                        case decimal num:
                            doc.Add(new DoubleField(name, (double)num, attr.Store));
                            break;
                        case double num:
                            doc.Add(new DoubleField(name, num, attr.Store));
                            break;
                        case DateTime _:
                            doc.Add(new TextField(name, value, attr.Store));
                            break;
                        default:
                            doc.Add(new TextField(name, value, attr.Store));
                            break;
                    }
                }
            }

            return doc;
        }
    }
}