allow bucketing by int attribute (using its string representation)

Author: eli-darkly

lib/ldclient-rb/evaluation.rb

@@ -223,7 +223,10 @@ def variation_for_user(rule, user, flag)
     def bucket_user(user, key, bucket_by, salt)
       return nil unless user[:key]
 
-      id_hash = user_value(user, bucket_by)
+      id_hash = bucketable_string_value(user_value(user, bucket_by))
+      if id_hash.nil?
+        return 0.0
+      end
 
       if user[:secondary]
         id_hash += "." + user[:secondary]
@@ -235,6 +238,12 @@ def bucket_user(user, key, bucket_by, salt)
       hash_val.to_i(16) / Float(0xFFFFFFFFFFFFFFF)
     end
 
+    def bucketable_string_value(value)
+      return value if value.is_a? String
+      return value.to_s if value.is_a? Integer
+      nil
+    end
+
     def user_value(user, attribute)
       attribute = attribute.to_sym
 

spec/evaluation_spec.rb

@@ -0,0 +1,45 @@
+require "spec_helper"
+
+describe LaunchDarkly::Evaluation do
+  subject { LaunchDarkly::Evaluation }
+
+  include LaunchDarkly::Evaluation
+
+  describe "bucket_user" do
+    it "can bucket by int value (equivalent to string)" do
+      user = {
+        key: "userkey",
+        custom: {
+          stringAttr: "33333",
+          intAttr: 33333
+        }
+      }
+      stringResult = bucket_user(user, "key", "stringAttr", "salt")
+      intResult = bucket_user(user, "key", "intAttr", "salt")
+      expect(intResult).to eq(stringResult)
+    end
+
+    it "cannot bucket by float value" do
+      user = {
+        key: "userkey",
+        custom: {
+          floatAttr: 33.5
+        }
+      }
+      result = bucket_user(user, "key", "floatAttr", "salt")
+      expect(result).to eq(0.0)
+    end
+
+
+    it "cannot bucket by bool value" do
+      user = {
+        key: "userkey",
+        custom: {
+          boolAttr: true
+        }
+      }
+      result = bucket_user(user, "key", "boolAttr", "salt")
+      expect(result).to eq(0.0)
+    end
+  end
+end