<?php
|
|
/**PHP7.x使用 */
|
|
namespace MongoDB7;
|
|
use MongoDB\Driver\Manager;
|
use MongoDB\Driver\BulkWrite;
|
use MongoDB\Driver\WriteConcern;
|
use MongoDB\Driver\Query;
|
use MongoDB\Driver\Command;
|
|
class MongoDb
|
{
|
|
protected $mongodb;
|
protected $database;
|
protected $collection;
|
protected $bulk;
|
protected $writeConcern;
|
// db默认配置
|
protected $defaultConfig = [
|
'hostname' => 'localhost',
|
'port' => '27017',
|
'username' => '',
|
'password' => '',
|
'database' => ''
|
];
|
|
public function __construct($collection, $dbconfig)
|
{
|
$config = array_merge($this->defaultConfig, $dbconfig);
|
$mongoServer = "mongodb://";
|
if ($config['username']) {
|
$mongoServer .= $config['username'] . ':' . $config['password'] . '@';
|
}
|
$mongoServer .= $config['hostname'];
|
if ($config['port']) {
|
$mongoServer .= ':' . $config['port'];
|
}
|
$mongoServer .= '/' . $config['database'];
|
|
$this->mongodb = new Manager($mongoServer);
|
$this->database = $config['database'];
|
//$this->collection = $config['collection'];
|
$this->collection = $collection;
|
$this->bulk = new BulkWrite();
|
$this->writeConcern = new WriteConcern(WriteConcern::MAJORITY, 100);
|
}
|
|
public function query($where = [], $option = [])
|
{
|
$query = new Query($where, $option);
|
$result = $this->mongodb->executeQuery("$this->database.$this->collection", $query);
|
|
// 返回值是 MongoDB\Driver\Cursor ，需要转化为数据数组，方便逻辑中使用
|
return json_decode(json_encode($result->toArray()), true);
|
}
|
|
public function count($where = [])
|
{
|
$command = new Command(['count' => $this->collection, 'query' => $where]);
|
$result = $this->mongodb->executeCommand($this->database, $command);
|
$res = $result->toArray();
|
$count = 0;
|
if ($res) {
|
$count = $res[0]->n;
|
}
|
|
return $count;
|
}
|
|
public function update($where = [], $update = [], $upsert = false)
|
{
|
$this->bulk->update($where, ['$set' => $update], ['multi' => true, 'upsert' => $upsert]);
|
$result = $this->mongodb->executeBulkWrite("$this->database.$this->collection", $this->bulk, $this->writeConcern);
|
|
return $result->getModifiedCount();
|
}
|
|
public function insert($data = [])
|
{
|
$this->bulk->insert($data);
|
$result = $this->mongodb->executeBulkWrite("$this->database.$this->collection", $this->bulk, $this->writeConcern);
|
|
return $result->getInsertedCount();
|
}
|
|
public function insertbatch($datas = [])
|
{
|
foreach ($datas as $data) {
|
$this->bulk->insert($data);
|
}
|
$result = $this->mongodb->executeBulkWrite("$this->database.$this->collection", $this->bulk, $this->writeConcern);
|
|
return $result->getInsertedCount();
|
}
|
|
public function delete($where = [], $limit = 1)
|
{
|
$this->bulk->delete($where, ['limit' => $limit]);
|
$result = $this->mongodb->executeBulkWrite("$this->database.$this->collection", $this->bulk, $this->writeConcern);
|
|
return $result->getDeletedCount();
|
}
|
}
|
