Java CloudSolrClient.close方法代码示例

import org.apache.solr.client.solrj.impl.CloudSolrClient; //导入方法依赖的package包/类
private static void setupSolrDocuments() throws SolrServerException, IOException {
	Builder builder = new CloudSolrClient.Builder();
	builder.withZkHost(
			Arrays.asList(new String[] { "192.168.204.181:2181,192.168.204.182:2181,192.168.204.183:2181" }));
	CloudSolrClient client = builder.build();
	client.setDefaultCollection("sql");
	client.deleteByQuery("*:*");
	insertDocument(client, 1, "bluejoe", 38);
	insertDocument(client, 2, "even", 35);
	insertDocument(client, 3, "alex", 8);
	client.commit();
	client.close();
}
 

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;
}