refactor: Move facet extraction

src/Examine.Lucene/Indexing/DateTimeType.cs

@@ -1,14 +1,18 @@
 using System;
+using System.Collections.Generic;
 using Examine.Lucene.Providers;
+using Examine.Lucene.Search;
+using Examine.Search;
 using Lucene.Net.Documents;
+using Lucene.Net.Facet;
 using Lucene.Net.Facet.SortedSet;
 using Lucene.Net.Search;
 using Microsoft.Extensions.Logging;
 
 namespace Examine.Lucene.Indexing
 {
 
-    public class DateTimeType : IndexFieldRangeValueType<DateTime>
+    public class DateTimeType : IndexFieldRangeValueType<DateTime>, IIndexFacetValueType
     {
         public DateResolution Resolution { get; }
 
@@ -66,5 +70,7 @@ public override Query GetQuery(DateTime? lower, DateTime? upper, bool lowerInclu
                 lower != null ? DateToLong(lower.Value) : (long?)null,
                 upper != null ? DateToLong(upper.Value) : (long?)null, lowerInclusive, upperInclusive);
         }
+
+        public virtual void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets, IFacetField field) => field.ExtractFacets(facetsCollector, sortedSetReaderState, facets);
     }
 }

src/Examine.Lucene/Indexing/DoubleType.cs

@@ -1,12 +1,16 @@
+using System.Collections.Generic;
 using Examine.Lucene.Providers;
+using Examine.Lucene.Search;
+using Examine.Search;
 using Lucene.Net.Documents;
+using Lucene.Net.Facet;
 using Lucene.Net.Facet.SortedSet;
 using Lucene.Net.Search;
 using Microsoft.Extensions.Logging;
 
 namespace Examine.Lucene.Indexing
 {
-    public class DoubleType : IndexFieldRangeValueType<double>
+    public class DoubleType : IndexFieldRangeValueType<double>, IIndexFacetValueType
     {
         private readonly bool _isFacetable;
 
@@ -46,5 +50,6 @@ public override Query GetQuery(double? lower, double? upper, bool lowerInclusive
                 lower ?? double.MinValue,
                 upper ?? double.MaxValue, lowerInclusive, upperInclusive);
         }
+        public virtual void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets, IFacetField field) => field.ExtractFacets(facetsCollector, sortedSetReaderState, facets);
     }
 }

src/Examine.Lucene/Indexing/FullTextType.cs

@@ -2,10 +2,13 @@
 using System.IO;
 using Examine.Lucene.Analyzers;
 using Examine.Lucene.Providers;
+using Examine.Lucene.Search;
+using Examine.Search;
 using Lucene.Net.Analysis;
 using Lucene.Net.Analysis.Miscellaneous;
 using Lucene.Net.Analysis.TokenAttributes;
 using Lucene.Net.Documents;
+using Lucene.Net.Facet;
 using Lucene.Net.Facet.SortedSet;
 using Lucene.Net.Index;
 using Lucene.Net.Search;
@@ -21,7 +24,7 @@ namespace Examine.Lucene.Indexing
     /// do an exact match search if the term is less than 4 chars, else it will do a full text search on the phrase
     /// with a higher boost, then 
     /// </remarks>
-    public class FullTextType : IndexFieldValueTypeBase
+    public class FullTextType : IndexFieldValueTypeBase, IIndexFacetValueType
     {
         private readonly bool _sortable;
         private readonly Analyzer _analyzer;
@@ -150,5 +153,6 @@ public override Query GetQuery(string query)
             return GenerateQuery(FieldName, query, _analyzer);
         }
 
+        public virtual void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets, IFacetField field) => field.ExtractFacets(facetsCollector, sortedSetReaderState, facets);
     }
 }

src/Examine.Lucene/Indexing/IIndexFacetValueType.cs

@@ -0,0 +1,20 @@
+using System.Collections.Generic;
+using Examine.Lucene.Search;
+using Examine.Search;
+using Lucene.Net.Facet;
+using Lucene.Net.Facet.SortedSet;
+
+namespace Examine.Lucene.Indexing
+{
+    public interface IIndexFacetValueType
+    {
+        /// <summary>
+        /// Extracts the facets from the field
+        /// </summary>
+        /// <param name="facetsCollector"></param>
+        /// <param name="sortedSetReaderState"></param>
+        /// <param name="facets"></param>
+        /// <param name="field"></param>
+        void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets, IFacetField field);
+    }
+}

src/Examine.Lucene/Indexing/Int32Type.cs

@@ -1,12 +1,16 @@
+using System.Collections.Generic;
 using Examine.Lucene.Providers;
+using Examine.Lucene.Search;
+using Examine.Search;
 using Lucene.Net.Documents;
+using Lucene.Net.Facet;
 using Lucene.Net.Facet.SortedSet;
 using Lucene.Net.Search;
 using Microsoft.Extensions.Logging;
 
 namespace Examine.Lucene.Indexing
 {
-    public class Int32Type : IndexFieldRangeValueType<int>
+    public class Int32Type : IndexFieldRangeValueType<int>, IIndexFacetValueType
     {
         private readonly bool _isFacetable;
 
@@ -46,5 +50,7 @@ public override Query GetQuery(int? lower, int? upper, bool lowerInclusive = tru
                 lower,
                 upper, lowerInclusive, upperInclusive);
         }
+
+        public virtual void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets, IFacetField field) => field.ExtractFacets(facetsCollector, sortedSetReaderState, facets);
     }
 }

