mongodb 隨機獲取一條記錄的方法
阿新 • • 發佈:2019-01-10
原理:
1.先查詢表中的記錄總數
2.隨機獲取偏移量為0~總記錄數-1
3.查詢時skip偏移量,再獲取1條記錄
因本人測試環境php已升級到7.0以上,mongodb擴充套件使用支援php7.0以上的擴充套件,很多方法與php5.6不同。因此程式碼必須在php7.0以上執行。如果是php5.6環境,需要修改程式碼才能執行。
程式碼如下:
function.php
<?php
// 連線mongodb
function conn($host, $user, $passwd){
$server = 'mongodb://'.$user.':'.$passwd.'@'.$host ;
try{
$conn = new MongoDB\Driver\Manager();
} catch (MongoDB\Driver\Exception\ConnectionException $e){
throw new ErrorException('Unable to connect to db server. Error:' . $e->getMessage(), 31);
}
return $conn;
}
// 插入資料
function add($conn, $dbname, $collname, $data , $index){
// 建立索引
$cmd = array(
'createIndexes' => $collname,
'indexes' => array(
array(
'name' => 'index',
'key' => $index,
'ns' => $dbname.'.'.$collname
)
)
);
$command = new MongoDB\Driver\Command($cmd);
$conn->executeCommand($dbname, $command);
// 插入資料
$bulk = new MongoDB\Driver\BulkWrite();
$inserted = 0;
if($data){
foreach($data as $k=>$v){
$bulk->insert($v);
}
$result = $conn->executeBulkWrite($dbname.'.'.$collname, $bulk);
$inserted = $result->getInsertedCount();
}
return $inserted;
}
// 獲取總記錄數
function getCount($conn, $dbname, $collname){
$cmd = array(
'count' => $collname,
'query' => array()
);
$command = new MongoDB\Driver\Command($cmd);
$result = $conn->executeCommand($dbname, $command);
$response = current($result->toArray());
if($response->ok==1){
return $response->n;
}
return 0;
}
// 隨機獲取一條記錄
function randOne($conn, $dbname, $collname){
// 總記錄數
$total = getCount($conn, $dbname, $collname);
// 隨機偏移
$skip = mt_rand(0, $total-1);
$filter = array();
$options = array('skip'=>$skip, 'limit'=>1);
$query = new MongoDB\Driver\Query($filter, $options);
$cursor = $conn->executeQuery($dbname.'.'.$collname, $query);
$result = array();
if($cursor){
foreach($cursor as $v){
$v = objectToArray($v);
unset($v['_id']);
$result[] = $v;
}
}
return $result? $result[0] : $result;
}
// 物件轉為陣列
function objectToArray($obj){
$arr = is_object($obj) ? get_object_vars($obj) : $obj;
if(is_array($arr)){
return array_map(__FUNCTION__, $arr);
}else{
return $arr;
}
}
?>
demo.php
<?php
require('function.php');
// 連線mongodb
$conn = conn('localhost','testdb','root','123456');
// 插入50條資料記錄
$data = array();
// 索引
$index = array('user'=>true);
for($i=0; $i<50; $i++){
$data[] = array(
'user' => 'test_user_'.str_pad($i, 4, '0', STR_PAD_LEFT)
);
}
$inserted = add($conn, 'testdb', 'user', $data, $index);
echo '成功插入'.$inserted.'條測試記錄數<br><br>';
// 隨機獲取一條記錄,抽5次
echo '隨機獲取一條記錄,抽5次<br>';
$result = array();
for($i=0; $i<5; $i++){
$result[] = randOne($conn, 'testdb', 'user');
}
echo '<pre>';
print_r($result);
echo '</pre>';
?>
輸出:
成功插入50條測試記錄數
隨機獲取一條記錄,抽5次
Array
(
[0] => Array
(
[user] => test_user_0017
)
[1] => Array
(
[user] => test_user_0026
)
[2] => Array
(
[user] => test_user_0004
)
[3] => Array
(
[user] => test_user_0043
)
[4] => Array
(
[user] => test_user_0023
)
)
測試php程式碼,首先需要在mongodb建立testdb及建立使用者和執行auth。方法如下:
use testdb
db.createUser(
{
"user":"root",
"pwd":"123456",
"roles":[{"role" : "readWrite", "db":"testdb"}]
}
)
db.auth(
{
"user":"root",
"pwd":"123456"
}
)
原始碼下載地址:點選檢視