gpt4 book ai didi

c - 如何有效地合并 Ruby C API 中的两个散列?

转载 作者:数据小太阳 更新时间:2023-10-29 07:49:06 26 4
gpt4 key购买 nike

我正在为确实需要合并两个散列的 Ruby 编写 C 扩展,但是 rb_hash_merge() 函数在 Ruby 1.8.6 中是静态的。我尝试改为使用:

rb_funcall(hash1, rb_intern("merge"), 1, hash2);

但这太慢了,而且性能在此应用程序中非常关键。

有谁知道如何在兼顾效率和速度的情况下执行此合并?

(请注意,我曾尝试简单地查看 rb_hash_merge() 的源代码并复制它,但它充满了其他静态函数,这些静态函数本身充满了更多的静态函数,因此似乎几乎不可能解开...我需要另一种方式)

最佳答案

好的,看起来可能无法在已发布的 API 中进行优化。

测试代码:

#extconf.rb
require 'mkmf'
dir_config("hello")
create_makefile("hello")


// hello.c
#include "ruby.h"

static VALUE rb_mHello;
static VALUE rb_cMyCalc;

static void calc_mark(void *f) { }
static void calc_free(void *f) { }
static VALUE calc_alloc(VALUE klass) { return Data_Wrap_Struct(klass, calc_mark, calc_free, NULL); }

static VALUE calc_init(VALUE obj) { return Qnil; }

static VALUE calc_merge(VALUE obj, VALUE h1, VALUE h2) {
return rb_funcall(h1, rb_intern("merge"), 1, h2);
}

static VALUE
calc_merge2(VALUE obj, VALUE h1, VALUE h2)
{
VALUE h3 = rb_hash_new();
VALUE keys;
VALUE akey;
keys = rb_funcall(h1, rb_intern("keys"), 0);
while (akey = rb_each(keys)) {
rb_hash_aset(h3, akey, rb_hash_aref(h1, akey));
}
keys = rb_funcall(h2, rb_intern("keys"), 0);
while (akey = rb_each(keys)) {
rb_hash_aset(h3, akey, rb_hash_aref(h2, akey));
}
return h3;
}

static VALUE
calc_merge3(VALUE obj, VALUE h1, VALUE h2)
{
VALUE keys;
VALUE akey;
keys = rb_funcall(h1, rb_intern("keys"), 0);
while (akey = rb_each(keys)) {
rb_hash_aset(h2, akey, rb_hash_aref(h1, akey));
}
return h2;
}

void
Init_hello()
{
rb_mHello = rb_define_module("Hello");
rb_cMyCalc = rb_define_class_under(rb_mHello, "Calculator", rb_cObject);
rb_define_alloc_func(rb_cMyCalc, calc_alloc);
rb_define_method(rb_cMyCalc, "initialize", calc_init, 0);
rb_define_method(rb_cMyCalc, "merge", calc_merge, 2);
rb_define_method(rb_cMyCalc, "merge2", calc_merge, 2);
rb_define_method(rb_cMyCalc, "merge3", calc_merge, 2);
}


# test.rb
require "hello"

h1 = Hash.new()
h2 = Hash.new()

1.upto(100000) { |x| h1[x] = x+1; }
1.upto(100000) { |x| h2["#{x}-12"] = x+1; }

c = Hello::Calculator.new()

puts c.merge(h1, h2).keys.length if ARGV[0] == "1"
puts c.merge2(h1, h2).keys.length if ARGV[0] == "2"
puts c.merge3(h1, h2).keys.length if ARGV[0] == "3"

现在测试结果:

$ time ruby test.rb

real 0m1.021s
user 0m0.940s
sys 0m0.080s
$ time ruby test.rb 1
200000

real 0m1.224s
user 0m1.148s
sys 0m0.076s
$ time ruby test.rb 2
200000

real 0m1.219s
user 0m1.132s
sys 0m0.084s
$ time ruby test.rb 3
200000

real 0m1.220s
user 0m1.128s
sys 0m0.092s

所以看起来我们可能会在 0.2 秒的操作中最多减少 ~0.004 秒。

鉴于除了设置值之外可能没有那么多,因此可能没有那么多的空间来进一步优化。也许尝试破解 ruby​​ 源代码本身——但那时你不再真正开发“扩展”而是改变语言,所以它可能行不通。

如果哈希的连接是您需要在 C 部分中多次执行的操作 - 那么可能使用内部数据结构并仅在最后一次将它们导出到 Ruby 哈希中将是优化事情的唯一方法。

附注从 this excellent tutorial 借用的代码的初始框架

关于c - 如何有效地合并 Ruby C API 中的两个散列?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/1256975/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com