当前位置: 首页 > 工具软件 > Constellio > 使用案例 >

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

谷梁振
2023-12-01

因为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());
    }

}
 类似资料: