现在的代码，贴一下

Posted 2020-07-25 兔六哥

<?php
/**
 * Created by PhpStorm.
 * User: user
 * Date: 14-9-19
 * Time: 上午11:12
 */
namespace Application\chinalao\esengine;
use Application\Config\config;
use Elasticsearch\Client;

/**
 * 禁止直接调用本类进行ES操作
 * 要在type对应的类中继承本类
 * 并且在type类里面要初始化$this->type变量
 * Class AbstractEngine
 * @package Application\chinalao\esengine
 */
class AbstractEngine {

    protected $client;
    protected $config;
    protected $index;//索引名称
    protected $search_data;
    protected $params;

    protected function __construct(){
        $this->config = config::elasticsearch();
        $this->client = new Client($this->config[‘construct‘]);
        $this->index = $this->config[‘index‘];
        $this->debug = config::debug();

    }
    //更改默认的index
    function setIndex($index){
        $this->index = $index;
    }

    /**
     * 返回接口操作对象，便于更自定义的操作
     * @return Client
     */
    function get_client(){
        return $this->client;
    }

    /**
     * 如果不需要返回数据，统一用这个对返回结果进行过滤
     * @param $return
     * @return bool成功返回true，失败返回false
     */
    protected function do_return($return){
        if(!isset($return[‘acknowledged‘])){
            if($this->debug){
                return $return;
            }
            return false;
        }
        return $return[‘acknowledged‘]==1;
    }

    /**
     * 建立索引
     * @param $index
     * @param int $shards
     * @param int $replicas
     * @return bool成功返回true，失败返回false
     */
    public function createIndex($index, $shards=null, $replicas=null){
        $indexParams[‘index‘] = $index;
        $indexParams[‘body‘][‘settings‘][‘number_of_shards‘] = is_null($shards)?$this->config[‘number_of_shards‘]:$shards;
        $indexParams[‘body‘][‘settings‘][‘number_of_replicas‘] = is_null($replicas)?$this->config[‘number_of_replicas‘]:$replicas;
//        $indexParams[‘body‘][‘settings‘][‘analysis‘] = array(
//            ‘analyzer‘=>array(
//                ‘ik‘=>array(
//                    ‘tokenizer‘=>‘ik‘
//                )
//            )
//        );
        $re = $this->client->indices()->create($indexParams);
        return $this->do_return($re);
    }

    /**
     * 给index设置别名
     * @param $index
     * @param $name
     * @param $body
     * @return bool成功返回true
     */
    function putAlias($index,$name,$body=array()){
        $params = array();
        $params[‘index‘] = $index;
        $params[‘name‘] = $name;
        $params[‘body‘] = $body;
        $re = $this->client->indices()->putAlias($params);
        return $this->do_return($re);
    }

    /**
     * 删除别名
     * @param $index
     * @param $name
     * @return bool成功返回true
     */
    function deleteAlias($index,$name){
        $params = array();
        $params[‘index‘] = $index;
        $params[‘name‘] = $name;
        $re = $this->client->indices()->deleteAlias($params);
        return $this->do_return($re);
    }
    protected function putSettings($index,$body){
        $this->client->indices()->close(array(‘index‘=>$index));
        $this->client->indices()->putSettings(array(‘index‘=>$index,‘body‘=>$body));
        $this->client->indices()->open(array(‘index‘=>$index));
        return true;
    }
    /**
     * 删除索引
     * @param $index
     * @return bool成功返回true，失败返回false
     */
    public function deleteIndex($index){
        $deleteParams[‘index‘] = $index;
        $re =   $this->client->indices()->delete($deleteParams);
        return $this->do_return($re);
    }

    /**
     * 建立Mapping，或者给现有的type添加字段
     * @param $index
     * @param $type
     * @param array $properties_arr
     * @return bool成功返回true，失败返回false
     */
    public function createMapping($type, $mapParam = array()){
        $indexParams[‘index‘] = $this->index;
        $indexParams[‘type‘] = $type;
        $indexParams[‘body‘][$indexParams[‘type‘]] = $mapParam;
        $ret = $this->client->indices()->putMapping($indexParams);
        return $this->do_return($ret);
    }

