Source code for eventsourcing.infrastructure.integersequencegenerators.redisincr

from uuid import uuid4

import os
from redis import StrictRedis

from eventsourcing.infrastructure.integersequencegenerators.base import AbstractIntegerSequenceGenerator


class RedisIncr(AbstractIntegerSequenceGenerator):
    """
    Generates a sequence of integers, using Redis' INCR command.

    Maximum number is 2**63, or 9223372036854775807, the maximum
    value of a 64 bit signed integer.
    """
    def __init__(self, redis=None, key=None):
        redis_host = os.getenv('REDIS_HOST', 'localhost')
        self.redis = redis or StrictRedis(host=redis_host)
        self.key = key or 'integer-sequence-generator-{}'.format(uuid4())

    def __next__(self):
        return self.redis.incr(self.key) - 1