将字节保存到固定大小的字节数组缓冲区中的通用 Go 函数

Question

如何编写一个通用的 Go 函数来将字节推入字节数组？

我认为附加到切片并不是一个完美的选择，因为数组不应该增长。 但是，该函数应处理任何数组大小。 我正在考虑对数组进行切片以使函数接受任何数组大小。 请参阅下面的草图。

有没有更清洁的方法？

播放：http://play.golang.org/p/Gii9-JM33E

func push(buf []byte, size int, b byte) (int, error) {
    max := len(buf)

    if max < 1 {
        return size, fmt.Errorf("buffer underflow: max=%d char=%d", max, b)
    }

    if size >= max {
        return size, fmt.Errorf("buffer overflow: size=%d max=%d char=%d", size, max, b)
    }

    buf[size] = b

    return size + 1, nil
}

用法：

buf := [3]byte{}
size := 0
var err error
size, err = push(buf[:], size, 'a')

Answer 1

您几乎肯定想要使用切片而不是数组。切片具有三样东西：底层支持数组、容量和长度。容量“cap”告诉您底层支持数组的大小。

因此，您可以在检查 len(s) < cap(s) 后使用 append 以避免任何重新分配。或者如果你想做一个函数，也许是这样的：

func pushByte(s []byte, b byte) ([]byte, error) {
    if len(s) == cap(s) {
        return s, errors.New("capacity reached")
    }
    return append(s, b), nil
}

你会像这样使用它：

s := make([]byte, 0, 10)
var err error
if s, err = pushByte(s, 10); err != nil {
    ... handle error
}

Answer 2

数组的大小是其类型的一部分。

如果有一组您知道自己会接受的尺寸，您可以使用buf interface{} 并在您想要的尺寸上使用类型开关。我假设您知道数组大小，因为它们在编译时必须保持不变。

除此之外，您还需要使用反射来实际查找数组大小，并将元素存储在数组中。

很少需要使用数组而不是切片。您可能需要重新考虑为什么必须传递不同大小的数组。