使用PHP7来批量更新MangoDB数据

直接贴代码

<?php

set_time_limit(0);
$manager = new MongoDBDriverManager('mongodb://localhost:27017');
$namespace = 'test.test';
$aQuery = array('status' => 200);
$oQuery = new MongoDBDriverQuery($aQuery);
$t1 = microtime(TRUE);
$cursor = $manager->executeQuery($namespace, $oQuery);
$i = 0;
$bulk = new MongoDBDriverBulkWrite;
foreach($cursor as $doc) {
    $i++;
    if ($i % 100000 === 0) {
        $manager->executeBulkWrite($namespace, $bulk);
        echo ($i / 10000) . 'W ' . number_format(microtime(TRUE) - $t1, 2) . " Seconds
";
        $bulk = new MongoDBDriverBulkWrite;
    }
    $set = array();
    $set['A'] = strToArray($doc->a);
    $set['B'] = strToArray($doc->b);
    $set['C'] = strToArray($doc->c);
    $bulk->update(
        array('_id' => $doc->_id),
        array('$set' => $set)
    );
}
$manager->executeBulkWrite($namespace, $bulk);
echo 'Total: ' . number_format(microtime(TRUE) - $t1, 2) . " Seconds
";

function strToArray($str) {
    $str = str_replace('，', ',', $str);
    $arr = explode(',', $str);
    $ret = array();
    foreach ($arr as $s) {
        $s = trim($s);
        if ($s) {
            $ret[] = $s;
        }
    }
    return $ret;
}