redis.rb in flipper-redis-0.5.0

- old
+ new
@@ -1,36 +1,127 @@
 require 'set'
 require 'redis'
+require 'flipper'
 
 module Flipper
   module Adapters
     class Redis
+      include Flipper::Adapter
+
+      # Private: The key that stores the set of known features.
+      FeaturesKey = :flipper_features
+
+      # Public: The name of the adapter.
+      attr_reader :name
+
+      # Public: Initializes a Redis flipper adapter.
+      #
+      # client - The Redis client to use. Feel free to namespace it.
       def initialize(client)
         @client = client
+        @name = :redis
       end
 
-      def read(key)
-        @client.get key.to_s
+      # Public: Gets the values for all gates for a given feature.
+      #
+      # Returns a Hash of Flipper::Gate#key => value.
+      def get(feature)
+        result = {}
+        doc = doc_for(feature)
+        fields = doc.keys
+
+        feature.gates.each do |gate|
+          result[gate.key] = case gate.data_type
+          when :boolean, :integer
+            doc[gate.key.to_s]
+          when :set
+            fields_to_gate_value fields, gate
+          else
+            unsupported_data_type gate.data_type
+          end
+        end
+
+        result
       end
 
-      def write(key, value)
-        @client.set key.to_s, value.to_s
+      # Public: Enables a gate for a given thing.
+      #
+      # feature - The Flipper::Feature for the gate.
+      # gate - The Flipper::Gate to disable.
+      # thing - The Flipper::Type being disabled for the gate.
+      #
+      # Returns true.
+      def enable(feature, gate, thing)
+        case gate.data_type
+        when :boolean, :integer
+          @client.hset feature.key, gate.key, thing.value.to_s
+        when :set
+          @client.hset feature.key, to_field(gate, thing), 1
+        else
+          unsupported_data_type gate.data_type
+        end
+
+        true
       end
 
-      def delete(key)
-        @client.del key.to_s
+      # Public: Disables a gate for a given thing.
+      #
+      # feature - The Flipper::Feature for the gate.
+      # gate - The Flipper::Gate to disable.
+      # thing - The Flipper::Type being disabled for the gate.
+      #
+      # Returns true.
+      def disable(feature, gate, thing)
+        case gate.data_type
+        when :boolean
+          @client.del feature.key
+        when :integer
+          @client.hset feature.key, gate.key, thing.value.to_s
+        when :set
+          @client.hdel feature.key, to_field(gate, thing)
+        else
+          unsupported_data_type gate.data_type
+        end
+
+        true
       end
 
-      def set_add(key, value)
-        @client.sadd(key.to_s, value.to_s)
+      # Public: Adds a feature to the set of known features.
+      def add(feature)
+        @client.sadd FeaturesKey, feature.name
+        true
       end
 
-      def set_delete(key, value)
-        @client.srem(key.to_s, value.to_s)
+      # Public: The set of known features.
+      def features
+        @client.smembers(FeaturesKey).to_set
       end
 
-      def set_members(key)
-        @client.smembers(key.to_s).to_set
+      # Private: Gets a hash of fields => values for the given feature.
+      #
+      # Returns a Hash of fields => values.
+      def doc_for(feature)
+        @client.hgetall(feature.key)
+      end
+
+      # Private: Converts gate and thing to hash key.
+      def to_field(gate, thing)
+        "#{gate.key}/#{thing.value}"
+      end
+
+      # Private: Returns a set of values given an array of fields and a gate.
+      #
+      # Returns a Set of the values enabled for the gate.
+      def fields_to_gate_value(fields, gate)
+        regex = /^#{Regexp.escape(gate.key)}\//
+        keys = fields.grep(regex)
+        values = keys.map { |key| key.split('/', 2).last }
+        values.to_set
+      end
+
+      # Private
+      def unsupported_data_type(data_type)
+        raise "#{data_type} is not supported by this adapter"
       end
     end
   end
 end