118 lines
4.1 KiB
PHP
118 lines
4.1 KiB
PHP
<?php
|
|
|
|
namespace Tests\Benchmarks\Services\Storage;
|
|
|
|
use CURLFile;
|
|
use PhpBench\Attributes\BeforeMethods;
|
|
use PhpBench\Attributes\ParamProviders;
|
|
use Tests\Benchmarks\Scopes\Scope;
|
|
use Tests\E2E\Client;
|
|
use Tests\E2E\Scopes\ProjectCustom;
|
|
use Utopia\Database\ID;
|
|
use Utopia\Database\Permission;
|
|
use Utopia\Database\Role;
|
|
|
|
abstract class Base extends Scope
|
|
{
|
|
use ProjectCustom;
|
|
|
|
protected static string $bucketId;
|
|
protected static string $fileId;
|
|
|
|
#[BeforeMethods(['createBucket'])]
|
|
public function benchFileCreate()
|
|
{
|
|
$this->client->call(Client::METHOD_POST, '/storage/buckets/' . static::$bucketId . '/files', array_merge([
|
|
'content-type' => 'multipart/form-data',
|
|
'x-appwrite-project' => $this->getProject()['$id'],
|
|
], $this->getHeaders()), [
|
|
'fileId' => ID::unique(),
|
|
'permissions' => [
|
|
Permission::read(Role::user($this->getUser()['$id'])),
|
|
Permission::write(Role::user($this->getUser()['$id'])),
|
|
],
|
|
|
|
'file' => new CURLFile(realpath(__DIR__ . '/../../../resources/logo.png'), 'image/png', 'logo.png'),
|
|
]);
|
|
}
|
|
|
|
#[ParamProviders(['provideCounts'])]
|
|
#[BeforeMethods(['createBucket', 'createFiles'])]
|
|
public function benchFileReadList(array $params)
|
|
{
|
|
$this->client->call(Client::METHOD_GET, '/storage/buckets/' . static::$bucketId . '/files', array_merge([
|
|
'content-type' => 'application/json',
|
|
'x-appwrite-project' => $this->getProject()['$id'],
|
|
], $this->getHeaders()), [
|
|
'queries' => ['limit(' . $params['files'] . ')'],
|
|
]);
|
|
}
|
|
|
|
#[BeforeMethods(['createBucket', 'createFiles'])]
|
|
public function benchFileRead()
|
|
{
|
|
$this->client->call(Client::METHOD_GET, '/storage/buckets/' . static::$bucketId . '/files/' . static::$fileId, array_merge([
|
|
'content-type' => 'application/json',
|
|
'x-appwrite-project' => $this->getProject()['$id'],
|
|
], $this->getHeaders()));
|
|
}
|
|
|
|
#[BeforeMethods(['createBucket', 'createFiles'])]
|
|
public function benchFileUpdate()
|
|
{
|
|
$this->client->call(Client::METHOD_PUT, '/storage/buckets/' . static::$bucketId . '/files/' . static::$fileId, array_merge([
|
|
'content-type' => 'application/json',
|
|
'x-appwrite-project' => $this->getProject()['$id'],
|
|
], $this->getHeaders()), [
|
|
'name' => 'Updated name',
|
|
'permissions' => [],
|
|
]);
|
|
}
|
|
|
|
public function provideCounts(): array
|
|
{
|
|
return [
|
|
'10 Files' => ['files' => 10],
|
|
'100 Files' => ['files' => 100],
|
|
];
|
|
}
|
|
|
|
public function createBucket(array $params = [])
|
|
{
|
|
// Create bucket
|
|
$bucket = $this->client->call(Client::METHOD_POST, '/storage/buckets', [
|
|
'content-type' => 'application/json',
|
|
'x-appwrite-project' => $this->getProject()['$id'],
|
|
'x-appwrite-key' => $this->getProject()['apiKey']
|
|
], [
|
|
'bucketId' => ID::unique(),
|
|
'name' => 'Test Bucket',
|
|
'permissions' => [
|
|
Permission::read(Role::user($this->getUser()['$id'])),
|
|
Permission::write(Role::user($this->getUser()['$id'])),
|
|
],
|
|
'fileSecurity' => true
|
|
]);
|
|
static::$bucketId = $bucket['body']['$id'];
|
|
}
|
|
|
|
public function createFiles(array $params = [])
|
|
{
|
|
$count = $params['files'] ?? 1;
|
|
|
|
// Create files
|
|
for ($i = 0; $i < $count; $i++) {
|
|
$response = $this->client->call(Client::METHOD_POST, '/storage/buckets/' . static::$bucketId . '/files', [
|
|
'content-type' => 'multipart/form-data',
|
|
'x-appwrite-project' => $this->getProject()['$id'],
|
|
'x-appwrite-key' => $this->getProject()['apiKey']
|
|
], [
|
|
'fileId' => ID::unique(),
|
|
'file' => new CURLFile(realpath(__DIR__ . '/../../../resources/logo.png'), 'image/png', 'logo.png'),
|
|
]);
|
|
|
|
static::$fileId = $response['body']['$id'];
|
|
}
|
|
}
|
|
}
|