121 lines
2.9 KiB
C#
121 lines
2.9 KiB
C#
using Raven.Client.Documents;
|
|
|
|
namespace zero.Search;
|
|
|
|
public class SearchIndexMap
|
|
{
|
|
internal Type Type;
|
|
internal string _Icon;
|
|
protected string _group;
|
|
protected string[] _fields;
|
|
protected float _boost = 0;
|
|
protected Func<ZeroEntity, SearchResult, IZeroContext, Task> _modify;
|
|
|
|
const string mapTemplate = @"map('{collection}', function (x) {
|
|
return {
|
|
Id: x.Id,
|
|
Group: '{group}',
|
|
Name: boost(x.Name, {boost}),
|
|
IsActive: x.IsActive,
|
|
Fields: [{fields}]
|
|
};
|
|
});";
|
|
|
|
internal SearchIndexMap(Type type, string icon = null)
|
|
{
|
|
Type = type;
|
|
_group = "__TODO";
|
|
_Icon = icon;
|
|
}
|
|
|
|
internal string BuildInstruction(IDocumentStore store)
|
|
{
|
|
return TokenReplacement.Apply(mapTemplate, new()
|
|
{
|
|
{ "collection", store.Conventions.GetCollectionName(Type) },
|
|
{ "group", _group },
|
|
{ "fields", BuildFieldArray(_fields) },
|
|
{ "boost", _boost.ToString() }
|
|
});
|
|
}
|
|
|
|
internal string BuildFieldArray(string[] fields)
|
|
{
|
|
if (fields == null || !fields.Any())
|
|
{
|
|
return String.Empty;
|
|
}
|
|
|
|
return String.Join(", ", fields.Select(x => $"x.{x}")); //$"boost(x.{x}, {_boost})"));
|
|
}
|
|
|
|
internal bool CanModify(Type type)
|
|
{
|
|
return Type.IsAssignableFrom(type);
|
|
}
|
|
|
|
internal async Task Modify(ZeroEntity entity, SearchResult result, IZeroContext context)
|
|
{
|
|
if (_modify != null)
|
|
{
|
|
await _modify(entity, result, context);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
public class SearchIndexMap<T> : SearchIndexMap where T : ZeroEntity
|
|
{
|
|
internal SearchIndexMap(string icon = null) : base(typeof(T), icon) { }
|
|
|
|
public virtual SearchIndexMap<T> Icon(string icon)
|
|
{
|
|
_Icon = icon;
|
|
return this;
|
|
}
|
|
|
|
public SearchIndexMap<T> Display(Action<T, SearchResult> modify = null)
|
|
{
|
|
_modify = (x, res, opts) =>
|
|
{
|
|
modify?.Invoke(x as T, res);
|
|
return Task.CompletedTask;
|
|
};
|
|
return this;
|
|
}
|
|
|
|
public SearchIndexMap<T> Display(Func<T, SearchResult, Task> modify = null)
|
|
{
|
|
_modify = (x, res, ctx) => modify?.Invoke(x as T, res);
|
|
return this;
|
|
}
|
|
|
|
public SearchIndexMap<T> Display(Func<T, SearchResult, IZeroContext, Task> modify = null)
|
|
{
|
|
_modify = (x, res, ctx) => modify?.Invoke(x as T, res, ctx);
|
|
return this;
|
|
}
|
|
|
|
public SearchIndexMap<T> Display(Action<T, SearchResult, IZeroContext> modify = null)
|
|
{
|
|
_modify = (x, res, ctx) =>
|
|
{
|
|
modify?.Invoke(x as T, res, ctx);
|
|
return Task.CompletedTask;
|
|
};
|
|
return this;
|
|
}
|
|
|
|
public SearchIndexMap<T> Fields(params string[] fieldNames)
|
|
{
|
|
_fields = fieldNames;
|
|
return this;
|
|
}
|
|
|
|
public SearchIndexMap<T> Boost(float boostValue)
|
|
{
|
|
_boost = boostValue;
|
|
return this;
|
|
}
|
|
}
|