blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-24 02:16:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
withchao 2023-02-02 17:54:02 +08:00
parent 778e3e5cd2
commit e8a24ab59e
2 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pkg/utilsv2/slice.go → pkg/utils/utils_v2.go
View File

@ -1,4 +1,4 @@
package utilsv2 package utils
import ( import (
"sort" "sort"
@ -26,8 +26,8 @@ func Distinct[T comparable](ts []T) []T {
}) })
} }
// DeleteAt delete slice element, support negative number to delete the penultimate // Delete delete slice element, support negative number to delete the penultimate
func DeleteAt[E any](es []E, index ...int) []E { func Delete[E any](es []E, index ...int) []E {
switch len(index) { switch len(index) {
case 0: case 0:
return es return es
@ -58,6 +58,13 @@ func DeleteAt[E any](es []E, index ...int) []E {
} }
} }
// DeleteAt delete slice element, support negative number to delete the penultimate
func DeleteAt[E any](es *[]E, index ...int) []E {
v := Delete(*es, index...)
*es = v
return v
}
// IndexAny get the index of the element // IndexAny get the index of the element
func IndexAny[E any, K comparable](es []E, e E, fn func(e E) K) int { func IndexAny[E any, K comparable](es []E, e E, fn func(e E) K) int {
k := fn(e) k := fn(e)

12
pkg/utilsv2/slice_test.go → pkg/utils/utils_v2_test.go
View File

@ -1,4 +1,4 @@
package utilsv2 package utils
import ( import (
"fmt" "fmt"
@ -12,9 +12,9 @@ func TestDistinct(t *testing.T) {
func TestDeleteAt(t *testing.T) { func TestDeleteAt(t *testing.T) {
arr := []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9} arr := []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
fmt.Println(DeleteAt(arr, 0, 1, -1, -2)) fmt.Println(Delete(arr, 0, 1, -1, -2))
fmt.Println(DeleteAt(arr)) fmt.Println(Delete(arr))
fmt.Println(DeleteAt(arr, 1)) fmt.Println(Delete(arr, 1))
} }
func TestSliceToMap(t *testing.T) { func TestSliceToMap(t *testing.T) {
@ -72,8 +72,8 @@ func TestCompleteAny(t *testing.T) {
}) })
} }
list = DeleteAt(list, -1) DeleteAt(&list, -1)
ids = DeleteAt(ids, -1) DeleteAt(&ids, -1)
ok := CompleteAny(ids, list, func(t Item) int { ok := CompleteAny(ids, list, func(t Item) int {
return t.ID return t.ID