abak-press/redis_counters

# coding: utf-8
require 'redis_counters/unique_values_lists/base'

module RedisCounters
  module UniqueValuesLists

    # Список уникальных значений, на основе не блокирующего алгоритма.
    #
    # Особенности:
    #   * 2-х кратный расход памяти в случае использования партиций;
    #   * Не ведет список партиций;
    #   * Не транзакционен;
    #   * Методы delete_partitions! и delete_partition_direct!, удаляют только дублирующие партиции,
    #     но не удаляют данные из основной партиции.
    #     Для удаления основной партиции необходимо вызвать delete_main_partition!
    #     или воспользоваться методами delete_all! или delete_all_direct!,
    #     для удаления всех партиций кластера включая основную.

    class NonBlocking < Base

      # Public: Проверяет существует ли заданное значение.
      #
      # params - Hash - параметры кластера и значения.
      #
      # Returns Boolean.
      #
      def has_value?(params)
        set_params(params)
        redis.sismember(main_partition_key, value)
      end

      # Public: Нетранзакционно удаляет все данные счетчика в кластере, включая основную партицию.
      # Если кластеризация не используется, то удаляет все данные.
      #
      # cluster       - Hash - хеш параметров, определяющих кластер.
      # write_session - Redis - соединение с Redis, в рамках которого
      #                 будет производится удаление (опционально).
      #                 По умолчанию - основное соединение счетчика.
      #
      # Returns Nothing.
      #
      def delete_all_direct!(cluster, write_session = redis, parts = partitions(cluster))
        super(cluster, write_session, parts)
        delete_main_partition!(cluster, write_session)
      end

      # Public: Удаляет основную партицию.
      #
      # cluster       - Hash - хеш параметров, определяющих кластер.
      #                 Опционально, если кластеризация не используется.
      # write_session - Redis - соединение с Redis, в рамках которого
      #                 будет производится удаление (опционально).
      #                 По умолчанию - основное соединение счетчика.
      #
      # Returns Nothing.
      #
      def delete_main_partition!(cluster = {}, write_session = redis)
        cluster = ::RedisCounters::Cluster.new(self, cluster).params
        key = key(main_partition, cluster)
        write_session.del(key)
      end

      protected

      def process_value
        return unless add_value
        yield redis if block_given?
        true
      end

      def add_value
        return unless redis.sadd(main_partition_key, value)
        redis.sadd(current_partition_key, value) if use_partitions?
        true
      end

      def main_partition_key
        key(main_partition)
      end

      def current_partition_key
        key
      end

      def main_partition
        []
      end
    end

  end
end