    /**
     * 删除Type表
     * @param $type
     * @return array
     */
    function deleteType($type){
        $indexParams[‘index‘]  = $this->index;
        $indexParams[‘type‘]  = $type;
        $ret = $this->client->indices()->deleteMapping($indexParams);
        return $this->do_return($ret);
    }



    /**
     * 批量添加记录
     * @param $index
     * @param $type
     * @param array $arr
     * @return bool成功返回true，失败返回false
     */
    public function insertBulk($type ,$arr = array()){
        $bulkParams[‘index‘]  = $this->index;
        $bulkParams[‘type‘]  = $type;
        $bulkParams[‘body‘]  = $arr;
        //print_r($bulkParams);die;
        $re = $this->client->bulk($bulkParams);
        //var_dump($re);die;
        if($re[‘errors‘]===false)return true;
        if($this->debug)var_dump($re);
        return false;
    }

    /**
     * 添加(更新)一条记录
     * @param $index
     * @param $type
     * @param $id
     * @param array $arr
     * @return bool成功返回true，失败返回false
     */
    public function insertOneById($type,$id,$arr = array()){
        $params = array();
        $params[‘index‘] = $this->index;
        $params[‘type‘]  = $type;
        $params[‘id‘]    = $id;
        $params[‘body‘]  = $arr;
        $re = $this->client->index($params);
        return $this->do_return($re);
    }
    /**
     * 添加(更新)一条记录
     * @param $index
     * @param $type
     * @param array $arr
     * @return bool成功返回true，失败返回false
     */
    public function insertOne($type,$arr = array()){
        $params = array();
        $params[‘index‘] = $this->index;
        $params[‘type‘]  = $type;
        $params[‘body‘]  = $arr;
        $re = $this->client->index($params);
//        if($re[‘created‘]===false && $this->debug){
//            return $re;
//        }
        return $re[‘created‘];
    }


    /**
     * 获得某条记录的值
     * @param $index
     * @param $type
     * @param $id
     * @return array查询结果
     */
    public function getOne($type, $id){
        $getParams = array();
        $getParams[‘index‘] = $this->index;
        $getParams[‘type‘]  = $type;
        $getParams[‘id‘]  = $id;
        $getParams[‘ignore‘] = ‘404‘;
        $re = $this->client->get($getParams);
        //print_r($re);die;
        $json_re = @json_decode($re,true);
        if(!empty($json_re) && $json_re[‘found‘]===false)return null;
        return $re[‘_source‘];
    }

    /**
     * 更新某条记录
     * @param $index
     * @param $type
     * @param $id
     * @param array $arr
     * @return bool成功返回true，失败返回false
     */
    public function  updateOne($type, $id, $arr = array()){
        $updateParams[‘index‘] = $this->index;
        $updateParams[‘type‘]  = $type;
        $updateParams[‘id‘]    = $id;
        $updateParams[‘ignore‘] = ‘404,400‘;
        $updateParams[‘body‘][‘doc‘]  = $arr;
        //print_r($updateParams);die;
        $re = $this->client->update($updateParams);
        $json_re = @json_decode($re,true);
        if(!empty($json_re) ){
            if($json_re[‘status‘]===404)
                return null;
            else{//status==400的情况
                return false;
            }
        }
        //var_dump($re);die;
        return true;
    }

    /**
     * 删除某条记录
     * @param $index
     * @param $type
     * @param $id
     * @return bool成功返回true，失败返回false null为改条信息不存在
     */
    public function deleteOne($type, $id){
        $deleteParams = array();
        $deleteParams[‘index‘] = $this->index;
        $deleteParams[‘type‘] = $type;
        $deleteParams[‘id‘] = $id;
        $deleteParams[‘ignore‘] = ‘404,400‘;
        $re = $this->client->delete($deleteParams);
        $json_re = @json_decode($re,true);
        if(!empty($json_re) ){
            if($json_re[‘found‘]===false)
                return null;
            else{//status==400的情况
                return false;
            }
        }
        if($re[‘found‘]==true)return true;
        return false;
    }

