From 62ef35a518c259142679d171f53060d0cef79d13 Mon Sep 17 00:00:00 2001 From: kercylan98 Date: Fri, 8 Sep 2023 13:27:25 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20slice=20=E5=8C=85=E6=96=B0=E5=A2=9E=20Z?= =?UTF-8?q?oom=20=E5=87=BD=E6=95=B0=EF=BC=8C=20stream=20=E5=8C=85=E6=94=AF?= =?UTF-8?q?=E6=8C=81=20Zoom=20=E5=87=BD=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/slice/zoom.go | 11 +++++++++++ utils/stream/slice.go | 12 +++++++++++- 2 files changed, 22 insertions(+), 1 deletion(-) create mode 100644 utils/slice/zoom.go diff --git a/utils/slice/zoom.go b/utils/slice/zoom.go new file mode 100644 index 0000000..567d3d3 --- /dev/null +++ b/utils/slice/zoom.go @@ -0,0 +1,11 @@ +package slice + +// Zoom 将切片的长度缩放到指定的大小,如果 newSize 小于 slice 的长度,则会截断 slice,如果 newSize 大于 slice 的长度,则会在 slice 的末尾添加零值数据 +func Zoom[V any](newSize int, slice []V) []V { + if newSize < 0 { + newSize = 0 + } + var s = make([]V, newSize) + copy(s, slice) + return s +} diff --git a/utils/stream/slice.go b/utils/stream/slice.go index 00fe381..460e9ea 100644 --- a/utils/stream/slice.go +++ b/utils/stream/slice.go @@ -9,7 +9,17 @@ func (slf Slice[V]) Slice() []V { return slf } -// Chunk 的快捷方式 +// Copy 复制一份切片 +func (slf Slice[V]) Copy() Slice[V] { + return slice.Copy(slf) +} + +// Zoom 是 slice.Zoom 的快捷方式 +func (slf Slice[V]) Zoom(newSize int) Slice[V] { + return slice.Zoom(newSize, slf) +} + +// Chunk 是 slice.Chunk 的快捷方式 func (slf Slice[V]) Chunk(size int) Slices[V] { chunks := slice.Chunk(slf, size) result := make(Slices[V], len(chunks))