Java CloudSolrClient.connect方法代码示例

import org.apache.solr.client.solrj.impl.CloudSolrClient; //导入方法依赖的package包/类
@BeforeClass
public static void startCluster() throws Exception {
  File solrXml = new File("src/test/resources/solr.xml");

  tempDir = Files.createTempDirectory("MiniSolrCloudCluster");

  try {
    cluster = new MiniSolrCloudCluster(1, null, tempDir, MiniSolrCloudCluster.DEFAULT_CLOUD_SOLR_XML, null, null);
  } catch (Exception exc) {
    log.error("Failed to initialize a MiniSolrCloudCluster due to: " + exc, exc);
    throw exc;
  }

  cloudSolrClient = new CloudSolrClient(cluster.getZkServer().getZkAddress(), true);
  cloudSolrClient.connect();

  assertTrue(!cloudSolrClient.getZkStateReader().getClusterState().getLiveNodes().isEmpty());
}
 

import org.apache.solr.client.solrj.impl.CloudSolrClient; //导入方法依赖的package包/类
public SolrCloudFixture(String solrHome) throws Exception {
    String xml = IOHelper.loadText(new FileInputStream(new File(solrHome, "solr-no-core.xml")));
    miniCluster = new MiniSolrCloudCluster(1, "/solr", new File("target/tmp").toPath(), xml, null, null);
    String zkAddr = miniCluster.getZkServer().getZkAddress();
    String zkHost = miniCluster.getZkServer().getZkHost();

    buildZooKeeper(zkHost, zkAddr, new File(solrHome), "solrconfig.xml", "schema.xml");
    List<JettySolrRunner> jettys = miniCluster.getJettySolrRunners();
    for (JettySolrRunner jetty : jettys) {
        if (!jetty.isRunning()) {
            log.warn("JETTY NOT RUNNING!");
        } else {
            log.info("JETTY RUNNING AT " + jetty.getBaseUrl() + " PORT " + jetty.getLocalPort());
        }
    }

    solrClient = new CloudSolrClient(zkAddr, true);
    solrClient.connect();

    createCollection(solrClient, "collection1", 1, 1, "conf1");
    Thread.sleep(1000); // takes some time to setup the collection...
                        // otherwise you'll get no live solr servers
    solrClient.setDefaultCollection("collection1");

    SolrInputDocument doc = new SolrInputDocument();
    doc.setField("id", "1");

    solrClient.add(doc);
    solrClient.commit();
}
 

import org.apache.solr.client.solrj.impl.CloudSolrClient; //导入方法依赖的package包/类
public FakeRetrySolrServer(String zkString, boolean alwaysFail, int retries) {
  CloudSolrClient client  = new CloudSolrClient(zkString);
  client.setDefaultCollection(DEFAULT_COLLECTION);
  client.connect();
  this.delegate = client;
  this.alwaysFail = alwaysFail;
  this.retries = retries;
}
 

import org.apache.solr.client.solrj.impl.CloudSolrClient; //导入方法依赖的package包/类
public static CloudSolrClient getCloudSolrClient(String url) throws MalformedURLException {
  CloudSolrClient sc = new CloudSolrClient(url.replace('|', ','));
  sc.setParallelUpdates(true);
  sc.connect();
  return sc;
}
 

import org.apache.solr.client.solrj.impl.CloudSolrClient; //导入方法依赖的package包/类
@Override
public InputSplit[] getSplits(JobConf job, int numSplits) throws IOException {
  this.conf = job;
  boolean isZk = false;
  String connectionUri = job.get(SOLR_ZKHOST);
  String collection = job.get(SOLR_COLLECTION);
  List<String> shards = null;

  if (collection == null) {
    throw new IOException(SOLR_COLLECTION + " not provided or is empty");
  }

  if (connectionUri != null && connectionUri != "") {
    isZk = true;
    CloudSolrClient solrCloud = null;
    try {
      solrCloud = new CloudSolrClient.Builder()
          .withZkHost(connectionUri)
          .build();
      solrCloud.setDefaultCollection(collection);
      solrCloud.connect();
      // first get a list of replicas to query for this collection
      shards = buildShardList(solrCloud, collection);
    } catch (Exception ex) {
      log.warn("Error getting the Shard: " + ex.getMessage());
    } finally {
      if (solrCloud != null) {
        solrCloud.close();
      }
    }

  } else {
    connectionUri = job.get(SOLR_SERVER_URL);
  }

  if (connectionUri == null || connectionUri == "") {
    throw new IOException(SOLR_ZKHOST + " nor " + SOLR_SERVER_URL + " provided or is empty");
  }

  InputSplit[] splits = null;
  if (shards != null) {
    //Create more splits.
    splits = new InputSplit[shards.size()];
    for (int i = 0; i < shards.size(); i++) {
      splits[i] = new LWInputSplit(isZk, true, shards.get(i));
    }
  } else {
    splits = new InputSplit[] { new LWInputSplit(isZk, false, connectionUri) };
  }

  return splits;
}
 

import org.apache.solr.client.solrj.impl.CloudSolrClient; //导入方法依赖的package包/类
/**
 * Returns the list of shards of the default collection.
 *
 * @param zkHost            ZooKeeper URL
 * @param chronixCollection Solr collection name for chronix time series data
 * @return the list of shards of the default collection
 */
public List<String> getShardList(String zkHost, String chronixCollection) throws IOException {

    CloudSolrClient cloudSolrClient = new CloudSolrClient(zkHost);
    List<String> shards = new ArrayList<>();

    try {
        cloudSolrClient.connect();

        ZkStateReader zkStateReader = cloudSolrClient.getZkStateReader();

        ClusterState clusterState = zkStateReader.getClusterState();

        String[] collections;
        if (clusterState.hasCollection(chronixCollection)) {
            collections = new String[]{chronixCollection};
        } else {
            // might be a collection alias?
            Aliases aliases = zkStateReader.getAliases();
            String aliasedCollections = aliases.getCollectionAlias(chronixCollection);
            if (aliasedCollections == null)
                throw new IllegalArgumentException("Collection " + chronixCollection + " not found!");
            collections = aliasedCollections.split(",");
        }

        Set<String> liveNodes = clusterState.getLiveNodes();
        Random random = new Random(5150);

        for (String coll : collections) {
            for (Slice slice : clusterState.getSlices(coll)) {
                List<String> replicas = new ArrayList<>();
                for (Replica r : slice.getReplicas()) {
                    if (r.getState().equals(Replica.State.ACTIVE)) {
                        ZkCoreNodeProps replicaCoreProps = new ZkCoreNodeProps(r);
                        if (liveNodes.contains(replicaCoreProps.getNodeName()))
                            replicas.add(replicaCoreProps.getCoreUrl());
                    }
                }
                int numReplicas = replicas.size();
                if (numReplicas == 0)
                    throw new IllegalStateException("Shard " + slice.getName() + " in collection " +
                            coll + " does not have any active replicas!");

                String replicaUrl = (numReplicas == 1) ? replicas.get(0) : replicas.get(random.nextInt(replicas.size()));
                shards.add(replicaUrl);
            }
        }
    } finally {
        cloudSolrClient.close();
    }

    return shards;
}