<?php

final class MongoDB
{
    private static $ins = [];

    private static $def = "default";

    private $_conn = null;

    private static $_config = [
        "default" => ["url" => "mongodb://192.168.1.233:27017", "dbName" => "test"],
        "localhost" => ["url" => "mongodb://localhost:27017", "dbName" => "test"],
        "db1" => ["url" => "mongodb://username:password@localhost:27017", "dbName" => "myDb1"],
    ];

    private $_db = null;


    /**
     * 创建实例
     * @param  string $confKey
     * @return \MongoDB
     */
    static function getInstance($confKey = NULL)
    {
        if (!$confKey) {
            $confKey = self::$def;
        }
        if (!isset(self::$ins[$confKey]) && ($conf = self::$_config[$confKey])) {
            $m = new MongoDB($conf);
            self::$ins[$confKey] = $m;
        }
        return self::$ins[$confKey];
    }


    /**
     * 构造方法-单例模式
     * MongoDB constructor.
     * @param array $conf
     */
    private function __construct(array $conf)
    {
        $this->_conn = new \MongoDB\Driver\Manager($conf["url"] . "/{$conf["dbName"]}");
        $this->_db = $conf["dbName"];
    }


    /**
     * 插入数据
     * @param  string $collName
     * @param  array $documents [["name"=>"values", ...], ...]
     * @param  array $writeOps ["ordered"=>boolean,"writeConcern"=>array]
     * @return \MongoDB\Driver\Cursor
     */
    function insert($collName, array $documents, array $writeOps = [])
    {
        $cmd = [
            "insert" => $collName,
            "documents" => $documents,
        ];
        $cmd += $writeOps;
        return $this->command($cmd);
    }


    /**
     * 删除数据
     * @param  string $collName
     * @param  array $deletes [["q"=>query,"limit"=>int], ...]
     * @param  array $writeOps ["ordered"=>boolean,"writeConcern"=>array]
     * @return \MongoDB\Driver\Cursor
     */
    function del($collName, array $deletes, array $writeOps = [])
    {
        foreach ($deletes as &$_) {
            if (isset($_["q"]) && !$_["q"]) {
                $_["q"] = (Object)[];
            }
            if (isset($_["limit"]) && !$_["limit"]) {
                $_["limit"] = 0;
            }
        }
        $cmd = [
            "delete" => $collName,
            "deletes" => $deletes,
        ];
        $cmd += $writeOps;
        return $this->command($cmd);
    }


    /**
     * 更新数据
     * @param  string $collName
     * @param  array $updates [["q"=>query,"u"=>update,"upSet"=>boolean,"multi"=>boolean], ...]
     * @param  array $writeOps ["ordered"=>boolean,"writeConcern"=>array]
     * @return \MongoDB\Driver\Cursor
     */
    function update($collName, array $updates, array $writeOps = [])
    {
        $cmd = [
            "update" => $collName,
            "updates" => $updates,
        ];
        $cmd += $writeOps;
        return $this->command($cmd);
    }

    /**
     * 查询
     * @param  string $collName
     * @param  array $filter [query]
     * @param  array $writeOps
     * @return \MongoDB\Driver\Cursor
     */
    function query($collName, array $filter, array $writeOps = [])
    {
        $cmd = [
            "find" => $collName,
            "filter" => $filter
        ];
        $cmd += $writeOps;
        return $this->command($cmd);
    }

    /**
     * 执行MongoDB命令
     * @param array $param
     * @return \MongoDB\Driver\Cursor
     */
    function command(array $param)
    {
        $cmd = new \MongoDB\Driver\Command($param);
        // print_r($cmd);exit();
        return $this->_conn->executeCommand($this->_db, $cmd);
    }

    /**
     * 获取当前mongoDB Manager
     * @return \MongoDB\Driver\Manager
     */
    function getMongoManager()
    {
        return $this->_conn;
    }

    /**
     * 聚合查询
     * @param $collName
     * @param array $where
     * @param array $group
     * @return \MongoDB\Driver\Cursor
     */
    function aggregate($collName, array $where, array $group)
    {
        $cmd = [
            'aggregate' => $collName,
            'pipeline' => [
                ['$match' => $where],
                ['$group' => $group]
            ]
        ];
        //print_r($cmd);exit();
        $result = $this->command($cmd)->toArray();
        // print_r($result);exit();
        return $result[0]->result;
    }

    /**
     * 同mysql中的distinct功能
     *
     * @param string $collName collection名
     * @param string $key 要进行distinct的字段名
     * @param array $where 条件
     * @return array
     * Array
     * (
     * [0] => 1.0
     * [1] => 1.1
     * )
     */
    function distinct($collName, $key, array $where)
    {
        $result = [];
        $cmd = [
            'distinct' => $collName,
            'key' => $key,
            'query' => $where
        ];
        $arr = $this->command($cmd)->toArray();
        if (!empty($arr)) {
            $result = $arr[0]->values;
        }
        return $result;
    }

    /**
     * 按条件计算个数
     *
     * @param string $collName 集合名
     * @param array $where 条件
     * @return int
     */
    function count($collName, array $where)
    {
        $result = 0;
        $cmd = [
            'count' => $collName,
            'query' => $where
        ];
        $arr = $this->command($cmd)->toArray();
        if (!empty($arr)) {
            $result = $arr[0]->n;
        }
        return $result;
    }


}