1. 程式人生 > >mongodb 隨機獲取一條記錄的方法

mongodb 隨機獲取一條記錄的方法

原理:

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條測試記錄數

隨機獲取一條記錄,抽5Array
(
    [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"  
    }  
) 



原始碼下載地址:點選檢視