public IndexManager(IMemoryCache memoryCache, IElasticClientFactory elasticClientFactory, ILogger <IndexManager> logger)
{
_memoryCache = memoryCache;
_logger = logger;
_elasticClient = elasticClientFactory.Create();
Console.WriteLine("ElasticsearchIndex初始化");
}
public ElasticSearchServiceQuery(IElasticClientFactory elasticClientFactory, IIndexManager indexManager)
{
_indexManager = indexManager;
_elasticClient = elasticClientFactory.Create();
}
public ElasticsearchSpanStorage(IElasticClientFactory elasticClientFactory, IIndexManager indexManager)
{
_elasticClient = elasticClientFactory?.Create() ?? throw new ArgumentNullException(nameof(elasticClientFactory));
_indexManager = indexManager ?? throw new ArgumentNullException(nameof(indexManager));
}
public ElasticsearchServiceStorage(IElasticClientFactory elasticClientFactory, IIndexManager indexManager, IMemoryCache memoryCache)
{
_elasticClient = elasticClientFactory?.Create() ?? throw new ArgumentNullException(nameof(elasticClientFactory));
_indexManager = indexManager ?? throw new ArgumentNullException(nameof(indexManager));
_memoryCache = memoryCache ?? throw new ArgumentNullException(nameof(memoryCache));
}
public IndexManager(IMemoryCache memoryCache, IElasticClientFactory elasticClientFactory, ILogger <IndexManager> logger)
{
_memoryCache = memoryCache;
_logger = logger;
_elasticClient = elasticClientFactory.Create();
}
public async Task <DataSourceResponseModel> GetMessages(int page = 0, int take = 50, string?search = null)
{
var response = await _elasticClientFactory
.Create()
.SearchAsync <VkMessageModel>(z => z
.From(page * take)
.Take(take)
.Query(f =>
{
var q = f
.Bool(e => e
.Must(g => g
.MatchPhrase(n => n
.Field(message => message.Text)
.Query(search)
)
)
);
return(q);
})
.Highlight(a => a
.Fields(descriptor => descriptor
.Field(h => h.Text)
.NumberOfFragments(0)
.PreTags("<strong>")
.PostTags("</strong>")
)
)
);
await using var scope = _transactionScopeFactory.CreateQuery();
var keys = response.Documents.Select(f => f.OwnerId + "_" + f.Id);
var items = await scope.Query <MessageVk>()
.Select(r =>
new { Key = r.OwnerId.ToString() + "_" + r.MessageId.ToString(), item = r })
.Where(e => keys.Contains(e.Key))
.Select(f => new { message = f.item, category = f.item.MessageCategory.Title })
.ToListAsync();
var reposts = await scope.Query <VkRepost>()
.Where(f => f.Status == VkRepostStatus.Complete)
.Select(r =>
new { Key = r.OwnerId.ToString() + "_" + r.MessageId.ToString(), r.OwnerId, r.MessageId })
.Where(f => keys.Contains(f.Key))
.ToListAsync();
MessagePredictResponse?predicted = null;
try
{
predicted = await _messagePredictingClient.PredictAsync(new MessagePredictRequest
{
Messages =
{
response.Documents.Select(e => new MessagePredictRequest.Types.MessagePredict
{
Id = e.Id,
Text = e.Text,
OwnerId = e.OwnerId
})
}
});
}
catch (Exception e)
{
_logger.LogError(e, "Error to access predict client");
}
foreach (var document in response.Hits)
{
var messageCategory =
items.FirstOrDefault(f =>
f.message.MessageId == document.Source.Id && f.message.OwnerId == document.Source.OwnerId);
document.Source.MessageCategoryId = messageCategory?.message?.MessageCategoryId;
document.Source.MessageCategory = messageCategory?.category;
if (document.Highlight.ContainsKey("text"))
{
document.Source.Text = document.Highlight["text"].FirstOrDefault();
}
document.Source.UserReposted = reposts.Any(f =>
f.MessageId == document.Source.Id && f.OwnerId == document.Source.OwnerId);
document.Source.MessageCategoryPredict = predicted?.Messages
.FirstOrDefault(e => e.Id == document.Source.Id && e.OwnerId == document.Source.OwnerId)?.Category;
}
var result = response.Hits
.OrderByDescending(a => a.Score)
.Select(f => f.Source)
.GroupBy(f => new { f.OwnerId, f.Id })
.Select(f => f.FirstOrDefault());
return(new DataSourceResponseModel(result, response.Total));
}
