c# - 如何在 NEST C# 中编写动态 Elasticsearch 查询

Question

我有一个函数可以根据 bool 标志在时间间隔内搜索文档，使用 的 NEST 查询将有所不同小于等于 在 if 语句和 小于 在 else 语句中。

public IEnumerable<Book> GetBooks(DateTime startTime, DateTime endTime,  string author, bool includeEnd)
{
  var readRecords;
  if (includeEnd){
      readRecords = elastic.Search<Book>(s => s.Index(IndexName)
      .Query(q => q.Term(book => book.Author, author) &&
            q.DateRange(i => i.Field(book => book.Author)    
            .GreaterThanOrEquals(startTime)
            .LessThanOrEquals(endTime)))).Documents.ToArray();      
  }
  else{
    readRecords = elastic.Search<Book>(s => s.Index(IndexName)
           .Query(q => q.Term(book => book.Author, author) &&
            q.DateRange(i => i.Field(book => book.Author)
            .GreaterThanOrEquals(startTime)
            .LessThan(endTime)))).Documents.ToArray();
    }   
    return readRecords;
}

如何使用 bool 标志“includeEnd”使这个 NEST 查询动态化，这样我就不需要使用 if 语句？

Answer 1

看看 LessThanOrEquals 和 LessThan 泛型扩展方法的作用。基本上，它们扩展了 T 的 DateRangeQueryDescriptor(在这种情况下，T 是一本书)，接受 DateMath 参数，并返回另一个 DateRangeQueryDescriptor。因此，我们可以分解出一个函数，根据 includeEnd 标志，返回正确的函数以期望查询描述符。

 public IEnumerable<Book> GetBooks(DateTime startTime, DateTime endTime, string author, bool includeEnd)
    {
        var dateFilter = includeEnd ?
            // you have to do a little casting for the lambdas to know what type we're returning
            (Func<DateRangeQueryDescriptor<Book>, DateRangeQueryDescriptor<Book>>)
            (q => q.LessThanOrEquals(endTime))
            : q => q.LessThan(endTime);

        return elastic.Search<Book>(s => s
            .Index(IndexName)
            .Query(q => q.Term(book => book.Author, author) &&
                        q.DateRange(i =>
                        {
                            i = i.Field(book => book.Author)
                            .GreaterThanOrEquals(startTime);
                            return dateFilter(i);
                        }
                )))
                .Documents
                .ToArray();
    }

为了进一步了解这个想法，您可以编写自己的扩展方法:

 public static class DateQueryExtensions
{
    public static DateRangeQueryDescriptor<T> LessThanWithOption<T>(this DateRangeQueryDescriptor<T> q, DateMath to, bool includeEnd)
        where T : class
    {
        return includeEnd ? q.LessThanOrEquals(to) : q.LessThan(to);
    }
}

然后你可以像这样使用它:

 public IEnumerable<Book> GetBooksUsingExtension(DateTime startTime, DateTime endTime, string author, bool includeEnd)
    {
        return elastic.Search<Book>(s => s
           .Index(IndexName)
           .Query(q => q.Term(book => book.Author, author) &&
                       q.DateRange(i => i.Field(book => book.Author)
                           .GreaterThanOrEquals(startTime)
                           .LessThanWithOption(endTime, includeEnd)
                       )))
               .Documents
               .ToArray();

    }

大多数 Nest fluent 接口(interface)扩展都以类似的方式对查询描述符进行操作。