PHPLIB-1236 Implement GridFS Benchmarks

Author: GromNaN

benchmark/DriverBench/GridFSBench.php

@@ -0,0 +1,67 @@
+<?php
+
+namespace MongoDB\Benchmark\DriverBench;
+
+use MongoDB\Benchmark\Fixtures\Data;
+use MongoDB\Benchmark\Utils;
+use MongoDB\GridFS\Bucket;
+use PhpBench\Attributes\AfterMethods;
+use PhpBench\Attributes\BeforeMethods;
+
+/**
+ * For accurate results, run benchmarks on a standalone server.
+ *
+ * @see https://github.com/mongodb/specifications/blob/ddfc8b583d49aaf8c4c19fa01255afb66b36b92e/source/benchmarking/benchmarking.rst#multi-doc-benchmarks
+ */
+#[AfterMethods('afterAll')]
+final class GridFSBench
+{
+    /** @var resource */
+    private $stream;
+    private Bucket $bucket;
+    private mixed $id;
+
+    /** @see https://github.com/mongodb/specifications/blob/ddfc8b583d49aaf8c4c19fa01255afb66b36b92e/source/benchmarking/benchmarking.rst#gridfs-upload */
+    #[BeforeMethods('beforeUpload')]
+    public function benchUpload(): void
+    {
+        $this->bucket->uploadFromStream('test', $this->stream);
+    }
+
+    public function beforeUpload(): void
+    {
+        $database = Utils::getDatabase();
+        $database->drop();
+
+        $this->bucket = $database->selectGridFSBucket();
+        // Init the GridFS bucket
+        $this->bucket->uploadFromStream('init', Data::getStream(1));
+        // Prepare the 50MB stream to upload
+        $this->stream = Data::getStream(50 * 1024 * 1024);
+    }
+
+    /** @see https://github.com/mongodb/specifications/blob/ddfc8b583d49aaf8c4c19fa01255afb66b36b92e/source/benchmarking/benchmarking.rst#gridfs-download */
+    #[BeforeMethods('beforeDownload')]
+    public function benchDownload(): void
+    {
+        $this->bucket->downloadToStream($this->id, $this->stream);
+    }
+
+    public function beforeDownload(): void
+    {
+        $database = Utils::getDatabase();
+        $database->drop();
+
+        $this->bucket = $database->selectGridFSBucket();
+        // Upload a 50MB file
+        $this->id = $this->bucket->uploadFromStream('init', Data::getStream(50 * 1024 * 1024));
+        // Prepare the stream to receive the download
+        $this->stream = Data::getStream(0);
+    }
+
+    public function afterAll(): void
+    {
+        unset($this->bucket, $this->stream, $this->id);
+        Utils::getDatabase()->drop();
+    }
+}

benchmark/DriverBench/MultiDocBench.php

@@ -6,6 +6,7 @@
 use MongoDB\Benchmark\Fixtures\Data;
 use MongoDB\Benchmark\Utils;
 use MongoDB\BSON\Document;
+use MongoDB\GridFS\Bucket;
 use PhpBench\Attributes\BeforeMethods;
 use PhpBench\Attributes\ParamProviders;
 
@@ -17,19 +18,18 @@
  *
  * @see https://github.com/mongodb/specifications/blob/ddfc8b583d49aaf8c4c19fa01255afb66b36b92e/source/benchmarking/benchmarking.rst#multi-doc-benchmarks
  */
