<?php if (!defined('BASEPATH')) exit('No direct script access allowed');

/**
 * Class MongoDB操作类
 */
Class Mongo_db
{

    private $CI;
    private $config = array();
    private $manager;
    private $param;
    private $db;
    private $hostname;
    private $port;
    private $database;
    private $username;
    private $password;
    private $debug;
    private $filter = array();
    private $option = array();
    private $selects = array();
    private $updates = array();
    private $pipeline = array();
    private $limit = 999999;
    private $offset = 0;
    private $db_version;


    /**
     * --------------------------------------------------------------------------------
     * Class Constructor
     * --------------------------------------------------------------------------------
     *
     * Automatically check if the Mongo PECL extension has been installed/enabled.
     * Get Access to all CodeIgniter available resources.
     * Load mongodb config file from application/config folder.
     * Prepare the connection variables and establish a connection to the MongoDB.
     * Try to connect on MongoDB server.
     */

    function __construct($param)
    {

        if (!class_exists('MongoDB\Driver\Manager') && !class_exists('MongoDB\Driver\Command')) {
            show_error("The MongoDB Driver extension has not been installed or enabled", 500);
        }
        $this->CI =& get_instance();
        $this->CI->load->config('mongo_db');
        $this->config = $this->CI->config->item('mongo_db');
        $this->param = $param;
        $this->connect();
    }

    /*
    * --------------------------------------------------------------------------------
    * Prepare configuration for mongoDB connection
    * --------------------------------------------------------------------------------
    *
    * Validate group name or autoload default group name from config file.
    * Validate all the properties present in config file of the group.
    */

    private function prepare()
    {
        if (is_array($this->param) && count($this->param) > 0 && isset($this->param['activate']) == TRUE) {
            $this->activate = $this->param['activate'];
        } else if (isset($this->config['active']) && !empty($this->config['active'])) {
            $this->activate = $this->config['active'];
        } else {
            show_error("MongoDB configuration is missing.", 500);
        }

        if (isset($this->config[$this->activate]) == TRUE) {
            if (empty($this->config[$this->activate]['hostname'])) {
                show_error("Hostname missing from mongodb config group : {$this->activate}", 500);
            } else {
                $this->hostname = trim($this->config[$this->activate]['hostname']);
            }

            if (empty($this->config[$this->activate]['port'])) {
                show_error("Port number missing from mongodb config group : {$this->activate}", 500);
            } else {
                $this->port = trim($this->config[$this->activate]['port']);
            }

			if($this->config[$this->activate]['no_auth'] == FALSE){			
				if (empty($this->config[$this->activate]['username'])) {
					show_error("Username missing from mongodb config group : {$this->activate}", 500);
				} else {
					$this->username = trim($this->config[$this->activate]['username']);
				}

				if (empty($this->config[$this->activate]['password'])) {
					show_error("Password missing from mongodb config group : {$this->activate}", 500);
				} else {
					$this->password = trim($this->config[$this->activate]['password']);
				}
			}

            if (empty($this->config[$this->activate]['database'])) {
                show_error("Database name missing from mongodb config group : {$this->activate}", 500);
            } else {
                $this->database = trim($this->config[$this->activate]['database']);
            }

            if (empty($this->config[$this->activate]['db_debug'])) {
                $this->debug = FALSE;
            } else {
                $this->debug = $this->config[$this->activate]['db_debug'];
            }
        } else {
            show_error("mongodb config group :  <strong>{$this->activate}</strong> does not exist.", 500);
        }
    }

    /**
     * --------------------------------------------------------------------------------
     * Connect to MongoDB Database
     * --------------------------------------------------------------------------------
     *
     * Connect to mongoDB database or throw exception with the error message.
     */

    private function connect()
    {
        $this->prepare();
        try {
            $dns = "mongodb://{$this->hostname}:{$this->port}";
            if (isset($this->config[$this->activate]['no_auth']) == TRUE && $this->config[$this->activate]['no_auth'] == TRUE) {
                $options = array();
            } else {
                $options = array('username' => $this->username, 'password' => $this->password);
            }
            $this->manager = new MongoDB\Driver\Manager($dns, $options);
            $this->db = $this->database;
            $this->db_version = $this->get_db_version();
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error("Unable to connect to MongoDB: {$e->getMessage()}", 500);
            } else {
                show_error("Unable to connect to MongoDB", 500);
            }
        }
    }

    public function get_manager(){
        return $this->manager;
    }

    public function get_db_version(){
        $command = new MongoDB\Driver\Command(array("buildinfo" => 1));
        try {
            $cursor = $this->manager->executeCommand("admin", $command);
            $buildinfo = (array)$cursor->toArray()[0];
            if(array_key_exists("version",$buildinfo)){
                return $buildinfo['version'];
            }else{
                return "";
            } 
        }catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error("MongoDB query failed: {$e->getMessage()}", 500);
            } else {
                Throw new Exception("查询数据发生异常！");
            }
        }
    }

    /**
     * 查询数据
     * @param $tales
     */
    public function get($table)
    {
        if (empty($table)) {
            show_error("In order to retrieve documents from MongoDB, a collection name must be passed", 500);
        }
        try {
            if (is_array($this->selects) && count($this->selects) > 0) {
                $this->option['projection'] = $this->selects;
            }
            $query = new MongoDB\Driver\Query($this->filter, $this->option);
            $table = $this->database . '.' . $table;
            $data = $this->manager->executeQuery($table, $query);
            $this->_clear();
            $returns = [];
            foreach ($data as $doc) {
                $returns[] = $this->object_array($doc);
            }
            return $returns;
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error("MongoDB query failed: {$e->getMessage()}", 500);
            } else {
                Throw new Exception("查询数据发生异常！");
            }
        }
    }

    /**
     * 根据条件查询
     * @param string $table
     * @param array $filter
     * @return mixed
     */
    public function get_where($table = "", $filter = array())
    {
        if (is_array($filter) && count($filter) > 0) {
            return $this->where($filter)->get($table);
        } else {
            show_error("Nothing passed to perform search or value is empty.", 500);
        }
    }

    public function select($includes = array(), $excludes = array())
    {
        if (!is_array($includes)) {
            $includes = array();
        }
        if (!is_array($excludes)) {
            $excludes = array();
        }
        if (!empty($includes)) {
            foreach ($includes as $key => $col) {
                $this->selects[$col] = 1;
            }
        }
        if (!empty($excludes)) {
            foreach ($excludes as $col) {
                $this->selects[$col] = 0;
            }
        }
        return ($this);
    }

    public function where($filter, $value = null)
    {
        if (is_array($filter)) {
            foreach ($filter as $wh => $val) {
                $this->filter[$wh] = $val;
            }
        } elseif($filter) {
            $this->filter[$filter] = $value;
        }
        return $this;
    }

    public function where_or($filter = array())
    {
        if (is_array($filter) && count($filter) > 0) {
            if (!isset($this->filter['$or']) || !is_array($this->filter['$or'])) {
                $this->filter['$or'] = array();
            }
            foreach ($filter as $wh => $val) {
                $this->filter['$or'][] = array($wh => $val);
            }
            return ($this);
        } else {
            show_error("Where value should be an array.", 500);
        }
    }

    public function where_in($field = "", $in = array())
    {
        if (empty($field)) {
            show_error("Mongo field is require to perform where in query.", 500);
        }

        if (is_array($in) && count($in) > 0) {
            $this->_w($field);
            $this->filter[$field]['$in'] = $in;
            return ($this);
        } else {
            show_error("in value should be an array.", 500);
        }
    }

    public function where_in_all($field = "", $in = array())
    {
        if (empty($field)) {
            show_error("Mongo field is require to perform where all in query.", 500);
        }

        if (is_array($in) && count($in) > 0) {
            $this->_w($field);
            $this->filter[$field]['$all'] = $in;
            return ($this);
        } else {
            show_error("in value should be an array.", 500);
        }
    }

    public function where_not_in($field = "", $in = array())
    {
        if (empty($field)) {
            show_error("Mongo field is require to perform where not in query.", 500);
        }

        if (is_array($in) && count($in) > 0) {
            $this->_w($field);
            $this->filter[$field]['$nin'] = $in;
            return ($this);
        } else {
            show_error("in value should be an array.", 500);
        }
    }

    public function where_gt($field = "", $x)
    {
        if (!isset($field)) {
            show_error("Mongo field is require to perform greater then query.", 500);
        }

        if (!isset($x)) {
            show_error("Mongo field's value is require to perform greater then query.", 500);
        }

        $this->_w($field);
        $this->filter[$field]['$gt'] = $x;
        return ($this);
    }

    /**
     * --------------------------------------------------------------------------------
     * Where less than
     * --------------------------------------------------------------------------------
     *
     * Get the documents where the value of a $field is less than $x
     *
     * @usage : $this->mongo_db->where_lt('foo', 20);
     */
    public function where_lt($field = "", $x)
    {
        if (!isset($field)) {
            show_error("Mongo field is require to perform less then query.", 500);
        }

        if (!isset($x)) {
            show_error("Mongo field's value is require to perform less then query.", 500);
        }

        $this->_w($field);
        $this->filter[$field]['$lt'] = $x;
        return ($this);
    }

    /**
     * --------------------------------------------------------------------------------
     * Where less than or equal to
     * --------------------------------------------------------------------------------
     *
     * Get the documents where the value of a $field is less than or equal to $x
     *
     * @usage : $this->mongo_db->where_lte('foo', 20);
     */
    public function where_lte($field = "", $x)
    {
        if (!isset($field)) {
            show_error("Mongo field is require to perform less then or equal to query.", 500);
        }

        if (!isset($x)) {
            show_error("Mongo field's value is require to perform less then or equal to query.", 500);
        }

        $this->_w($field);
        $this->filter[$field]['$lte'] = $x;
        return ($this);
    }

    /**
     * --------------------------------------------------------------------------------
     * Where between
     * --------------------------------------------------------------------------------
     *
     * Get the documents where the value of a $field is between $x and $y
     *
     * @usage : $this->mongo_db->where_between('foo', 20, 30);
     */
    public function where_between($field = "", $x, $y)
    {
        if (!isset($field)) {
            show_error("Mongo field is require to perform greater then or equal to query.", 500);
        }

        if (!isset($x)) {
            show_error("Mongo field's start value is require to perform greater then or equal to query.", 500);
        }

        if (!isset($y)) {
            show_error("Mongo field's end value is require to perform greater then or equal to query.", 500);
        }

        $this->_w($field);
        $this->filter[$field]['$gte'] = $x;
        $this->filter[$field]['$lte'] = $y;
        return ($this);
    }

    /**
     * --------------------------------------------------------------------------------
     * Where between and but not equal to
     * --------------------------------------------------------------------------------
     *
     * Get the documents where the value of a $field is between but not equal to $x and $y
     *
     * @usage : $this->mongo_db->where_between_ne('foo', 20, 30);
     */
    public function where_between_ne($field = "", $x, $y)
    {
        if (!isset($field)) {
            show_error("Mongo field is require to perform between and but not equal to query.", 500);
        }

        if (!isset($x)) {
            show_error("Mongo field's start value is require to perform between and but not equal to query.", 500);
        }

        if (!isset($y)) {
            show_error("Mongo field's end value is require to perform between and but not equal to query.", 500);
        }

        $this->_w($field);
        $this->filter[$field]['$gt'] = $x;
        $this->filter[$field]['$lt'] = $y;
        return ($this);
    }

    /**
     * --------------------------------------------------------------------------------
     * Where not equal
     * --------------------------------------------------------------------------------
     *
     * Get the documents where the value of a $field is not equal to $x
     *
     * @usage : $this->mongo_db->where_ne('foo', 1)->get('foobar');
     */
    public function where_ne($field = '', $x)
    {
        if (!isset($field)) {
            show_error("Mongo field is require to perform Where not equal to query.", 500);
        }

        if (!isset($x)) {
            show_error("Mongo field's value is require to perform Where not equal to query.", 500);
        }

        $this->_w($field);
        $this->filter[$field]['$ne'] = $x;
        return ($this);
    }

    /**
     * --------------------------------------------------------------------------------
     * Like
     * --------------------------------------------------------------------------------
     *
     * Get the documents where the (string) value of a $field is like a value. The defaults
     * allow for a case-insensitive search.
     *
     * @param $flags
     * Allows for the typical regular expression flags:
     * i = case insensitive
     * m = multiline
     * x = can contain comments
     * l = locale
     * s = dotall, "." matches everything, including newlines
     * u = match unicode
     *
     * @param $enable_start_wildcard
     * If set to anything other than TRUE, a starting line character "^" will be prepended
     * to the search value, representing only searching for a value at the start of
     * a new line.
     *
     * @param $enable_end_wildcard
     * If set to anything other than TRUE, an ending line character "$" will be appended
     * to the search value, representing only searching for a value at the end of
     * a line.
     *
     * @usage : $this->mongo_db->like('foo', 'bar', 'im', FALSE, TRUE);
     */
    public function like($field = "", $value = "", $flags = "i")
    {
        if (empty($field)) {
            show_error("Mongo field is require to perform like query.", 500);
        }

        if (empty($value)) {
            show_error("Mongo field's value is require to like query.", 500);
        }

        $field = (string)trim($field);
        $this->_w($field);
        $value = (string)trim($value);
        $this->filter[$field] = $this->get_regex($value, $flags);
        return ($this);
    }

    public function create_like($value = "", $flags = "i")
    {
        if (empty($value)) {
            show_error("Mongo field's value is require to like query.", 500);
        }
        $value = (string)trim($value);
        return $this->get_regex($value, $flags);
    }

    /**
     * --------------------------------------------------------------------------------
     * Count
     * --------------------------------------------------------------------------------
     *
     * Count the documents based upon the passed parameters
     *
     * @usage : $this->mongo_db->count('foo');
     */
    public function count($collection = "")
    {
        if (empty($collection)) {
            show_error("In order to retrieve a count of documents from MongoDB, a collection name must be passed", 500);
        }
        $count = $this->query_count($collection, $this->filter);
        $this->_clear();
        return ($count);
    }


    public function order_by($fields = array())
    {
        if(is_array($fields)) {
            foreach ($fields as $col => $val) {
                if ($val == -1 || $val === FALSE || strtolower($val) == 'desc') {
                    $this->option['sort'][$col] = -1;
                } else {
                    $this->option['sort'][$col] = 1;
                }
            }
        }
        return ($this);
    }

    public function limit($x = 99999)
    {
        if ($x !== NULL && is_numeric($x) && $x >= 1) {
            $this->option['limit'] = (int)$x;
        }
        return ($this);
    }

    public function offset($x = 0)
    {
        if ($x !== NULL && is_numeric($x) && $x >= 1) {
            $this->option['skip'] = (int)$x;
        }
        return ($this);
    }

    public function set_wheres($wheres)
    {
        if (is_array($wheres) && count($wheres) > 0) {
            $this->filter = $wheres;
        }
        return ($this);
    }

    public function set($fields, $value = NULL)
    {
        if (is_string($fields)) {
            $this->updates['$set'][$fields] = $value;
        } elseif (is_array($fields)) {
            foreach ($fields as $field => $value) {
                $this->updates['$set'][$field] = $value;
            }
        }
        return $this;
    }

    public function update_all($table, $option = [])
    {
        try {
            $bulk = new MongoDB\Driver\BulkWrite(['ordered' => true]);
            $this->option = array('multi' => true);
            $bulk->update($this->filter, $this->updates, $this->option);
            $table = $this->database . '.' . $table;
            $ret = $this->manager->executeBulkWrite($table, $bulk);
            $this->_clear();
            return $ret;
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error('table: ' . $table . '. pipeline: ' . json_encode($this->filter) . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("更新数据发生异常！");
            }
        }
    }


    /**
     * --------------------------------------------------------------------------------
     * _clear
     * --------------------------------------------------------------------------------
     *
     * Resets the class variables to default settings
     */
    private function _clear()
    {
        $this->filter = array();
        $this->option = array();
        $this->pipeline = array();
        $this->selects = array();
        $this->limit = 999999;
        $this->offset = 0;
    }

    /**
     * --------------------------------------------------------------------------------
     * Where initializer
     * --------------------------------------------------------------------------------
     *
     * Prepares parameters for insertion in $wheres array().
     */
    private function _w($param)
    {
        if (!isset($this->filter[$param])) {
            $this->filter[$param] = array();
        }
    }


    /**
     * 获取MongoDB ObjectID
     * @param $_id
     * @return \MongoDB\BSON\objectID
     */
    public function get_mongo_id($_id)
    {
        if (is_string($_id)) {
            try {
                $_id = new  MongoDB\BSON\objectID($_id);
            } catch (Exception $e) {
                if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                    show_error('get_mongo_id:' . $_id . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
                } else {
                    Throw new Exception("获取MongoDB ObjectID 发生异常！");
                }
            }
        }
        return $_id;
    }

    /**
     * ָ插入一或多条数据
     * @param string $table 数据库表单名称
     * @param array $data 需要插入的数据
     * @param bool $is_multi 是否批处理
     * @return bool ִ执行结果
     * @throws Exception
     */
    public function insert($table, $data, $is_multi = FALSE)
    {
        try {
            $bulk = new MongoDB\Driver\BulkWrite(['ordered' => true]);
            if (!$is_multi) {
                $bulk->insert($data);
            } else {
                foreach ($data as $one_data) {
                    $bulk->insert($one_data);
                }
            }
            $table = $this->database . '.' . $table;
            return $this->manager->executeBulkWrite($table, $bulk);
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error('table: ' . $table . '. data: ' . json_encode($data) . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("插入数据发生异常！");
            }
        }
    }


    /**
     * 获取单条数据库数据
     * @param string $table 数据库表单名称
     * @param array $filter 过滤条件
     * @param array $option 列、排序、分页等
     * @return objects 符合条件数据
     * @throws Exception
     */
    public function find_one($table)
    {
        if (empty($table)) {
            show_error("In order to retrieve documents from MongoDB, a collection name must be passed", 500);
        }
        try {
            $option = array('limit' => 1);
            $query = new MongoDB\Driver\Query($this->filter, $option);
            $table = $this->database . '.' . $table;
            $cursor = $this->manager->executeQuery($table, $query);
            $this->_clear();
            $return = $cursor->toArray();
            if ($return) {
                return $this->object_array($return[0]);
            } else {
                return null;
            }
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error('table: ' . $table . '. filter: ' . json_encode($$this->filter) . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("获取数据发生异常！");
            }
        }
    }

    /**
     * 将stdClass转换为array
     * @param $array
     * @return array
     */
    public function object_array($array)
    {
        if (is_object($array) && get_class($array) == "stdClass") {
            $array = (array)$array;
        }

        if(is_object($array) && get_class($array) == "MongoDB\BSON\ObjectID"){
            $array = (string)$array;
        }

        if (is_array($array)) {
            foreach ($array as $key => $value) {
                $array[$key] = $this->object_array($value);
            }
        }
        return $array;
    }

    public function set_pipeline($pipeline){
        $this->pipeline = $pipeline;
        return ($this);
    }

    public function lookup($lookup){
        if(is_array($lookup)){
            $this->pipeline['$lookup'] = $lookup;
        }
        return ($this);
    }


    /**
     * 聚合管道查询
     * @param string $table 数据库表单名称
     * @param array $pipeline 聚合管道操作
     * @return array
     * @throws Exception
     */
    public function aggregate($table, $group ,$project=NULL ,$is_cursor = FALSE)
    {
        try {
            $commands = array();
            $commands['aggregate'] = $table;
            if(is_array($project)) {
                $this->pipeline[] = array('$project'=>$project);
            }
            if(count($this->filter)>=1) {
                $this->pipeline[] = array('$match' => $this->filter);
            }
            if(array_key_exists("\$limit",$this->option)){
                $this->pipeline[] = array('$limit'=>$this->option['$limit']);
            }
            if(array_key_exists("\$skip",$this->option)){
                $this->pipeline[] = array('$skip'=>$this->option['$skip']);
            }
            if(is_array($group)) {
                $this->pipeline[] = array('$group'=>$group);
            }
            if(array_key_exists("sort",$this->option)){
                $this->pipeline[] = array('$sort'=>$this->option['sort']);
            }
            $commands['pipeline'] = $this->pipeline;
            $commands['allowDiskUse'] = true;
            if($this->db_version){
                $version = explode(".",$this->db_version);
                if($version[0] == 3 && $version[1]>2){
                    $is_cursor =TRUE;
                }
            }
            if ($is_cursor) {
                $commands['cursor'] = new stdClass;
            }
            $command = new MongoDB\Driver\Command($commands);
            $database_name = str_replace('.', '', $this->database);
            $cursor = $this->manager->executeCommand($database_name, $command);
            if ($is_cursor) {
                $response = [];
                foreach ($cursor as $document) {
                    $response[] = $document;
                }
            } else {
                $response = $cursor->toArray()[0]->result;
            }
            if (!empty($response)) {
                foreach ($response as &$value) {
                    $value = (array)$value;
                }
            }
            $this->_clear();
            return $this->object_array($response);
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error('table: ' . $table . '. pipeline: ' . json_encode($this->pipeline) . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("聚合查询发生异常！");
            }
        }
    }


    /**
     * 创建(多个)索引
     * @param $table string 表名
     * @param array $indexes , 例:
     * [
     *     [
     *         'key' => [
     *             'update_time' => 1       //要索引的字段, 升序还是降序
     *         ],
     *         'name' => 'update_time',     //索引名称
     *         'expireAfterSeconds' => 600  //可选, 索引参数, 这里的expireAfterSeconds表示这是一个ttl索引, 将在600s后被自动删除
     *     ],
     * ]
     * @throws Exception
     */
    public function create_index($table, array $indexes)
    {
        $cmd = [
            'createIndexes' => $table,
            'indexes' => $indexes,
        ];

        $command = new MongoDB\Driver\Command($cmd);
        try {
            $this->manager->executeCommand($this->database, $command);
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                log_message(ERROR_LEVE, 'table: ' . $table . '. pipeline: ' . json_encode($cmd) . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("创建索引发生异常！");
            }
        }
    }

    /**
     * 更新指定记录
     * @param string $table 数据库表单名称
     * @param array $data 数据记录
     * @param array $option 参数选项
     * @param bool $is_multi 是否同时更新多条记录
     * @return \MongoDB\Driver\WriteResult
     * @throws Exception
     */
    public function update($table, $data, $option = [], $is_multi = FALSE)
    {
        try {
            $bulk = new MongoDB\Driver\BulkWrite(['ordered' => true]);
            if (!$is_multi) {
                if(array_key_exists('_id',$data['update'])) {
                    $data['update']['_id'] = $this->get_mongo_id($data['update']['_id']);
                }
                $bulk->update($data['filter'], array('$set' => $data['update']), $option);
            } else {
                foreach ($data as $one_data) {
                    $bulk->update($one_data['filter'], array('$set' => $data['update']), $option);
                }
            }
            $table = $this->database . '.' . $table;
            $ret = $this->manager->executeBulkWrite($table, $bulk);
            $this->_clear();
            return $ret;
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error('table: ' . $table . '. pipeline: ' . json_encode($data) . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("更新数据发生异常！");
            }
        }
    }

    /**
     *
     * 删除记录
     * @param string $table 数据库表单名称
     * @param array $filter 更新的数据
     * @param bool $is_multi 是否同时删除多条记录
     * @param array $opinion mongo语句参数
     * @return \MongoDB\Driver\WriteResult
     * @throws Exception
     */
    public function delete($table)
    {
        try {
            $bulk = new MongoDB\Driver\BulkWrite(['ordered' => true]);
            $bulk->delete($this->filter, $this->option);
            $table = $this->database . '.' . $table;
            $ret = $this->manager->executeBulkWrite($table, $bulk);
            $this->_clear();
            return $ret;
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error('table: ' . $table . '. pipeline: ' . json_encode($this->filter) . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("删除数据发生异常！");
            }
        }
    }

    public function delete_all($table){
        try {
            $bulk = new MongoDB\Driver\BulkWrite(['ordered' => true]);
            $bulk->delete($this->filter, $this->option);
            $table = $this->database . '.' . $table;
            $ret = $this->manager->executeBulkWrite($table, $bulk);
            $this->_clear();
            return $ret;
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error('table: ' . $table . '. pipeline: ' . json_encode($this->filter) . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("删除数据发生异常！");
            }
        }
    }

    /**
     * 获取正则表达式
     * @param $pattern
     * @param string $flags
     * @param bool $preg 是否转义
     * @return \MongoDB\BSON\Regex
     * @throws Exception
     */
    public function get_regex($pattern, $flags = 'i', $preg = true)
    {
        try {
            #TD82216
            if ($preg) {
                $pattern = preg_quote($pattern);
            }
            return new MongoDB\BSON\Regex($pattern, $flags);
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error('pattern: ' . $pattern . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("获取正则表达式发生异常！");
            }
        }
    }

    /**
     * 获取字段不同值的个数
     * @param string $table 表的名称
     * @param string $key 字段
     * @param array $filter 过滤条件
     * @return int 数值
     * @throws Exception
     */
    public function query_distinct_count($table, $key, $filter)
    {
        try {
            $param['distinct'] = $table;
            $param['key'] = $key;
            $param['query'] = $filter;
            $command = new MongoDB\Driver\Command($param);
            $cursor = $this->manager->executeCommand($this->database, $command);
            $result = $cursor->toArray();
            if (empty($result)) {
                return 0;
            }
            return count($result[0]->values);
        } catch (Exception $e) {
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                log_message(ERROR_LEVE, 'pattern: ' . $table . ' key: ' . $key . '. MongoDB Exception: ' . $e->getMessage(), $e->getFile(), $e->getLine());
            } else {
                Throw new Exception("获取指定字段的数量发生异常！");
            }
        }
    }

    /*
    * 获取数据查询记录数
    * @param string $table 数据库表单名称
    * @param array $filter 数据查询过滤条件
    * @return int 查询记录数
    * @throws Exception 数据库查询异常
    */
    public function query_nodistinct_count($table, $filter)
    {
        $param['aggregate'] = $table;
        $param['pipeline'] = array(
            array('$match' => $filter),
            array('$group' => array(
                '_id' => null,
                'sum' => array(
                    '$sum' => 1
                )
            )
            )
        );

        try {
            $param['cursor'] = new stdClass();
            $command = new MongoDB\Driver\Command($param);
            $cursor = $this->manager->executeCommand($this->database, $command);
        } catch (Exception $e) {
            $err_msg = sprintf("MongoDB Exception: %s", $e->getMessage());
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error($err_msg . $e->getFile() . $e->getLine());
            } else {
                throw new Exception("获取数据查询记录数发生异常！");
            }
        }
        $result = $cursor->toArray();
        if (empty($result)) {
            return 0;
        }

        return $result[0]->sum;
    }

    /**
     * 获取所有的记录数
     * @param string $table 表的名称
     * @param array $filter 过滤条件
     * @return int
     * @throws Exception
     */
    public function query_count($table, $filter)
    {
        $param['count'] = $table;
        $param['query'] = $filter;
        $command = new MongoDB\Driver\Command($param);
        try {
            $cursor = $this->manager->executeCommand($this->database, $command);
        } catch (Exception $e) {
            $err_msg = sprintf("MongoDB Exception: %s", $e->getMessage());
            if (isset($this->debug) == TRUE && $this->debug == TRUE) {
                show_error($err_msg . $e->getFile() . $e->getLine());
            } else {
                throw new Exception("获取所有的记录数发生异常！");
            }
        }
        $result = current($cursor->toArray());
        // Older server versions may return a float
        if (!isset($result->n) || !(is_integer($result->n) || is_float($result->n))) {
            throw new UnexpectedValueException('count command did not return a numeric "n" value');
        }

        return (integer)$result->n;

    }
}