ClassCastException during percolation query, closes #1905.

src/main/java/org/elasticsearch/index/cache/bloom/simple/SimpleBloomCache.java

@@ -123,7 +123,9 @@ public class SimpleBloomCache extends AbstractIndexComponent implements BloomCac
             synchronized (creationMutex) {
                 fieldCache = cache.get(reader.getCoreCacheKey());
                 if (fieldCache == null) {
+                    if (reader instanceof SegmentReader) {
                         ((SegmentReader) reader).addCoreClosedListener(this);
+                    }
                     fieldCache = ConcurrentCollections.newConcurrentMap();
                     cache.put(reader.getCoreCacheKey(), fieldCache);
                 }

src/main/java/org/elasticsearch/index/cache/field/data/support/AbstractConcurrentMapFieldDataCache.java

@@ -112,7 +112,9 @@ public abstract class AbstractConcurrentMapFieldDataCache extends AbstractIndexC
                 fieldDataCache = cache.get(reader.getCoreCacheKey());
                 if (fieldDataCache == null) {
                     fieldDataCache = buildFieldDataMap();
+                    if (reader instanceof SegmentReader) {
                         ((SegmentReader) reader).addCoreClosedListener(this);
+                    }
                     cache.put(reader.getCoreCacheKey(), fieldDataCache);
                 }
             }

src/main/java/org/elasticsearch/index/cache/filter/weighted/WeightedFilterCache.java

@@ -166,7 +166,7 @@ public class WeightedFilterCache extends AbstractIndexComponent implements Filte
             if (cacheValue == null) {
                 if (!cache.seenReaders.containsKey(reader.getCoreCacheKey())) {
                     Boolean previous = cache.seenReaders.putIfAbsent(reader.getCoreCacheKey(), Boolean.TRUE);
-                    if (previous == null) {
+                    if (previous == null && (reader instanceof SegmentReader)) {
                         ((SegmentReader) reader).addCoreClosedListener(cache);
                         cache.seenReadersCount.inc();
                     }

src/main/java/org/elasticsearch/index/cache/id/simple/SimpleIdCache.java

@@ -109,7 +109,9 @@ public class SimpleIdCache extends AbstractIndexComponent implements IdCache, Se
                         continue;
                     }
 
+                    if (reader instanceof SegmentReader) {
                         ((SegmentReader) reader).addCoreClosedListener(this);
+                    }
                     HashMap<String, TypeBuilder> readerBuilder = new HashMap<String, TypeBuilder>();
                     builders.put(reader.getCoreCacheKey(), readerBuilder);
 

src/main/java/org/elasticsearch/index/percolator/PercolatorExecutor.java

@@ -249,7 +249,7 @@ public class PercolatorExecutor extends AbstractIndexComponent {
         }
     }
 
-    public synchronized void addQuery(String name, Query query) {
+    private synchronized void addQuery(String name, Query query) {
         Preconditions.checkArgument(query != null, "query must be provided for percolate request");
         this.queries = MapBuilder.newMapBuilder(queries).put(name, query).immutableMap();
     }
@@ -349,8 +349,9 @@ public class PercolatorExecutor extends AbstractIndexComponent {
         }
 
         final IndexSearcher searcher = memoryIndex.createSearcher();
-
         List<String> matches = new ArrayList<String>();
+
+        try {
             if (request.query() == null) {
                 Lucene.ExistsCollector collector = new Lucene.ExistsCollector();
                 for (Map.Entry<String, Query> entry : queries.entrySet()) {
@@ -383,8 +384,10 @@ public class PercolatorExecutor extends AbstractIndexComponent {
                     percolatorSearcher.release();
                 }
             }
-
+        } finally {
+            // explicitly clear the reader, since we can only register on callback on SegmentReader
             indexCache.clear(searcher.getIndexReader());
+        }
 
         return new Response(matches, request.doc().mappersAdded());
     }

src/test/java/org/elasticsearch/test/unit/index/percolator/PercolatorExecutorTests.java

@@ -36,6 +36,7 @@ import org.elasticsearch.index.cache.IndexCacheModule;
 import org.elasticsearch.index.engine.IndexEngineModule;
 import org.elasticsearch.index.mapper.MapperServiceModule;
 import org.elasticsearch.index.percolator.PercolatorExecutor;
+import org.elasticsearch.index.query.FilterBuilders;
 import org.elasticsearch.index.query.IndexQueryParserModule;
 import org.elasticsearch.index.settings.IndexSettingsModule;
 import org.elasticsearch.index.similarity.SimilarityModule;
@@ -45,6 +46,7 @@ import org.elasticsearch.threadpool.ThreadPoolModule;
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
 
+import static org.elasticsearch.index.query.QueryBuilders.constantScoreQuery;
 import static org.elasticsearch.index.query.QueryBuilders.termQuery;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.*;
@@ -60,7 +62,7 @@ public class PercolatorExecutorTests {
     @BeforeClass
     public void buildPercolatorService() {
         Settings settings = ImmutableSettings.settingsBuilder()
-                .put("index.cache.filter.type", "none")
+                //.put("index.cache.filter.type", "none")
                 .build();
         Index index = new Index("test");
         Injector injector = new ModulesBuilder().add(
@@ -128,5 +130,13 @@ public class PercolatorExecutorTests {
         percolate = percolatorExecutor.percolate(new PercolatorExecutor.SourceRequest("type1", source));
         assertThat(percolate.matches(), hasSize(1));
         assertThat(percolate.matches(), hasItems("test1"));
+
+        // add a range query (cached)
+        // add a query
+        percolatorExecutor.addQuery("test1", constantScoreQuery(FilterBuilders.rangeFilter("field2").from("value").includeLower(true)));
+
+        percolate = percolatorExecutor.percolate(new PercolatorExecutor.SourceRequest("type1", source));
+        assertThat(percolate.matches(), hasSize(1));
+        assertThat(percolate.matches(), hasItem("test1"));
     }
 }