-#[BeforeMethods('prepareDatabase')]
-class MultiDocBench
+final class MultiDocBench
 {
-    public function prepareDatabase(): void
-    {
-        Utils::getCollection()->drop();
-    }
+    /** @var resource */
+    private $stream;
+    private Bucket $bucket;
+    private mixed $id;
 
     /**
      * @see https://github.com/mongodb/specifications/blob/master/source/benchmarking/benchmarking.rst#find-many-and-empty-the-cursor
      * @param array{options: array} $params
      */
-    #[BeforeMethods('setupFindMany')]
+    #[BeforeMethods('beforeFindMany')]
     #[ParamProviders('provideFindManyParams')]
     public function benchFindMany(array $params): void
     {
@@ -40,11 +40,14 @@ public function benchFindMany(array $params): void
         foreach ($collection->find([], $params['options']) as $document);
     }
 
-    public function setupFindMany(): void
+    public function beforeFindMany(): void
     {
+        $collection = Utils::getCollection();
+        $collection->drop();
+
         $tweet = Data::readJsonFile(Data::TWEET_FILE_PATH);
         $documents = array_fill(0, 9_999, $tweet);
-        Utils::getCollection()->insertMany($documents);
+        $collection->insertMany($documents);
     }
 
     public static function provideFindManyParams(): Generator
@@ -63,7 +66,7 @@ public static function provideFindManyParams(): Generator
      * @see https://github.com/mongodb/specifications/blob/ddfc8b583d49aaf8c4c19fa01255afb66b36b92e/source/benchmarking/benchmarking.rst#large-doc-bulk-insert
      * @param array{documents: array} $params
      */
-    #[BeforeMethods('setupBulkInsert')]
+    #[BeforeMethods('beforeBulkInsert')]
     #[ParamProviders('provideBulkInsertParams')]
     public function benchBulkInsert(array $params): void
     {
@@ -74,12 +77,11 @@ public function benchBulkInsert(array $params): void
         $collection->insertMany($params['documents']);
     }
 
-    public function setupBulkInsert(): void
+    public function beforeBulkInsert(): void
     {
-        $collectionName = Utils::getCollection()->getCollectionName();
-        Utils::getClient()
-            ->selectDatabase(Utils::getDatabase())
-            ->createCollection($collectionName);
+        $database = Utils::getDatabase();
+        $database->dropCollection(Utils::getCollectionName());
+        $database->createCollection(Utils::getCollectionName());
     }
 
     public static function provideBulkInsertParams(): Generator

benchmark/Extension/EnvironmentProvider.php

@@ -61,7 +61,7 @@ private function getServerInfo(Manager $manager): array
     private function getBuildInfo(Manager $manager): array
     {
         $buildInfo = $manager->executeCommand(
-            Utils::getDatabase(),
+            Utils::getDatabaseName(),
             new Command(['buildInfo' => 1]),
             new ReadPreference(ReadPreference::PRIMARY),
         )->toArray()[0];

benchmark/Fixtures/Data.php

@@ -3,7 +3,11 @@
 namespace MongoDB\Benchmark\Fixtures;
 
 use function file_get_contents;
+use function fopen;
+use function fwrite;
 use function json_decode;
+use function rewind;
+use function str_repeat;
 
 use const JSON_THROW_ON_ERROR;
 
@@ -17,4 +21,18 @@ public static function readJsonFile(string $path): array
     {
         return json_decode(file_get_contents($path), true, 512, JSON_THROW_ON_ERROR);
     }
+
+    /**
+     * Generates an in-memory stream of the given size.
+     *
+     * @return resource
+     */
+    public static function getStream(int $size)
+    {
+        $stream = fopen('php://memory', 'w+');
+        fwrite($stream, str_repeat("\0", $size));
+        rewind($stream);
+
+        return $stream;
+    }
 }

benchmark/Utils.php

@@ -4,41 +4,43 @@
 
 use MongoDB\Client;
 use MongoDB\Collection;
+use MongoDB\Database;
 
 use function getenv;
 
 final class Utils
 {
     private static ?Client $client;
+    private static ?Database $database;
     private static ?Collection $collection;
 
     public static function getClient(): Client
     {
-        return self::$client ??= self::createClient();
+        return self::$client ??= new Client(self::getUri());
+    }
+
+    public static function getDatabase(): Database
+    {
+        return self::$database ??= self::getClient()->selectDatabase(self::getDatabaseName());
     }
 
     public static function getCollection(): Collection
     {
-        return self::$collection ??= self::createCollection();
+        return self::$collection ??= self::getDatabase()->selectCollection(self::getCollectionName());
     }
 
     public static function getUri(): string
     {
         return getenv('MONGODB_URI') ?: 'mongodb://localhost:27017/';
     }
 
-    public static function getDatabase(): string
+    public static function getDatabaseName(): string
     {
         return getenv('MONGODB_DATABASE') ?: 'phplib_test';
     }
 
-    private static function createClient(): Client
-    {
-        return new Client(self::getUri());
-    }
-
-    private static function createCollection(): Collection
+    public static function getCollectionName(): string
     {
-        return self::getClient()->selectCollection(self::getDatabase(), 'perftest');
+        return 'perftest';
     }
 }

phpbench.json.dist

@@ -5,6 +5,7 @@
     "runner.bootstrap": "vendor/autoload.php",
     "runner.file_pattern": "*Bench.php",
     "runner.path": "benchmark",
+    "runner.php_config": { "memory_limit": "1G" },
     "runner.iterations": 3,
     "runner.revs": 10
 }