Examples with IndexFieldDataCache org.opensearch.index.fielddata.IndexFieldDataCache used on opensource projects

Example 1 with IndexFieldDataCache

use of org.opensearch.index.fielddata.IndexFieldDataCache in project OpenSearch by opensearch-project.

the class PercolateQueryBuilder method wrap.

static QueryShardContext wrap(QueryShardContext shardContext) {
    return new QueryShardContext(shardContext) {

        @Override
        public IndexReader getIndexReader() {
            // the reader of the MemoryIndex. We just use `null` for simplicity.
            return null;
        }

        @Override
        public BitSetProducer bitsetFilter(Query query) {
            return context -> {
                final IndexReaderContext topLevelContext = ReaderUtil.getTopLevelContext(context);
                final IndexSearcher searcher = new IndexSearcher(topLevelContext);
                searcher.setQueryCache(null);
                final Weight weight = searcher.createWeight(searcher.rewrite(query), ScoreMode.COMPLETE_NO_SCORES, 1f);
                final Scorer s = weight.scorer(context);
                if (s != null) {
                    return new BitDocIdSet(BitSet.of(s.iterator(), context.reader().maxDoc())).bits();
                } else {
                    return null;
                }
            };
        }

        @Override
        @SuppressWarnings("unchecked")
        public <IFD extends IndexFieldData<?>> IFD getForField(MappedFieldType fieldType) {
            IndexFieldData.Builder builder = fieldType.fielddataBuilder(shardContext.getFullyQualifiedIndex().getName(), shardContext::lookup);
            IndexFieldDataCache cache = new IndexFieldDataCache.None();
            CircuitBreakerService circuitBreaker = new NoneCircuitBreakerService();
            return (IFD) builder.build(cache, circuitBreaker);
        }
    };
}