Files
compras/tests/Concurrency/run-budgets.php
T

136 lines
7.0 KiB
PHP

<?php
use App\Models\Budget;
use App\Models\BudgetEntry;
use App\Services\BudgetWorkflow;
use App\Services\ConflictGuard;
use Illuminate\Contracts\Console\Kernel;
use Illuminate\Support\Facades\Artisan;
use Illuminate\Support\Facades\DB;
use Illuminate\Support\Facades\Schema;
use Symfony\Component\Process\Process;
require __DIR__.'/../../vendor/autoload.php';
foreach (['APP_ENV' => 'testing', 'CACHE_STORE' => 'array', 'SESSION_DRIVER' => 'array', 'QUEUE_CONNECTION' => 'sync', 'MAIL_MAILER' => 'array'] as $key => $value) {
putenv($key.'='.$value);
$_ENV[$key] = $_SERVER[$key] = $value;
}
$app = require __DIR__.'/../../bootstrap/app.php';
$app->make(Kernel::class)->bootstrap();
require __DIR__.'/../Support/ReceiptFixture.php';
$original = DB::connection();
if ($original->getDriverName() !== 'pgsql') {
throw new LogicException('Exige PostgreSQL.');
}
$config = $original->getConfig();
$admin = $original->getPdo();
$database = 'compras_receipts_test_'.bin2hex(random_bytes(6));
$created = false;
$processes = [];
$report = ['tested_at_utc' => gmdate('c'), 'separate_database' => true, 'transport' => 'Independent PHP service transactions and PostgreSQL connections', 'scenarios' => []];
try {
$admin->exec('CREATE DATABASE "'.$database.'"');
$created = true;
$isolated = [...$config, 'database' => $database, 'url' => null];
config(['database.connections.pgsql' => $isolated]);
DB::purge('pgsql');
DB::setDefaultConnection('pgsql');
Schema::clearResolvedInstance('db.schema');
app()->forgetInstance('db.schema');
if (DB::connection()->getDatabaseName() !== $database || Schema::getConnection()->getDatabaseName() !== $database) {
throw new LogicException('Migração fora do isolamento recusada.');
}
Artisan::call('migrate', ['--database' => 'pgsql', '--force' => true]);
$workerEnv = ['APP_ENV' => 'testing', 'DB_CONNECTION' => 'pgsql', 'DB_URL' => '', 'DB_DATABASE' => $database, 'DB_HOST' => $config['host'], 'DB_PORT' => (string) $config['port'], 'DB_USERNAME' => $config['username'], 'DB_PASSWORD' => $config['password'], 'CACHE_STORE' => 'array', 'SESSION_DRIVER' => 'array'];
foreach (['over_reserve', 'same_reserve', 'same_commit', 'same_release', 'over_award'] as $scenario) {
$f = receiptFixture();
$user = $f['issuer'];
auth()->guard()->setUser($user);
$budget = Budget::query()->create(['tenant_id' => $user->tenant_id, 'cost_center_id' => $f['r']->cost_center_id, 'year' => now()->year, 'amount_minor' => 10000, 'includes_tax' => true, 'created_by' => $user->id, 'approved_by' => $f['receiver']->id, 'reason' => 'Fixture isolada']);
$first = $f['r'];
$second = $first;
if (in_array($scenario, ['over_reserve', 'over_award'])) {
$second = $first->replicate();
$second->number = 'REQ-TEST-'.bin2hex(random_bytes(5));
$second->save();
}
if (in_array($scenario, ['same_commit', 'same_release', 'over_award'])) {
foreach ($scenario === 'over_award' ? [$first, $second] : [$first] as $request) {
DB::transaction(function () use ($user, $request) {
app(ConflictGuard::class)->lockCompany($user);
app(BudgetWorkflow::class)->record($user, $request, 'reserve', 2000, 2000);
});
}
}
$kind = match ($scenario) {
'same_commit' => 'commit', 'same_release' => 'release', 'over_award' => 'award', default => 'reserve'
};
$connection = DB::connection();
$connection->beginTransaction();
$connection->table('tenants')->where('id', $user->tenant_id)->lockForUpdate()->first();
$processes = [];
foreach ([$first, $second] as $index => $request) {
$process = new Process([PHP_BINARY, __DIR__.'/budget-worker.php', $database, (string) $user->id, (string) $request->id, $index === 0 ? 'A' : 'B'], base_path(), $workerEnv);
$process->setInput(json_encode(['kind' => $kind, 'amount' => 6000], JSON_THROW_ON_ERROR));
$process->setTimeout(40);
$process->start();
$processes[] = $process;
}
$statement = $admin->prepare("SELECT count(*) FROM pg_stat_activity WHERE datname = ? AND application_name IN ('sgc_budget_A','sgc_budget_B') AND wait_event_type = 'Lock'");
$deadline = microtime(true) + 20;
do {
$statement->execute([$database]);
$waits = (int) $statement->fetchColumn();
if ($waits === 2) {
break;
}
foreach ($processes as $process) {
if (! $process->isRunning()) {
throw new RuntimeException('Worker terminou antes do bloqueio: '.$process->getErrorOutput());
}
}
usleep(20000);
} while (microtime(true) < $deadline);
$connection->commit();
if ($waits !== 2) {
throw new RuntimeException('Sem sobreposição de bloqueios.');
}
$results = [];
foreach ($processes as $process) {
$process->wait();
if (! $process->isSuccessful()) {
throw new RuntimeException('Falha do worker: '.$process->getErrorOutput());
}
$results[] = json_decode($process->getOutput(), true, 512, JSON_THROW_ON_ERROR);
}
$statuses = array_column($results, 'status');
sort($statuses);
$reserved = (int) BudgetEntry::query()->sum('reserved_minor');
$committed = (int) BudgetEntry::query()->sum('committed_minor');
[$expectedReserve, $expectedCommit] = match ($scenario) {
'same_commit' => [0, 6000], 'same_release' => [0, 0], 'over_award' => [8000, 0], default => [6000, 0]
};
if ($statuses !== (in_array($scenario, ['over_reserve', 'over_award']) ? [200, 422] : [200, 200]) || $reserved !== $expectedReserve || $committed !== $expectedCommit || count(array_unique(array_column($results, 'backend_pid'))) !== 2) {
throw new RuntimeException('Resultado inesperado: '.json_encode([$statuses, $reserved, $committed]));
}
$report['scenarios'][] = ['name' => $scenario, 'passed' => true, 'overlapping_lock_waits' => $waits, 'reserved_minor' => $reserved, 'committed_minor' => $committed];
}
$report['passed'] = true;
} finally {
foreach ($processes as $process) {
if ($process->isRunning()) {
$process->stop();
}
}
if (isset($connection) && $connection->transactionLevel() > 0) {
$connection->rollBack();
}
DB::disconnect('pgsql');
if ($created && $database !== $original->getDatabaseName() && preg_match('/^compras_receipts_test_[a-f0-9]{12}$/', $database)) {
$admin->exec('DROP DATABASE "'.$database.'"');
$report['test_database_dropped'] = true;
}
}
file_put_contents(base_path('docs/orcamento-concorrencia-postgresql.json'), json_encode($report, JSON_PRETTY_PRINT | JSON_THROW_ON_ERROR).PHP_EOL);
echo json_encode($report, JSON_PRETTY_PRINT | JSON_THROW_ON_ERROR).PHP_EOL;