| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.solr.search; |
| |
| import org.apache.lucene.util.TestUtil; |
| import org.apache.solr.SolrTestCaseJ4; |
| |
| import org.apache.solr.metrics.MetricsMap; |
| import org.apache.solr.metrics.SolrMetricManager; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import java.lang.invoke.MethodHandles; |
| import java.util.Map; |
| import java.util.Random; |
| |
| public class TestSolrFieldCacheBean extends SolrTestCaseJ4 { |
| |
| private static final Logger log = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass()); |
| |
| @BeforeClass |
| public static void beforeClass() throws Exception { |
| initCore("solrconfig.xml","schema-minimal.xml"); |
| } |
| |
| @Test |
| public void testEntryList() throws Exception { |
| // ensure entries to FieldCache |
| assertU(adoc("id", "id0")); |
| assertU(commit()); |
| assertQ(req("q", "*:*", "sort", "id asc"), "//*[@numFound='1']"); |
| |
| // Test with entry list enabled |
| assertEntryListIncluded(false); |
| |
| // Test again with entry list disabled |
| System.setProperty("disableSolrFieldCacheMBeanEntryList", "true"); |
| try { |
| assertEntryListNotIncluded(false); |
| } finally { |
| System.clearProperty("disableSolrFieldCacheMBeanEntryList"); |
| } |
| |
| // Test with entry list enabled for jmx |
| assertEntryListIncluded(true); |
| |
| // Test with entry list disabled for jmx |
| System.setProperty("disableSolrFieldCacheMBeanEntryListJmx", "true"); |
| try { |
| assertEntryListNotIncluded(true); |
| } finally { |
| System.clearProperty("disableSolrFieldCacheMBeanEntryListJmx"); |
| } |
| } |
| |
| private void assertEntryListIncluded(boolean checkJmx) { |
| SolrFieldCacheBean mbean = new SolrFieldCacheBean(); |
| Random r = random(); |
| String registryName = TestUtil.randomSimpleString(r, 1, 10); |
| SolrMetricManager metricManager = h.getCoreContainer().getMetricManager(); |
| mbean.initializeMetrics(metricManager, registryName, "foo", null); |
| MetricsMap metricsMap = (MetricsMap)((SolrMetricManager.GaugeWrapper)metricManager.registry(registryName).getMetrics().get("CACHE.fieldCache")).getGauge(); |
| Map<String, Object> metrics = checkJmx ? metricsMap.getValue(true) : metricsMap.getValue(); |
| assertTrue(((Number)metrics.get("entries_count")).longValue() > 0); |
| assertNotNull(metrics.get("total_size")); |
| assertNotNull(metrics.get("entry#0")); |
| } |
| |
| private void assertEntryListNotIncluded(boolean checkJmx) { |
| SolrFieldCacheBean mbean = new SolrFieldCacheBean(); |
| Random r = random(); |
| String registryName = TestUtil.randomSimpleString(r, 1, 10); |
| SolrMetricManager metricManager = h.getCoreContainer().getMetricManager(); |
| mbean.initializeMetrics(metricManager, registryName, "foo", null); |
| MetricsMap metricsMap = (MetricsMap)((SolrMetricManager.GaugeWrapper)metricManager.registry(registryName).getMetrics().get("CACHE.fieldCache")).getGauge(); |
| Map<String, Object> metrics = checkJmx ? metricsMap.getValue(true) : metricsMap.getValue(); |
| assertTrue(((Number)metrics.get("entries_count")).longValue() > 0); |
| assertNull(metrics.get("total_size")); |
| assertNull(metrics.get("entry#0")); |
| } |
| } |