xunsearch使用笔记

<?php
namespace ServiceTools;

require_once(dirname(__FILE__)."/XunSearch/XS.php");
class XunSearch
{
    static $instace;

    private $project;
    private $index;
    private $search;
    private $tokenizer;

    private $table_config=array(
        "company"=>array(
            "tablename"=>"k_company",
            "use_field"=>"id,title",
            "condition"=>"id=%d",
            "primary_key"=>"id",
        ),

    );

    public static function getInstance($project=null)
    {
        return self::$instace ?: (self::$instace = new self($project));
    }

    public function __construct($project)
    {
        $this->project=$project;
        $this->setSearchConn($project);
    }

    public function setSearchConn($project)
    {
        if($project=="company"){
            $file_name=COMPANY_SEARCH_CONFIG;
        }
        $config_path=dirname(__FILE__)."/XunSearch/config/".$file_name;
        if (!file_exists($config_path)){
            throw new Exception("缺少对应配置文件");
        }

        $xs=new XS($config_path);
        $this->index =$xs->index;
        $this->search=$xs->search;

        // 开启模糊搜索
        //$this->search->setFuzzy();
        $this->search->setCharset('UTF-8');
        $this->search->setTimeout(0);

        $this->tokenizer = new XSTokenizerScws;
    }

    public function executeIndex($id,$type){

        $condition=sprintf($this->table_config[$this->project]["condition"],$id);
        $field=$this->table_config[$this->project]["use_field"];
        $info=M()->table($this->table_config[$this->project]["tablename"])->where($condition)->field($field)->find();
        if(empty($info)){
            return ["error"=>"1","msg"=>"未找到相关数据"];
        }
        // 创建文档对象
        $doc = new XSDocument;
        $doc->setFields($info);
        switch ($type){
            case 'add':
                // 添加到索引数据库中
                $this->index->clean();
                $this->index->add($doc);
                break;
            case "edit":
                // 更新到索引数据库中
                $this->index->update($doc);
                break;
            case "del":
                $this->index->del($info[$this->table_config[$this->project]["primary_key"]]);
            default:
                break;
        }
        $this->index->flushIndex();
        return ["error"=>"0","msg"=>"数据处理成功"];
    }

    public function executeIndexInfo($arr,$type){
        $field_str=$this->table_config[$this->project]["use_field"];
        $field_arr=explode(",",$field_str);
        $info=array();
        foreach((array)$field_arr as $value){
            if(!isset($arr[$value])){
                return ["error"=>"1","msg"=>"未找到相关数据"];
                exit;
            }
            $info[$value]=$arr[$value];
        }

        if(empty($info)){
            return ["error"=>"1","msg"=>"未找到相关数据"];
            exit;
        }

        // 创建文档对象
        $doc = new XSDocument;
        $doc->setFields($info);
        switch ($type){
            case 'add':
                // 添加到索引数据库中
                $this->index->add($doc);
                break;
            case "edit":
                // 更新到索引数据库中
                $this->index->update($doc);
                break;
            default:
                return ["error"=>"1","msg"=>"未找到相关数据"];
                exit;
                break;
        }
        $this->index->flushIndex();
        return ["error"=>"0","msg"=>"数据处理成功"];

    }
    //清空索引
    public function cleanIndex(){
        $this->index->clean();
        return ["error"=>"0","msg"=>"数据处理成功"];
    }

    public function rebuildIndex($model=1,$condition=array()){
        if($model==1){
            // 宣布开始重建索引
            $this->index->beginRebuild();
            $this->addAllIndex($condition);
            //重建完比
            $this->index->endRebuild();
            return ["error"=>"0","msg"=>"数据重建成功"];
        }else{
            $this->index->clean();
            $this->addAllIndex($condition);
            $this->index->flushIndex();
            return ["error"=>"0","msg"=>"数据重建成功"];
        }
    }

    public function addAllIndex($condition){
        $num=0;
        $page=1000;
        while(true){
            $field=$this->table_config[$this->project]["use_field"];
            if($condition){
                $list=M()->table($this->table_config[$this->project]["tablename"])->where($condition)->field($field)->page($num.','.$page)->select();
            }else{
                $list=M()->table($this->table_config[$this->project]["tablename"])->field($field)->page($num.','.$page)->select();
            }
            $num++;
            if(empty($list)){
                break;
            }

            foreach($list as $value){
                $doc = new XSDocument;
                $doc->setFields($value);
                $this->index->add($doc);
            }
        }
    }

    public function searchIndexKey($keywords, $page_num, $page_size){
        $words = $this->tokenizer->getResult($keywords);
        $wordsArr=array();
        foreach($words as $value){
            $wordsArr[]=$value["word"];
        }
        $queryStr = '';
        $queryStr.=join(' OR ', $wordsArr);
        $offset = ($page_num - 1) * $page_size;
        $this->search->addQueryString($queryStr);
        $this->search->setLimit($page_size, $offset);
        $result = $this->search->search();

        $list = $this->processResult($result,$this->project);
        $total = $this->search->getLastCount();

        return array(
            'list' => $list,
            'total' => $total,
        );
    }
    /**
     *
     * @param type $keywords
     * @param type $page_num
     * @param type $page_size
     * @return array('total' => int,'list' => array())
     */
    public function searchIndex($keywords = array(), $page_num = 1, $page_size = 10)
    {

        $queryStr = '';
        if (!empty($keywords)) {
            $queryStr.=join(' OR ', $keywords);
        }

        $offset = ($page_num - 1) * $page_size;
        $this->search->addQueryString($queryStr);
        $this->search->setSort($this->table_config[$this->project]["primary_key"]);
        $this->search->setLimit($page_size, $offset);
        $result = $this->search->search();

        $list = $this->processResult($result,$this->project);
        $total = $this->search->getLastCount();

        return array(
            'list' => $list,
            'total' => $total,
        );
    }

    public function processResult($list,$type)
    {
        $docs = array();
        foreach ($list as $item) {
            if($type=='company'){
                $row['id']=$item['id'];
                //$row['title']=$this->search->highlight($item['title'],true);
                $row['title']=$item['title'];
                $docs[] = $row;
            }elseif($type==""){
                //$row['update_time']=$row['create_time'];
                //$row['content']=$this->search->highlight($row['content'],true);
                //$row['title']=$this->search->highlight($row['title'],true);
                //$docs[] = $row;
            }
        }
        return $docs;
    }


}