Java tutorial
/** * 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.client.solrj; import junit.framework.Assert; import org.apache.commons.httpclient.HttpClient; import org.apache.commons.httpclient.MultiThreadedHttpConnectionManager; import org.apache.commons.io.FileUtils; import org.apache.lucene.util.LuceneTestCase; import org.apache.solr.SolrTestCaseJ4; import org.apache.solr.client.solrj.embedded.JettySolrRunner; import org.apache.solr.client.solrj.impl.CommonsHttpSolrServer; import org.apache.solr.client.solrj.impl.LBHttpSolrServer; import org.apache.solr.client.solrj.response.QueryResponse; import org.apache.solr.client.solrj.response.UpdateResponse; import org.apache.solr.common.SolrInputDocument; import org.apache.solr.util.AbstractSolrTestCase; import java.io.File; import java.io.IOException; import java.util.ArrayList; import java.util.HashSet; import java.util.List; import java.util.Set; /** * Test for LBHttpSolrServer * * @version $Id: TestLBHttpSolrServer.java 1053509 2010-12-29 02:55:21Z rmuir $ * @since solr 1.4 */ public class TestLBHttpSolrServer extends LuceneTestCase { SolrInstance[] solr = new SolrInstance[3]; HttpClient httpClient; public void setUp() throws Exception { super.setUp(); httpClient = new HttpClient(new MultiThreadedHttpConnectionManager()); httpClient.getParams().setParameter("http.connection.timeout", new Integer(1000)); for (int i = 0; i < solr.length; i++) { solr[i] = new SolrInstance("solr" + i, 0); solr[i].setUp(); solr[i].startJetty(); addDocs(solr[i]); } } private void addDocs(SolrInstance solrInstance) throws IOException, SolrServerException { List<SolrInputDocument> docs = new ArrayList<SolrInputDocument>(); for (int i = 0; i < 10; i++) { SolrInputDocument doc = new SolrInputDocument(); doc.addField("id", i); doc.addField("name", solrInstance.name); docs.add(doc); } CommonsHttpSolrServer solrServer = new CommonsHttpSolrServer(solrInstance.getUrl(), httpClient); UpdateResponse resp = solrServer.add(docs); assertEquals(0, resp.getStatus()); resp = solrServer.commit(); assertEquals(0, resp.getStatus()); } @Override public void tearDown() throws Exception { for (SolrInstance aSolr : solr) { aSolr.tearDown(); } super.tearDown(); } public void testSimple() throws Exception { String[] s = new String[solr.length]; for (int i = 0; i < solr.length; i++) { s[i] = solr[i].getUrl(); } LBHttpSolrServer lbHttpSolrServer = new LBHttpSolrServer(httpClient, s); lbHttpSolrServer.setAliveCheckInterval(500); SolrQuery solrQuery = new SolrQuery("*:*"); Set<String> names = new HashSet<String>(); QueryResponse resp = null; for (String value : s) { resp = lbHttpSolrServer.query(solrQuery); assertEquals(10, resp.getResults().getNumFound()); names.add(resp.getResults().get(0).getFieldValue("name").toString()); } assertEquals(3, names.size()); // Kill a server and test again solr[1].jetty.stop(); solr[1].jetty = null; names.clear(); for (String value : s) { resp = lbHttpSolrServer.query(solrQuery); assertEquals(10, resp.getResults().getNumFound()); names.add(resp.getResults().get(0).getFieldValue("name").toString()); } assertEquals(2, names.size()); assertFalse(names.contains("solr1")); // Start the killed server once again solr[1].startJetty(); // Wait for the alive check to complete Thread.sleep(1200); names.clear(); for (String value : s) { resp = lbHttpSolrServer.query(solrQuery); assertEquals(10, resp.getResults().getNumFound()); names.add(resp.getResults().get(0).getFieldValue("name").toString()); } assertEquals(3, names.size()); } public void testTwoServers() throws Exception { LBHttpSolrServer lbHttpSolrServer = new LBHttpSolrServer(httpClient, solr[0].getUrl(), solr[1].getUrl()); lbHttpSolrServer.setAliveCheckInterval(500); SolrQuery solrQuery = new SolrQuery("*:*"); Set<String> names = new HashSet<String>(); QueryResponse resp = null; solr[0].jetty.stop(); solr[0].jetty = null; resp = lbHttpSolrServer.query(solrQuery); String name = resp.getResults().get(0).getFieldValue("name").toString(); Assert.assertEquals("solr1", name); resp = lbHttpSolrServer.query(solrQuery); name = resp.getResults().get(0).getFieldValue("name").toString(); Assert.assertEquals("solr1", name); solr[1].jetty.stop(); solr[1].jetty = null; solr[0].startJetty(); Thread.sleep(1200); try { resp = lbHttpSolrServer.query(solrQuery); } catch (SolrServerException e) { // try again after a pause in case the error is lack of time to start server Thread.sleep(3000); resp = lbHttpSolrServer.query(solrQuery); } name = resp.getResults().get(0).getFieldValue("name").toString(); Assert.assertEquals("solr0", name); } public void testReliability() throws Exception { String[] s = new String[solr.length]; for (int i = 0; i < solr.length; i++) { s[i] = solr[i].getUrl(); } HttpClient myHttpClient = new HttpClient(new MultiThreadedHttpConnectionManager()); myHttpClient.getParams().setParameter("http.connection.timeout", new Integer(100)); myHttpClient.getParams().setParameter("http.socket.timeout", new Integer(100)); LBHttpSolrServer lbHttpSolrServer = new LBHttpSolrServer(myHttpClient, s); lbHttpSolrServer.setAliveCheckInterval(500); // Kill a server and test again solr[1].jetty.stop(); solr[1].jetty = null; // query the servers for (String value : s) lbHttpSolrServer.query(new SolrQuery("*:*")); // Start the killed server once again solr[1].startJetty(); // Wait for the alive check to complete waitForServer(30000, lbHttpSolrServer, 3, "solr1"); } // wait maximum ms for serverName to come back up private void waitForServer(int maximum, LBHttpSolrServer server, int nServers, String serverName) throws Exception { long endTime = System.currentTimeMillis() + maximum; while (System.currentTimeMillis() < endTime) { QueryResponse resp = server.query(new SolrQuery("*:*")); String name = resp.getResults().get(0).getFieldValue("name").toString(); if (name.equals(serverName)) return; } } private class SolrInstance { String name; File homeDir; File dataDir; File confDir; int port; JettySolrRunner jetty; public SolrInstance(String name, int port) { this.name = name; this.port = port; } public String getHomeDir() { return homeDir.toString(); } public String getUrl() { return "http://localhost:" + port + "/solr"; } public String getSchemaFile() { return "." + File.separator + "solr" + File.separator + "conf" + File.separator + "schema-replication1.xml"; } public String getConfDir() { return confDir.toString(); } public String getDataDir() { return dataDir.toString(); } public String getSolrConfigFile() { String fname = ""; fname = "." + File.separator + "solr" + File.separator + "conf" + File.separator + "solrconfig-slave1.xml"; return fname; } public void setUp() throws Exception { File home = new File(SolrTestCaseJ4.TEMP_DIR, getClass().getName() + "-" + System.currentTimeMillis()); homeDir = new File(home, name); dataDir = new File(homeDir, "data"); confDir = new File(homeDir, "conf"); homeDir.mkdirs(); dataDir.mkdirs(); confDir.mkdirs(); File f = new File(confDir, "solrconfig.xml"); FileUtils.copyFile(SolrTestCaseJ4.getFile(getSolrConfigFile()), f); f = new File(confDir, "schema.xml"); FileUtils.copyFile(SolrTestCaseJ4.getFile(getSchemaFile()), f); } public void tearDown() throws Exception { try { jetty.stop(); } catch (Exception e) { } AbstractSolrTestCase.recurseDelete(homeDir); } public void startJetty() throws Exception { jetty = new JettySolrRunner("/solr", port); System.setProperty("solr.solr.home", getHomeDir()); System.setProperty("solr.data.dir", getDataDir()); jetty.start(); int newPort = jetty.getLocalPort(); if (port != 0 && newPort != port) { fail("TESTING FAILURE: could not grab requested port."); } this.port = newPort; // System.out.println("waiting........."); // Thread.sleep(5000); } } }