浅谈Go Slice 高级实践

go 语言切片是对数组的抽象。

go 数组的长度不可改变，在特定场景中这样的集合就不太适用，go中提供了一种灵活，功能强悍的内置类型切片("动态数组"),与数组相比切片的长度是不固定的，可以追加元素，在追加时可能使切片的容量增大。

定义切片

你可以声明一个未指定大小的数组来定义切片：

var identifier []type

切片不需要说明长度。

或使用make()函数来创建切片:

var slice1 []type = make([]type, len)

也可以简写为

slice1 := make([]type, len)

也可以指定容量，其中capacity为可选参数。

make([]t, length, capacity)

这里 len 是数组的长度并且也是切片的初始长度。

以下用法中，类型均使用 int64 做为示例，不处理 interface 。

代码只是展示实现思路，不一定完善。

合并两个有序切片，新切片仍然有序

func mergesortedslice(s1, s2 []int64) []int64 {
  // 从末尾元素开始遍历
  i := len(s1) - 1
  j := len(s2) - 1
  // 合并后的长度
  newlen := len(s1) + len(s2)
  // 合并后的索引，也从末尾元素开始
  newidx := newlen - 1
  // 创建一个新切片，代表合并后的
  news := make([]int64, newlen)
  // 将 s1 的内容拷贝到新切片
  for k, v := range s1 {
    news[k] = v
  }
  // 开始遍历
  for i >= 0 && j >= 0 {
    // 新元素
    var newnum int64
    // 将较大的值赋给新元素，同时向前移动指针
    if news[i] > s2[j] {
      newnum = news[i]
      i--
    } else {
      newnum = s2[j]
      j--
    }
    news[newidx] = newnum
    newidx--
  }
  // 如果 s2 还有剩余元素，则剩余元素一定都是最小的，直接放到头部即可
  for j >= 0 {
    news[newidx] = s2[j]
    j--
    newidx--
  }
  return news
}

根据特定规则过滤元素

func filterslice(s []int64, filter func(x int64) bool) []int64 {
  // 返回的新切片
  // s[:0] 这种写法是创建了一个 len 为 0，cap 为 len(s) 即和原始切片最大容量一致的切片
  // 因为是过滤，所以新切片的元素总个数一定不大于比原始切片，这样做减少了切片扩容带来的影响
  // 同时，也有一个问题，因为 news 和 s 共享底层数组，那么过滤后 s 也会被修改！
  news := s[:0]
  // 遍历，对每个元素执行 filter，符合条件的加入新切片中
  for _, x := range s {
    if !filter(x) {
      news = append(news, x)
    }
  }
  return news
}

去重

两种思路，循环顺序查找和使用 map 加快查找（引入一个 map 在各方面也是有开销的）。选用哪种，可以通过具体场景的 benchmark 决定

func removeduplicates(s []int64) []int64 {
  var ret []int64
  for _, v := range s {
    found := false
    for _, v2 := range ret {
      if v == v2 {
        found = true
        break
      }
    }
    if !found {
      ret = append(ret, v)
    }
  }
  return ret
}

func removeduplicates2(s []int64) []int64 {
  ret := s[:0]
  // 利用 struct{}{} 减少内存占用
  assist := map[int64]struct{}{}
  for _, v := range s {
    if _, ok := assist[v]; !ok {
      assist[v] = struct{}{}
      ret = append(ret, v)
    }
  }
  return ret
}

反转

func reversing(s []int64) []int64 {
  for left, right := 0, len(s)-1; left < right; left, right = left+1, right-1 {
    s[left], s[right] = s[right], s[left]
  }
  return s
}

分块

主要用于当单个切片过大，需要分多次使用的时候，比如网络调用等。

func slicechunk(s []int64, size int) [][]int64 {
  var ret [][]int64
  for size < len(s) {
    // s[:size:size] 表示 len 为 size，cap 也为 size，第二个冒号后的 size 表示 cap
    s, ret = s[size:], append(ret, s[:size:size])
  }
  ret = append(ret, s)
  return ret
}

类型转换

rpc 中，不同下游接收的类型可能不一样，还有自定义类型，这里提供一个快速转换的方法

s := []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
var news []int64
// 做法是利用 reflect 直接替换数据指针
// 但是这个不保证在以后的版本中一直可用 ╮(╯▽╰)╭
*(*reflect.sliceheader)(unsafe.pointer(&news)) = *(*reflect.sliceheader)(unsafe.pointer(&s))
fmt.printf("type:%t value:%v", news, news)

主要参考：https://github.com/golang/go/wiki/slicetricks

官方使用技巧，建议多看看。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。