@Override
protected void extractUnknownQuery(Query query,
                                   Map<String, WeightedSpanTerm> terms) throws IOException {
    if (query instanceof FunctionScoreQuery) {
        query = ((FunctionScoreQuery) query).getSubQuery();
        extract(query, query.getBoost(), terms);
    } else if (query instanceof FiltersFunctionScoreQuery) {
        query = ((FiltersFunctionScoreQuery) query).getSubQuery();
        extract(query, query.getBoost(), terms);
    } else if (terms.isEmpty()) {
        extractWeightedTerms(terms, query, query.getBoost());
    }
}
 

@Override
protected void extractUnknownQuery(Query query,
    Map<String, WeightedSpanTerm> terms) throws IOException {
  float boost = 1f;
  while (query instanceof BoostQuery) {
    BoostQuery bq = (BoostQuery) query;
    boost *= bq.getBoost();
    query = bq.getQuery();
  }
  if (query instanceof CustomQuery) {
    extractWeightedTerms(terms, new TermQuery(((CustomQuery) query).term), boost);
  }
}
 

@Override
protected void extract(Query query, float boost, Map<String, WeightedSpanTerm> terms) throws IOException {
  // these queries are not supported in lucene highlighting out of the box since 8.0
  if (query instanceof ToParentBlockJoinQuery) {
    extract(((ToParentBlockJoinQuery) query).getChildQuery(), boost, terms);
  } else if (query instanceof ToChildBlockJoinQuery) {
    extract(((ToChildBlockJoinQuery) query).getParentQuery(), boost, terms);
  } else {
    super.extract(query, boost, terms);
  }
}
 

public CustomQueryScorer(WeightedSpanTerm[] weightedTerms) {
    super(weightedTerms);
}