src/Examine.Lucene/Indexing/Int64Type.cs

@@ -1,12 +1,16 @@
+using System.Collections.Generic;
 using Examine.Lucene.Providers;
+using Examine.Lucene.Search;
+using Examine.Search;
 using Lucene.Net.Documents;
+using Lucene.Net.Facet;
 using Lucene.Net.Facet.SortedSet;
 using Lucene.Net.Search;
 using Microsoft.Extensions.Logging;
 
 namespace Examine.Lucene.Indexing
 {
-    public class Int64Type : IndexFieldRangeValueType<long>
+    public class Int64Type : IndexFieldRangeValueType<long>, IIndexFacetValueType
     {
         private readonly bool _isFacetable;
 
@@ -46,5 +50,6 @@ public override Query GetQuery(long? lower, long? upper, bool lowerInclusive = t
                 lower,
                 upper, lowerInclusive, upperInclusive);
         }
+        public virtual void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets, IFacetField field) => field.ExtractFacets(facetsCollector, sortedSetReaderState, facets);
     }
 }

src/Examine.Lucene/Indexing/SingleType.cs

@@ -1,12 +1,16 @@
+using System.Collections.Generic;
 using Examine.Lucene.Providers;
+using Examine.Lucene.Search;
+using Examine.Search;
 using Lucene.Net.Documents;
+using Lucene.Net.Facet;
 using Lucene.Net.Facet.SortedSet;
 using Lucene.Net.Search;
 using Microsoft.Extensions.Logging;
 
 namespace Examine.Lucene.Indexing
 {
-    public class SingleType : IndexFieldRangeValueType<float>
+    public class SingleType : IndexFieldRangeValueType<float>, IIndexFacetValueType
     {
         private readonly bool _isFacetable;
 
@@ -46,5 +50,7 @@ public override Query GetQuery(float? lower, float? upper, bool lowerInclusive =
                 lower ?? float.MinValue,
                 upper ?? float.MaxValue, lowerInclusive, upperInclusive);
         }
+
+        public virtual void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets, IFacetField field) => field.ExtractFacets(facetsCollector, sortedSetReaderState, facets);
     }
 }

src/Examine.Lucene/Search/FacetDoubleField.cs

@@ -0,0 +1,39 @@
+using Lucene.Net.Facet.Range;
+using Lucene.Net.Facet;
+using System.Collections.Generic;
+using Examine.Search;
+using System.Linq;
+using Lucene.Net.Facet.SortedSet;
+
+namespace Examine.Lucene.Search
+{
+    public readonly struct FacetDoubleField : IFacetField
+    {
+        public Examine.Search.DoubleRange[] DoubleRanges { get; }
+
+        public string Field { get; }
+
+        public string FacetField { get; }
+
+        public FacetDoubleField(string field, Examine.Search.DoubleRange[] doubleRanges, string facetField)
+        {
+            Field = field;
+            DoubleRanges = doubleRanges;
+            FacetField = facetField;
+        }
+
+        public void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets)
+        {
+            var doubleFacetCounts = new DoubleRangeFacetCounts(Field, facetsCollector, DoubleRanges.AsLuceneRange().ToArray());
+
+            var doubleFacets = doubleFacetCounts.GetTopChildren(0, Field);
+
+            if (doubleFacets == null)
+            {
+                return;
+            }
+
+            facets.Add(Field, new Examine.Search.FacetResult(doubleFacets.LabelValues.Select(labelValue => new FacetValue(labelValue.Label, labelValue.Value) as IFacetValue)));
+        }
+    }
+}

src/Examine.Lucene/Search/FacetFloatField.cs

@@ -0,0 +1,40 @@
+using System.Collections.Generic;
+using System.Linq;
+using Examine.Search;
+using Lucene.Net.Facet;
+using Lucene.Net.Facet.Range;
+using Lucene.Net.Facet.SortedSet;
+using Lucene.Net.Queries.Function.ValueSources;
+
+namespace Examine.Lucene.Search
+{
+    public readonly struct FacetFloatField : IFacetField
+    {
+        public FloatRange[] FloatRanges { get; }
+
+        public string Field { get; }
+
+        public string FacetField { get; }
+
+        public FacetFloatField(string field, FloatRange[] floatRanges, string facetField)
+        {
+            Field = field;
+            FloatRanges = floatRanges;
+            FacetField = facetField;
+        }
+
+        public void ExtractFacets(FacetsCollector facetsCollector, SortedSetDocValuesReaderState sortedSetReaderState, Dictionary<string, IFacetResult> facets)
+        {
+            var floatFacetCounts = new DoubleRangeFacetCounts(Field, new SingleFieldSource(Field), facetsCollector, FloatRanges.AsLuceneRange().ToArray());
+
+            var floatFacets = floatFacetCounts.GetTopChildren(0, Field);
+
+            if (floatFacets == null)
+            {
+                return;
+            }
+
+            facets.Add(Field, new Examine.Search.FacetResult(floatFacets.LabelValues.Select(labelValue => new FacetValue(labelValue.Label, labelValue.Value) as IFacetValue)));
+        }
+    }
+}