如何确定slice接口的元素类型{}？

Question

我有以下代码将切片加倍。

func doubleSlice(s []int) []int  {
  t := make([]int, len(s), (cap(s) + 1) * 2 )
  for i := range s {
    t[i] = s[i]
  }
  return t
}

我想使函子加倍任何类型的切片。我首先需要知道元素类型。

func showInterfaceItem(s interface{}) interface{} {
  if reflect.TypeOf(s).Kind() != reflect.Slice {
    fmt.Println("The interface is not a slice.")
    return
  }

  var t interface{}
  newLen := reflect.ValueOf(s).Len()
  newCap := (cap(reflect.ValueOf(s).Cap()) + 1) * 2
  t = make([]reflect.TypeOf(s), newLen, newCap)

  return t
}

所述reflect.TypeOf(s)返回接口{类型}，元件的不类型。如何获取slice接口的元素类型？

Answer 1

您可以reflect.TypeOf(s).Elem() 用来获取slice元素的类型。

package main

import (
    "fmt"
    "reflect"
)

func doubleSlice(s interface{}) interface{} {
    if reflect.TypeOf(s).Kind() != reflect.Slice {
        fmt.Println("The interface is not a slice.")
        return nil
    }

    v := reflect.ValueOf(s)
    newLen := v.Len()
    newCap := (v.Cap() + 1) * 2
    typ := reflect.TypeOf(s).Elem()

    t := reflect.MakeSlice(reflect.SliceOf(typ), newLen, newCap)
    reflect.Copy(t, v)
    return t.Interface()
}

func main() {
    xs := doubleSlice([]string{"foo", "bar"}).([]string)
    fmt.Println("data =", xs, "len =", len(xs), "cap =", cap(xs))

    ys := doubleSlice([]int{3, 1, 4}).([]int)
    fmt.Println("data =", ys, "len =", len(ys), "cap =", cap(ys))
}

输出将是：

data = [foo bar] len = 2 cap = 6
data = [3 1 4] len = 3 cap = 8

签入：去游乐场