Java SolrInfoMBean类代码示例

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
private void addMBean(SolrQueryRequest req, NamedList<NamedList<NamedList<Object>>> cats, Set<String> requestedKeys, String key, SolrInfoMBean m) {
  if ( ! ( requestedKeys.isEmpty() || requestedKeys.contains(key) ) ) return;
  NamedList<NamedList<Object>> catInfo = cats.get(m.getCategory().name());
  if ( null == catInfo ) return;
  NamedList<Object> mBeanInfo = new SimpleOrderedMap<>();
  mBeanInfo.add("class", m.getName());
  mBeanInfo.add("version", m.getVersion());
  mBeanInfo.add("description", m.getDescription());
  mBeanInfo.add("src", m.getSource());

  // Use an external form
  URL[] urls = m.getDocs();
  if(urls!=null) {
    List<String> docs = new ArrayList<>(urls.length);
    for(URL url : urls) {
      docs.add(url.toExternalForm());
    }
    mBeanInfo.add("docs", docs);
  }

  if (req.getParams().getFieldBool(key, "stats", false))
    mBeanInfo.add("stats", m.getStatistics());

  catInfo.add(key, mBeanInfo);
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
private List<SolrIndexSearcher> getRegisteredSearchers()
{
    List<SolrIndexSearcher> searchers = new ArrayList<SolrIndexSearcher>();
    for (Entry<String, SolrInfoMBean> entry : core.getInfoRegistry().entrySet())
    {
        if (entry.getValue() != null)
        {
            if (entry.getValue().getName().equals(SolrIndexSearcher.class.getName()))
            {
                if (!entry.getKey().equals("searcher"))
                {
                    searchers.add((SolrIndexSearcher) entry.getValue());
                }
            }
        }
    }
    return searchers;
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
protected List<SolrIndexSearcher> getRegisteredSearchers()
{
    List<SolrIndexSearcher> searchers = new ArrayList<SolrIndexSearcher>();
    for (String key : core.getInfoRegistry().keySet())
    {
        SolrInfoMBean mBean = core.getInfoRegistry().get(key);
        if (mBean != null)
        {
            if (mBean.getName().equals(SolrIndexSearcher.class.getName()))
            {
                if (!key.equals("searcher"))
                {
                    searchers.add((SolrIndexSearcher) mBean);
                }

            }
        }
    }

    return searchers;
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
@Override
public int getRegisteredSearcherCount()
{
    HashSet<String> keys = new HashSet<String>();

    for (String key : core.getInfoRegistry().keySet())
    {
        SolrInfoMBean mBean = core.getInfoRegistry().get(key);
        if (mBean != null)
        {
            if (mBean.getName().equals(SolrIndexSearcher.class.getName()))
            {
                if (!key.equals("searcher"))
                {
                    keys.add(key);
                }
            }
        }
    }

    log.info(".... registered Searchers for " + core.getName() + " = " + keys.size());
    return keys.size();
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
protected NamedList<NamedList<NamedList<Object>>> getMBeanInfo(SolrQueryRequest req) {

    NamedList<NamedList<NamedList<Object>>> cats = new NamedList<>();
    
    String[] requestedCats = req.getParams().getParams("cat");
    if (null == requestedCats || 0 == requestedCats.length) {
      for (SolrInfoMBean.Category cat : SolrInfoMBean.Category.values()) {
        cats.add(cat.name(), new SimpleOrderedMap<NamedList<Object>>());
      }
    } else {
      for (String catName : requestedCats) {
        cats.add(catName,new SimpleOrderedMap<NamedList<Object>>());
      }
    }
         
    Set<String> requestedKeys = arrayToSet(req.getParams().getParams("key"));
    
    Map<String, SolrInfoMBean> reg = req.getCore().getInfoRegistry();
    for (Map.Entry<String, SolrInfoMBean> entry : reg.entrySet()) {
      addMBean(req, cats, requestedKeys, entry.getKey(),entry.getValue());
    }

    for (SolrInfoMBean infoMBean : req.getCore().getCoreDescriptor().getCoreContainer().getResourceLoader().getInfoMBeans()) {
      addMBean(req,cats,requestedKeys,infoMBean.getName(),infoMBean);
    }
    return cats;
  }
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
private static SimpleOrderedMap<Object> getSolrInfoBeans( SolrCore core, boolean stats )
{
  SimpleOrderedMap<Object> list = new SimpleOrderedMap<>();
  for (SolrInfoMBean.Category cat : SolrInfoMBean.Category.values()) 
  {
    SimpleOrderedMap<Object> category = new SimpleOrderedMap<>();
    list.add( cat.name(), category );
    Map<String, SolrInfoMBean> reg = core.getInfoRegistry();
    for (Map.Entry<String,SolrInfoMBean> entry : reg.entrySet()) {
      SolrInfoMBean m = entry.getValue();
      if (m.getCategory() != cat) continue;

      String na = "Not Declared";
      SimpleOrderedMap<Object> info = new SimpleOrderedMap<>();
      category.add( entry.getKey(), info );

      info.add( "name",        (m.getName()       !=null ? m.getName()        : na) );
      info.add( "version",     (m.getVersion()    !=null ? m.getVersion()     : na) );
      info.add( "description", (m.getDescription()!=null ? m.getDescription() : na) );
      info.add( "source",      (m.getSource()     !=null ? m.getSource()      : na) );

      URL[] urls = m.getDocs();
      if ((urls != null) && (urls.length > 0)) {
        ArrayList<String> docs = new ArrayList<>(urls.length);
        for( URL u : urls ) {
          docs.add( u.toExternalForm() );
        }
        info.add( "docs", docs );
      }

      if( stats ) {
        info.add( "stats", m.getStatistics() );
      }
    }
  }
  return list;
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
/**
 * Gets a list of everything we can find in the classpath and makes sure it has
 * a name, description, etc...
 */
public void testCallMBeanInfo() throws Exception {
  List<Class> classes = new ArrayList<>();
  classes.addAll(getClassesForPackage(StandardRequestHandler.class.getPackage().getName()));
  classes.addAll(getClassesForPackage(SearchHandler.class.getPackage().getName()));
  classes.addAll(getClassesForPackage(SearchComponent.class.getPackage().getName()));
  classes.addAll(getClassesForPackage(LukeRequestHandler.class.getPackage().getName()));
  classes.addAll(getClassesForPackage(DefaultSolrHighlighter.class.getPackage().getName()));
  classes.addAll(getClassesForPackage(LRUCache.class.getPackage().getName()));
 // System.out.println(classes);
  
  int checked = 0;
  for( Class clazz : classes ) {
    if( SolrInfoMBean.class.isAssignableFrom( clazz ) ) {
      try {
        SolrInfoMBean info = (SolrInfoMBean)clazz.newInstance();
        
        //System.out.println( info.getClass() );
        assertNotNull( info.getName() );
        assertNotNull( info.getDescription() );
        assertNotNull( info.getCategory() );
        
        if( info instanceof LRUCache ) {
          continue;
        }
        
        assertNotNull( info.toString() );
        // increase code coverage...
        assertNotNull( info.getDocs() + "" );
        assertNotNull( info.getStatistics()+"" );
        checked++;
      }
      catch( InstantiationException ex ) {
        // expected...
        //System.out.println( "unable to initalize: "+clazz );
      }
    }
  }
  assertTrue( "there are at least 10 SolrInfoMBean that should be found in the classpath, found " + checked, checked > 10 );
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
@Override
public void execute(AgentOrder agentOrder) {
    SolrEventBuilder solrEventBuilder = SolrEventBuilder.createStatisticsBuilder(this.getEventBuilderData(),
        this.getCoreName());

    Map<String, SolrInfoMBean> infoRegistry = this.solrCore.getInfoRegistry();
    this.addIndexStatistics(solrEventBuilder, infoRegistry);
    this.addUpdateStatistics(solrEventBuilder, infoRegistry);
    this.addReplicationStatistics(solrEventBuilder, infoRegistry);
    this.addCacheStatistics(solrEventBuilder, infoRegistry);

    this.sendEvent(solrEventBuilder.toEvent());
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
private void addReplicationStatistics(SolrEventBuilder solrEventBuilder, Map<String, SolrInfoMBean> infoRegistry) {
    SolrInfoMBean mBean = infoRegistry.get("/replication");
    if (mBean == null) {
        return;
    }

    NamedList<?> statistics = mBean.getStatistics();

    solrEventBuilder.setIndexSize(getIndexSize(statistics));
    solrEventBuilder.setGeneration(getLong(statistics, "generation"));
    solrEventBuilder.setIndexVersion(getLong(statistics, "indexVersion"));
    solrEventBuilder.setIsMaster(getBoolean(statistics, "isMaster"));
    solrEventBuilder.setIsSlave(getBoolean(statistics, "isSlave"));
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
private void addUpdateStatistics(SolrEventBuilder solrEventBuilder, Map<String, SolrInfoMBean> infoRegistry) {
    SolrInfoMBean mBean = infoRegistry.get("updateHandler");
    if (mBean == null) {
        return;
    }

    NamedList<?> statistics = mBean.getStatistics();

    solrEventBuilder.setUpdates(getLong(statistics, "cumulative_adds"));
    solrEventBuilder.setDeletes(getLong(statistics, "cumulative_deletesByQuery") + getLong(statistics, "cumulative_deletesById"));
    solrEventBuilder.setCommits(getLong(statistics, "commits"));
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
@Before
public void setUpBase() throws Exception
{
    coreContainer = new CoreContainer();
    coreDescriptor = new CoreDescriptor(coreContainer, "name", "instanceDir");
    when(resourceLoader.getCoreProperties()).thenReturn(new Properties());
    
    // SolrCore is final, we can't mock with mockito
    core = new SolrCore("name", coreDescriptor);
    FieldUtils.writeField(core, "updateHandler", updateHandler, true);
    FieldUtils.writeField(core, "resourceLoader", resourceLoader, true);
    infoRegistry = new HashMap<String, SolrInfoMBean>();
    FieldUtils.writeField(core, "infoRegistry", infoRegistry, true);
    reqHandlers = new RequestHandlers(core);
    reqHandlers.register("/select", selectRequestHandler);
    reqHandlers.register("/afts", aftsRequestHandler);
    FieldUtils.writeField(core, "reqHandlers", reqHandlers, true);

    Map<String, UpdateRequestProcessorChain> map = new HashMap<>();
    UpdateRequestProcessorFactory[] factories = new UpdateRequestProcessorFactory[] { runUpdateProcessorFactory };
    when(runUpdateProcessorFactory.getInstance(any(SolrQueryRequest.class), any(SolrQueryResponse.class),
                            any(UpdateRequestProcessor.class))).thenReturn(processor);
    UpdateRequestProcessorChain def = new UpdateRequestProcessorChain(factories, core);
    map.put(null, def);
    map.put("", def);
    FieldUtils.writeField(core, "updateProcessorChains", map, true);
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
private static SimpleOrderedMap<Object> getSolrInfoBeans( SolrCore core, boolean stats )
{
  SimpleOrderedMap<Object> list = new SimpleOrderedMap<Object>();
  for (SolrInfoMBean.Category cat : SolrInfoMBean.Category.values()) 
  {
    SimpleOrderedMap<Object> category = new SimpleOrderedMap<Object>();
    list.add( cat.name(), category );
    Map<String, SolrInfoMBean> reg = core.getInfoRegistry();
    for (Map.Entry<String,SolrInfoMBean> entry : reg.entrySet()) {
      SolrInfoMBean m = entry.getValue();
      if (m.getCategory() != cat) continue;

      String na = "Not Declared";
      SimpleOrderedMap<Object> info = new SimpleOrderedMap<Object>();
      category.add( entry.getKey(), info );

      info.add( "name",        (m.getName()       !=null ? m.getName()        : na) );
      info.add( "version",     (m.getVersion()    !=null ? m.getVersion()     : na) );
      info.add( "description", (m.getDescription()!=null ? m.getDescription() : na) );
      info.add( "source",      (m.getSource()     !=null ? m.getSource()      : na) );

      URL[] urls = m.getDocs();
      if ((urls != null) && (urls.length > 0)) {
        ArrayList<String> docs = new ArrayList<String>(urls.length);
        for( URL u : urls ) {
          docs.add( u.toExternalForm() );
        }
        info.add( "docs", docs );
      }

      if( stats ) {
        info.add( "stats", m.getStatistics() );
      }
    }
  }
  return list;
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
@Override
public final Category getCategory() {
  return SolrInfoMBean.Category.OTHER;
}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
/**
 * test that our assumptions about how caches are affected hold true
 */
public void testCacheImpacts() throws Exception {
  // cursor queryies can't live in the queryResultCache, but independent filters
  // should still be cached & reused

  // don't add in order of any field to ensure we aren't inadvertantly 
  // counting on internal docid ordering
  assertU(adoc("id", "9", "str", "c", "float", "-3.2", "int", "42"));
  assertU(adoc("id", "7", "str", "c", "float", "-3.2", "int", "-1976"));
  assertU(adoc("id", "2", "str", "c", "float", "-3.2", "int", "666"));
  assertU(adoc("id", "0", "str", "b", "float", "64.5", "int", "-42"));
  assertU(adoc("id", "5", "str", "b", "float", "64.5", "int", "2001"));
  assertU(adoc("id", "8", "str", "b", "float", "64.5", "int", "4055"));
  assertU(adoc("id", "6", "str", "a", "float", "64.5", "int", "7"));
  assertU(adoc("id", "1", "str", "a", "float", "64.5", "int", "7"));
  assertU(adoc("id", "4", "str", "a", "float", "11.1", "int", "6"));
  assertU(adoc("id", "3", "str", "a", "float", "11.1", "int", "3"));
  assertU(commit());

  final Collection<String> allFieldNames = getAllSortFieldNames();

  final SolrInfoMBean filterCacheStats 
    = h.getCore().getInfoRegistry().get("filterCache");
  assertNotNull(filterCacheStats);
  final SolrInfoMBean queryCacheStats 
    = h.getCore().getInfoRegistry().get("queryResultCache");
  assertNotNull(queryCacheStats);

  final long preQcIn = (Long) queryCacheStats.getStatistics().get("inserts");
  final long preFcIn = (Long) filterCacheStats.getStatistics().get("inserts");
  final long preFcHits = (Long) filterCacheStats.getStatistics().get("hits");

  SentinelIntSet ids = assertFullWalkNoDups
    (10, params("q", "*:*",
                "rows",""+TestUtil.nextInt(random(),1,11),
                "fq", "-id:[1 TO 2]",
                "fq", "-id:[6 TO 7]",
                "fl", "id",
                "sort", buildRandomSort(allFieldNames)));
  
  assertEquals(6, ids.size());

  final long postQcIn = (Long) queryCacheStats.getStatistics().get("inserts");
  final long postFcIn = (Long) filterCacheStats.getStatistics().get("inserts");
  final long postFcHits = (Long) filterCacheStats.getStatistics().get("hits");
  
  assertEquals("query cache inserts changed", preQcIn, postQcIn);
  // NOTE: use of pure negative filters causees "*:* to be tracked in filterCache
  assertEquals("filter cache did not grow correctly", 3, postFcIn-preFcIn);
  assertTrue("filter cache did not have any new cache hits", 0 < postFcHits-preFcHits);

}
 

import org.apache.solr.core.SolrInfoMBean; //导入依赖的package包/类
protected NamedList<NamedList<NamedList<Object>>> getMBeanInfo(SolrQueryRequest req) {

    NamedList<NamedList<NamedList<Object>>> cats = new NamedList<NamedList<NamedList<Object>>>();
    
    String[] requestedCats = req.getParams().getParams("cat");
    if (null == requestedCats || 0 == requestedCats.length) {
      for (SolrInfoMBean.Category cat : SolrInfoMBean.Category.values()) {
        cats.add(cat.name(), new SimpleOrderedMap<NamedList<Object>>());
      }
    } else {
      for (String catName : requestedCats) {
        cats.add(catName,new SimpleOrderedMap<NamedList<Object>>());
      }
    }
         
    Set<String> requestedKeys = arrayToSet(req.getParams().getParams("key"));
    
    Map<String, SolrInfoMBean> reg = req.getCore().getInfoRegistry();
    for (Map.Entry<String, SolrInfoMBean> entry : reg.entrySet()) {
      String key = entry.getKey();
      SolrInfoMBean m = entry.getValue();

      if ( ! ( requestedKeys.isEmpty() || requestedKeys.contains(key) ) ) continue;

      NamedList<NamedList<Object>> catInfo = cats.get(m.getCategory().name());
      if ( null == catInfo ) continue;

      NamedList<Object> mBeanInfo = new SimpleOrderedMap<Object>();
      mBeanInfo.add("class", m.getName());
      mBeanInfo.add("version", m.getVersion());
      mBeanInfo.add("description", m.getDescription());
      mBeanInfo.add("src", m.getSource());
      
      // Use an external form
      URL[] urls = m.getDocs();
      if(urls!=null) {
        List<String> docs = new ArrayList<String>(urls.length);
        for(URL url : urls) {
          docs.add(url.toExternalForm());
        }
        mBeanInfo.add("docs", docs);
      }
      
      if (req.getParams().getFieldBool(key, "stats", false))
        mBeanInfo.add("stats", m.getStatistics());
      
      catInfo.add(key, mBeanInfo);
    }
    return cats;
  }