summaryrefslogtreecommitdiff
path: root/jni/ruby/ext/-test-/st/update/update.c
blob: 979ad3e33420c1a180d61c57e9192aed142145c0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#include <ruby.h>
#include <ruby/st.h>

static int
update_func(st_data_t *key, st_data_t *value, st_data_t arg, int existing)
{
    VALUE ret = rb_yield_values(existing ? 2 : 1, (VALUE)*key, (VALUE)*value);
    switch (ret) {
      case Qfalse:
        return ST_STOP;
      case Qnil:
        return ST_DELETE;
      default:
        *value = ret;
        return ST_CONTINUE;
    }
}

static VALUE
test_st_update(VALUE self, VALUE key)
{
    if (st_update(RHASH_TBL(self), (st_data_t)key, update_func, 0))
        return Qtrue;
    else
        return Qfalse;
}

void
Init_update(void)
{
    VALUE st = rb_define_class_under(rb_define_module("Bug"), "StTable", rb_cHash);
    rb_define_method(st, "st_update", test_st_update, 1);
}