mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-11-21 14:44:00 +00:00
87338633b1
The added helper functions - SetLength() and ExtendCapacity() - replace error-prone code with simple function calls.
23 lines
639 B
Go
23 lines
639 B
Go
package slicesutil
|
|
|
|
// ExtendCapacity returns a with the capacity extended to len(a)+itemsToAdd.
|
|
//
|
|
// It may allocate new slice if cap(a) is smaller than len(a)+itemsToAdd.
|
|
func ExtendCapacity[T any](a []T, itemsToAdd int) []T {
|
|
aLen := len(a)
|
|
if n := aLen + itemsToAdd - cap(a); n > 0 {
|
|
a = append(a[:cap(a)], make([]T, n)...)
|
|
return a[:aLen]
|
|
}
|
|
return a
|
|
}
|
|
|
|
// SetLength sets len(a) to newLen and returns the result.
|
|
//
|
|
// It may allocate new slice if cap(a) is smaller than newLen.
|
|
func SetLength[T any](a []T, newLen int) []T {
|
|
if n := newLen - cap(a); n > 0 {
|
|
a = append(a[:cap(a)], make([]T, n)...)
|
|
}
|
|
return a[:newLen]
|
|
}
|