如何修改嵌套结构的属性值

Question

我正在尝试修改 Go 中嵌套结构变量的值。基本上，我想修改RsvpString 属性，但GetRsvp() 似乎返回Rsvp 的值而不是引用，所以当我修改它的属性值时，它不会反映在Event 实例中。

测试如下。

type Event struct {
    Rsvps     []Rsvp `json:"rsvps"`
}

type Rsvp struct {
    UserId          string `json:"userId"`
    RsvpString      string `json:"rsvp"`
}

func (e *Event) GetRsvp(userId string) (rsvp *Rsvp, err error) {
    for _, element := range e.Rsvps {
        if element.UserId == userId {
            return &element, nil
        }
    }
    return &Rsvp{}, fmt.Errorf("could not find RSVP based on UserID")
}

func (e *Event) UpdateExistingRsvp(userId string, rsvpString string) {
    rsvp, err := e.GetRsvp(userId)
    if err == nil {
        rsvp.RsvpString = rsvpString
    }
}

这是测试代码：

func TestEvent_UpdateExistingRsvp(t *testing.T) {
    e := Event{[]Rsvp{
        {Name:      "Bill",
            UserId:    "bill",
            Rsvp:      "yes"}}}

    e.UpdateExistingRsvp("bill", "no")
    assert.Equal(t, "no", e.Rsvps[0].Rsvp, "RSVP should be switched to no") // fails
}

Answer 1

GetRsvp 正在返回循环变量的地址，而不是数组中元素的地址。修复：

    for i, element := range e.Rsvps {
        if element.UserId == userId {
            return &e.Rsvps[i], nil
        }
    }

循环变量保留一份 e.Rsvps[i] 的副本，并在每次迭代时被覆盖。如果返回循环变量的地址，则返回该副本的地址。

Answer 2

在切片上进行测距时，每次迭代都会返回两个值。第一个是索引，第二个是该索引处元素的副本。

所以从技术上讲，您正在尝试修改 Rsvp 的副本。 相反，从 GetRsvp() 方法返回索引并进行更新。

func (e *Event) GetRsvp(userId string) (int, error) {
    for index , element := range e.Rsvps {
        if element.UserId == userId {
            return index, nil
        }
    }
    return -1 , fmt.Errorf("could not find RSVP based on UserID")
}

func (e *Event) UpdateExistingRsvp(userId string, rsvpString string) {
    index, err := e.GetRsvp(userId)

    if err != nil || index == -1  {
        fmt.Println("no such user")
    }
    e.Rsvps[index].RsvpString = rsvpString
}