MongoDB随机获取记录的原理及代码

发布时间：2023-09-16 14:54:16 所属栏目：系统来源：

导读：原理：

1.先查询表中的记录总数

2.随机获取偏移量为0~总记录数-1

3.查询时skip偏移量，再获取1条记录

因本人测试环境PHP已升级到7.0以上，mongodb扩展使用支持php7.0以上的扩展，很多方法与php5.6不

原理：

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"

  }

)

（编辑：聊城站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!