Question

今天我遇到了一些奇怪的Array元素赋值行为：

arr = ["a","b"]
arr2 = [1,2]
arr.unshift(arr2) #= [[1, 2], "a", "b"] 
arr.push(arr2) #=> ["a", "b", [1, 2]]

然而，这是有道理的：

arr[0,0] = arr2 #=> [1, 2, "a", "b"]

我知道在[0,0]中，第一个零是index，第二个是从index开始在该数组中生效的元素数。

在我的想法中它应该与unshift相同，但事实并非如此。

任何人都可以解释这种行为吗？

Answer 1

如果我们dive into the ruby source code，我们会在rb_ary_splice找到一个名为array assignment happens的函数，该函数有三个参数（即索引，长度和新值）：

static VALUE
rb_ary_aset(int argc, VALUE *argv, VALUE ary)
{
    long offset, beg, len;

    if (argc == 3) {
        rb_ary_modify_check(ary);
        beg = NUM2LONG(argv[0]);
        len = NUM2LONG(argv[1]);
        rb_ary_splice(ary, beg, len, argv[2]);
        return argv[2];
    }
[...]

如果我们跟随rb_ary_splice，我们会发生魔法发生的地方：

static void
rb_ary_splice(VALUE ary, long beg, long len, VALUE rpl)
{
    long rlen;
    long olen;

    if (len < 0) rb_raise(rb_eIndexError, "negative length (%ld)", len);
    olen = RARRAY_LEN(ary);

    [...]

        if (len != rlen) {
            RARRAY_PTR_USE(ary, ptr,
                   MEMMOVE(ptr + beg + rlen, ptr + beg + len,
                       VALUE, olen - (beg + len)));
            ARY_SET_LEN(ary, alen);
        }
        if (rlen > 0) {
            MEMMOVE(RARRAY_PTR(ary) + beg, RARRAY_CONST_PTR(rpl), VALUE, rlen);
        }
    }
    RB_GC_GUARD(rpl);
}

首先，它在数组中为新元素腾出空间并更新长度：

RARRAY_PTR_USE(ary, ptr,
    MEMMOVE(ptr + beg + rlen, ptr + beg + len,
        VALUE, olen - (beg + len)));
ARY_SET_LEN(ary, alen);

然后通过C指针的魔力，它插入新元素：

MEMMOVE(RARRAY_PTR(ary) + beg, RARRAY_CONST_PTR(rpl), VALUE, rlen);

Array元素赋值的奇怪行为

1 个答案: