private void Merge(string dim, string[] path)
{
int index = 0;
if (drillDownDims.TryGetValue(dim, out int?idx) && idx.HasValue)
{
index = idx.Value;
}
if (query.GetClauses().Length == drillDownDims.Count + 1)
{
index++;
}
ConstantScoreQuery q = (ConstantScoreQuery)query.GetClauses()[index].Query;
if ((q.Query is BooleanQuery) == false)
{
// App called .add(dim, customQuery) and then tried to
// merge a facet label in:
throw new Exception("cannot merge with custom Query");
}
string indexedField = config.GetDimConfig(dim).IndexFieldName;
BooleanQuery bq = (BooleanQuery)q.Query;
bq.Add(new TermQuery(Term(indexedField, dim, path)), Occur.SHOULD);
}
public virtual void TestTermNonDefault()
{
string aField = config.GetDimConfig("a").IndexFieldName;
Term termA = DrillDownQuery.Term(aField, "a");
Assert.AreEqual(new Term(aField, "a"), termA);
string bField = config.GetDimConfig("b").IndexFieldName;
Term termB = DrillDownQuery.Term(bField, "b");
Assert.AreEqual(new Term(bField, "b"), termB);
}
/// <summary>
/// Note: if you use a counting <see cref="Facets"/> implementation, you can amortize the
/// sampled counts by calling this method. Uses the <see cref="FacetsConfig"/> and
/// the <see cref="IndexSearcher"/> to determine the upper bound for each facet value.
/// </summary>
public virtual FacetResult AmortizeFacetCounts(FacetResult res, FacetsConfig config, IndexSearcher searcher)
{
if (res == null || totalHits <= sampleSize)
{
return(res);
}
LabelAndValue[] fixedLabelValues = new LabelAndValue[res.LabelValues.Length];
IndexReader reader = searcher.IndexReader;
DimConfig dimConfig = config.GetDimConfig(res.Dim);
// +2 to prepend dimension, append child label
string[] childPath = new string[res.Path.Length + 2];
childPath[0] = res.Dim;
Array.Copy(res.Path, 0, childPath, 1, res.Path.Length); // reuse
for (int i = 0; i < res.LabelValues.Length; i++)
{
childPath[res.Path.Length + 1] = res.LabelValues[i].Label;
string fullPath = FacetsConfig.PathToString(childPath, childPath.Length);
int max = reader.DocFreq(new Term(dimConfig.IndexFieldName, fullPath));
int correctedCount = (int)((double)res.LabelValues[i].Value / samplingRate);
correctedCount = Math.Min(max, correctedCount);
fixedLabelValues[i] = new LabelAndValue(res.LabelValues[i].Label, correctedCount);
}
// cap the total count on the total number of non-deleted documents in the reader
int correctedTotalCount = (int)res.Value;
if (correctedTotalCount > 0)
{
correctedTotalCount = Math.Min(reader.NumDocs, (int)((double)res.Value / samplingRate));
}
return(new FacetResult(res.Dim, res.Path, correctedTotalCount, fixedLabelValues, res.ChildCount));
}
/// <summary>
/// Note: if you use a counting <seealso cref="Facets"/> implementation, you can amortize the
/// sampled counts by calling this method. Uses the <seealso cref="FacetsConfig"/> and
/// the <seealso cref="IndexSearcher"/> to determine the upper bound for each facet value.
/// </summary>
public virtual FacetResult AmortizeFacetCounts(FacetResult res, FacetsConfig config, IndexSearcher searcher)
{
if (res == null || totalHits <= sampleSize)
{
return res;
}
LabelAndValue[] fixedLabelValues = new LabelAndValue[res.LabelValues.Length];
IndexReader reader = searcher.IndexReader;
DimConfig dimConfig = config.GetDimConfig(res.Dim);
// +2 to prepend dimension, append child label
string[] childPath = new string[res.Path.Length + 2];
childPath[0] = res.Dim;
Array.Copy(res.Path, 0, childPath, 1, res.Path.Length); // reuse
for (int i = 0; i < res.LabelValues.Length; i++)
{
childPath[res.Path.Length + 1] = res.LabelValues[i].label;
string fullPath = FacetsConfig.PathToString(childPath, childPath.Length);
int max = reader.DocFreq(new Term(dimConfig.IndexFieldName, fullPath));
int correctedCount = (int)((double)res.LabelValues[i].value / samplingRate);
correctedCount = Math.Min(max, correctedCount);
fixedLabelValues[i] = new LabelAndValue(res.LabelValues[i].label, correctedCount);
}
// cap the total count on the total number of non-deleted documents in the reader
int correctedTotalCount = (int)res.Value;
if (correctedTotalCount > 0)
{
correctedTotalCount = Math.Min(reader.NumDocs, (int)((double)res.Value / samplingRate));
}
return new FacetResult(res.Dim, res.Path, correctedTotalCount, fixedLabelValues, res.ChildCount);
}