    /**
     * 根据query查询语句删除记录
     * @param $type
     * @param $body
     * @return bool成功返回true，失败返回false
     */
    function delete($type,$body){
        $data = array();
        $data[‘index‘] = $this->index;
        $data[‘type‘] = $type;
        $data[‘body‘] = $body;
        $re = $this->client->deleteByQuery($data);
        return $this->do_return($re);
    }

    /**
     * 统一搜索接口，
     * 其中from是搜索的结果起始位置，size是搜索结果数
     * 也可以给from参数传0,20这样的参数，类似于mysql的limit语法，会自动把20赋值给size
     * @param $type
     * @param $body
     * @param int $from可以是一个整数，也可以是“整数,整数”格式，如果是后者，size参数将不再起作用
     * @param int $size
     * @param array $sort
     * @return array 返回查询结果
     */
    function select($type,$body,$from=0,$size=10,$sort=array()){
        $params[‘index‘] = $this->index;
        $params[‘type‘] = $type;
        $params[‘body‘] = $body;

        if(false !== strpos($from,‘,‘)){
            list($from,$size) = explode(‘,‘,$from);
        }
        $params[‘from‘] = $from;
        $params[‘size‘] = $size;
        if(!empty($sort))$params[‘body‘][‘sort‘] = $sort;
        //print_r($params);die;
        $this->params = $params;
        $this->search_data = $this->client->search($params);
        return $this->toArray();
    }
    function getParams(){
        return $this->params;
    }

    /**
     * 返回查询结果总数（这个结果，不受查询结果的from和size限制）
     * @return mixed
     */
    function count(){
        return isset($this->search_data[‘hits‘][‘total‘])?$this->search_data[‘hits‘][‘total‘]:0;
    }

    /**
     * 直接返回es查询的结果
     * @return mixed
     */
    function getSearchData(){
        return $this->search_data;
    }

    /**
     * 将es查询结果过滤掉多余的标记，直接返回数据的二维数组
     * @param bool $complete
     * @param null $search_data
     * @return array
     */
    function toArray($complete=false,$search_data=null){
        !is_null($search_data) && $this->search_data = $search_data;
        if($complete)return $this->search_data[‘hits‘][‘hits‘];
        $rs = array();
        foreach($this->search_data[‘hits‘][‘hits‘] as $data){
            $rs[] = $data[‘_source‘];
        }
        return $rs;
    }

    /**
     * 添加一条数据
     * @param $data
     * @return mixed
     */
    function addOne($data){
        return $this->insertOne($this->type,$data);
    }
    /**
     * 删除该mapping
     */
    function deleteMap(){
        return $this->deleteType($this->type);
    }
    /**
     * 更新一条数据，data是array(字段名=>字段值);的格式
     * 不更新的字段，不需要传递
     * @param $id
     * @param $data
     * @return mixed
     */
    function update($id,$data){
        return $this->updateOne($this->type,$id,$data);
    }

    /**
     * 删除一条记录
     * @param $id 主键
     * @return bool成功返回true
     */
    function remove($id){
        return $this->deleteOne($this->type,$id);
    }

    /**
     * 根据id获取当前type的记录
     * @param $id
     * @return array查询结果
     */
    function getDataById($id){
        return $this->getOne($this->type,$id);
    }
    /**
     * 根据多个id获取当前type的记录信息
     * @param string type 所查表
     * @param array $ids 数组
     * @param array $fields 要获取的字段，默认是所有字段
     * @param int total 如果已经知道ids的数量，就赋值给TA
     * @return array
     */
    function getDataByIds($type,array $ids,$fields = array(),$total = 0){
        $body = array(
            ‘query‘=>array(
                ‘ids‘=>array(
                    ‘values‘=>$ids,
                ),
            ),
        );
        if(!empty($fields))$body[‘_source‘] = $fields;
        if(empty($total))$total = count($ids);
        return $this->select($type,$body,0,$total);
    }

}