php7 mongo 操作类

<?php
/**
* 这是关于mongo数据操作的类
* song
* 2022.01.20
*/

declare(strict_types=1);

class MongoDb
{
public static $_instance = null;
public $conf = null;
public $handle = null;
public $_and = [];
public $_or = [];
public $collection = "";
public $limit = 0;
public $skip = 0;
public $_field = [];
public $_sort = [];

public static function getInstance($conf = null){
if(self::$_instance === null){
self::$_instance = new self($conf);
}
return self::$_instance;
}

private function __construct($conf=null)
{
$this->conf = $conf;
if(!isset($this->conf["type"]) || empty($this->conf["type"])){
$this->conf["type"] = "mongodb";
}
$this->connect();
}

public function connect(){
if(!empty($this->conf["user"]) && !empty($this->conf["password"])){
$uri = "{$this->conf["type"]}://{$this->conf["user"]}:{$this->conf["password"]}@";
}else{
$uri = "{$this->conf["type"]}://";
}
$uri .= "{$this->conf['host']}:{$this->conf['port']}/{$this->conf["db"]}";
try {
$this->handle = new MongoDB\Driver\Manager($uri);
}catch (Exception $exception){
throw new Exception($exception->getMessage());
}
}

public function collection(string $collection){
$this->collection = $this->conf["db"] . "." . $collection;
return $this;
}

public function where(array $where){
if(empty($where)){
$this->echoError(new Exception("param where is empty"));
}
$this->_and = array_merge($this->_and,$where);
return $this;
}

public function orWhere(array $where){
if(empty($where)){
$this->echoError(new Exception("param where is empty"));
}
$this->_or = array_merge($this->_or,$where);
return $this;
}

public function limit(int $limit){
$this->limit = $limit;
return $this;
}

public function skip(int $skip){
$this->skip = $skip;
return $this;
}

public function field(string $field,bool $_id=true){
if(!empty($field)){
$fieldArr = explode(",",$field);
if(is_array($fieldArr)){
foreach($fieldArr as $val){
$this->_field[$val] = 1;
}
}
}
if(!$_id){
$this->_field["_id"] = 0;
}
return $this;
}

public function sort(string $field,int $sort){
$this->_sort["field"] = $field;
$this->_sort["rule"] = $sort;
return $this;
}

public function find(){
return $this->query();
}

public function findOne(){
$this->limit = 1;
$this->skip = 0;
return $this->query();
}

public function count(){
$filter = $this->getWhere();
$aggregate = [
"aggregate"=>str_replace($this->conf["db"].".","",$this->collection),
"pipeline" => [
['$match'=>$filter],
['$count'=>"count"],
],
"cursor" => (object)array()
];
$command = new \MongoDB\Driver\Command($aggregate);
$result = $this->command($this->conf["db"],$command);
return $result ? $result->toArray()[0]->count : false;
}

public function sum(string $field){
$filter = $this->getWhere();
$aggregate = [
"aggregate"=>str_replace($this->conf["db"].".","",$this->collection),
"pipeline" => [
['$match' => $filter],
[
'$group' => [
'_id' => '',
'total' => ['$sum' => '$' . $field],
]
],
],
"cursor" => (object)array()
];
$command = new \MongoDB\Driver\Command($aggregate);
$result = $this->command($this->conf["db"],$command);
return $result ? $result->toArray()[0]->total : false;
}

public function getWhere(){
$filter = [];
if(!empty($this->_and)){
$filter = array_merge($filter,$this->_and);
}
if(!empty($this->_or)){
foreach($this->_or as $key =>$val) {
$filter['$or'][][$key] = $val;
}
}
return $filter;
}

public function getQuery(){
$filter = $this->getWhere();
if(!empty($this->_field)){
$queryOptions["projection"] = $this->_field;
}
if(!empty($this->_sort)){
$queryOptions["sort"] = [$this->_sort["field"]=>$this->_sort["rule"]];
}
if($this->limit > 0){
$queryOptions["limit"] = $this->limit;
}
if($this->skip > 0){
$queryOptions["skip"] = $this->skip;
}
$query = new MongoDB\Driver\Query($filter,$queryOptions);
return $query;
}

public function query(){
$query = $this->getQuery();
try {
$result = $this->handle->executeQuery($this->collection,$query);
}catch (\Exception $exception){
$this->echoError($exception);
$result = false;
}
$this->init();
return $result?:false;
}

/**
* @param array $data
* @param bool $batch 是否批量
* @return mixed
*/
public function insert(array $data,bool $batch=false){
$write = new MongoDB\Driver\BulkWrite();
if($batch){
foreach($data as $val){
$write->insert($val);
}
}else{
$write->insert($data);
}
$result = $this->execute($this->collection,$write);
return $result ? $result->getInsertedCount() : false;
}

/**
* @param array $update
* @param bool $multi true 匹配所有文档 false 匹配一个
* @param bool $upsert true 匹配不到将插入
* @return bool
* @throws Exception
*/
public function update(array $update,bool $multi = false, bool $upsert = true){
if(empty($this->_and)){
$this->echoError(new Exception("update where is empty"));
}
$write = new MongoDB\Driver\BulkWrite();
$write->update(
$this->_and,
['$set'=>$update],
['multi' => $multi, 'upsert' => $upsert]
);
$result = $this->execute($this->collection,$write);
return $result ? $result->getUpsertedCount() + $result->getMatchedCount() : false;
}

public function delete(bool $all=false){
if(empty($this->_and)){
$this->echoError(new Exception("delete where is empty"));
}
$write = new MongoDB\Driver\BulkWrite();
$write->delete($this->_and,['limit'=>$all]);
$result = $this->execute($this->collection,$write);
return $result ? $result->getDeletedCount() : false;
}

/**
* @param array $pipeline
* $pipeline 参数说明:
* [
* [
* '$match' => [
* 'time' => ['$lt'=>1598864580]
* ],
* ],
* [
* '$group' => [
* "_id"=>'$time', "total" => ['$sum' => 1]
* ],
* ],
* [
* '$limit' => 3
* ],
* [
* '$sort' => ['total' => -1] //1升序 -1降序
* ]
* ]
* @return bool
*/
public function aggregate(array $pipeline){
$aggregate = [
"aggregate"=>str_replace($this->conf["db"].".","",$this->collection),
"pipeline" => $pipeline,
"cursor" => (object)array()
];
$command = new \MongoDB\Driver\Command($aggregate);
$result = $this->command($this->conf["db"],$command);
return $result ? $result->toArray() : false;
}

public function execute($namespace,$object){
try {
$result = $this->handle->executeBulkWrite($namespace,$object);
}catch (\Exception $exception){
$this->echoError($exception);
}
$this->init();
return $result;
}

public function command($db,$command){
try {
$result = $this->handle->executeCommand($db,$command);
}catch (\Exception $exception){
$this->echoError($exception);
}
$this->init();
return $result;
}

public function echoError(Exception $exception){
throw new Exception($exception->getMessage());
}

public function init(){
$this->_and = [];
$this->_or = [];
$this->collection = "";
$this->limit = 0;
$this->skip = 0;
$this->_field = [];
$this->_sort = [];
}

}

