Implement Hash#choice

Author: baban

hash.c

@@ -1348,6 +1348,30 @@ rb_hash_values_at(int argc, VALUE *argv, VALUE hash)
     return result;
 }
 
+/*
+ * call-seq:
+ *   hsh.choice(key, ...)   -> Hash
+ *
+ * Return a key-value pair associated with given keys. 
+ *
+ * { :a => 1, 2 => "2", "c" => true }.choice(:a, 2) # => { :a => 1, 2 => "2" }
+ * { :a => 1, 2 => "2", "c" => true }.choice("c", 10000) # => { "c" => true, 10000 => nil }
+ */
+
+VALUE
+rb_hash_choice(int argc, VALUE *argv, VALUE self){
+  int i;
+  VALUE result, value;
+  result = rb_hash_new();
+
+  for(i=0; i < argc; i++){
+    value = rb_hash_aref(self, argv[i]);
+    rb_hash_aset(result, argv[i], value);
+  }
+
+  return result;
+}
+
 /*
  * call-seq:
  *   hsh.fetch_values(key, ...)                 -> array
@@ -4476,6 +4500,7 @@ Init_Hash(void)
     rb_define_method(rb_cHash, "keys", rb_hash_keys, 0);
     rb_define_method(rb_cHash, "values", rb_hash_values, 0);
     rb_define_method(rb_cHash, "values_at", rb_hash_values_at, -1);
+    rb_define_method(rb_cHash, "choice", rb_hash_choice, -1);
     rb_define_method(rb_cHash, "fetch_values", rb_hash_fetch_values, -1);
 
     rb_define_method(rb_cHash, "shift", rb_hash_shift, 0);

test/ruby/test_hash.rb

@@ -572,6 +572,17 @@ def test_values_at
     assert_equal ['three', nil, 'one', 'nil'], res
   end
 
+  def test_choice
+    res = @h.choice()
+    assert_equal({}, res)
+
+    res = @h.choice(1,2)
+    assert_equal({1 => "one", 2 => "two"}, res)
+
+    res = @h.choice(1,2,1000)
+    assert_equal({1 => "one", 2 => "two", 1000 => nil}, res)
+  end
+
   def test_fetch_values
     res = @h.fetch_values
     assert_equal(0, res.length)