Можно ли в Golang изменить длину среза напрямую?

Question

[bozedai]

Проблема такая: в функцию передается срез

func foo(bar []int){
    // Как изменить length у среза?
}
input := []int{1, 2, 3}
foo(input)
fmt.Println(len(input)) // Например: 2

И нужно внутри функции изменить длину срез, не создавая новый, чтобы по старому указателю все было доступно.

Так вот не работает
*&bar = bar[:2]

Answer 1

[Евгений Мамонов]

Можно попробовать сделать через reflect, например так

package main

import (
    "fmt"
    "reflect"
)

func modify(input *[]int, newLen int) {
    val := reflect.Indirect(reflect.ValueOf(input))
    val.SetLen(newLen)
}

func main() {
    input := []int{1, 2, 3}

    fmt.Printf("Before: %p, value: %v\n", &input, input)

    modify(&input, len(input) - 1)

    fmt.Printf("After: %p, value: %v\n", &input, input)
}

Вывод:

Before: 0xc00000c060, value: [1 2 3]
After: 0xc00000c060, value: [1 2]

Comments

[bozedai]

Вот видите, ссылка то не поменялась, а возвращать и перезаписывать в переменную ее вам пришлось. Попробуйте реализовать мой пример, чтобы слайс изменился, но без возврата его из функции.

[bozedai]

@EvgenyMamonov
В том то и дело, что не такой :-)

[bozedai]

Евгений Мамонов,
А вот это вообще мозг взрывает, что куда там указывает, что происходит вообще?

[bozedai]

Евгений Мамонов,

Собственно нифига ссылка не та же самая

package main

import "fmt"

func modify(input []int) {
    fmt.Printf("Addr: %p, value: %v\n", &input, input)
}

func main() {
    input := []int{1, 2, 3}

    fmt.Printf("Addr: %p, value: %v\n", &input, input)

    modify(input)
}

[Евгений Мамонов]

Задача именно уменьшить размер слайса (укоротить с конца)?

[Евгений Мамонов]

Можно попробовать сделать через reflect, например так

package main

import (
    "fmt"
    "reflect"
)

func modify(input *[]int, newLen int) {
    val := reflect.Indirect(reflect.ValueOf(input))
    val.SetLen(newLen)
}

func main() {
    input := []int{1, 2, 3}

    fmt.Printf("Before: %p, value: %v\n", &input, input)

    modify(&input, len(input) - 1)

    fmt.Printf("After: %p, value: %v\n", &input, input)
}

Вывод:

Before: 0xc00000c060, value: [1 2 3]
After: 0xc00000c060, value: [1 2]

То, что нужно?

[bozedai]

Евгений Мамонов, Да, спасибо большое!

[Евгений Мамонов]

Рад помочь :)

[bozedai]

Евгений Мамонов, Ах нет(( Вы передаете по ссылке срез, а меня именно это не устраивает, если так передавать, то и без рефлексии работает

[Евгений Мамонов]

Можете описать задачу подробнее, возможно я предложу более подходящее решение?

[bozedai]

Евгений Мамонов, спасибо за помощь, вопрос именно возник в этом моменте, хотя я вроде нашла ответ. Срезы не передаются по умолчанию по ссылке, копируется вся структура, поэтому внутрь функции попадет адрес на массив, но копия длины и вместимости, а переменная будет уже ссылаться на новую скопированную структуру и изменения в ней не будут видны снаружи в любом случае.

[Евгений Мамонов]

В go слайсы выглядят примерно так

type Slice struct {
    Data uintptr
    Len  int
    Cap  int
}

При передаче слайса при вызове функции - будет создан новый слайс, но указатель на Data у них будет одинаковый. Т.е. полного копирования данных не происходит, копируются только указатель на данные, Cap и Len.

Соответственно внутри функции по другому не получится изменить слайс, или возвращать значение или передавать в функцию указатель на слайс.

[Евгений Мамонов]

По опыту могу сказать, что менять переменную, которая приходит в параметрах - это не очень хорошая практика. Такой код крайне сложно отлаживать и искать в нём ошибки.

Имеет смысл менять значения переменных только когда, тип которых не известен (interface{})

Так делают, например в пакете database/sql, в функции Scan

var name string
rows.Scan(&name);

Во всех остальных случаях - лучше возвращать значения из функции.

[bozedai]

Евгений Мамонов, Вы безусловно и абсолютно правы, было недопонимание тут с указателями.

Answer 2

[Klars]

Еще вариант

package main

import (
	"fmt"
)

func squeeze(sl *[]int, splitter int) {
	s := *sl
	s = s[:splitter]
	*sl = s
}

func main() {
	sl := []int{1, 2, 3, 4, 5, 6}
	fmt.Println(len(sl), cap(sl), sl)
	fmt.Printf("%p\n\n", &sl)

	squeeze(&sl, 4)

	fmt.Println(len(sl), cap(sl), sl)
	fmt.Printf("%p", &sl)
}

output:
6 6 [1 2 3 4 5 6]
0xc00000c0a0

4 6 [1 2 3 4]
0xc00000c0a0

Comments

[bozedai]

Да весь прикол был в том, чтобы не передавать ссылку в функцию, теперь то понимаю, что это бессмысленно, но спасибо :-)