// $db = MongoDB::getInstance([
// "type" => "mongodb",
// "host" => "127.0.0.1",
// "port" => "27017",
// "db" => "gengsan",
// "user" => "",
// "password" => ""
// ]);

// print_r($db);


//查询
// $result = $db->collection("ecms_news")->Where(["id"=>7])->find();

// $result = $db->collection("ecms_news")->Where(["id"=>['$gt'=>0]])->count();
// $result = $db->collection("ecms_news")->Where(["id"=>['$lte'=>2]])->sort('id',1)->find();
// $result = $db->collection("ecms_news")->Where(["id"=>['$gt'=>0]])->count();
// $result = $result->toArray();
// print_r($result);
//$result = $db->collection("message")->Where(["time"=>['$lte'=>1598864449]])->sum("time");
//写入
// $result = $db->collection("ecms_news")->insert([
// "id" => 7,
// "title" => "测试",
// "titleurl" => "gengsan.com",
// ]);

// print_r($result);

//更新
// $result = $db->collection("message")->where(["id"=>"8"])->update(["title"=>"ssd"]);
// print_r($result);
//删除
// $result = $db->collection("ecms_news")->where(["id"=>['$gt'=>1]])->delete();
//aggregate 聚合
// $result = $db->collection("ecms_news")->aggregate([
// ['$match'=>['id'=>['$gt'=>1]]]
// ]);

// foreach ($result as $key => $value) {
// echo "<pre>";
// print_r($value);
// // exit();
// }
// $result = $result->toArray();
// print_r($result[0]->title);
// var_dump($result);

posted @ 2022-01-24 16:11  宋先生日记  阅读(116)  评论(0编辑  收藏  举报