From d68c3ecf98bf3b5802a6b0f9a6bcf7825addd9e5 Mon Sep 17 00:00:00 2001 From: nobu Date: Tue, 10 Nov 2015 05:02:02 +0000 Subject: [PATCH] hash.c: compare methods * hash.c (rb_hash_{le,lt,ge,gt}): new methods, Hash#<=, Hash#<, Hash#>=, Hash#>, to test if all elements of a hash are also included in another hash, and vice versa. [ruby-core:68561] [Feature #10984] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@52518 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ChangeLog | 7 ++++++ NEWS | 1 + hash.c | 57 ++++++++++++++++++++++++++++++++++++++++++ test/ruby/test_hash.rb | 25 ++++++++++++++++++ 4 files changed, 90 insertions(+) diff --git a/ChangeLog b/ChangeLog index 5697e3edb7..eeb22b2ebf 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,10 @@ +Tue Nov 10 14:01:59 2015 Nobuyoshi Nakada + + * hash.c (rb_hash_{le,lt,ge,gt}): new methods, Hash#<=, Hash#<, + Hash#>=, Hash#>, to test if all elements of a hash are also + included in another hash, and vice versa. + [ruby-core:68561] [Feature #10984] + Tue Nov 10 11:25:29 2015 NARUSE, Yui * time.c (rb_timespec_now): added. diff --git a/NEWS b/NEWS index 704d58206d..c48f0f4aed 100644 --- a/NEWS +++ b/NEWS @@ -59,6 +59,7 @@ with all sufficient information, see the ChangeLog file. * Hash#fetch_values [Feature #10017] * Hash#dig [Feature #11643] + * Hash#<=, Hash#<, Hash#>=, Hash#> [Feature #10984] * IO diff --git a/hash.c b/hash.c index ae65aa4a1e..beaaad6818 100644 --- a/hash.c +++ b/hash.c @@ -2714,6 +2714,58 @@ rb_hash_dig(int argc, VALUE *argv, VALUE self) return rb_obj_dig(argc, argv, self, Qnil); } +static int +hash_le_i(VALUE key, VALUE value, VALUE arg) +{ + VALUE *args = (VALUE *)arg; + VALUE v = rb_hash_lookup2(args[0], key, Qundef); + if (v != Qundef && rb_equal(value, v)) return ST_CONTINUE; + args[1] = Qfalse; + return ST_STOP; +} + +static VALUE +hash_le(VALUE hash1, VALUE hash2) +{ + VALUE args[2]; + args[0] = hash2; + args[1] = Qtrue; + rb_hash_foreach(hash1, hash_le_i, (VALUE)args); + return args[1]; +} + +static VALUE +rb_hash_le(VALUE hash, VALUE other) +{ + other = to_hash(other); + if (RHASH_SIZE(hash) > RHASH_SIZE(other)) return Qfalse; + return hash_le(hash, other); +} + +static VALUE +rb_hash_lt(VALUE hash, VALUE other) +{ + other = to_hash(other); + if (RHASH_SIZE(hash) >= RHASH_SIZE(other)) return Qfalse; + return hash_le(hash, other); +} + +static VALUE +rb_hash_ge(VALUE hash, VALUE other) +{ + other = to_hash(other); + if (RHASH_SIZE(hash) < RHASH_SIZE(other)) return Qfalse; + return hash_le(other, hash); +} + +static VALUE +rb_hash_gt(VALUE hash, VALUE other) +{ + other = to_hash(other); + if (RHASH_SIZE(hash) <= RHASH_SIZE(other)) return Qfalse; + return hash_le(other, hash); +} + static int path_tainted = -1; static char **origenviron; @@ -4139,6 +4191,11 @@ Init_Hash(void) rb_define_method(rb_cHash, "any?", rb_hash_any_p, 0); rb_define_method(rb_cHash, "dig", rb_hash_dig, -1); + rb_define_method(rb_cHash, "<=", rb_hash_le, 1); + rb_define_method(rb_cHash, "<", rb_hash_lt, 1); + rb_define_method(rb_cHash, ">=", rb_hash_ge, 1); + rb_define_method(rb_cHash, ">", rb_hash_gt, 1); + /* Document-class: ENV * * ENV is a hash-like accessor for environment variables. diff --git a/test/ruby/test_hash.rb b/test/ruby/test_hash.rb index ee51c8ed7d..4f48d3b90c 100644 --- a/test/ruby/test_hash.rb +++ b/test/ruby/test_hash.rb @@ -1308,6 +1308,31 @@ class TestHash < Test::Unit::TestCase assert_nil(h.dig(:c, 1)) end + def test_cmp + h1 = {a:1, b:2} + h2 = {a:1, b:2, c:3} + + assert_operator(h1, :<=, h1) + assert_operator(h1, :<=, h2) + assert_not_operator(h2, :<=, h1) + assert_operator(h2, :<=, h2) + + assert_operator(h1, :>=, h1) + assert_not_operator(h1, :>=, h2) + assert_operator(h2, :>=, h1) + assert_operator(h2, :>=, h2) + + assert_not_operator(h1, :<, h1) + assert_operator(h1, :<, h2) + assert_not_operator(h2, :<, h1) + assert_not_operator(h2, :<, h2) + + assert_not_operator(h1, :>, h1) + assert_not_operator(h1, :>, h2) + assert_operator(h2, :>, h1) + assert_not_operator(h2, :>, h2) + end + class TestSubHash < TestHash class SubHash < Hash def reject(*)