<?php

namespace Amp\Postgres\Internal;

use Amp\Loop;

trait Operation {
    /** @var bool */
    private $complete = false;

    /** @var callable[] */
    private $onComplete = [];

    public function onComplete(callable $onComplete) {
        if ($this->complete) {
            $onComplete();
            return;
        }

        $this->onComplete[] = $onComplete;
    }

    private function complete() {
        if ($this->complete) {
            return;
        }

        $this->complete = true;
        foreach ($this->onComplete as $callback) {
            try {
                $callback();
            } catch (\Throwable $exception) {
                Loop::defer(function () use ($exception) {
                    throw $exception; // Rethrow to event loop error handler.
                });
            }
        }
        $this->onComplete = null;
    }
}