Package org.apdplat.platform.search

Source Code of org.apdplat.platform.search.IndexManager

/**
*
* APDPlat - Application Product Development Platform
* Copyright (c) 2013, 杨尚川, yang-shangchuan@qq.com
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program.  If not, see <http://www.gnu.org/licenses/>.
*
*/

package org.apdplat.platform.search;

import org.apdplat.module.system.service.PropertyHolder;
import org.apdplat.platform.log.APDPlatLogger;
import org.apdplat.platform.model.Model;
import org.apdplat.platform.util.ConvertUtils;
import java.io.File;
import java.util.Locale;
import javax.annotation.Resource;
import org.apdplat.platform.log.APDPlatLoggerFactory;
import org.compass.core.CompassSession;
import org.compass.core.CompassTemplate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

/**
* 实时索引管理,包括:新增、修改、删除
* @author 杨尚川
*/
@Service
public class IndexManager {
    private static final APDPlatLogger LOG = APDPlatLoggerFactory.getAPDPlatLogger(IndexManager.class);
   
    @Resource(name = "compassTemplate")
    private CompassTemplate compassTemplate;
    @Resource(name = "indexRebuilder")
    private IndexRebuilder indexRebuilder;
    private volatile boolean buiding=false;

    /**
     * 异步重建索引(线程安全且为异步调用)
     * 只有一个重建操作完成后才可开始另一个重建操作
     * 利用buiding变量来进行控制
     * 因为涉及到多个线程对同一个变量的读写,所以
     * buiding变量要加volatile
     * 保证线程的可见性
     */
    public void rebuidAll(){
        if(buiding){
            LOG.info("正在重建索引,请求自动取消");
            LOG.info("Rebuilding index is in progress, request auto cancel", Locale.ENGLISH);
            return;
        }
        buiding=true;
        LOG.info("开始重建索引");
        LOG.info("Begin to rebuild index", Locale.ENGLISH);
        new Thread(new Runnable(){
            @Override
            public void run(){
                indexRebuilder.build();
                LOG.info("结束重建索引");
                LOG.info("Finish rebuild index", Locale.ENGLISH);
                buiding=false;
            }
        }).start();
    }
    public static File getIndexDir(){
        String userDir = System.getProperty("user.dir");
        String indexDir=PropertyHolder.getProperty("index.dictionary").replace("/", File.separator);
        File file=new File(userDir,indexDir);
        LOG.info("获取索引文件目录:"+file.getAbsolutePath());
        LOG.info("Get index file path:", Locale.ENGLISH);
        return file;
    }
    private CompassSession getCompassSession() {
        return compassTemplate.getCompass().openSession();

    }
    private void closeCompassSession(CompassSession session) {
        session.close();
    }
    @Transactional
    public void createIndex(Model model) {
        try{
            CompassSession session = getCompassSession();
            try {
                session.create(model);
            } catch (Exception e) {
                String info=e.getMessage();
                LOG.error("创建索引失败", e);
                LOG.error("Failed in building index", e, Locale.ENGLISH);

                if(info.indexOf("LockObtainFailedException")!=-1){
                    int index=info.indexOf("@");
                    String path=info.substring(index+1);
                    File file=new File(path);
                    file.delete();
                    fixIndex();
                    session.create(model);
                }
            } finally{
                closeCompassSession(session);
            }
        }catch(Exception e){
            LOG.error("创建索引失败", e);
            LOG.error("Failed in building index", e, Locale.ENGLISH);
        }
    }
    @Transactional
    public void updateIndex(Class<? extends Model> type, Model model) {
        try{
            deleteIndex(type,model.getId());
            CompassSession session = getCompassSession();
            try {
                session.save(model);
            } catch (Exception e) {
                String info=e.getMessage();
                LOG.error("更新索引失败", e);
                LOG.error("Failed to update index", e, Locale.ENGLISH);
                if(info.indexOf("LockObtainFailedException")!=-1){
                    int index=info.indexOf("@");
                    String path=info.substring(index+1);
                    File file=new File(path);
                    file.delete();
                    fixIndex();
                    session.save(model);
                }
            } finally{
                closeCompassSession(session);
            }
        }catch(Exception e){
            LOG.error("更新索引失败", e);
            LOG.error("Failed to update index", e, Locale.ENGLISH);
        }
    }
    @Transactional
    public void deleteIndex(Class<? extends Model> type, Object objectID) {
        try{
            CompassSession session = getCompassSession();
            try {
                session.delete(session.load(type, objectID));
            } catch (Exception e) {
                String info=e.getMessage();
                LOG.error("删除索引失败", e);
                LOG.error("Failed to delete index", e , Locale.ENGLISH);
                if(info.indexOf("LockObtainFailedException") != -1){
                    int index=info.indexOf("@");
                    String path=info.substring(index+1);
                    File file=new File(path);
                    file.delete();
                    fixIndex();
                    //再次删除
                    session.delete(session.load(type, objectID));
                }
            } finally{
                closeCompassSession(session);
            }
        }catch(Exception e){
            LOG.error("删除索引失败",e);
            LOG.error("Failed to delete index", e, Locale.ENGLISH);
        }
    }
    private void fixIndex(){
        LOG.info("开始修复索引");
        LOG.info("Begin repair index", Locale.ENGLISH);
        long beginTime = System.currentTimeMillis();
        File file=IndexManager.getIndexDir();
        clearWriteLock(file);
        long costTime = System.currentTimeMillis() - beginTime;
        LOG.info("耗时:" + ConvertUtils.getTimeDes(costTime));
        LOG.info("Elapsed:", Locale.ENGLISH);
        LOG.info("结束修复索引");
        LOG.info("Finish repair index", Locale.ENGLISH);
    }
    private void clearWriteLock(File file){
        if(file.isFile()){
            if(file.getName().equals("write.lock")){
                file.delete();
            }
        }else if(file.isDirectory()){
            File[] files=file.listFiles();
            if(files.length>0){
                for(File f : files){
                    clearWriteLock(f);
                }
            }
        }
    }
}
TOP

Related Classes of org.apdplat.platform.search.IndexManager

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.