constellio——基于solr的开源搜索引擎系统源码研究（二）

因为constellio系统的类比较多，先从搜索引擎最本质的功能模块索引与检索入手，看constellio系统实现索引与检索功能的源码

首先是获取solr上下文，下面是SolrCoreContext.java源码

/**
* Constellio, Open Source Enterprise Search
* Copyright (C) 2010 DocuLibre inc.
*
* This copyrighted material is made available to anyone wishing to use, modify,
* copy, or redistribute it subject to the terms and conditions of the GNU
* Lesser General Public License, as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
* for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this distribution; if not, write to:
* Free Software Foundation, Inc.
* 51 Franklin Street, Fifth Floor
* Boston, MA  02110-1301  USA
*/
package com.doculibre.constellio.solr.context;

import java.io.File;
import java.io.IOException;
import java.net.MalformedURLException;
import java.util.HashMap;
import java.util.Map;

import javax.xml.parsers.ParserConfigurationException;

import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.embedded.EmbeddedSolrServer;
import org.apache.solr.client.solrj.impl.CommonsHttpSolrServer;
import org.apache.solr.core.CoreContainer;
import org.apache.solr.core.SolrCore;
import org.xml.sax.SAXException;

import com.doculibre.constellio.entities.RecordCollection;
import com.doculibre.constellio.opensearch.OpenSearchSolrServer;
import com.doculibre.constellio.services.RecordCollectionServices;
import com.doculibre.constellio.utils.ClasspathUtils;
import com.doculibre.constellio.utils.ConstellioSpringUtils;

public class SolrCoreContext {

private static File solrCoresRootDir;

private static CoreContainer cores = new CoreContainer();

public static final String DEFAULT_COLLECTION_NAME = "constellio_default";

private static Map<String, SolrServer> coreServers = new HashMap<String, SolrServer>();

public static synchronized void init() {
if (solrCoresRootDir == null) {
solrCoresRootDir = ClasspathUtils.getCollectionsRootDir();
}
init(solrCoresRootDir);
}

public static synchronized void init(File solrCoresRootDir) {
SolrCoreContext.solrCoresRootDir = solrCoresRootDir;
try {
File solrXml = new File(solrCoresRootDir, "solr.xml");
cores.load(solrCoresRootDir.getPath(), solrXml);

for (String coreName : cores.getCoreNames()) {
// Default
setEmbeddedSolrServer(coreName);
}
} catch (ParserConfigurationException e) {
throw new RuntimeException(e);
} catch (IOException e) {
throw new RuntimeException(e);
} catch (SAXException e) {
throw new RuntimeException(e);
}
}

public static synchronized void shutdown() {
if (cores != null) {
cores.shutdown();
}
}

public static synchronized void clear() {
solrCoresRootDir = null;
cores = new CoreContainer();
coreServers = new HashMap<String, SolrServer>();
}

public static File getSolrCoresRootDir() {
return solrCoresRootDir;
}

public static File getSolrCoreRootDir(String coreName) {
String coreDirName;
if (coreName != null) {
coreDirName = coreName;
} else {
coreDirName = DEFAULT_COLLECTION_NAME;
}
return new File(getSolrCoresRootDir(), coreDirName);
}

public static File getSolrCoreIndexDir(String coreName) {
File solrCoreRootDir = getSolrCoreRootDir(coreName);
return new File(solrCoreRootDir, "data" + File.separator + "index");
}

public static CoreContainer getCores() {
return cores;
}

public static SolrServer getSolrServer(String coreName) {
RecordCollectionServices collectionServices = ConstellioSpringUtils.getRecordCollectionServices();
RecordCollection collection = collectionServices.get(coreName);
if (collection != null && collection.isOpenSearch()) {
return new OpenSearchSolrServer();
} else {
return coreServers.get(coreName);
}
}

//Pour tests sans collection
public static SolrServer getSolrServerUtil(String coreName) {
return coreServers.get(coreName);
}

public static void registerSolrServer(String coreName, SolrServer solrServer) {
coreServers.put(coreName, solrServer);
}

public static void setEmbeddedSolrServer(String coreName) {
registerSolrServer(coreName, new EmbeddedSolrServer(cores, coreName));
}

public static void setHttpSolrServer(String coreName, String url) {
try {
registerSolrServer(coreName, new CommonsHttpSolrServer(url));
} catch (MalformedURLException e) {
throw new RuntimeException(e);
}
}

public static void removeCore(String coreName) {
SolrCore core = cores.getCore(coreName);
if (core != null) {
core.closeSearcher();
core.close();
}
cores.remove(coreName);
coreServers.remove(coreName);
}

}

下面是类路径工具类ClasspathUtils.java

/**
* Constellio, Open Source Enterprise Search
* Copyright (C) 2010 DocuLibre inc.
*
* This copyrighted material is made available to anyone wishing to use, modify,
* copy, or redistribute it subject to the terms and conditions of the GNU
* Lesser General Public License, as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
* for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this distribution; if not, write to:
* Free Software Foundation, Inc.
* 51 Franklin Street, Fifth Floor
* Boston, MA  02110-1301  USA
*/
package com.doculibre.constellio.utils;

import java.io.File;
import java.net.URISyntaxException;
import java.net.URL;

import com.doculibre.constellio.entities.RecordCollection;

public class ClasspathUtils {

public static File getClassesDir() {
URL classesOrBinDirURL = ClasspathUtils.class.getClassLoader().getResource("");
File classesOrBinDir;
try {
classesOrBinDir = new File(classesOrBinDirURL.toURI());
} catch (URISyntaxException e) {
throw new RuntimeException(e);
}
return classesOrBinDir;
}

public static File getWebinfDir() {
File webinfDir;
File classesOrBinDir = getClassesDir();
if (classesOrBinDir.getName().equals("bin")) {
File projectDir = classesOrBinDir.getParentFile();
File webContentDir = new File(projectDir, "WebContent");
webinfDir = new File(webContentDir, "WEB-INF");
} else {
webinfDir = classesOrBinDir.getParentFile();
}
return webinfDir;
}

/**
* For a web application, it's the web-inf dir. For tests, it's a temporary folder.
* We don't want unit test to modify contellio project files
*/
public static File getWorkDir() {
File workDir;
File classesOrBinDir = getClassesDir();
if (classesOrBinDir.getName().equals("bin")) {
File webInfDir = getWebinfDir();
workDir = new File(classesOrBinDir, "temp");
if (!workDir.exists()) {
workDir.mkdir();
}
//default Solr core
File constellioDir = new File(workDir, "constellio");
if (!constellioDir.exists()) {
FileUtils.copyDirectory(new File(webInfDir, "constellio"), constellioDir);
}
workDir.deleteOnExit();
} else {
workDir = getWebinfDir();
}
return workDir;
}

public static File getCollectionsRootDir() {
File webinfDir = getWorkDir();
File constellioDir = new File(webinfDir, "constellio");
File collectionsDir = new File(constellioDir, "collections");
return collectionsDir;
}

public static File getCollectionRootDir(RecordCollection collection) {
File collectionsDir = getCollectionsRootDir();
File collectionDir = new File(collectionsDir, collection.getName());
return collectionDir;
}

public static File getSynonymsFile(RecordCollection collection) {
File collectionDir = getCollectionRootDir(collection);
File synonymsFile = new File(collectionDir, "conf" + File.separator + "synonyms.txt");
return synonymsFile;
}

public static void main(String[] args) {
System.out.println(getClassesDir());
}

}