init

Author: starfork

.gitattributes

@@ -0,0 +1,2 @@
+*.go linguist-language=Go
+go.sum linguist-generated

.gitignore

@@ -0,0 +1,21 @@
+.DS_Store
+*.DS_Store
+*/*.DS_Store
+
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+go.sum
+go.work

go.mod

@@ -0,0 +1,5 @@
+module github.com/starfork/go-slice
+
+go 1.21.4
+
+require github.com/starfork/stargo v0.0.6

item.go

@@ -0,0 +1,32 @@
+package slice
+
+// 数字类型特有的
+type Item interface {
+	int | float32 | float64 | uint32 | uint64 | string
+}
+
+type SliceItem[T Item] []T
+
+func NewNumber[T Item](a []T) SliceItem[T] {
+	return a
+}
+
+// 求和
+func (s SliceItem[T]) Sum() T {
+	var sum T
+	for _, v := range s {
+		sum += v
+	}
+	return sum
+}
+
+// 取最大值。
+func (s SliceItem[T]) Max() T {
+	var max T = s[0]
+	for _, item := range s {
+		if item > max {
+			max = item
+		}
+	}
+	return max
+}

job.go

@@ -0,0 +1,19 @@
+package slice
+
+import "time"
+
+// 分批执行一个slice的任务
+func RangeJob[T Item](ids []T, step int, job func([]T), duration ...time.Duration) {
+	l := len(ids)
+	for i := 0; i < l; {
+		h := i + step
+		if h > l {
+			h = l
+		}
+		if len(duration) > 0 {
+			time.Sleep(duration[0])
+		}
+		go job(ids[i:h])
+		i = i + step
+	}
+}

job_test.go

@@ -0,0 +1,16 @@
+package slice
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestRangeFunc(t *testing.T) {
+	ids := []uint64{123, 4343, 343, 343, 555}
+	index := "abcd"
+	testJob := func(ids []uint64) {
+		fmt.Println(index, ids)
+	}
+
+	RangeJob(ids, 2, testJob)
+}

slice.go

@@ -0,0 +1,158 @@
+package slice
+
+import "github.com/starfork/stargo/util/number"
+
+type Slice[T comparable] []T
+
+func New[T comparable](a []T) Slice[T] {
+	return a
+}
+
+// 包含，f过滤函数
+func (s Slice[T]) Contains(key T, f ...func(k T) bool) bool {
+
+	fn := func(v T) bool {
+		return key == v
+	}
+	if len(f) > 0 {
+		fn = f[0]
+	}
+	for _, v := range s {
+		if fn(v) {
+			return true
+		}
+	}
+	return false
+}
+
+// 默认返回.不包含k，则返回v
+func (s Slice[T]) Default(k, v T) T {
+	if s.Contains(k) {
+		return k
+	}
+	return v
+}
+
+// 取一个.非随机取一个
+func (s Slice[T]) One(index ...int) T {
+	max := len(s)
+	if max == 1 {
+		return s[0]
+	}
+	var idx int = 0
+	if len(index) > 0 {
+		idx = index[0]
+	}
+	if idx >= max {
+		idx = max - 1
+	}
+	return s[idx : idx+1][0]
+}
+
+func (s Slice[T]) Rand() T {
+	max := len(s)
+	idx, _ := number.RangeRand(0, int64(max-1))
+	return s[idx : idx+1][0]
+}
+
+func (s Slice[T]) Map(fn func(k T) T) Slice[T] {
+	var res []T
+	for _, item := range s {
+		res = append(res, fn(item))
+	}
+	return res
+}
+
+// 过滤
+func (s Slice[T]) Filter(fn func(T) bool) Slice[T] {
+	var res []T
+	for _, item := range s {
+		if fn(item) {
+			res = append(res, item)
+		}
+	}
+	return res
+}
+
+// 去重
+func (s Slice[T]) Unique() Slice[T] {
+
+	tmp := map[T]T{}
+
+	for _, item := range s {
+		tmp[item] = item
+	}
+	var res []T
+	for _, k := range tmp {
+		res = append(res, k)
+	}
+	s = res
+	return s
+}
+
+// Tail 获取切片尾部元素
+// dv: 空切片默认值
+func (s Slice[T]) Tail(dv ...T) T {
+	if s == nil && len(dv) > 0 {
+		return dv[0]
+	}
+	return s[len(s)-1]
+}
+
+// 交集
+func (s Slice[T]) Intersect(b Slice[T]) Slice[T] {
+	inter := make([]T, 0)
+	mp := make(map[T]bool)
+	for _, sa := range s {
+		if _, ok := mp[sa]; !ok {
+			mp[sa] = true
+		}
+	}
+	for _, sb := range b {
+		if _, ok := mp[sb]; ok {
+			inter = append(inter, sb)
+		}
+	}
+	return inter
+}
+
+// 差集
+func (s Slice[T]) Diff(b Slice[T]) Slice[T] {
+	diff := make([]T, 0)
+	mp := make(map[T]bool)
+	for _, sa := range s {
+		if _, ok := mp[sa]; !ok {
+			mp[sa] = true
+		}
+	}
+	for _, sb := range b {
+		if ok := mp[sb]; ok {
+			delete(mp, sb)
+		}
+	}
+	for k := range mp {
+		diff = append(diff, k)
+	}
+
+	return diff
+}
+
+// 并集
+func (s Slice[T]) Union(b Slice[T]) Slice[T] {
+	union := make([]T, 0)
+	mp := make(map[T]bool)
+	s = append(s, b...)
+	for _, v := range s {
+		if ok := mp[v]; ok {
+			continue
+		}
+		mp[v] = true
+		union = append(union, v)
+	}
+
+	return union
+}
+
+func (s Slice[T]) IsEmpty() bool {
+	return s == nil
+}

slice_test.go

@@ -0,0 +1,89 @@
+package slice_test
+
+import (
+	"fmt"
+	"strings"
+	"testing"
+
+	"github.com/starfork/go-slice"
+)
+
+func TestContains(t *testing.T) {
+	noAuth := slice.New([]string{"/v1/passport/", "/v1/public/"})
+	path := "/v1/passport/login"
+	rs := noAuth.Contains(path, func(key string) bool {
+		fmt.Println(path, key)
+		return strings.Contains(path, key)
+	})
+	fmt.Println(rs)
+	//rs1 := noAuth.Contains("/v1/passport/")
+	//fmt.Println(rs1)
+}
+
+func TestAsSlice(t *testing.T) {
+	str := ""
+	s := slice.New(strings.Split(str, ","))
+	fmt.Println(s)
+	if !s.Contains("abc") {
+		s = append(s, "abc")
+	}
+	if !s.Contains("abc") {
+		s = append(s, "abc")
+	}
+	ns := slice.New(strings.Split(strings.Join(s, ","), ","))
+	fmt.Println(ns)
+	if !ns.Contains("def") {
+		ns = append(ns, "def")
+	}
+	if !ns.Contains("def") {
+		ns = append(ns, "def")
+	}
+	fmt.Println(len(ns))
+
+	fmt.Println(strings.Join(ns, ","))
+}
+
+func TestIntersect(t *testing.T) {
+	a := slice.New([]string{"/v1/passport/", "/v1/public/"})
+	b := slice.New([]string{"/v1/passport/", "/v1/public/1"})
+	fmt.Println(a.Intersect(b))
+}
+
+func TestDiff(t *testing.T) {
+	a := slice.New([]string{"/v1/passport/", "/v1/public/", "/v1/public/2"})
+	b := slice.New([]string{"/v1/passport/", "/v1/public/1"})
+	fmt.Println(a.Diff(b))
+}
+func TestUnion(t *testing.T) {
+	a := slice.New([]string{"/v1/passport/", "/v1/public/", "/v1/public/2"})
+	b := slice.New([]string{"/v1/passport/", "/v1/public/1"})
+	fmt.Println(a.Union(b))
+}
+
+func TestOne(t *testing.T) {
+	a := slice.New([]string{"/v1/passport/", "/v1/public/", "/v1/public/2", "/v1/public/4"})
+
+	fmt.Println(a.One(5))
+}
+
+func TestCalc(t *testing.T) {
+	a := slice.New([]string{}) //老权限
+	//删除2，新增7，8之后
+	b := slice.New([]string{"/v1/estate/estates$$GET", "/v1/service/display/services$$GET", "/v1/service/evaluate$$GET", "/v1/service/service$$PUT", "/v1/user/auth/rule$$PUT", "/v1/user/log/login$$GET", "/v1/user/friends$$GET", "/v1/user/auth/roles$$GET", "/v1/user/certifications$$GET"})
+	inter := a.Intersect(b)
+	fmt.Printf("交集:%+v", inter)
+
+	//需要删除的
+	fmt.Printf("需要删除的节点:%+v", a.Diff(inter))
+	add := b.Diff(inter)
+	fmt.Printf("需要增加的节点:%+v", add)
+	for k, v := range add {
+		fmt.Printf("add-v,%d,%+v\n", k, v)
+	}
+}
+
+func TestUnique(t *testing.T) {
+	a := slice.New([]string{"/v1/passport/", "/v1/public/", "/v1/public/2", "/v1/public/", "/v1/public/", "/v1/public/2"})
+
+	fmt.Println(a.Unique())
+}