ackintosh / ganesha

:elephant: A Circuit Breaker pattern implementation for PHP applications.
https://ackintosh.github.io/ganesha/
MIT License
593 stars 44 forks source link

Array adapter #117

Open TheCelavi opened 1 month ago

TheCelavi commented 1 month ago

Would it be possible to write array adapter?

Seams kinda missing for unit/integration tests, right?

Would you accept such PR?

TheCelavi commented 1 month ago

Ok, I wrote adapter, you can use it if you like. Only thing which is missing here is to reset adapter when test ends. In Symfony, that would be to implement ResetInterface.

To be framework agnostic, this could be implemented as singletone pattern, and then to use PHPUnit hooks to call reset().

/**
 * @phpstan-type Record = array{
 *     service: string,
 *     count: int,
 *     lastFailureTime: int|null,
 *     status: int,
 * }
 *
 * @SuppressWarnings(PHPMD.TooManyPublicMethods)
 */
final class ArrayAdapter implements AdapterInterface, TumblingTimeWindowInterface, SlidingTimeWindowInterface
{
   /**
     * Storage must be shared among all instances of the adapter.
     *
     * @var array<string, int|null>
     */
    private static array $storage = [];

    /**
     * {@inheritdoc}
     */
    public function supportCountStrategy(): bool
    {
        return true;
    }

    /**
     * {@inheritdoc}
     */
    public function supportRateStrategy(): bool
    {
        return true;
    }

    /**
     * {@inheritdoc}
     */
    public function setContext(Context $context): void
    {
        // This adapter doesn't use the context.
    }

    /**
     * {@inheritdoc}
     */
    public function setConfiguration(Configuration $configuration): void
    {
        // noop
    }

    /**
     * {@inheritdoc}
     */
    public function load(string $service): int
    {
        self::$storage[$service] = self::$storage[$service] ?? 0;
        return self::$storage[$service] ?? 0;
    }

    /**
     * {@inheritdoc}
     */
    public function save(string $service, int $count): void
    {
        self::$storage[$service] = $count;
    }

    /**
     * {@inheritdoc}
     */
    public function increment(string $service): void
    {
        self::$storage[$service] = (self::$storage[$service] ?? 0) + 1;
    }

    /**
     * {@inheritdoc}
     */
    public function decrement(string $service): void
    {
        self::$storage[$service] = \max(0, (self::$storage[$service] ?? 0) - 1);
    }

    /**
     * {@inheritdoc}
     */
    public function saveLastFailureTime(string $service, int $lastFailureTime): void
    {
        // WHY??? Count strategy does not work without this, why?
        $service = \sprintf('%s_last_failure_time', $service);

        self::$storage[$service] = $lastFailureTime;
    }

    /**
     * {@inheritdoc}
     */
    public function loadLastFailureTime(string $service)
    {
        // WHY???
        $service = \sprintf('%s_last_failure_time', $service);

        return self::$storage[$service] ?? null;
    }

    /**
     * {@inheritdoc}
     */
    public function saveStatus(string $service, int $status): void
    {
        self::$storage[$service] = $status;
    }

    /**
     * {@inheritdoc}
     */
    public function loadStatus(string $service): int
    {
        if (!isset(self::$storage[$service])) {
            self::$storage[$service] = Ganesha::STATUS_CALMED_DOWN;
        }

        return self::$storage[$service];
    }

    /**
     * {@inheritdoc}
     */
    public function reset(): void
    {
        self::$storage = [];
    }
}