Java Code Examples for org.apache.solr.client.solrj.request.CollectionAdminRequest#AsyncCollectionAdminRequest
The following examples show how to use
org.apache.solr.client.solrj.request.CollectionAdminRequest#AsyncCollectionAdminRequest .
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source File: ReplaceNodeTest.java From lucene-solr with Apache License 2.0 | 6 votes |
public static CollectionAdminRequest.AsyncCollectionAdminRequest createReplaceNodeRequest(String sourceNode, String targetNode, Boolean parallel) { if (random().nextBoolean()) { return new CollectionAdminRequest.ReplaceNode(sourceNode, targetNode).setParallel(parallel); } else { // test back compat with old param names // todo remove in solr 8.0 return new CollectionAdminRequest.AsyncCollectionAdminRequest(CollectionParams.CollectionAction.REPLACENODE) { @Override public SolrParams getParams() { ModifiableSolrParams params = (ModifiableSolrParams) super.getParams(); params.set("source", sourceNode); params.setNonNull("target", targetNode); if (parallel != null) params.set("parallel", parallel.toString()); return params; } }; } }
Example 2
Source File: TestSimExecutePlanAction.java From lucene-solr with Apache License 2.0 | 4 votes |
@Test // commented out on: 24-Dec-2018 @LuceneTestCase.BadApple(bugUrl="https://issues.apache.org/jira/browse/SOLR-12028") // 28-June-2018 public void testExecute() throws Exception { SolrClient solrClient = cluster.simGetSolrClient(); String collectionName = "testExecute"; CollectionAdminRequest.Create create = CollectionAdminRequest.createCollection(collectionName, "conf", 1, 2); create.setMaxShardsPerNode(1); create.process(solrClient); if (log.isInfoEnabled()) { log.info("Collection ready after {} ms", CloudUtil.waitForState(cluster, collectionName, 120, TimeUnit.SECONDS, CloudUtil.clusterShape(1, 2, false, true))); } String sourceNodeName = cluster.getSimClusterStateProvider().simGetRandomNode(); ClusterState clusterState = cluster.getClusterStateProvider().getClusterState(); DocCollection docCollection = clusterState.getCollection(collectionName); List<Replica> replicas = docCollection.getReplicas(sourceNodeName); assertNotNull(replicas); assertFalse(replicas.isEmpty()); List<String> otherNodes = cluster.getClusterStateProvider().getLiveNodes().stream() .filter(node -> !node.equals(sourceNodeName)).collect(Collectors.toList()); assertFalse(otherNodes.isEmpty()); String survivor = otherNodes.get(0); try (ExecutePlanAction action = new ExecutePlanAction()) { action.configure(cluster.getLoader(), cluster, Collections.singletonMap("name", "execute_plan")); // used to signal if we found that ExecutePlanAction did in fact create the right znode before executing the operation AtomicBoolean znodeCreated = new AtomicBoolean(false); CollectionAdminRequest.AsyncCollectionAdminRequest moveReplica = new CollectionAdminRequest.MoveReplica(collectionName, replicas.get(0).getName(), survivor); CollectionAdminRequest.AsyncCollectionAdminRequest mockRequest = new CollectionAdminRequest.AsyncCollectionAdminRequest(CollectionParams.CollectionAction.OVERSEERSTATUS) { @Override public void setAsyncId(String asyncId) { super.setAsyncId(asyncId); String parentPath = ZkStateReader.SOLR_AUTOSCALING_TRIGGER_STATE_PATH + "/xyz/execute_plan"; try { if (cluster.getDistribStateManager().hasData(parentPath)) { java.util.List<String> children = cluster.getDistribStateManager().listData(parentPath); if (!children.isEmpty()) { String child = children.get(0); VersionedData data = cluster.getDistribStateManager().getData(parentPath + "/" + child); @SuppressWarnings({"rawtypes"}) Map m = (Map) Utils.fromJSON(data.getData()); if (m.containsKey("requestid")) { znodeCreated.set(m.get("requestid").equals(asyncId)); } } } } catch (Exception e) { throw new RuntimeException(e); } } }; List<CollectionAdminRequest.AsyncCollectionAdminRequest> operations = Lists.asList(moveReplica, new CollectionAdminRequest.AsyncCollectionAdminRequest[]{mockRequest}); NodeLostTrigger.NodeLostEvent nodeLostEvent = new NodeLostTrigger.NodeLostEvent(TriggerEventType.NODELOST, "mock_trigger_name", Collections.singletonList(SIM_TIME_SOURCE.getTimeNs()), Collections.singletonList(sourceNodeName), CollectionParams.CollectionAction.MOVEREPLICA.toLower()); ActionContext actionContext = new ActionContext(cluster, null, new HashMap<>(Collections.singletonMap("operations", operations))); action.process(nodeLostEvent, actionContext); // assertTrue("ExecutePlanAction should have stored the requestid in ZK before executing the request", znodeCreated.get()); @SuppressWarnings({"unchecked"}) List<NamedList<Object>> responses = (List<NamedList<Object>>) actionContext.getProperty("responses"); assertNotNull(responses); assertEquals(2, responses.size()); NamedList<Object> response = responses.get(0); assertNull(response.get("failure")); assertNotNull(response.get("success")); } if (log.isInfoEnabled()) { log.info("Collection ready after {} ms", CloudUtil.waitForState(cluster, collectionName, 300, TimeUnit.SECONDS, CloudUtil.clusterShape(1, 2, false, true))); } }
Example 3
Source File: ExecutePlanActionTest.java From lucene-solr with Apache License 2.0 | 4 votes |
@Test public void testExecute() throws Exception { CloudSolrClient solrClient = cluster.getSolrClient(); String collectionName = "testExecute"; CollectionAdminRequest.Create create = CollectionAdminRequest.createCollection(collectionName, "conf", 1, 2); create.setMaxShardsPerNode(1); create.process(solrClient); cluster.waitForActiveCollection(collectionName, 1, 2); waitForState("Timed out waiting for replicas of new collection to be active", collectionName, clusterShape(1, 2)); JettySolrRunner sourceNode = cluster.getRandomJetty(random()); String sourceNodeName = sourceNode.getNodeName(); ClusterState clusterState = solrClient.getZkStateReader().getClusterState(); DocCollection docCollection = clusterState.getCollection(collectionName); List<Replica> replicas = docCollection.getReplicas(sourceNodeName); assertNotNull(replicas); assertFalse(replicas.isEmpty()); List<JettySolrRunner> otherJetties = cluster.getJettySolrRunners().stream() .filter(jettySolrRunner -> jettySolrRunner != sourceNode).collect(Collectors.toList()); assertFalse(otherJetties.isEmpty()); JettySolrRunner survivor = otherJetties.get(0); try (ExecutePlanAction action = new ExecutePlanAction()) { action.configure(loader, cloudManager, Collections.singletonMap("name", "execute_plan")); // used to signal if we found that ExecutePlanAction did in fact create the right znode before executing the operation AtomicBoolean znodeCreated = new AtomicBoolean(false); CollectionAdminRequest.AsyncCollectionAdminRequest moveReplica = new CollectionAdminRequest.MoveReplica(collectionName, replicas.get(0).getName(), survivor.getNodeName()); CollectionAdminRequest.AsyncCollectionAdminRequest mockRequest = new CollectionAdminRequest.AsyncCollectionAdminRequest(CollectionParams.CollectionAction.OVERSEERSTATUS) { @Override public void setAsyncId(String asyncId) { super.setAsyncId(asyncId); String parentPath = ZkStateReader.SOLR_AUTOSCALING_TRIGGER_STATE_PATH + "/xyz/execute_plan"; try { if (zkClient().exists(parentPath, true)) { java.util.List<String> children = zkClient().getChildren(parentPath, null, true); if (!children.isEmpty()) { String child = children.get(0); byte[] data = zkClient().getData(parentPath + "/" + child, null, null, true); @SuppressWarnings({"rawtypes"}) Map m = (Map) Utils.fromJSON(data); if (m.containsKey("requestid")) { znodeCreated.set(m.get("requestid").equals(asyncId)); } } } } catch (Exception e) { throw new RuntimeException(e); } } }; List<CollectionAdminRequest.AsyncCollectionAdminRequest> operations = Lists.asList(moveReplica, new CollectionAdminRequest.AsyncCollectionAdminRequest[]{mockRequest}); NodeLostTrigger.NodeLostEvent nodeLostEvent = new NodeLostTrigger.NodeLostEvent (TriggerEventType.NODELOST, "mock_trigger_name", Collections.singletonList(cloudManager.getTimeSource().getTimeNs()), Collections.singletonList(sourceNodeName), CollectionParams.CollectionAction.MOVEREPLICA.toLower()); ActionContext actionContext = new ActionContext(survivor.getCoreContainer().getZkController().getSolrCloudManager(), null, new HashMap<>(Collections.singletonMap("operations", operations))); action.process(nodeLostEvent, actionContext); // assertTrue("ExecutePlanAction should have stored the requestid in ZK before executing the request", znodeCreated.get()); @SuppressWarnings({"unchecked"}) List<NamedList<Object>> responses = (List<NamedList<Object>>) actionContext.getProperty("responses"); assertNotNull(responses); assertEquals(2, responses.size()); NamedList<Object> response = responses.get(0); assertNull(response.get("failure")); assertNotNull(response.get("success")); } waitForState("Timed out waiting for replicas of new collection to be active", collectionName, clusterShape(1, 2)); }
Example 4
Source File: MigrateRouteKeyTest.java From lucene-solr with Apache License 2.0 | 4 votes |
protected void invokeCollectionMigration(CollectionAdminRequest.AsyncCollectionAdminRequest request) throws IOException, SolrServerException, InterruptedException { request.processAndWait(cluster.getSolrClient(), 60000); }