/**
* 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.servlet;
import java.io.File;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import org.apache.solr.core.Config;
import org.apache.solr.core.SolrConfig;
import org.apache.solr.core.SolrCore;
import org.apache.solr.core.SolrException;
import org.apache.solr.request.MapSolrParams;
import org.apache.solr.request.QueryResponseWriter;
import org.apache.solr.request.SolrParams;
import org.apache.solr.request.SolrQueryRequest;
import org.apache.solr.request.SolrQueryResponse;
import org.apache.solr.request.SolrRequestHandler;
import org.apache.solr.schema.IndexSchema;
import org.apache.solr.util.ContentStream;
import org.apache.solr.util.ContentStreamBase;
/**
* DirectSolrConnection provides an interface to solr that is similar to
* the the HTTP interface, but does not require an HTTP connection.
*
* This class is designed to be as simple as possible and alow for more flexibility
* in how you interface to solr.
*
* @author ryan
* @version $Id: DirectSolrConnection.java 542679 2007-05-29 22:28:21Z ryan $
* @since solr 1.2
*/
public class DirectSolrConnection
{
final SolrCore core;
final SolrRequestParsers parser;
/**
* Initialize using the static singleton SolrCore.getSolrCore().
*/
public DirectSolrConnection()
{
core = SolrCore.getSolrCore();
parser = new SolrRequestParsers( core, SolrConfig.config );
}
/**
* Initialize using an explicit SolrCore
*/
public DirectSolrConnection( SolrCore c )
{
core = c;
parser = new SolrRequestParsers( core, SolrConfig.config );
}
/**
* This constructor is designed to make it easy for JNI embedded applications
* to setup the entire solr environment with a simple interface. It takes three parameters:
*
* <code>instanceDir:</code> The solr instance directory. If null, it will check the standard
* places first (JNDI,properties,"solr" directory)
*
* <code>dataDir:</code> where the index is stored.
*
* <code>loggingPath:</code> Path to a java.util.logging.config.file. If the path represents
* an absolute path or is relative to the CWD, it will use that. Next it will try a path
* relative to the instanceDir. If none of these files exist, it will error.
*/
public DirectSolrConnection( String instanceDir, String dataDir, String loggingPath )
{
// If a loggingPath is specified, try using that (this needs to happen first)
if( loggingPath != null ) {
File loggingConfig = new File( loggingPath );
if( !loggingConfig.exists() && instanceDir != null ) {
loggingConfig = new File( new File(instanceDir), loggingPath );
}
if( loggingConfig.exists() ) {
System.setProperty("java.util.logging.config.file", loggingConfig.getAbsolutePath() );
}
else {
throw new SolrException( SolrException.ErrorCode.SERVER_ERROR, "can not find logging file: "+loggingConfig );
}
}
// Set the instance directory
if( instanceDir != null ) {
if( Config.isInstanceDirInitialized() ) {
String dir = Config.getInstanceDir();
if( !dir.equals( instanceDir ) ) {
throw new SolrException( SolrException.ErrorCode.SERVER_ERROR, "already initalized: "+dir );
}
}
Config.setInstanceDir( instanceDir );
}
// If the Data directory is specified, initalize SolrCore directly
if( dataDir != null ) {
core = new SolrCore( dataDir, new IndexSchema(instanceDir+"/conf/schema.xml"));
}
else {
core = SolrCore.getSolrCore();
}
parser = new SolrRequestParsers( core, SolrConfig.config );
}
/**
* For example:
*
* String json = solr.request( "/select?qt=dismax&wt=json&q=...", null );
* String xml = solr.request( "/update", "<add><doc><field ..." );
*
*/
public String request( String pathAndParams, String body ) throws Exception
{
String path = null;
SolrParams params = null;
int idx = pathAndParams.indexOf( '?' );
if( idx > 0 ) {
path = pathAndParams.substring( 0, idx );
params = SolrRequestParsers.parseQueryString( pathAndParams.substring(idx+1) );
}
else {
path= pathAndParams;
params = new MapSolrParams( new HashMap<String, String>() );
}
// Extract the handler from the path or params
SolrRequestHandler handler = core.getRequestHandler( path );
if( handler == null ) {
if( "/select".equals( path ) || "/select/".equalsIgnoreCase( path) ) {
String qt = params.get( SolrParams.QT );
handler = core.getRequestHandler( qt );
if( handler == null ) {
throw new SolrException( SolrException.ErrorCode.BAD_REQUEST, "unknown handler: "+qt);
}
}
}
if( handler == null ) {
throw new SolrException( SolrException.ErrorCode.BAD_REQUEST, "unknown handler: "+path );
}
// Make a stream for the 'body' content
List<ContentStream> streams = new ArrayList<ContentStream>( 1 );
if( body != null && body.length() > 0 ) {
streams.add( new ContentStreamBase.StringStream( body ) );
}
SolrQueryRequest req = parser.buildRequestFrom( params, streams );
SolrQueryResponse rsp = new SolrQueryResponse();
core.execute( handler, req, rsp );
if( rsp.getException() != null ) {
throw rsp.getException();
}
// Now write it out
QueryResponseWriter responseWriter = core.getQueryResponseWriter(req);
StringWriter out = new StringWriter();
responseWriter.write(out, req, rsp);
return out.toString();
}
}